dsh-desktop-statusbar 2.1.1 → 2.2.0

package/lib/balance-api.js

@@ -242,7 +242,38 @@ const adapters = [
       if (Number.isFinite(total) !== true) return null;
       /* 积分不是货币：currency 留空让客户端不拼货币符号；不给 parts，
          气泡因此只显示总数、不列套餐构成。 */
-      return { currency: '', total: formatCredits(total), parts: [], isCredits: true };
+      const out = { currency: '', total: formatCredits(total), parts: [], isCredits: true };
+      /* 套餐列表：新插件（dsh-connect-workbuddy）的 /usage 带 credits.packages，就是官方客户端
+         「积分套餐」那一栏的数据。老插件 /status 只有 credits.total，**不产生 rewards** ——
+         客户端据此保持现在的纯总数显示，不会因为字段缺失去猜。 */
+      const packages = Array.isArray(credits.packages) ? credits.packages : null;
+      if (packages === null) return out;
+      const rewards = [];
+      for (let i = 0; i < packages.length; i += 1) {
+        const one = packages[i];
+        if (one === null || one === undefined || typeof one !== 'object') continue;
+        const size = Number(one.size);
+        const remain = Number(one.remain);
+        /* size/remain 缺一个就没法算「已用/总量」：跳过这个包，
+           不让一条坏数据把整条气泡弄崩（旧插件/上游漂移都可能缺字段）。 */
+        if (Number.isFinite(size) !== true || Number.isFinite(remain) !== true || size < 0) continue;
+        const name = typeof one.packageName === 'string' ? one.packageName : '';
+        if (name.length === 0) continue;
+        const monthly = one.monthly === true;
+        rewards.push({
+          name,
+          /* 「已用」与官方同口径：总量 - 剩余（connect 插件已按 CapacitySize/Remain 算好）；
+             剩余超过总量这种异常数据夹住到 0，不显示负数。 */
+          used: Math.max(0, size - remain),
+          total: size,
+          /* 一次性包看到期日；月度包看「下期刷新」（官方对月度包不显示「到期」，别编造到期时间） */
+          expireAt: monthly ? null : dateOfMs(one.expiresAtMs),
+          refreshAt: monthly ? dateOfMs(one.cycleRefreshMs) : null,
+          monthly,
+        });
+      }
+      out.rewards = rewards;
+      return out;
     }
   }
 ];
@@ -262,6 +293,23 @@ function formatCredits(value) {
   return whole.replace(/\B(?=(\d{3})+(?!\d))/g, ',') + rest;
 }
 
+/**
+ * 毫秒时间戳 → 北京时间 'YYYY-MM-DD'（套餐气泡里的到期日/刷新日）。
+ *
+ * 为什么固定按北京时间算、不看运行环境时区：这个项目的所有「日」口径（峰谷计价、
+ * 签到日）都按北京时间；套餐到期时间给用户看的是**中国的日期**，跟着机器时区走
+ * 会在海外机器上差一天（服务端实测值如 1794056975000 = 北京时间 2026-11-07 21:09）。
+ * 取不到/非法值给 null —— 客户端显示「—」，绝不猜一个日期。
+ */
+function dateOfMs(value) {
+  const ms = Number(value);
+  if (Number.isFinite(ms) !== true || ms <= 0) return null;
+  const bj = new Date(ms + 8 * 3600 * 1000);
+  const month = String(bj.getUTCMonth() + 1).padStart(2, '0');
+  const day = String(bj.getUTCDate()).padStart(2, '0');
+  return bj.getUTCFullYear() + '-' + month + '-' + day;
+}
+
 /**
  * 会话线索 → 平台适配器。
  *
@@ -371,52 +419,17 @@ function pluginRoutesIn(table) {
  *   · qoder    ：`usage.userQuota.total` 是**配额容量**，`remaining` 才是余额（语义正好相反）
  *   · trae     ：同时有 `total` / `available` / `workAvailable`
  * 若写成"找任意名为 total 的数字"，在 qoder 上会稳定显示一个**看起来合理的错数**。
- * 本文件第 253 行那条实测教训（拿错平台的 key 去问余额，稳定 401）与
- * "宁可不显示，也不显示错数"是同一个原则。
+ * 与 workbuddy 适配器注释里那条实测教训（拿错平台的 key 去问余额，稳定 401）是同一个原则：
+ * 宁可不显示，也不显示错数。
  *
- * 抽取器按优先级排列，命中即返回；全部不命中返回 null（调用方显示空）。
- */
-const LOCAL_SHAPE_EXTRACTORS = [
-  /* workbuddy（新 /usage 与老 /status 都是这个形状）：顶层 credits.total 是余额 */
-  (data) => {
-    const credits = data !== null && data !== undefined ? data.credits : null;
-    if (credits === null || credits === undefined || typeof credits !== 'object') return null;
-    const total = Number(credits.total);
-    if (Number.isFinite(total) !== true) return null;
-    return { currency: '', total: formatCredits(total), parts: [], isCredits: true };
-  },
-  /* 配额百分比（智谱那条路的形状）：明确带 limits 数组才算，避免把别的 total 当配额 */
-  (data) => {
-    const inner = data !== null && data !== undefined && data.data !== null && data.data !== undefined
-      ? data.data
-      : null;
-    if (inner === null || inner === undefined || Array.isArray(inner.limits) !== true) return null;
-    let percent = null;
-    for (let i = 0; i < inner.limits.length; i += 1) {
-      const one = inner.limits[i];
-      if (one === null || one === undefined) continue;
-      const kind = typeof one.type === 'string' ? one.type.toUpperCase() : '';
-      if (kind !== 'TOKENS_LIMIT' && kind !== 'CREDIT_LIMIT') continue;
-      const value = Number(one.percentage);
-      if (Number.isFinite(value) !== true) continue;
-      if (percent === null || value > percent) percent = value;
-    }
-    if (percent === null) return null;
-    return { currency: '', total: percent.toFixed(2) + '%', parts: [], isPercent: true };
-  }
-];
-
-/**
- * 本机路由文档 → 余额。抽取器全部不命中就给 null（**不猜**）。
- * 导出是为了让测试能直接钉住"qoder 那种结构必须给 null"这条防线。
+ * 这条防线**唯一的实现处就是各适配器自己的 `parse`**：`index.js` 的 `queryLocalAdapter`
+ * 拿到本机路由响应后直接调 `adapter.parse(data)` —— workbuddy 只认 `credits.total`、
+ * 智谱配额只认带 `limits` 数组的形状，认不出一律给 null（既不猜字段，也不硬凑）。
+ * 回归测试（tests/verify-balance-api.js）打的就是这条生产路径。
+ *
+ * 历史上这里还有一套 `LOCAL_SHAPE_EXTRACTORS` + `balanceFromLocalDoc`：它只被测试引用、
+ * 生产从未调用（2026-10 删除）。测试打生产 `parse` 才是真防线 —— 打只给测试用的旁路函数，
+ * 绿了也不代表线上安全。
  */
-function balanceFromLocalDoc(data) {
-  if (data === null || data === undefined || typeof data !== 'object') return null;
-  for (let i = 0; i < LOCAL_SHAPE_EXTRACTORS.length; i += 1) {
-    const value = LOCAL_SHAPE_EXTRACTORS[i](data);
-    if (value !== null) return value;
-  }
-  return null;
-}
 
-export { adapters, adapterFor, adaptersFor, pluginRoutesIn, balanceFromLocalDoc, LOCAL_READONLY_VERBS, LOCAL_NEVER_PROBE };
+export { adapters, adapterFor, adaptersFor, pluginRoutesIn, LOCAL_READONLY_VERBS, LOCAL_NEVER_PROBE };