balance-dsh 1.0.8 → 1.0.9

package/lib/index.js

@@ -1,9 +1,55 @@
 // dsh 插件：余额与当前会话消耗
 // 服务端：注册 /balance 命令 + /balance-api HTTP 接口（供 Web UI 插件直接渲染）
-import { inspectApiRemoteSession } from "@deepseek-ai/dsh-api-remotes";
-
 export const name = "balance-dsh";
 
+let legacySessionInspector;
+let legacySessionInspectorLoaded = false;
+
+/**
+ * 跨 DSH 版本读取会话，且不在模块加载阶段绑定可能被删除的命名导出。
+ * 新版优先使用 sessionController；其次直接使用 sessionQuery；旧版最后回退
+ * @deepseek-ai/dsh-api-remotes.inspectApiRemoteSession。接口全部不可用时抛错，
+ * 由子代理统计调用方跳过该会话，不影响余额插件本身启动。
+ */
+async function inspectPersistedSession(ctx, sessionId) {
+  const getService = typeof ctx?.get === "function" ? ctx.get.bind(ctx) : null;
+
+  const sessionController = getService?.("sessionController");
+  if (typeof sessionController?.inspect === "function") {
+    return sessionController.inspect(sessionId);
+  }
+
+  const sessionQuery = getService?.("sessionQuery");
+  if (typeof sessionQuery?.observeSession === "function") {
+    const observation = await sessionQuery.observeSession(sessionId, { projectionMode: "none" });
+    try {
+      return {
+        meta: observation.header,
+        events: [...(observation.events ?? [])],
+      };
+    } finally {
+      observation?.[Symbol.dispose]?.();
+    }
+  }
+
+  if (!legacySessionInspectorLoaded) {
+    legacySessionInspectorLoaded = true;
+    try {
+      const remotes = await import("@deepseek-ai/dsh-api-remotes");
+      if (typeof remotes.inspectApiRemoteSession === "function") {
+        legacySessionInspector = remotes.inspectApiRemoteSession;
+      }
+    } catch {
+      // 新版可能不再提供旧模块或旧导出；由下方统一处理。
+    }
+  }
+  if (typeof legacySessionInspector === "function") {
+    return legacySessionInspector(ctx, sessionId);
+  }
+
+  throw new Error("当前 DSH 未提供兼容的持久化会话读取接口");
+}
+
 const DEEPSEEK_BALANCE_URL = "https://api.deepseek.com/user/balance";
 
 // DeepSeek 目录价（人民币/百万 tokens），按模型区分。默认取 2026-08-17 生效的
@@ -429,12 +475,12 @@ async function buildUsageStats(ctx, config) {
 }
 
 /** 汇总某会话的所有子代理后代会话的 usage（跨时段精确计费，与主会话分开）。
- *  @param ctx DSH 上下文（用于 subagents 服务 + inspectApiRemoteSession）
+ *  @param ctx DSH 上下文（用于 subagents 服务 + 跨版本持久化会话读取）
  *  @param sessionId 父会话 id
  *  @param config 插件配置
  *  说明：子代理会话结束后会被从 live 会话 store 移除，故不能用 store.list() 枚举；
  *  这里用 subagents 服务的 listDescendants（含已结束/嵌套子代理）取后代 id，
- *  再对每个子代理用 inspectApiRemoteSession 读取持久化事件计算 usage。无子代理返回 {usage:null,debug}。 */
+ *  再对每个子代理用兼容读取器取得持久化事件并计算 usage。无子代理返回 {usage:null,debug}。 */
 async function sumSubagentUsage(ctx, sessionId, config) {
   if (ctx == null || sessionId === "") return { usage: null };
   // 用 ctx.get("subagents") 安全获取（非注入，可空）；不要直接读 ctx.subagents 属性，
@@ -466,7 +512,7 @@ async function sumSubagentUsage(ctx, sessionId, config) {
   let count = 0;
   for (const id of childIds) {
     try {
-      const inspected = await inspectApiRemoteSession(ctx, id);
+      const inspected = await inspectPersistedSession(ctx, id);
       const events = Array.isArray(inspected?.events) ? inspected.events : [];
       const u = sumEventsUsage(events, config);
       if (u === null || u === void 0) continue;
@@ -548,6 +594,11 @@ function writeJson(res, status, payload) {
 
 export function apply(ctx, config = {}) {
   let statsCache = { expiresAt: 0, value: null };
+  // DSH 热重载(reload)时会重新 apply 本插件；若旧实例注册的 /balance-api 路由
+  // 未被清理，新实例再次 register 就会抛出 duplicate route，导致整个 apply 失败、
+  // 余额不显示。这里收集每次 register 返回的 disposer，并在插件卸载(ctx.on("dispose"))
+  // 时统一清理，保证重复 apply 不冲突。
+  const disposers = [];
 
   // 1) /balance 命令：保留命令行查询入口
   ctx.inject(["commands"], (commandCtx) => {
@@ -591,7 +642,7 @@ export function apply(ctx, config = {}) {
   //      /balance-api 不注册（404）。改为只声 inject ["webServer"]，子代理在
   //      sumSubagentUsage 内部用可空 ctx.get("subagents") 获取。
   ctx.inject(["webServer"], (httpCtx) => {
-    httpCtx.webServer.register({
+    const balanceRouteDisposer = httpCtx.webServer.register({
       kind: "exact",
       path: "/balance-api",
       handler: async (req, res) => {
@@ -705,6 +756,15 @@ export function apply(ctx, config = {}) {
         }
       },
     });
-
+    if (typeof balanceRouteDisposer === "function") disposers.push(balanceRouteDisposer);
   });
+
+  // 插件卸载/dispose 时清理本实例注册的路由，避免 DSH 热重载时旧实例残留
+  // 的路由导致新实例 register /balance-api 抛 duplicate route、插件启动失败。
+  const release = () => {
+    for (const d of disposers) {
+      try { d(); } catch { /* ignore */ }
+    }
+  };
+  if (typeof ctx.on === "function") ctx.on("dispose", release);
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "balance-dsh",
-  "version": "1.0.8",
+  "version": "1.0.9",
   "description": "查询 DeepSeek 余额并显示当前会话 token 消耗。DeepSeek 账户余额 + 会话 token/费用显示，含高峰/空闲峰谷计费。",
   "author": "Rannist",
   "type": "module",
@@ -26,10 +26,10 @@
       "patch": "./cordis.patch.yml"
     },
     "client": {
-      "platform": "web",
       "inject": [
         "@deepseek-ai/dsh-client-runtime"
       ],
+      "platform": "web",
       "immediately": true
     }
   },