dsh-deepseek-balance-widget 1.2.2 → 1.2.4

package/lib/client.js

@@ -14,6 +14,8 @@ window.__ModuleLoader__.load({
     const UPDATE_ENDPOINT = "/deepseek-balance-update";
     /** Auto-refresh cadence, milliseconds. */
     const REFRESH_MS = 30000;
+    /** How long the "update succeeded, restart dsh" message stays visible, milliseconds. */
+    const UPDATE_SUCCESS_MS = 5000;
 
     /** Inline icon (matches the shell's 16px nav-icon look). */
     const ICON = '<svg viewBox="0 0 16 16" width="14" height="14" fill="none" stroke="currentColor" stroke-width="1.3" stroke-linecap="round" stroke-linejoin="round" aria-hidden="true"><path d="M4.5 4L8 8.5 11.5 4M8 8.5v4M5.5 9.5h5M5.5 11.5h5"/></svg>';
@@ -210,8 +212,9 @@ window.__ModuleLoader__.load({
       if (result.ok) {
         btn.textContent = "成功，请重启 dsh";
         btn.disabled = true;
-        // Re-check version so the footer shows the new state (until dsh restarts).
-        await checkVersion();
+        // Keep the success message visible for a while before reverting to version info.
+        updateSuccessUntil = Date.now() + UPDATE_SUCCESS_MS;
+        setTimeout(() => checkVersion(), UPDATE_SUCCESS_MS);
       } else {
         btn.textContent = "更新失败";
         btn.title = String(result.error || "更新失败") + (result.output ? "\n" + result.output.slice(-200) : "");
@@ -398,6 +401,9 @@ window.__ModuleLoader__.load({
 
     /** Build the npm version snippet shown in the footer (always a clickable button). */
     function versionHtmlOf(versionEnvelope) {
+      if (Date.now() < updateSuccessUntil) {
+        return '<span class="dshBalanceVersion" data-state="ok">成功，请重启 dsh</span>';
+      }
       if (!versionEnvelope) {
         return '<span class="dshBalanceVersion" data-state="ok"><button class="dshBalanceVersionBtn" type="button" data-update-cmd>检查更新</button></span>';
       }
@@ -469,6 +475,7 @@ window.__ModuleLoader__.load({
     let lastUpdated = null;
     let timer = null;
     let versionTimer = null;
+    let updateSuccessUntil = 0;
 
     /** One refresh cycle: fetch balance + usage, update entry stats and popover. */
     async function refresh(manual) {
@@ -562,6 +569,7 @@ window.__ModuleLoader__.load({
             document.removeEventListener("mousedown", outsideHandler);
             outsideHandler = null;
           }
+          updateSuccessUntil = 0;
           if (popRef !== null) popRef.remove();
           entry.remove();
           entryRef = null;

package/lib/index.js

@@ -1,5 +1,5 @@
 import { launchEnvironmentOf } from "@deepseek-ai/dsh-launch-environment";
-import { readFile } from "node:fs/promises";
+import { readFile, stat } from "node:fs/promises";
 import { fileURLToPath } from "node:url";
 import { dirname, resolve } from "node:path";
 import { spawn } from "node:child_process";
@@ -101,17 +101,43 @@ async function queryNpmVersion() {
   return { ok: true, local: LOCAL_VERSION, latest, updateAvailable };
 }
 
+/** @returns {Promise<boolean>} whether the given path exists. */
+async function fileExists(path) {
+  try {
+    await stat(path);
+    return true;
+  } catch {
+    return false;
+  }
+}
+
+/**
+ * Detect the package manager used by UPDATE_CWD.
+ * @returns {Promise<"pnpm" | "npm">}
+ */
+async function detectPackageManager() {
+  const pnpmLock = resolve(UPDATE_CWD, "pnpm-lock.yaml");
+  const pnpmLockYml = resolve(UPDATE_CWD, "pnpm-lock.yml");
+  if (await fileExists(pnpmLock) || await fileExists(pnpmLockYml)) return "pnpm";
+  return "npm";
+}
+
 /**
- * Run `npm install <pkg>@latest` in the discovered project root.
+ * Run the package-manager update command for this package in the discovered project root.
+ * Uses shell mode on Windows so that .cmd scripts (npm.cmd / pnpm.cmd) can actually spawn.
  * @returns {Promise<{ok:boolean, output:string, error?:string}>}
  */
-async function runNpmUpdate() {
-  const cmd = process.platform === "win32" ? "npm.cmd" : "npm";
-  const args = ["install", "dsh-deepseek-balance-widget@latest"];
+async function runPackageUpdate() {
+  const pm = await detectPackageManager();
+  const isWin = process.platform === "win32";
+  const cmd = isWin ? (pm === "pnpm" ? "pnpm.cmd" : "npm.cmd") : pm;
+  const args = pm === "pnpm"
+    ? ["add", "dsh-deepseek-balance-widget@latest"]
+    : ["install", "dsh-deepseek-balance-widget@latest"];
   return new Promise((resolve) => {
     const child = spawn(cmd, args, {
       cwd: UPDATE_CWD,
-      shell: false,
+      shell: isWin,
       env: { ...process.env, NPM_CONFIG_FUND: "false", NPM_CONFIG_AUDIT: "false" }
     });
     let stdout = "";
@@ -126,7 +152,7 @@ async function runNpmUpdate() {
       if (code === 0) {
         resolve({ ok: true, output: output.slice(-800) });
       } else {
-        resolve({ ok: false, output, error: `npm install exited with code ${code}` });
+        resolve({ ok: false, output, error: `${pm} ${args.join(" ")} exited with code ${code}` });
       }
     });
   });
@@ -533,7 +559,7 @@ function apply(ctx) {
             res.end(JSON.stringify({ ok: true, noOp: true, local: before.local, latest: before.latest, message: "已是最新版本" }));
             return;
           }
-          const result = await runNpmUpdate();
+          const result = await runPackageUpdate();
           if (result.ok) {
             res.writeHead(200, { "content-type": "application/json; charset=utf-8" });
             res.end(JSON.stringify({ ok: true, message: "更新成功，请彻底重启 dsh 以加载新版本", output: result.output }));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-deepseek-balance-widget",
-  "version": "1.2.2",
+  "version": "1.2.4",
   "type": "module",
   "description": "DeepSeek API balance widget for the dsh web sidebar: a live, auto-refreshing balance pill (balance / today spend / today tokens) with a detail popover (cumulative spend, monthly usage, request count). Every user sees only their own balance — keys are resolved per-machine from the local credential seam, never hardcoded.",
   "keywords": [