creditdaddy 1.6.0 → 1.6.3

package/src/traeGateway.js

@@ -7,7 +7,7 @@
  * 架构特性：
  * 1. 账号多路轮换（provider=trae）：自动在有效账号间按剩余积分加权轮询（SWRR）
  * 2. 凭据热对齐：401 时若为本机导入账号，尝试重读 storage.json 最新登录态
- * 3. 429 限流保护：429 限流自动冷却 5 分钟
+ * 3. 429 限流保护：429 限流自动冷却（退避阶梯 30s → 1min → 2min → 5min → 10min 封顶，成功一次即清零）
  * 4. 局域网访问控制：继承相同的白名单与回环检查机制
  * 5. SSE 增量转换：将 Trae SOLO 累积流转换为标准 Anthropic 流式 SSE 或一次性 JSON 响应
  */
@@ -18,9 +18,12 @@ import { fetchTraeQuota } from './traeClient.js';
 import { gatewayHostSuggestion } from './tenrouter.js';
 import { logger, summarizeAttempts } from './logger.js';
 import { t } from './i18n.js';
+import { formatDurationMs } from './formatDuration.js';
 
 const TRAE_SOLO_BASE = 'https://solo.trae.cn/api/remote/v1';
 const ACCOUNT_COOLING_MS = 5 * 60_000;
+// 429 退避阶梯：30s → 1min → 2min → 5min → 10min 封顶（与 ZCode 网关同款）。
+const RATE_LIMIT_BACKOFF_MS = [30_000, 60_000, 120_000, 300_000, 600_000];
 const MAX_ATTEMPTS = 5;
 const UPSTREAM_TIMEOUT_MS = 600_000;
 const UPSTREAM_CONNECT_MS = 15_000;
@@ -38,12 +41,15 @@ export async function setGatewayEnabled(v) {
   if (on) {
     dead.clear();
     cooling.clear();
+    coolingStreak.clear();
   }
 }
 
 // ── 账号轮换状态（进程内） ──
 
 const cooling = new Map();  // accountId → 冷却截止(ms)
+// accountId → 连续 429 次数（成功即清零）：决定本次 429 走到退避阶梯的哪一档
+const coolingStreak = new Map();
 const dead = new Map();     // accountId → 拉黑时的凭据指纹（指纹变化 = 已重新授权/导入，自动复活）
 const weights = new Map();  // accountId → { w: 剩余积分权重, at: 查询时刻 }
 const WEIGHT_TTL_MS = 10 * 60_000;  // 超过 10 分钟重查额度，让积分变化反映到分流上
@@ -64,6 +70,16 @@ function markCooling(id, ms = ACCOUNT_COOLING_MS) {
   cooling.set(id, Date.now() + ms);
 }
 
+function markRateLimited(id) {
+  const streak = (coolingStreak.get(id) || 0) + 1;
+  coolingStreak.set(id, streak);
+  const ms = RATE_LIMIT_BACKOFF_MS[Math.min(streak, RATE_LIMIT_BACKOFF_MS.length) - 1];
+  cooling.set(id, Date.now() + ms);
+  return ms;
+}
+
+function resetCoolingStreak(id) { coolingStreak.delete(id); }
+
 function markDead(account) {
   dead.set(account.id, credFingerprint(account));
   cooling.set(account.id, Number.MAX_SAFE_INTEGER);
@@ -75,12 +91,20 @@ function isDead(account) {
   if (fp === credFingerprint(account)) return true;
   dead.delete(account.id);
   cooling.delete(account.id);
+  coolingStreak.delete(account.id);
   logger.info('TRAE-GW', '{label} 凭据已更新,解除拉黑', { label: account.name || account.uid || account.id });
   return false;
 }
 
+/**
+ * 测试专用：把某账号的冷却截止拨到指定时间戳（0 = 立即到期，用于快进冷却窗口）。
+ * 生产不走这里——冷却窗口只能自然到期。与 zcodeGateway.__setCoolingForTests 同款。
+ */
+export function __setCoolingForTests(accountId, untilMs) { cooling.set(accountId, untilMs); }
+
 export function __resetForTests() {
   cooling.clear();
+  coolingStreak.clear();
   dead.clear();
   weights.clear();
   swrr.clear();
@@ -397,9 +421,11 @@ export async function handleGateway(req, res) {
     }
 
     if (sessionRes.status === 429) {
-      markCooling(account.id, ACCOUNT_COOLING_MS);
-      attempted.push({ account: label, ok: false, error: '429 限流，冷却 5 分钟' });
-      logger.info('TRAE-GW', '{label} 429 限流,冷却 5 分钟', { label });
+      const ms = markRateLimited(account.id);
+      const dur = formatDurationMs(ms);
+      const streak = coolingStreak.get(account.id);
+      attempted.push({ account: label, ok: false, error: `429 限流，冷却 ${dur}` });
+      logger.info('TRAE-GW', '{label} 429 限流,冷却 {dur}(连续第 {streak} 次)', { label, dur, streak });
       continue;
     }
 
@@ -446,11 +472,22 @@ export async function handleGateway(req, res) {
 
     if (!eventsRes.ok || !eventsRes.body) {
       cleanupSession();
+      // 事件流被 429（会话建好了、拉流被限流）同样按阶梯冷却——不判的话这个号
+      // 下一轮会话创建仍从 30 秒档起算，继续撞限流
+      if (eventsRes.status === 429) {
+        const ms = markRateLimited(account.id);
+        const dur = formatDurationMs(ms);
+        const streak = coolingStreak.get(account.id);
+        attempted.push({ account: label, ok: false, error: `429 限流（事件流），冷却 ${dur}` });
+        logger.info('TRAE-GW', '{label} 429 限流(事件流),冷却 {dur}(连续第 {streak} 次)', { label, dur, streak });
+        continue;
+      }
       attempted.push({ account: label, ok: false, error: `事件流 HTTP ${eventsRes.status}` });
       continue;
     }
 
     stats.lastAccount = label;
+      resetCoolingStreak(account.id);   // 成功一次即清零连续 429 计数
       logger.debug('TRAE-GW', '{label} 连接成功,开始流式输出 ({model})', { label, model: requestedModel });
 
     // 成功建立流，按 isStream 分发
@@ -691,7 +728,18 @@ export async function gatewayStatus() {
       lastAccount: stats.lastAccount,
     },
     accounts: all.length,
+    // cooling 保持字符串数组（历史形状）；冷却细节走 coolingDetail（与 zcodeGateway 同款）
     cooling: all.filter((a) => !isDead(a) && (cooling.get(a.id) || 0) > now).map((a) => a.name || a.uid || a.id),
+    coolingDetail: all.filter((a) => !isDead(a) && (cooling.get(a.id) || 0) > now)
+      .map((a) => {
+        const remainMs = (cooling.get(a.id) || 0) - now;
+        return {
+          id: a.id, name: a.name || a.uid || a.id,
+          remainMs, remainSeconds: Math.max(0, Math.ceil(remainMs / 1000)),
+          remainText: formatDurationMs(remainMs),
+          streak: coolingStreak.get(a.id) || 0,
+        };
+      }),
     dead: all.filter((a) => isDead(a)).map((a) => a.name || a.uid || a.id),
     endpoint: '/gateway/trae/v1/messages',
   };

package/src/workbuddyLocal.js

@@ -99,6 +99,46 @@ export function currentWorkbuddyUid() {
   return readWorkbuddySessions().accounts.find((a) => a.current)?.uid || null;
 }
 
+// 客户端可执行文件候选（VS Code 系安装惯例：%LOCALAPPDATA%\Programs\<名>\<名>.exe）。
+// 「已安装」只认 exe：凭据目录不能当依据——CreditDaddy 自己切换时 mkdir 也会把目录建出来
+// （无客户端时静默切换留下的空壳目录），拿目录判断会把没装客户端的机器误报成已安装。
+const CLIENT_CANDIDATES = process.platform === 'win32'
+  ? [
+      ['Programs', 'CodeBuddy', 'CodeBuddy.exe'],
+      ['Programs', 'WorkBuddy', 'WorkBuddy.exe'],
+      ['CodeBuddy', 'CodeBuddy.exe'],
+      ['WorkBuddy', 'WorkBuddy.exe'],
+    ]
+  : process.platform === 'darwin'
+    ? ['/Applications/CodeBuddy.app', '/Applications/WorkBuddy.app']
+    : [];
+
+/**
+ * 本机 WorkBuddy / CodeBuddy 客户端探测（面板「客户端：未安装」提示与切换拦截用）。
+ * Linux / NAS 没有已知安装路径，退而用「凭据目录里有会话文件」判定——
+ * 会话文件只有客户端登录后才写，纯 Token 导入的机器不会产生。
+ */
+export function detectWorkbuddy() {
+  const dir = workbuddyAuthDir();
+  let dirExists = false;
+  try { dirExists = fs.statSync(dir).isDirectory(); } catch { /* 未安装 / 从未登录 */ }
+  let clientPath = null;
+  for (const cand of CLIENT_CANDIDATES) {
+    const p = process.platform === 'win32' && Array.isArray(cand)
+      ? path.join(process.env.LOCALAPPDATA || path.join(os.homedir(), 'AppData', 'Local'), ...cand)
+      : cand;
+    try { if (fs.existsSync(p)) { clientPath = p; break; } } catch { /* 下一个候选 */ }
+  }
+  const sessions = dirExists ? readWorkbuddySessions() : { accounts: [] };
+  return {
+    dir,
+    dirExists,
+    clientPath,
+    installed: Boolean(clientPath) || (CLIENT_CANDIDATES.length === 0 && sessions.accounts.length > 0),
+    signedIn: sessions.accounts.some((a) => a.current),
+  };
+}
+
 /**
  * 把账号写成 WorkBuddy 当前会话（切换账号）。需要账号带有导入时保存的会话信息。
  * @returns {{ file, previousUid }}

package/src/zcodeAutoClaim.js

@@ -67,7 +67,10 @@ export async function zcodeAutoClaim(account) {
     // 第 2 步：先试不带验证码
     try {
       const r = await claimPlan(account, plan.planId, {});
-      claims.push({ plan: planName, via: 'direct', ok: true });
+      // startsAt 是「到账时间」：活动权益常带 effective_at（如当天 23:00），
+      // 领到了但余额里还没有——不把这个时间带出去，面板只能显示冷冰冰的
+      // 「待生效」，用户以为没领上。
+      claims.push({ plan: planName, via: 'direct', ok: true, startsAt: r?.startsAt || null });
       anySuccess = true;
             logger.info('ZCODE-CLAIM', '{label} 已领取「{plan}」(免验证码){grants}', { label, plan: planName, grants: grantsText(plan) ? ':' + grantsText(plan) : '' });
       continue;
@@ -97,8 +100,8 @@ export async function zcodeAutoClaim(account) {
       const cfg = await fetchCaptchaConfig();
       if (!cfg.enabled || !cfg.sceneId) throw new Error('验证码配置不可用');
       const { captchaParam, region } = await captchaProvider(cfg);
-      await claimPlan(account, plan.planId, { captchaParam, region });
-      claims.push({ plan: planName, via: 'captcha', ok: true });
+      const r2 = await claimPlan(account, plan.planId, { captchaParam, region });
+      claims.push({ plan: planName, via: 'captcha', ok: true, startsAt: r2?.startsAt || null });
       anySuccess = true;
             logger.info('ZCODE-CLAIM', '{label} 已领取「{plan}」(静默验证码){grants}', { label, plan: planName, grants: grantsText(plan) ? ':' + grantsText(plan) : '' });
     } catch (e2) {
@@ -128,8 +131,22 @@ export async function zcodeAutoClaim(account) {
   const status = names.length ? 'checked-in'
     : claims.every((c) => c.already) ? 'already'
     : 'failed';
+  // 到账时间：新领的活动若带 effective_at（当天稍晚才进余额），把时间写进消息——
+  // 否则面板只显示「待生效」，用户无法判断是没领上还是要等。
+  const fmtAt = (iso) => {
+    if (!iso) return '';
+    const d = new Date(iso);
+    if (Number.isNaN(d.getTime())) return '';
+    const p2 = (n) => String(n).padStart(2, '0');
+    const now = new Date();
+    const sameDay = d.toDateString() === now.toDateString();
+    const hm = `${p2(d.getHours())}:${p2(d.getMinutes())}`;
+    return sameDay ? hm : `${d.getMonth() + 1}-${p2(d.getDate())} ${hm}`;
+  };
+  const starts = claims.filter((c) => c.ok && c.startsAt && new Date(c.startsAt).getTime() > Date.now());
+  const startsText = starts.length ? `（${fmtAt(starts[0].startsAt)} 到账）` : '';
   const message = status === 'checked-in'
-    ? `已领取：${names.join('、')}${claims.some((c) => c.via === 'captcha') ? '（静默验证码）' : ''}`
+    ? `已领取：${names.join('、')}${claims.some((c) => c.via === 'captcha') ? '（静默验证码）' : ''}${startsText}`
     : status === 'already' ? '活动均已领取过'
     : needManual ? `有 ${claims.filter((c) => c.needManual).length} 个活动需要验证码，请在面板手动领取`
     : `领取失败：${claims.filter((c) => !c.ok).map((c) => c.message).filter(Boolean)[0] || '未知错误'}`;

package/src/zcodeClient.js

@@ -502,6 +502,8 @@ export function normalizeBalance(balanceResp) {
         const pending = Boolean(startsAt && new Date(startsAt).getTime() > now);
         return {
           name: e.show_name || e.entitlement_id || '活动额度',
+          // 活动名（如「ZCode 1024 Dev Day」）：未生效权益靠它告诉用户「这是哪个活动发的」
+          planName: String(p.name || p.plan_id || '') || null,
           unit: 'Token',
           total: grant,
           used: 0,
@@ -514,6 +516,10 @@ export function normalizeBalance(balanceResp) {
       }),
   );
   const parts = balanceParts.length ? balanceParts : entitlementParts;
+  // 未生效权益汇总：面板用它显示「待生效 N Token · 活动名 · HH:MM 到账」，
+  // 而不是把 0 当成「已用完」——活动权益通常要到生效点才进 balances
+  const pendingParts = parts.filter((p) => p.pending);
+  const pendingAt = pendingParts.map((p) => p.startsAt).filter(Boolean).sort()[0] || null;
   // 套餐有效期卡片上显示在「有效期至」列之外，这里记录最近的套餐结束时间
   const planEndsAt = plans.map((p) => epochMs(p.ends_at)).filter(Boolean).sort()[0] || null;
   const sum = (k) => parts.reduce((s, p) => s + (p[k] || 0), 0);
@@ -523,6 +529,8 @@ export function normalizeBalance(balanceResp) {
     plan: plans[0]?.name || plans[0]?.plan_id || null,
     planEndsAt,
     parts,
+    pendingTotal: pendingParts.reduce((s, p) => s + (p.total || 0), 0),
+    pendingAt,
     empty: parts.length === 0 && plans.length === 0,
     exceeded: false,
   };