modelstat 0.5.1 → 0.7.0

package/dist/cli.mjs

@@ -37086,7 +37086,7 @@ var init_scan = __esm({
     init_api();
     init_config2();
     init_pipeline2();
-    DAEMON_VERSION = true ? "daemon-0.5.1" : "daemon-dev";
+    DAEMON_VERSION = true ? "daemon-0.7.0" : "daemon-dev";
     BATCH_MAX_EVENTS = INGEST_BATCH_MAX_EVENTS;
     BATCH_MAX_TOOL_CALLS = 2e4;
     BATCH_BUFFER_HARD_CAP = BATCH_MAX_EVENTS * 2;
@@ -37239,6 +37239,90 @@ var init_lock = __esm({
   }
 });
 
+// src/update.ts
+import { spawn as spawn2, spawnSync as spawnSync3 } from "child_process";
+import { mkdirSync as mkdirSync6, readFileSync as readFileSync8, renameSync as renameSync4, writeFileSync as writeFileSync6 } from "fs";
+import { platform as platform4 } from "os";
+function prefPath() {
+  return homePath("auto-update.json");
+}
+function storedAutoUpdate() {
+  try {
+    const o = JSON.parse(readFileSync8(prefPath(), "utf8"));
+    return o.autoUpdate ?? true;
+  } catch {
+    return true;
+  }
+}
+function setStoredAutoUpdate(enabled) {
+  mkdirSync6(modelstatHome(), { recursive: true, mode: 448 });
+  const tmp = `${prefPath()}.${process.pid}.tmp`;
+  writeFileSync6(tmp, JSON.stringify({ autoUpdate: enabled }), { mode: 384 });
+  renameSync4(tmp, prefPath());
+}
+function parseFlag(v) {
+  if (v == null) return null;
+  const s = v.trim().toLowerCase();
+  if (["0", "off", "false", "no", "disable", "disabled"].includes(s)) return false;
+  if (["1", "on", "true", "yes", "enable", "enabled"].includes(s)) return true;
+  return null;
+}
+function autoUpdateEnabled() {
+  const env = parseFlag(process.env.MODELSTAT_AUTO_UPDATE);
+  return env ?? storedAutoUpdate();
+}
+function autoUpdatePinnedByEnv() {
+  return parseFlag(process.env.MODELSTAT_AUTO_UPDATE) !== null;
+}
+function canSelfUpdate() {
+  try {
+    return spawnSync3(NPM, ["--version"], { stdio: "ignore", timeout: 1e4 }).status === 0;
+  } catch {
+    return false;
+  }
+}
+function runUpgrade() {
+  if (!canSelfUpdate()) {
+    return { started: false, reason: "npm not found on PATH" };
+  }
+  try {
+    const env = { ...process.env };
+    delete env.npm_config_global;
+    delete env.npm_config_prefix;
+    const child = spawn2(NPM, ["install", "-g", `${PACKAGE}@latest`], {
+      detached: true,
+      stdio: "ignore",
+      env
+    });
+    child.unref();
+    return { started: true };
+  } catch (e) {
+    return { started: false, reason: e.message };
+  }
+}
+function maybeAutoUpdate(verdict, target) {
+  if (verdict !== "update_available" && verdict !== "upgrade_required") return null;
+  const key = `${verdict}:${target ?? ""}`;
+  if (handled.has(key)) return null;
+  handled.add(key);
+  const required = verdict === "upgrade_required";
+  if (!autoUpdateEnabled()) {
+    return `${required ? "upgrade required" : "update available"} (latest ${target ?? "?"}); auto-update is off \u2014 run \`modelstat upgrade\` or \`npm i -g modelstat@latest\``;
+  }
+  const r = runUpgrade();
+  return r.started ? `auto-updating to ${target ?? "latest"} \u2014 \`npm i -g modelstat@latest\`; the service will restart on the new version` : `auto-update could not start (${r.reason}); upgrade manually: \`npm i -g modelstat@latest\``;
+}
+var NPM, PACKAGE, handled;
+var init_update = __esm({
+  "src/update.ts"() {
+    "use strict";
+    init_paths();
+    NPM = platform4() === "win32" ? "npm.cmd" : "npm";
+    PACKAGE = "modelstat";
+    handled = /* @__PURE__ */ new Set();
+  }
+});
+
 // src/reconcile.ts
 import { stat as stat3 } from "fs/promises";
 function utcDay(ts) {
@@ -39418,6 +39502,10 @@ function noteEventAt(iso) {
   status.lastEventAt = iso;
   scheduleLocalFlush();
 }
+function setUpdate(u) {
+  status.update = u;
+  scheduleLocalFlush();
+}
 function snapshotBody() {
   return {
     device_id: state.deviceId ?? null,
@@ -39429,7 +39517,9 @@ function snapshotBody() {
     stats: status.stats,
     last_event_at: status.lastEventAt,
     daemon_version: DAEMON_VERSION2,
-    machine_id: machineKey()
+    machine_id: machineKey(),
+    update: status.update,
+    auto_update: autoUpdateEnabled()
   };
 }
 function scheduleLocalFlush() {
@@ -39447,6 +39537,26 @@ function scheduleLocalFlush() {
   }, LOCAL_FLUSH_THROTTLE_MS);
   localFlushTimer.unref();
 }
+function handleRelease(rel) {
+  const verdict = rel?.verdict ?? "ok";
+  const latest = rel?.latest ?? null;
+  if (verdict === "ok") {
+    if (status.update) setUpdate(null);
+    lastVerdict = "ok";
+    return;
+  }
+  setUpdate({ verdict, latest });
+  if (verdict !== lastVerdict) {
+    console.log(
+      `[modelstat] release ${verdict}: this daemon ${DAEMON_VERSION2}, latest ${latest ?? "?"}`
+    );
+  }
+  lastVerdict = verdict;
+  const note = maybeAutoUpdate(verdict, latest);
+  if (note) {
+    console.log(`[modelstat] ${note}`);
+  }
+}
 async function sendHeartbeat() {
   const bearer = state.bearer;
   const deviceId = state.deviceId;
@@ -39460,6 +39570,12 @@ async function sendHeartbeat() {
     });
     if (res.statusCode >= 300) {
       await res.body.text();
+    } else {
+      try {
+        const data = await res.body.json();
+        handleRelease(data?.daemon_release);
+      } catch {
+      }
     }
   } catch {
   }
@@ -39639,7 +39755,7 @@ async function runDaemon(opts = {}) {
     setMessage(`policy refresh unavailable: ${err.message}`);
   }
   const chokidar = (await Promise.resolve().then(() => (init_esm2(), esm_exports))).default;
-  const { homedir: homedir9, platform: platform6 } = await import("os");
+  const { homedir: homedir9, platform: platform7 } = await import("os");
   const { join: join14 } = await import("path");
   const home2 = homedir9();
   const dirs = [
@@ -39647,7 +39763,7 @@ async function runDaemon(opts = {}) {
     join14(home2, ".codex/sessions"),
     join14(home2, ".cursor/ai-tracking"),
     join14(home2, ".gemini"),
-    ...platform6() === "darwin" ? [
+    ...platform7() === "darwin" ? [
       join14(home2, "Library/Application Support/Cursor/User/workspaceStorage"),
       join14(home2, "Library/Application Support/Claude")
     ] : [join14(home2, ".config/Cursor/User/workspaceStorage")]
@@ -39701,7 +39817,7 @@ async function runDaemon(opts = {}) {
   await new Promise(() => {
   });
 }
-var import_undici2, DAEMON_VERSION2, HEARTBEAT_INTERVAL_MS, SCAN_INTERVAL_MS, DISCOVERY_INTERVAL_MS, status, LOCAL_FLUSH_THROTTLE_MS, localFlushTimer, localFlushPending, LOG_MAX_BYTES, LOG_TAIL_KEEP_BYTES, lastStatusPath, scanRunner;
+var import_undici2, DAEMON_VERSION2, HEARTBEAT_INTERVAL_MS, SCAN_INTERVAL_MS, DISCOVERY_INTERVAL_MS, status, LOCAL_FLUSH_THROTTLE_MS, localFlushTimer, localFlushPending, lastVerdict, LOG_MAX_BYTES, LOG_TAIL_KEEP_BYTES, lastStatusPath, scanRunner;
 var init_daemon = __esm({
   "src/daemon.ts"() {
     "use strict";
@@ -39715,7 +39831,8 @@ var init_daemon = __esm({
     init_reconcile();
     init_scan();
     init_single_flight();
-    DAEMON_VERSION2 = true ? "daemon-0.5.1" : "daemon-dev";
+    init_update();
+    DAEMON_VERSION2 = true ? "daemon-0.7.0" : "daemon-dev";
     HEARTBEAT_INTERVAL_MS = 1e4;
     SCAN_INTERVAL_MS = 5 * 60 * 1e3;
     DISCOVERY_INTERVAL_MS = 6e4;
@@ -39726,11 +39843,13 @@ var init_daemon = __esm({
       progressTotal: 0,
       queueSize: 0,
       stats: {},
-      lastEventAt: null
+      lastEventAt: null,
+      update: null
     };
     LOCAL_FLUSH_THROTTLE_MS = 400;
     localFlushTimer = null;
     localFlushPending = false;
+    lastVerdict = "ok";
     LOG_MAX_BYTES = 64 * 1024 * 1024;
     LOG_TAIL_KEEP_BYTES = 4 * 1024 * 1024;
     lastStatusPath = null;
@@ -39744,7 +39863,7 @@ __export(watch_exports, {
   watchForever: () => watchForever
 });
 import { existsSync as existsSync12 } from "fs";
-import { homedir as homedir8, platform as platform4 } from "os";
+import { homedir as homedir8, platform as platform5 } from "os";
 import { join as join13 } from "path";
 function resolveWatchDirs() {
   const home2 = homedir8();
@@ -39765,7 +39884,7 @@ function resolveWatchDirs() {
     join13(xdgConfig, "Code - Insiders/User/workspaceStorage"),
     join13(xdgData, "claude/projects"),
     // macOS
-    ...platform4() === "darwin" ? [
+    ...platform5() === "darwin" ? [
       join13(home2, "Library/Application Support/Cursor/User/workspaceStorage"),
       join13(home2, "Library/Application Support/Claude"),
       join13(home2, "Library/Application Support/Code/User/workspaceStorage"),
@@ -39840,8 +39959,8 @@ init_config2();
 init_identity();
 init_machine_key();
 init_scan();
-import { spawn as spawn2 } from "child_process";
-import { arch as cpuArch, hostname as hostname2, platform as platform5, release } from "os";
+import { spawn as spawn3 } from "child_process";
+import { arch as cpuArch, hostname as hostname2, platform as platform6, release } from "os";
 import { createInterface as createInterface3 } from "readline";
 
 // src/service.ts
@@ -40287,6 +40406,7 @@ function ageMs(iso, now) {
 }
 
 // src/cli.ts
+init_update();
 async function confirmPrompt(question, defaultYes) {
   if (process.stdin.isTTY !== true) return defaultYes;
   const rl = createInterface3({ input: process.stdin, output: process.stdout });
@@ -40302,11 +40422,11 @@ async function confirmPrompt(question, defaultYes) {
   }
 }
 function tryOpenBrowser(url) {
-  const p = platform5();
+  const p = platform6();
   const cmd = p === "darwin" ? "open" : p === "win32" ? "cmd" : "xdg-open";
   const args = p === "win32" ? ["/c", "start", "", url] : [url];
   try {
-    const child = spawn2(cmd, args, {
+    const child = spawn3(cmd, args, {
       stdio: "ignore",
       detached: true
     });
@@ -40316,9 +40436,9 @@ function tryOpenBrowser(url) {
     return false;
   }
 }
-var DAEMON_VERSION3 = true ? "daemon-0.5.1" : "daemon-dev";
+var DAEMON_VERSION3 = true ? "daemon-0.7.0" : "daemon-dev";
 function osFamily() {
-  const p = platform5();
+  const p = platform6();
   if (p === "darwin") return "macos";
   if (p === "linux") return "linux";
   return "other";
@@ -40542,7 +40662,7 @@ async function cmdConnect(opts) {
     claim_url: claimUrl,
     agent_url: agentUrl
   });
-  if (platform5() === "darwin") {
+  if (platform6() === "darwin") {
     step("Installing menu-bar tray (macOS)");
     const buildUi = createTrayBuildUi(opts);
     try {
@@ -40593,7 +40713,7 @@ async function cmdConnect(opts) {
       logs: svc.logs,
       summariser_ready: modelReady
     });
-    ok(`${platform5() === "darwin" ? "launchd" : "systemd --user"}: ${svc.path}`);
+    ok(`${platform6() === "darwin" ? "launchd" : "systemd --user"}: ${svc.path}`);
   } catch (e) {
     emitEvent(opts, "service_install_failed", { error: e.message });
     warn(`couldn't install service: ${e.message}`);
@@ -40636,7 +40756,7 @@ async function cmdConnect(opts) {
         `    detected: \x1B[32m${discovered.installations} installs \xB7 ${discovered.identities} accounts\x1B[0m`
       );
     }
-    if (platform5() === "darwin") {
+    if (platform6() === "darwin") {
       console.log(
         `    tray    : \x1B[${tray.installed ? "32" : "2"}m${tray.installed ? "menu-bar icon ready" : "not installed"}\x1B[0m`
       );
@@ -40755,6 +40875,44 @@ async function cmdStatus() {
   console.log(`logs:    ${logsDir()}`);
   console.log(`state:   ${state.storePath}`);
   console.log(`api:     ${state.apiUrl}`);
+  console.log(
+    `auto-update: ${autoUpdateEnabled() ? "on" : "off"}${autoUpdatePinnedByEnv() ? " (pinned by env)" : ""}`
+  );
+  const ls = await readLocalStatus();
+  const upd = ls?.update;
+  if (upd?.verdict && upd.verdict !== "ok") {
+    const what = upd.verdict === "upgrade_required" ? "REQUIRED" : "available";
+    console.log(`update:  ${what} \u2014 latest ${upd.latest ?? "?"} (run \`modelstat upgrade\`)`);
+  }
+}
+function cmdAutoUpdate(args) {
+  const sub = (args[0] ?? "").toLowerCase();
+  if (sub === "on" || sub === "enable") {
+    setStoredAutoUpdate(true);
+  } else if (sub === "off" || sub === "disable") {
+    setStoredAutoUpdate(false);
+  } else if (sub === "toggle") {
+    setStoredAutoUpdate(!storedAutoUpdate());
+  } else if (sub !== "" && sub !== "status") {
+    console.log("usage: modelstat autoupdate [on|off|toggle|status]");
+    return;
+  }
+  console.log(`auto-update: ${autoUpdateEnabled() ? "on" : "off"}`);
+  if (autoUpdatePinnedByEnv()) {
+    console.log("  (pinned by MODELSTAT_AUTO_UPDATE env \u2014 the stored toggle is ignored)");
+  }
+}
+function cmdUpgrade() {
+  console.log("upgrading modelstat to the latest published version\u2026");
+  const r = runUpgrade();
+  if (r.started) {
+    console.log(
+      "  started `npm install -g modelstat@latest` \u2014 the service restarts on the new build."
+    );
+  } else {
+    console.log(`  couldn't start the upgrade: ${r.reason}`);
+    console.log("  upgrade manually: npm install -g modelstat@latest");
+  }
 }
 function fmtInt(n) {
   return Number(n).toLocaleString("en-US");
@@ -40994,6 +41152,12 @@ async function main() {
       return cmdSelfRegister();
     case "await-claim":
       return cmdAwaitClaim();
+    case "upgrade":
+      cmdUpgrade();
+      return;
+    case "autoupdate":
+      cmdAutoUpdate(rest);
+      return;
     default:
       console.log("usage:");
       console.log(
@@ -41011,6 +41175,10 @@ async function main() {
       console.log();
       console.log("Diagnostics:");
       console.log("  npx modelstat@latest status         \u2014 show pairing + service state");
+      console.log("  npx modelstat@latest upgrade        \u2014 update to the latest version now");
+      console.log(
+        "  npx modelstat@latest autoupdate     \u2014 show/set auto-update: on|off|toggle"
+      );
       console.log(
         "  npx modelstat@latest stats          \u2014 live device summary: sessions \xB7 tokens \xB7 cost (--json)"
       );