balance-dsh 1.3.1 → 2.0.0

package/lib/index.js

@@ -47,7 +47,7 @@ async function inspectPersistedSession(ctx, sessionId) {
     return legacySessionInspector(ctx, sessionId);
   }
 
-  throw new Error("当前 DSH 未提供兼容的持久化会话读取接口");
+  throw new Error("当前 DeepSeek Harness 未提供兼容的持久化会话读取接口");
 }
 
 const DEEPSEEK_BALANCE_URL = "https://api.deepseek.com/user/balance";
@@ -159,8 +159,10 @@ function isPeakAt(timeMs, config) {
   if (kind === "holiday") return false; // 法定节假日：全天按空闲计费
   const hour = d.getUTCHours();
   const weekday = d.getUTCDay(); // 0=周日 … 6=周六
-  const onDuty = weekdays.includes(weekday) || kind === "workday"; // 调休上班的周末按工作日算
-  return onDuty && windows.some((w) => hour >= w.start && hour < w.end);
+  // 官方定价页口径（2026-10 核对）：周一至周五为工作日高峰区，其余时段
+  // 「包括周末及中国法定节假日全天」均为空闲。所以**调休上班的周末也按周末处理**，
+  // 不参与峰谷判定（CN_CALENDAR 里的 workday 条目只保留作数据，不再影响计费）。
+  return weekdays.includes(weekday) && windows.some((w) => hour >= w.start && hour < w.end);
 }
 
 /** 把模型字符串规整到定价表键；未识别返回 ""（用 Flash 价兜底）。
@@ -343,7 +345,7 @@ function formatBalance(balance) {
 //                          | { status:"failed" }  （网络失败；官方明确“失败绝不变成 0 余额”）
 // client 是 AccountClientMetadata（version / locale / timezoneOffsetSeconds），
 // 缺字段会在平台侧组装请求头时抛错，所以三个字段都要给。
-const PLUGIN_VERSION = "1.3.1";
+const PLUGIN_VERSION = "2.0.0";
 
 /**
  * 读取官方账号余额并归一化成与 normalizeBalance 相同的形状。
@@ -629,6 +631,30 @@ const DEFAULT_SETTINGS = {
   showUsagePill: true, // 输入框下方「本会话消耗」胶囊
   showPeriodLabel: true, // 高峰/空闲标签
   showFab: true, // 收起态悬浮圆钮
+  // 2.0.0：余额行外观。classic = 早期那版「钱包图标 + 余额 + 金额」纯文字行；
+  // timeline = 现在的「meta 行 + 扁刻度条 + 金额」。
+  footerStyle: "timeline",
+  // 2.0.0：左侧导航栏里的插件设置入口（点一下直接打开插件配置页）
+  showSidebarEntry: true,
+  // 2.0.0：右键刷新时，余额行从左往右扫过一道泛光亮带
+  refreshWave: true,
+  // 侧栏「余额与用量」入口是否已被点开过。false = 显示红点。
+  // 与版本号无关：只有用户真正点开过这个面板才会被写成 true，之后永不回退。
+  entrySeen: false,
+  // NEW 徽标的计时起点：记录「当前版本首次启动」的时间戳与版本号。
+  // 版本一变（用户升级了）就重置计时；满 3 天后客户端自动不再显示 NEW。
+  newTagVersion: "",
+  newTagSince: 0,
+  // 升级前的版本（用户上一次跑的是哪个版本）。
+  // 客户端用它和「某项功能的引入版本」比较：只有比它新的功能才标 NEW。
+  // 这样跨几个版本一次升上来的用户能看到中间所有新功能，
+  // 已经跑过某个版本的用户不会被重复提示。
+  newTagFrom: "",
+  // 逐项计时：{ [配置项 key]: { v: 引入版本, t: 首次看到的时间戳 } }。
+  // 每个新功能各自走自己的 3 天，互不影响：
+  // 旧功能没过完 3 天时又升级，也不会被这次升级重新计时——
+  // 因为它记录的版本已经是用户见过的版本了。
+  newTagSeen: {},
   // 收起态圆钮外观
   fabSize: 60, // 直径（px）
   fabLeft: 44, // 距窗口左缘（px）
@@ -655,6 +681,23 @@ function clampNumber(value, min, max, fallback) {
   return Math.min(max, Math.max(min, n));
 }
 
+/** NEW 逐项计时表：只收 { v: 非空版本串, t: 非负时间戳 } 的条目，
+ *  其余一律丢弃（宁可丢一条记录重新计时，也不要带着脏数据跑）。 */
+function normalizeNewTagSeen(raw) {
+  if (raw == null || typeof raw !== "object" || Array.isArray(raw)) return {};
+  const out = {};
+  for (const key of Object.keys(raw)) {
+    if (typeof key !== "string" || key === "") continue;
+    const rec = raw[key];
+    if (rec == null || typeof rec !== "object" || Array.isArray(rec)) continue;
+    const v = typeof rec.v === "string" ? rec.v.trim() : "";
+    const t = Number(rec.t);
+    if (v === "" || !Number.isFinite(t) || t <= 0) continue;
+    out[key] = { v, t: Math.floor(t) };
+  }
+  return out;
+}
+
 /** 一组覆盖价：逐个字段校验（有限非负数才收），不填的字段留空表示「用默认价」。
  *  四个字段都没填则整组丢弃（等价于不覆盖）。 */
 function normalizePriceOverride(raw) {
@@ -683,6 +726,14 @@ function normalizeSettings(raw) {
     showUsagePill: bool(src.showUsagePill, d.showUsagePill),
     showPeriodLabel: bool(src.showPeriodLabel, d.showPeriodLabel),
     showFab: bool(src.showFab, d.showFab),
+    footerStyle: src.footerStyle === "classic" ? "classic" : d.footerStyle,
+    showSidebarEntry: bool(src.showSidebarEntry, d.showSidebarEntry),
+    refreshWave: bool(src.refreshWave, d.refreshWave),
+    entrySeen: bool(src.entrySeen, d.entrySeen),
+    newTagVersion: typeof src.newTagVersion === "string" ? src.newTagVersion : d.newTagVersion,
+    newTagFrom: typeof src.newTagFrom === "string" ? src.newTagFrom : d.newTagFrom,
+    newTagSeen: normalizeNewTagSeen(src.newTagSeen),
+    newTagSince: clampNumber(src.newTagSince, 0, Number.MAX_SAFE_INTEGER, d.newTagSince),
     fabSize: clampNumber(src.fabSize, 32, 120, d.fabSize),
     fabLeft: clampNumber(src.fabLeft, 0, 600, d.fabLeft),
     statsSessionLimit: clampNumber(src.statsSessionLimit, 10, 2000, d.statsSessionLimit),
@@ -731,21 +782,35 @@ function settingsFilePath(ctx) {
   return `${home.replace(/[\\/]+$/, "")}${sep}${SETTINGS_FILENAME}`;
 }
 
-/** 读设置文件；不存在 / JSON 损坏 / 无权限一律返回 null（调用方保持默认值）。 */
+/** 读设置文件；不存在 / JSON 损坏 / 无权限一律返回 null（调用方保持默认值）。
+ *  额外做一件事：老版本写下的文件里没有 showSidebarEntry 这个键，
+ *  读到这种「升级前」的文件时把侧边栏入口强制打开一次，让用户第一次升级后
+ *  能在左侧导航看到新入口（用户要求：第一次更新的用户强制显示在侧边栏）。
+ *  之后再关掉就按用户的选择来，不会反复弹回来。 */
 async function readSettingsFile(file) {
   if (file === "") return null;
   try {
     const { readFile } = await import("node:fs/promises");
     const text = await readFile(file, "utf8");
-    return normalizeSettings(JSON.parse(text));
-  } catch {
+    const parsed = JSON.parse(text);
+    const settings = normalizeSettings(parsed);
+    if (parsed == null || typeof parsed !== "object" || !("showSidebarEntry" in parsed)) {
+      settings.showSidebarEntry = true;
+    }
+    // entrySeen 不按版本强制：文件里没写 true 的旧用户（升级上来的）
+    // 读出来就是未读，红点 + NEW 会提示他「这版加了新东西」。
+    return settings;
+  } catch (error) {
+    // 文件不存在 = 全新安装，这里不能当「未读」处理（否则新用户一装就看到一堆 NEW，
+    // 对他来说每一项都是新的，标 NEW 毫无信息量）。用字符串标记区分于「读失败」。
+    if (error && error.code === "ENOENT") return "missing";
     return null;
   }
 }
 
 /** 写设置文件：先写临时文件再改名，避免写一半把配置写坏。 */
 async function writeSettingsFile(file, settings) {
-  if (file === "") throw new Error("无法定位 DSH 配置目录");
+  if (file === "") throw new Error("无法定位 DeepSeek Harness 配置目录");
   const { mkdir, writeFile, rename } = await import("node:fs/promises");
   const { dirname } = await import("node:path");
   await mkdir(dirname(file), { recursive: true });
@@ -929,6 +994,76 @@ function recentDayKeys(days, now = Date.now()) {
   return result;
 }
 
+/** 只算「今天」的消耗总额，供余额行上方的 meta 行用。
+ *  与 buildUsageStats 的区别：只扫 updatedAt 落在今天的会话，通常只有几个，
+ *  所以可以随余额行刷新（缓存 60s），不必等 5 分钟的完整统计。
+ *  @returns {Promise<{ cost:number, totalTokens:number, sessions:number }>} */
+async function buildTodayCost(ctx, config) {
+  const getService = typeof ctx.get === "function" ? ctx.get.bind(ctx) : null;
+  const controller = getService?.("sessionController");
+  const persistence = getService?.("sessionPersistence");
+  const now = Date.now();
+  // 北京当日 0 点对应的真实时间戳（用 UTC 分量读北京日期，再减回去）
+  const bj = new Date(now + 8 * 60 * 60 * 1000);
+  const dayStart = Date.UTC(bj.getUTCFullYear(), bj.getUTCMonth(), bj.getUTCDate()) - 8 * 60 * 60 * 1000;
+  const todayKey = bj.getUTCFullYear() + "-" + String(bj.getUTCMonth() + 1).padStart(2, "0") + "-" + String(bj.getUTCDate()).padStart(2, "0");
+  // 只挑今天动过的会话
+  let candidates = [];
+  if (typeof controller?.list === "function") {
+    const listed = await controller.list({});
+    const items = Array.isArray(listed) ? listed : (listed?.items ?? []);
+    candidates = items
+      .map((item) => (item && typeof item === "object" ? { id: item.sessionId ?? item.id, at: Number(item.updatedAt ?? item.createdAt ?? 0) } : null))
+      .filter((x) => x && typeof x.id === "string" && x.id !== "");
+    // list() 已按活动降序；万一没有 updatedAt 字段就保守全扫（上限 STATS_MAX_SESSIONS）
+    if (candidates.some((x) => x.at > 0)) {
+      candidates = candidates.filter((x) => x.at >= dayStart);
+    }
+  } else if (typeof persistence?.list === "function") {
+    const headers = await persistence.list();
+    const list = Array.isArray(headers) ? headers : [];
+    candidates = list
+      .map((h) => (h && typeof h.id === "string" ? { id: h.id, at: Number(h.updatedAt ?? h.createdAt ?? 0) } : null))
+      .filter(Boolean);
+    if (candidates.some((x) => x.at > 0)) {
+      candidates = candidates.filter((x) => x.at >= dayStart);
+    }
+  } else {
+    throw new Error("会话持久化服务不可用");
+  }
+  const selected = candidates
+    .slice(0, clampNumber(config?.statsSessionLimit, 10, 2000, STATS_MAX_SESSIONS))
+    .map((x) => x.id);
+  const daily = new Map();
+  let cursor = 0;
+  const worker = async () => {
+    while (cursor < selected.length) {
+      const sessionId = selected[cursor++];
+      try {
+        const inspected = await inspectPersistedSession(ctx, sessionId);
+        collectDailyUsage(
+          inspected?.events ?? [],
+          config,
+          detectModel({ requestHeader: () => void 0, requestContext: () => void 0, events: inspected?.events ?? [] }),
+          dayStart,
+          daily,
+          sessionId,
+        );
+      } catch {
+        // 单个会话读不出来不影响总额
+      }
+    }
+  };
+  await Promise.all(Array.from({ length: Math.min(3, selected.length) }, () => worker()));
+  const day = daily.get(todayKey);
+  return {
+    date: todayKey,
+    cost: day?.cost ?? 0,
+    totalTokens: day?.totalTokens ?? 0,
+    sessions: day?.sessionIds?.size ?? 0,
+  };
+}
+
 async function buildUsageStats(ctx, config) {
   const getService = typeof ctx.get === "function" ? ctx.get.bind(ctx) : null;
   const controller = getService?.("sessionController");
@@ -1211,6 +1346,7 @@ function writeJson(res, status, payload) {
 }
 
 export function apply(ctx, config = {}) {
+  let todayCache = { expiresAt: 0, value: null };
   let statsCache = { expiresAt: 0, value: null };
   // 用户设置（「插件」页配置卡片读写）：先按出厂默认值跑，异步读到文件后热替换。
   // 读不到文件不是错误——全新安装就是没有文件，保持默认值即可。
@@ -1219,9 +1355,35 @@ export function apply(ctx, config = {}) {
   void (async () => {
     settingsState.file = settingsFilePath(ctx);
     const loaded = await readSettingsFile(settingsState.file);
-    if (loaded) {
+    let dirty = false;
+    if (loaded === "missing") {
+      // 全新安装：没有任何历史可回顾，红点不亮，
+      // 并立即落盘固化，保证后续每次启动都走「已读」分支。
+      settingsState.value = { ...DEFAULT_SETTINGS, entrySeen: true };
+      dirty = true;
+    } else if (loaded) {
       settingsState.value = loaded;
-      effectiveConfig = withSettings(config, loaded);
+    }
+    // NEW 徽标的计时：记下「这个版本首次启动」的时间。
+    // 版本号变了（用户升级了）就重置，否则沿用上次记录的起点——
+    // 否则每次启动都会把计时推到“现在”，永远过不了 3 天。
+    const tagVersion = settingsState.value.newTagVersion || "";
+    const tagSince = Number(settingsState.value.newTagSince) || 0;
+    if (tagVersion !== PLUGIN_VERSION || !(tagSince > 0)) {
+      // 升级：把「刚刚跑过的那个版本」存下来。首次安装时 tagVersion 为空串，
+      // 客户端见空串就把所有 NEW_TAGS 都当新功能（新用户确实都没见过）。
+      settingsState.value.newTagFrom = tagVersion || "";
+      settingsState.value.newTagVersion = PLUGIN_VERSION;
+      settingsState.value.newTagSince = Date.now();
+      dirty = true;
+    }
+    effectiveConfig = withSettings(config, settingsState.value);
+    if (dirty) {
+      try {
+        await writeSettingsFile(settingsState.file, settingsState.value);
+      } catch {
+        // 写盘失败（目录只读等）不影响这一次运行：内存里已经是最新值
+      }
     }
   })();
   // DSH 热重载(reload)时会重新 apply 本插件；若旧实例注册的 /balance-api 路由
@@ -1318,15 +1480,18 @@ export function apply(ctx, config = {}) {
         let model = "";
         let sessionParam = "";
         let wantStats = false;
+        let wantToday = false;
         try {
           const search = new URL(req.url, "http://localhost").searchParams;
           model = search.get("model") ?? "";
           sessionParam = search.get("session") ?? "";
           wantStats = search.get("stats") === "1";
+          wantToday = search.get("today") === "1";
         } catch {
           model = "";
           sessionParam = "";
           wantStats = false;
+          wantToday = false;
         }
         let stats = null;
         let statsError = "";
@@ -1343,6 +1508,18 @@ export function apply(ctx, config = {}) {
             statsError = error instanceof Error ? error.message : String(error);
           }
         }
+        // 今日消耗：只扫今天动过的会话，60 秒缓存，供余额行上方的 meta 行随余额一起刷新
+        let today = null;
+        if (wantToday) {
+          try {
+            if (todayCache.value == null || todayCache.expiresAt <= Date.now()) {
+              todayCache = { value: await buildTodayCost(httpCtx, effectiveConfig), expiresAt: Date.now() + 60 * 1000 };
+            }
+            today = todayCache.value;
+          } catch {
+            today = null; // 拿不到就不显示，不阻塞余额
+          }
+        }
         // 按 session id 取会话并分时段汇总 usage（拿不到则返回 null，客户端回退用本地投影）
         let usage = null;
         // 子代理后代会话的 usage 汇总（与主会话分开）
@@ -1413,6 +1590,7 @@ export function apply(ctx, config = {}) {
           subagentUsage,
           stats,
           statsError,
+          today,
         });
       },
     });

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "balance-dsh",
-  "version": "1.3.1",
-  "description": "在 DSH 中显示 DeepSeek 账户余额和当前会话 token/费用消耗（含高峰/空闲峰谷计费）。仅支持 DeepSeek 官方模型与账号，暂不支持其他大模型。",
+  "version": "2.0.0",
+  "description": "在 DeepSeek Harness 中显示 DeepSeek 账户余额和当前会话 token/费用消耗（含高峰/空闲峰谷计费）。仅支持 DeepSeek 官方模型与账号，暂不支持其他大模型。",
   "author": "Rannist",
   "type": "module",
   "main": "lib/index.js",