dsh-desktop-statusbar 2.0.2 → 2.1.1

package/lib/balance-api.js

@@ -137,7 +137,9 @@ const adapters = [
        鉴权：Authorization: Bearer <api_key> —— 注意与 coding plan 配额的「裸 key」不同
        响应 { code, msg, data: { availableBalance, balance, rechargeAmount, totalSpendAmount } }
        code === 200 才算成功；可用余额 = availableBalance ?? balance，单位 CNY。 */
-    keys: ['zhipu', 'glm', 'bigmodel'],
+    /* 'zai' 是 DSH 里 z.ai（智谱国际站）的 provider 名，模型为 glm-5.3-*（本机会话实测 2026-10-01）；
+       模型名本身也含 glm，这里补上 provider 名，免得模型改名后就匹配不到。 */
+    keys: ['zhipu', 'glm', 'bigmodel', 'zai'],
     refs: ['ZAI_API_KEY', 'ZHIPU_API_KEY', 'GLM_API_KEY', 'BIGMODEL_API_KEY'],
     request: (key) => ({
       /* 实测 2026-09-27：open.bigmodel.cn 与 api.z.ai 同后端，两个域名 + Bearer/裸 key 四种组合都返回 code=200 */
@@ -170,6 +172,11 @@ const adapters = [
        响应 { success, data: { level, limits: [{ type: TOKENS_LIMIT|CREDIT_LIMIT, percentage, ... }] } }
        percentage 就是配额使用率，直接当百分比显示，**不换算金额**。注意这是 coding plan
        配额，不是按量计费的现金余额 —— 现金余额只有控制台接口能给，需要网页登录态。 */
+    /* 它是 `zhipu` 的**第二条独立的路**，不是替代品：两者 keys 完全重叠，而 `adapterFor`
+       按数组顺序命中即返回，所以配额这条永远选不中 —— 必须靠 `fallbackOf` 显式挂成
+       链上的下一跳（见 adaptersFor），否则 coding plan 用户永远只看到现金余额那条路
+       的结果（实测症状：余额显示 ¥- 或 ¥0.00）。 */
+    fallbackOf: 'zhipu',
     keys: ['zhipu', 'glm', 'bigmodel'],
     refs: ['ZHIPU_API_KEY', 'GLM_API_KEY', 'BIGMODEL_API_KEY'],
     request: (key, base) => ({
@@ -197,28 +204,219 @@ const adapters = [
       if (percent === null) return null;
       return { currency: '', total: percent.toFixed(2) + '%', parts: [], isPercent: true };
     }
+  },
+  {
+    id: 'workbuddy',
+    label: 'WorkBuddy',
+    /* WorkBuddy / WorkBuddy AI 的余额是**积分**，且不在这里查：凭据由 connect 类插件
+       解密持有，上游鉴权与计费解析都由它实现，它把结论挂在本机 Web 路由上。
+       所以本适配器不发外部请求，只读那条本机路由（local: true），本插件不碰凭据、
+       也不重复实现上游协议。
+       实测 2026-10-07：国际版路由返回 credits.total = 177。
+       实测 2026-10-12：`credits.total` 这个口径在**新旧两个插件上一致**
+       （新 dsh-connect-workbuddy 的 /usage 与老 dsh-workbuddy-connect 的 /status
+       都是顶层 credits.total），所以解析器一份就够，变的只是**路径**。 */
+    local: true,
+    keys: ['workbuddy', 'codebuddy'],
+    refs: [],
+    /* 候选路由**按新旧插件依次尝试**（见 pathsOf）：第一个能解析出结果的胜出。
+       为什么不能只留一条：两个插件是不同作者的两套实现，用户可能同时装着、
+       也可能只用其中一个（实测本机同时装着，当前生效的是新插件）。
+       为什么新插件要带 region：它的路由按区域分账号（cn / global），
+       不带参数默认 cn，区域给错会拿到另一个区域的账号余额。 */
+    pathsOf: (provider) => {
+      const text = String(provider === undefined || provider === null ? '' : provider).toLowerCase()
+      /* 国际版：新插件用 region=global，老插件用 /ai/status；
+         实测 provider 名有三种形态：workbuddy（国内）、workbuddy-ai、workbuddy-global */
+      const global = text.includes('-ai') || text.includes('global')
+      const region = global ? 'global' : 'cn'
+      return [
+        '/plugins/dsh-connect-workbuddy/usage?region=' + region,
+        global ? '/plugins/dsh-workbuddy-connect/ai/status' : '/plugins/dsh-workbuddy-connect/status'
+      ]
+    },
+    parse: (data) => {
+      const credits = data !== null && data !== undefined && data.credits !== undefined ? data.credits : null;
+      if (credits === null) return null;
+      const total = Number(credits.total);
+      if (Number.isFinite(total) !== true) return null;
+      /* 积分不是货币：currency 留空让客户端不拼货币符号；不给 parts，
+         气泡因此只显示总数、不列套餐构成。 */
+      return { currency: '', total: formatCredits(total), parts: [], isCredits: true };
+    }
   }
 ];
 
-/** 会话线索（provider + 模型名）里出现这些词就认为是哪一家。 */
+/**
+ * 积分数值的显示文本：整数按千分位分组，非整数保留两位小数。
+ * 积分不带货币单位，所以这里不拼 ¥ / $（与 formatBalanceAmount 的区别就在这）。
+ */
+function formatCredits(value) {
+  const n = Number(value);
+  if (Number.isFinite(n) !== true) return '0';
+  const rounded = Number.isInteger(n) ? n : Math.round(n * 100) / 100;
+  const text = String(rounded);
+  const dot = text.indexOf('.');
+  const whole = dot < 0 ? text : text.slice(0, dot);
+  const rest = dot < 0 ? '' : text.slice(dot);
+  return whole.replace(/\B(?=(\d{3})+(?!\d))/g, ',') + rest;
+}
+
+/**
+ * 会话线索 → 平台适配器。
+ *
+ * **provider 优先于模型名**：先只用 provider 匹配一轮，没命中才把模型名并进来匹配。
+ * 为什么必须分两轮：模型名常常带别家厂商的词，混在一起匹配会让会话被认成错的平台 ——
+ * 实测 2026-10-07，WorkBuddy AI 会话（provider=workbuddy-ai）跑 deepseek-v4.1-flash-sg，
+ * 旧写法把 "deepseek" 从模型名里匹到，于是拿 DeepSeek 的 API key 去问官方余额接口，
+ * 稳定拿到 HTTP 401，底栏余额永远是横杠。provider 是会话实际走的计费通道，以它为准。
+ */
 function adapterFor(provider, model) {
-  const text = (String(provider === undefined || provider === null ? '' : provider) + ' '
-    + String(model === undefined || model === null ? '' : model)).toLowerCase();
-  if (text.trim().length === 0) return null;
-  for (let i = 0; i < adapters.length; i += 1) {
-    const one = adapters[i];
-    for (let k = 0; k < one.keys.length; k += 1) {
-      if (text.includes(one.keys[k])) return one;
+  const providerText = String(provider === undefined || provider === null ? '' : provider).toLowerCase().trim();
+  const modelText = String(model === undefined || model === null ? '' : model).toLowerCase().trim();
+  const pick = (text) => {
+    if (text.length === 0) return null;
+    for (let i = 0; i < adapters.length; i += 1) {
+      const one = adapters[i];
+      for (let k = 0; k < one.keys.length; k += 1) {
+        if (text.includes(one.keys[k])) return one;
+      }
     }
+    return null;
+  };
+  const byProvider = pick(providerText);
+  if (byProvider !== null) return byProvider;
+  /* provider 还没上报（或名字认不出）时才退回模型名，保持对自定义 provider 名的兼容 */
+  return pick(modelText);
+}
+
+/**
+ * 会话线索 → **要按顺序依次尝试的适配器链**（第一个是主选，其后是回退）。
+ *
+ * 为什么需要"链"而不是单个适配器：智谱有两种**互相独立**的计费通道，接口与响应结构
+ * 完全不同，而且 keys 完全重叠（都是 zhipu/glm/bigmodel）：
+ *   · 按量计费 → `zhipu`（现金余额，控制台接口，Bearer）
+ *   · coding plan → `zhipu-quota`（配额百分比，monitor 接口，裸 key）
+ * 单选的 `adapterFor` 按数组顺序命中即返回，`zhipu-quota` 永远选不中 —— 于是 coding plan
+ * 用户永远拿不到配额，客户端 `isPercent` 分支恒不可达（用户实测：智谱 coding plan 显示"余额 ¥-"）。
+ * 用户明确要求**按实际调用的模型/凭据决定显示余额还是配额**，两种是独立的路、不能二选一。
+ *
+ * 实现就是"两个端点都试、按返回结构决定显示什么"：调用方按这个顺序逐个请求，**第一个
+ * 解析出结果的胜出**（余额接口对 coding plan 返回不了 code=200 的余额 → parse 给 null →
+ * 自动落到配额；反过来按量计费账号在配额接口上拿不到 limits → 也给 null）。
+ * 判定完全交给各家自己的 `parse`（那是已经核实过上游协议的**唯一**权威），不在这里另立规则。
+ *
+ * 非智谱平台没有 `fallbackOf`，链长恒为 1 —— 行为与旧版逐字节一致。
+ */
+function adaptersFor(provider, model) {
+  const first = adapterFor(provider, model);
+  if (first === null) return [];
+  const chain = [first];
+  /* 跟着 `fallbackOf` 往下走：它的语义是"**我是**某条适配器的后备"
+     （`zhipu-quota.fallbackOf === 'zhipu'` = 配额是余额的后备），
+     所以这里找的是"谁声明了以当前这条为后备"，而不是当前这条的后备。
+     visited 兜住配置写错造成的环（否则会转不出来）。 */
+  const visited = new Set([first.id]);
+  let current = first;
+  for (;;) {
+    const next = adapters.filter((one) => one.fallbackOf === current.id && visited.has(one.id) !== true)[0];
+    if (next === undefined) break;
+    visited.add(next.id);
+    chain.push(next);
+    current = next;
   }
-  return null;
+  return chain;
+}
+
+/* ------------------------------------------------- connect 类插件的通用发现 */
+
+/**
+ * 允许探测的**动词白名单**。只探"读余额/状态"这一类，且都是只读语义。
+ *
+ * 为什么不放开：可行性调研实测到同族插件还有 `refresh` / `probe` / `checkin` /
+ * `transfer` / `oauth` / `__save` / `pool` 等路由，其中多数是**写操作**
+ * （签到会真的领奖励、probe 会真的花积分、__save 会写配置）。
+ * "GET 无副作用"是路由所有者的选择、不是保证 —— 所以这里只列明确只读的三条。
+ */
+const LOCAL_READONLY_VERBS = ['usage', 'status', 'account-credits'];
+
+/** 永不探测的路径：`/plugins/events` 是 SSE，GET 会挂住不返回（实测 15s 超时）。 */
+const LOCAL_NEVER_PROBE = ['/plugins/events'];
+
+/**
+ * 从一张路由表（Map）里挑出"形如 `/plugins/<id>/<动词>` 且动词在白名单里"的路径。
+ *
+ * 只做筛选、不发请求；调用方负责确认可达性。返回按插件 id 去重、保留全部动词候选。
+ * `table` 不是 Map（宿主改实现 / 字段被改名）时返回空数组 —— 调用方据此回退声明表。
+ */
+function pluginRoutesIn(table) {
+  const out = [];
+  if (table === null || table === undefined || typeof table.forEach !== 'function') return out;
+  table.forEach((value, key) => {
+    if (typeof key !== 'string') return;
+    if (LOCAL_NEVER_PROBE.indexOf(key) >= 0) return;
+    const hit = /^\/plugins\/([^/]+)\/([^/]+)$/.exec(key);
+    if (hit === null) return;
+    if (LOCAL_READONLY_VERBS.indexOf(hit[2]) < 0) return;
+    out.push({ pluginId: hit[1], verb: hit[2], path: key });
+  });
+  return out;
 }
 
-/** 会话模型属于 DeepSeek 时，优先走官方账号通道（账号登录没有 API key 也能查）。 */
-function isDeepSeek(provider, model) {
-  const text = (String(provider === undefined || provider === null ? '' : provider) + ' '
-    + String(model === undefined || model === null ? '' : model)).toLowerCase();
-  return text.includes('deepseek');
+/**
+ * 本机路由响应 → 余额/配额。**只认明确命名的结构，绝不自由猜字段。**
+ *
+ * 为什么这条约束是硬性的：可行性调研实测过同族三个插件的返回结构 ——
+ *   · workbuddy：顶层 `credits.total` **就是余额**
+ *   · qoder    ：`usage.userQuota.total` 是**配额容量**，`remaining` 才是余额（语义正好相反）
+ *   · trae     ：同时有 `total` / `available` / `workAvailable`
+ * 若写成"找任意名为 total 的数字"，在 qoder 上会稳定显示一个**看起来合理的错数**。
+ * 本文件第 253 行那条实测教训（拿错平台的 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"这条防线。
+ */
+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, isDeepSeek };
+export { adapters, adapterFor, adaptersFor, pluginRoutesIn, balanceFromLocalDoc, LOCAL_READONLY_VERBS, LOCAL_NEVER_PROBE };