balance-dsh 1.0.8 → 1.0.10

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 生效的
@@ -355,16 +401,35 @@ function recentDayKeys(days, now = Date.now()) {
 }
 
 async function buildUsageStats(ctx, config) {
-  const persistence = typeof ctx.get === "function" ? ctx.get("sessionPersistence") : void 0;
-  if (persistence == null || typeof persistence.list !== "function" || typeof persistence.inspect !== "function") {
-    throw new Error("会话持久化服务不可用");
-  }
+  const getService = typeof ctx.get === "function" ? ctx.get.bind(ctx) : null;
+  const controller = getService?.("sessionController");
+  const persistence = getService?.("sessionPersistence");
   const now = Date.now();
   const cutoffMs = now - (STATS_DAYS - 1) * 24 * 60 * 60 * 1000;
-  const headers = await persistence.list();
-  const selected = [...headers]
-    .filter((header) => header && typeof header.id === "string")
-    .sort((a, b) => Number(b.createdAt ?? 0) - Number(a.createdAt ?? 0))
+  // 列举会话要跨 DSH 版本：新版 sessionController.list() 返回 { items:[{ sessionId, updatedAt }] }
+  // （已按活动降序，字段叫 sessionId）；旧版 sessionPersistence.list() 返回 [{ id, createdAt }]
+  // （字段叫 id）。旧版 persistence 还提供 inspect()，新版改成 handle-based 的 open()，
+  // 所以读取一律走已做三级兼容的 inspectPersistedSession，不再直接调 persistence.inspect。
+  let sessionIds = [];
+  let availableSessions = 0;
+  if (typeof controller?.list === "function") {
+    const listed = await controller.list({});
+    const items = Array.isArray(listed) ? listed : (listed?.items ?? []);
+    sessionIds = items.map((item) => (item && typeof item === "object" ? item.sessionId ?? item.id : void 0));
+    availableSessions = sessionIds.length;
+  } else if (typeof persistence?.list === "function") {
+    const headers = await persistence.list();
+    const list = Array.isArray(headers) ? headers : [];
+    availableSessions = list.length;
+    sessionIds = [...list]
+      .filter((header) => header && typeof header.id === "string")
+      .sort((a, b) => Number(b.createdAt ?? 0) - Number(a.createdAt ?? 0))
+      .map((header) => header.id);
+  } else {
+    throw new Error("会话持久化服务不可用");
+  }
+  const selected = sessionIds
+    .filter((id) => typeof id === "string" && id !== "")
     .slice(0, STATS_MAX_SESSIONS);
   const daily = new Map();
   let cursor = 0;
@@ -372,16 +437,16 @@ async function buildUsageStats(ctx, config) {
   let failedSessions = 0;
   const worker = async () => {
     while (cursor < selected.length) {
-      const header = selected[cursor++];
+      const sessionId = selected[cursor++];
       try {
-        const inspected = await persistence.inspect(header.id);
+        const inspected = await inspectPersistedSession(ctx, sessionId);
         collectDailyUsage(
           inspected?.events ?? [],
           config,
           detectModel({ requestHeader: () => void 0, requestContext: () => void 0, events: inspected?.events ?? [] }),
           cutoffMs,
           daily,
-          header.id,
+          sessionId,
         );
         scannedSessions++;
       } catch {
@@ -422,19 +487,19 @@ async function buildUsageStats(ctx, config) {
     rangeDays: STATS_DAYS,
     scannedSessions,
     failedSessions,
-    availableSessions: headers.length,
-    limited: headers.length > selected.length,
+    availableSessions,
+    limited: availableSessions > selected.length,
     generatedAt: now,
   };
 }
 
 /** 汇总某会话的所有子代理后代会话的 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 +531,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 +613,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 +661,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 +775,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.10",
   "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
     }
   },