balance-dsh 2.1.1 → 2.1.2

package/lib/client.js

@@ -5910,6 +5910,11 @@ function UsageCostLine({ useProjection, sessionId }) {
               kind: "installed",
               to: data.to || "",
               failed: Array.isArray(data.failed) ? data.failed : [],
+              // 服务端会判断这次更新是否「持久」：按 npm 版本装进 profile 的，只覆盖文件
+              // 会被重启时的声明版本盖回去（用户 2026-10-31 报的版本回归）。
+              persisted: data.persisted !== false,
+              warning: typeof data.warning === "string" ? data.warning : "",
+              command: typeof data.command === "string" ? data.command : "",
             });
           })
           .catch((error) => {
@@ -5938,6 +5943,12 @@ function UsageCostLine({ useProjection, sessionId }) {
           if (failed.length > 0) {
             return "已更新到 " + updateCheck.to + "，但有 " + failed.length + " 个文件没覆盖成功（" + failed.join("；") + "）。重启后如仍异常，请先从插件市场重装。";
           }
+          // 不持久的那一次必须说清楚：重启会退回旧号，并给出能真正生效的命令。
+          if (updateCheck.persisted === false) {
+            return "已就地更新到 " + updateCheck.to + "，但本机是按 npm 版本装进 profile 的 —— 重启后可能退回旧号。"
+              + "请执行 " + (updateCheck.command || "dsh plugin install balance-dsh@" + updateCheck.to)
+              + "（或 设置 → 插件 里的更新）后再重启。";
+          }
           return "已更新到 " + updateCheck.to + "，重启 DeepSeek Harness 后生效。";
         }
         if (updateCheck.kind === "error") return "检查失败：" + updateCheck.error + "（可稍后重试）";

package/lib/index.js

@@ -758,7 +758,7 @@ function formatBalance(balance) {
 //                          | { status:"failed" }  （网络失败；官方明确“失败绝不变成 0 余额”）
 // client 是 AccountClientMetadata（version / locale / timezoneOffsetSeconds），
 // 缺字段会在平台侧组装请求头时抛错，所以三个字段都要给。
-const PLUGIN_VERSION = "2.1.1";
+const PLUGIN_VERSION = "2.1.2";
 
 /** 磁盘上**实际安装**的版本号。
  *
@@ -862,6 +862,188 @@ function updateTarballUrl(version) {
   return "https://registry.npmjs.org/balance-dsh/-/balance-dsh-" + version + ".tgz";
 }
 
+/** 插件目录所在的 profile 根（.../profiles/<名字>/node_modules/balance-dsh → .../profiles/<名字>）。
+ *
+ *  用户反馈（2026-10-31）：「在插件设置里更新之后重启会导致版本回归，更新完是 2.1.0，
+ *  重启了之后又变成 2.0.8」。根因就是这里 —— 就地覆盖文件只改了 node_modules 里的内容，
+ *  profile 的 package.json / pnpm-lock.yaml 里锁的还是旧版本；DSH 重启时按声明版本重新
+ *  落地那一份，于是又退回 2.0.8。所以必须同时把「声明的版本」也更新掉，见下面
+ *  installViaPackageManager。 */
+async function profileDirOf(pluginDir) {
+  if (typeof pluginDir !== "string" || pluginDir === "") return "";
+  const { dirname, basename, join } = await import("node:path");
+  const fs = await import("node:fs/promises");
+  /** 这一层是不是「声明了 balance-dsh 的 profile 根」。 */
+  const declares = async (dir) => {
+    try {
+      const raw = JSON.parse(await fs.readFile(join(dir, "package.json"), "utf8"));
+      return raw != null && typeof raw === "object" && raw.dependencies != null && Object.hasOwn(raw.dependencies, "balance-dsh");
+    } catch {
+      return false;
+    }
+  };
+  // 常规布局：<profile>/node_modules/<包>
+  const nodeModules = dirname(pluginDir);
+  if (basename(nodeModules) === "node_modules") {
+    const profile = dirname(nodeModules);
+    if (await declares(profile)) return profile;
+  }
+  // 兜底：pnpm 的虚拟仓是 <profile>/node_modules/.pnpm/<包>@<版本>/node_modules/<包>，
+  // 上一级不是 profile —— 往上找「package.json 里声明了 balance-dsh」的那一层，最多 6 级。
+  // 认不出来返回 ""（开发目录），调用方按「本地装法」处理，就地覆盖就是最终状态。
+  let dir = dirname(pluginDir);
+  for (let i = 0; i < 6 && dir !== ""; i += 1) {
+    if (await declares(dir)) return dir;
+    const parent = dirname(dir);
+    if (parent === dir) break;
+    dir = parent;
+  }
+  return "";
+}
+
+/** profile 里 balance-dsh 的依赖声明。只有**registry 版本号**才需要走包管理器：
+ *  `file:` / `link:` / `git` 这类指向本地目录的装法，就地覆盖文件就是最终状态，不会被回退。 */
+async function profileSpecOf(profileDir) {
+  if (profileDir === "") return "";
+  try {
+    const { readFile } = await import("node:fs/promises");
+    const { join } = await import("node:path");
+    const raw = JSON.parse(await readFile(join(profileDir, "package.json"), "utf8"));
+    const spec = raw?.dependencies?.["balance-dsh"];
+    if (typeof spec !== "string") return "";
+    const trimmed = spec.trim();
+    if (trimmed === "") return "";
+    // file:/link:/portal:/workspace:/git+https: 这些都不指向 registry 版本号。
+    // ⚠️ git 要写成 `git(?:\+[a-z]+)?`：实参是 `git+https://…`，光写 `git\+` 后面
+    // 跟的是 `https:` 而不是冒号，会漏过去被当成 registry 号（探针 2026-10-31 抓到）。
+    if (/^(file|link|portal|workspace|git(?:\+[a-z]+)?|https?):/i.test(trimmed)) return "";
+    if (trimmed.startsWith(".") || trimmed.startsWith("/") || /^[a-zA-Z]:[\\/]/.test(trimmed)) return "";
+    return trimmed;
+  } catch {
+    return "";
+  }
+}
+
+/** 用 DSH 自带的 pnpm 真正装一遍：它会把 profile 的 package.json + pnpm-lock.yaml + node_modules
+ *  一起改成新版本，重启不会再退回旧号。失败时把两个文件按原样恢复（对齐 DSH 插件管理器自己的做法）。 */
+async function installViaPackageManager(profileDir, version) {
+  const result = { ok: false, error: "", output: "" };
+  if (profileDir === "" || version === "") {
+    result.error = "没有 profile 目录";
+    return result;
+  }
+  const path = await import("node:path");
+  const fs = await import("node:fs/promises");
+  const roots = [];
+  const res = typeof process !== "undefined" && typeof process.resourcesPath === "string" ? process.resourcesPath : "";
+  if (res !== "") {
+    roots.push(path.join(res, "runtime", "primary-runtime", "dependencies"));
+    roots.push(path.join(res, "app", "runtime", "primary-runtime", "dependencies"));
+  }
+  const exe = typeof process !== "undefined" && typeof process.execPath === "string" ? process.execPath : "";
+  if (exe !== "") {
+    const base = path.dirname(exe);
+    roots.push(path.join(base, "runtime", "primary-runtime", "dependencies"));
+    roots.push(path.join(base, "resources", "runtime", "primary-runtime", "dependencies"));
+    roots.push(path.join(base, "..", "resources", "runtime", "primary-runtime", "dependencies"));
+  }
+  let nodeExe = "";
+  let pnpmMjs = "";
+  for (const root of roots) {
+    const nodeTry = process.platform === "win32"
+      ? path.join(root, "node", "bin", "node.exe")
+      : path.join(root, "node", "bin", "node");
+    const pnpmTry = path.join(root, "pnpm", "bin", "pnpm.mjs");
+    try {
+      await fs.access(nodeTry);
+      await fs.access(pnpmTry);
+      nodeExe = nodeTry;
+      pnpmMjs = pnpmTry;
+      break;
+    } catch {
+      // 继续试下一个候选根
+    }
+  }
+  if (nodeExe === "" || pnpmMjs === "") {
+    result.error = "找不到 DSH 自带的 pnpm";
+    return result;
+  }
+
+  // 备份两个会被 pnpm 改写的文件：失败要原样恢复，绝不能让 profile 停在半改状态
+  const saved = [];
+  for (const name of ["package.json", "pnpm-lock.yaml"]) {
+    const file = path.join(profileDir, name);
+    try {
+      saved.push({ file, text: await fs.readFile(file, "utf8") });
+    } catch {
+      saved.push({ file, text: null });
+    }
+  }
+  const restore = async () => {
+    for (const item of saved) {
+      try {
+        if (item.text === null) await fs.rm(item.file, { force: true });
+        else await fs.writeFile(item.file, item.text, "utf8");
+      } catch {
+        // 恢复失败也只能作罢，至少已经报错给用户
+      }
+    }
+  };
+
+  const { spawn } = await import("node:child_process");
+  const args = [
+    pnpmMjs,
+    "add",
+    "balance-dsh@" + version,
+    "--config.minimum-release-age=0",
+    "--config.confirmModulesPurge=false",
+    "--reporter=append-only",
+    "--registry",
+    "https://registry.npmjs.org/",
+  ];
+  const env = { ...process.env };
+  delete env.ELECTRON_RUN_AS_NODE;
+  delete env.ELECTRON_NO_ATTACH_CONSOLE;
+  const child = spawn(nodeExe, args, { cwd: profileDir, env, windowsHide: true, stdio: ["ignore", "pipe", "pipe"] });
+  let out = "";
+  child.stdout?.on("data", (chunk) => {
+    out += String(chunk);
+    if (out.length > 20000) out = out.slice(-20000);
+  });
+  child.stderr?.on("data", (chunk) => {
+    out += String(chunk);
+    if (out.length > 20000) out = out.slice(-20000);
+  });
+  const done = await new Promise((resolve) => {
+    const timer = setTimeout(() => {
+      try {
+        child.kill();
+      } catch {
+        // 进程可能已经退出
+      }
+      resolve({ code: -1, timedOut: true });
+    }, 240000);
+    child.on("error", (error) => {
+      clearTimeout(timer);
+      resolve({ code: -1, error: error instanceof Error ? error.message : String(error) });
+    });
+    child.on("close", (code) => {
+      clearTimeout(timer);
+      resolve({ code: typeof code === "number" ? code : -1, timedOut: false });
+    });
+  });
+  result.output = out.trim().split(/\r?\n/).slice(-6).join("\n");
+  if (done.code === 0) {
+    result.ok = true;
+    return result;
+  }
+  await restore();
+  result.error = done.timedOut
+    ? "pnpm 超过 4 分钟没有结束（已中止，profile 已恢复原状）"
+    : (done.error || ("pnpm 退出码 " + String(done.code)));
+  return result;
+}
+
 /** 就地更新本插件：下载 npm 上的最新 tarball → 解包 → 逐文件覆盖自己的安装目录。
  *
  *  几个刻意的取舍：
@@ -887,6 +1069,37 @@ async function installLatestVersion(ctx) {
   const root = await installedPluginDir();
   if (root === "") return { ok: false, error: "定位不到插件安装目录" };
 
+  // ★ 先看这一份装法是不是「按 registry 版本装进 profile 的」。
+  // 是的话，光覆盖 node_modules 里的文件没用 —— profile 的 package.json / pnpm-lock.yaml
+  // 还锁着旧版本，DSH 重启时按声明版本重新落地，版本就退回旧号（用户 2026-10-31 报的现象）。
+  // 所以这种情况先让 DSH 自带的 pnpm 真正装一遍，声明与文件一起变，重启才不丢。
+  const profileDir = await profileDirOf(root);
+  const registrySpec = await profileSpecOf(profileDir);
+  const manualCommand = "dsh plugin install balance-dsh@" + latest;
+  let pnpmFailure = "";
+  let pnpmOutput = "";
+  if (registrySpec !== "") {
+    const viaPnpm = await installViaPackageManager(profileDir, latest);
+    if (viaPnpm.ok) {
+      updateCache = { at: 0, value: null };
+      return {
+        ok: true,
+        installed: true,
+        from: currentVer,
+        to: latest,
+        via: "pnpm",
+        persisted: true,
+        files: 0,
+        failed: [],
+        restart: true,
+      };
+    }
+    // pnpm 跑不起来（没有自带 runtime / 断网 / 被策略拦）：继续走就地覆盖，
+    // 但必须如实告诉用户「这次不持久」，并给出能真正生效的命令。
+    pnpmFailure = viaPnpm.error;
+    pnpmOutput = viaPnpm.output;
+  }
+
   let gz;
   try {
     const signal =
@@ -972,6 +1185,17 @@ async function installLatestVersion(ctx) {
     installed: true,
     from: currentVer,
     to: latest,
+    via: "files",
+    // registry 装法只覆盖文件 = 重启会被 profile 里锁的旧版本覆盖回去，必须如实标出来
+    persisted: registrySpec === "",
+    warning: registrySpec === ""
+      ? ""
+      : "文件已覆盖到 " + latest + "，但本机是按 npm 版本装进 profile 的，profile 里锁定的还是旧版本；"
+        + "重启后可能退回旧号（" + (pnpmFailure || "自带 pnpm 不可用") + "）。"
+        + "请执行 " + manualCommand + " 后再重启，或用 设置 → 插件 里的更新。",
+    command: manualCommand,
+    pnpmError: pnpmFailure,
+    pnpmOutput,
     files: written.length,
     failed,
     backup: backupRoot,

package/package.json

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