modelstat 0.0.6 → 0.0.7

package/dist/cli.mjs

@@ -4424,6 +4424,17 @@ var init_redact = __esm({
   }
 });
 
+// ../../packages/core/src/billing.ts
+var MILLION, FREE_INCLUDED_TOKENS, TEAM_INCLUDED_PER_SEAT;
+var init_billing = __esm({
+  "../../packages/core/src/billing.ts"() {
+    "use strict";
+    MILLION = 1000000n;
+    FREE_INCLUDED_TOKENS = 100n * MILLION;
+    TEAM_INCLUDED_PER_SEAT = 250n * MILLION;
+  }
+});
+
 // ../../packages/core/src/index.ts
 var init_src = __esm({
   "../../packages/core/src/index.ts"() {
@@ -4432,6 +4443,7 @@ var init_src = __esm({
     init_schemas();
     init_ids();
     init_redact();
+    init_billing();
   }
 });
 
@@ -5512,7 +5524,7 @@ __export(daemon_exports, {
   setProgress: () => setProgress,
   setQueue: () => setQueue
 });
-import { existsSync as existsSync4, statSync as statSync2 } from "fs";
+import { existsSync as existsSync5, statSync as statSync2 } from "fs";
 import { request as request2 } from "undici";
 function setPhase(phase, message) {
   status.phase = phase;
@@ -5616,21 +5628,21 @@ async function runDaemon() {
   await runDiscovery();
   await runScanCycle("startup");
   const chokidar2 = (await import("chokidar")).default;
-  const { homedir: homedir4, platform: platform4 } = await import("os");
-  const { join: join4 } = await import("path");
-  const home = homedir4();
+  const { homedir: homedir5, platform: platform5 } = await import("os");
+  const { join: join5 } = await import("path");
+  const home2 = homedir5();
   const dirs = [
-    join4(home, ".claude/projects"),
-    join4(home, ".codex/sessions"),
-    join4(home, ".cursor/ai-tracking"),
-    join4(home, ".gemini"),
-    ...platform4() === "darwin" ? [
-      join4(home, "Library/Application Support/Cursor/User/workspaceStorage"),
-      join4(home, "Library/Application Support/Claude")
+    join5(home2, ".claude/projects"),
+    join5(home2, ".codex/sessions"),
+    join5(home2, ".cursor/ai-tracking"),
+    join5(home2, ".gemini"),
+    ...platform5() === "darwin" ? [
+      join5(home2, "Library/Application Support/Cursor/User/workspaceStorage"),
+      join5(home2, "Library/Application Support/Claude")
     ] : [
-      join4(home, ".config/Cursor/User/workspaceStorage")
+      join5(home2, ".config/Cursor/User/workspaceStorage")
     ]
-  ].filter((p) => existsSync4(p) && statSync2(p).isDirectory());
+  ].filter((p) => existsSync5(p) && statSync2(p).isDirectory());
   setPhase("watching", `Watching ${dirs.length} directories`);
   const watcher = chokidar2.watch(dirs, {
     persistent: true,
@@ -5695,37 +5707,37 @@ __export(watch_exports, {
   watchForever: () => watchForever
 });
 import chokidar from "chokidar";
-import { existsSync as existsSync5 } from "fs";
-import { homedir as homedir3, platform as platform2 } from "os";
-import { join as join3 } from "path";
+import { existsSync as existsSync6 } from "fs";
+import { homedir as homedir4, platform as platform3 } from "os";
+import { join as join4 } from "path";
 function resolveWatchDirs() {
-  const home = homedir3();
-  const xdgConfig = process.env.XDG_CONFIG_HOME ?? join3(home, ".config");
-  const xdgData = process.env.XDG_DATA_HOME ?? join3(home, ".local/share");
+  const home2 = homedir4();
+  const xdgConfig = process.env.XDG_CONFIG_HOME ?? join4(home2, ".config");
+  const xdgData = process.env.XDG_DATA_HOME ?? join4(home2, ".local/share");
   const candidates = [
     // universal (default HOME-rooted CLI data dirs)
-    join3(home, ".claude/projects"),
-    join3(home, ".codex/sessions"),
-    join3(home, ".cursor/ai-tracking"),
-    join3(home, ".gemini"),
-    join3(home, ".aider"),
+    join4(home2, ".claude/projects"),
+    join4(home2, ".codex/sessions"),
+    join4(home2, ".cursor/ai-tracking"),
+    join4(home2, ".gemini"),
+    join4(home2, ".aider"),
     // XDG / Linux
-    join3(xdgConfig, "claude/projects"),
-    join3(xdgConfig, "codex/sessions"),
-    join3(xdgConfig, "Cursor/User/workspaceStorage"),
-    join3(xdgConfig, "Code/User/workspaceStorage"),
-    join3(xdgConfig, "Code - Insiders/User/workspaceStorage"),
-    join3(xdgData, "claude/projects"),
+    join4(xdgConfig, "claude/projects"),
+    join4(xdgConfig, "codex/sessions"),
+    join4(xdgConfig, "Cursor/User/workspaceStorage"),
+    join4(xdgConfig, "Code/User/workspaceStorage"),
+    join4(xdgConfig, "Code - Insiders/User/workspaceStorage"),
+    join4(xdgData, "claude/projects"),
     // macOS
-    ...platform2() === "darwin" ? [
-      join3(home, "Library/Application Support/Cursor/User/workspaceStorage"),
-      join3(home, "Library/Application Support/Claude"),
-      join3(home, "Library/Application Support/Code/User/workspaceStorage"),
-      join3(home, "Library/Application Support/Windsurf/User/workspaceStorage"),
-      join3(home, "Library/Application Support/Zed")
+    ...platform3() === "darwin" ? [
+      join4(home2, "Library/Application Support/Cursor/User/workspaceStorage"),
+      join4(home2, "Library/Application Support/Claude"),
+      join4(home2, "Library/Application Support/Code/User/workspaceStorage"),
+      join4(home2, "Library/Application Support/Windsurf/User/workspaceStorage"),
+      join4(home2, "Library/Application Support/Zed")
     ] : []
   ];
-  return Array.from(new Set(candidates)).filter((p) => existsSync5(p));
+  return Array.from(new Set(candidates)).filter((p) => existsSync6(p));
 }
 async function safeScan(reason) {
   if (scanning) {
@@ -5788,10 +5800,224 @@ init_api();
 init_config();
 init_scan();
 import { spawn } from "child_process";
-import { platform as platform3, release, arch as cpuArch, hostname as hostname2 } from "os";
+import { platform as platform4, release, arch as cpuArch, hostname as hostname2 } from "os";
 import { setTimeout as delay } from "timers/promises";
+
+// src/service.ts
+import { spawnSync } from "child_process";
+import {
+  copyFileSync,
+  existsSync as existsSync4,
+  mkdirSync,
+  unlinkSync,
+  writeFileSync
+} from "fs";
+import { homedir as homedir3, platform as platform2, userInfo } from "os";
+import { dirname as dirname3, join as join3 } from "path";
+import { fileURLToPath as fileURLToPath2 } from "url";
+var SERVICE_LABEL = "ai.modelstat.agent";
+var SYSTEMD_UNIT = "modelstat";
+function home() {
+  return homedir3();
+}
+function stateDir() {
+  return join3(home(), ".modelstat");
+}
+function binDir() {
+  return join3(stateDir(), "bin");
+}
+function logDir() {
+  return join3(stateDir(), "logs");
+}
+function installedCliPath() {
+  return join3(binDir(), "modelstat.mjs");
+}
+function runningCliPath() {
+  return fileURLToPath2(import.meta.url).replace(/service\.(mjs|js|ts)$/, "cli.mjs");
+}
+function installBundle() {
+  mkdirSync(binDir(), { recursive: true });
+  mkdirSync(logDir(), { recursive: true });
+  const src = runningCliPath();
+  const dest = installedCliPath();
+  if (!existsSync4(src)) {
+    throw new Error(
+      `Can't find the CLI bundle to install from (${src}). Are you running a local dev build?`
+    );
+  }
+  copyFileSync(src, dest);
+  return dest;
+}
+function nodeBinary() {
+  return process.execPath;
+}
+function plistPath() {
+  return join3(home(), "Library", "LaunchAgents", `${SERVICE_LABEL}.plist`);
+}
+function writePlist(cliPath) {
+  const p = plistPath();
+  mkdirSync(dirname3(p), { recursive: true });
+  const plist = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>Label</key><string>${SERVICE_LABEL}</string>
+  <key>ProgramArguments</key>
+  <array>
+    <string>${nodeBinary()}</string>
+    <string>${cliPath}</string>
+    <string>start</string>
+  </array>
+  <key>RunAtLoad</key><true/>
+  <key>KeepAlive</key>
+  <dict><key>SuccessfulExit</key><false/></dict>
+  <key>ThrottleInterval</key><integer>30</integer>
+  <key>StandardOutPath</key><string>${join3(logDir(), "out.log")}</string>
+  <key>StandardErrorPath</key><string>${join3(logDir(), "err.log")}</string>
+  <key>EnvironmentVariables</key>
+  <dict>
+    <key>PATH</key><string>/usr/local/bin:/opt/homebrew/bin:/usr/bin:/bin</string>
+  </dict>
+  <key>WorkingDirectory</key><string>${home()}</string>
+</dict>
+</plist>
+`;
+  writeFileSync(p, plist, { mode: 420 });
+  return p;
+}
+function launchctl(args) {
+  const r = spawnSync("launchctl", args, { encoding: "utf8" });
+  return { ok: r.status === 0, out: r.stdout ?? "", err: r.stderr ?? "" };
+}
+function macInstall() {
+  const cliPath = installBundle();
+  const plist = writePlist(cliPath);
+  const uid = userInfo().uid;
+  const target = `gui/${uid}/${SERVICE_LABEL}`;
+  launchctl(["bootout", target]);
+  const boot = launchctl(["bootstrap", `gui/${uid}`, plist]);
+  if (!boot.ok && !/already loaded|service already bootstrapped/i.test(boot.err)) {
+    const load = launchctl(["load", "-w", plist]);
+    if (!load.ok) {
+      throw new Error(
+        `launchctl load failed:
+  bootstrap: ${boot.err.trim()}
+  load: ${load.err.trim()}`
+      );
+    }
+  }
+  launchctl(["kickstart", "-k", target]);
+}
+function macUninstall() {
+  const uid = userInfo().uid;
+  const target = `gui/${uid}/${SERVICE_LABEL}`;
+  launchctl(["bootout", target]);
+  const plist = plistPath();
+  if (existsSync4(plist)) {
+    try {
+      unlinkSync(plist);
+    } catch {
+    }
+  }
+}
+function macStatus() {
+  const uid = userInfo().uid;
+  const r = launchctl(["print", `gui/${uid}/${SERVICE_LABEL}`]);
+  return { running: r.ok, hint: r.ok ? "launchd managed" : "not installed" };
+}
+function systemdUnitPath() {
+  const xdg = process.env.XDG_CONFIG_HOME ?? join3(home(), ".config");
+  return join3(xdg, "systemd", "user", `${SYSTEMD_UNIT}.service`);
+}
+function writeSystemdUnit(cliPath) {
+  const unitPath = systemdUnitPath();
+  mkdirSync(dirname3(unitPath), { recursive: true });
+  const unit = `[Unit]
+Description=modelstat agent
+Documentation=https://modelstat.ai
+After=network-online.target
+Wants=network-online.target
+
+[Service]
+Type=simple
+ExecStart=${nodeBinary()} ${cliPath} start
+Restart=always
+RestartSec=10
+# Don't restart-storm if the backend is persistently unreachable.
+StartLimitIntervalSec=300
+StartLimitBurst=10
+StandardOutput=append:${join3(logDir(), "out.log")}
+StandardError=append:${join3(logDir(), "err.log")}
+
+[Install]
+WantedBy=default.target
+`;
+  writeFileSync(unitPath, unit, { mode: 420 });
+  return unitPath;
+}
+function systemctl(args) {
+  const r = spawnSync("systemctl", ["--user", ...args], { encoding: "utf8" });
+  return { ok: r.status === 0, out: r.stdout ?? "", err: r.stderr ?? "" };
+}
+function linuxInstall() {
+  const cliPath = installBundle();
+  writeSystemdUnit(cliPath);
+  systemctl(["daemon-reload"]);
+  const en = systemctl(["enable", "--now", `${SYSTEMD_UNIT}.service`]);
+  if (!en.ok) {
+    throw new Error(`systemctl enable failed: ${en.err.trim()}`);
+  }
+  systemctl(["restart", `${SYSTEMD_UNIT}.service`]);
+}
+function linuxUninstall() {
+  systemctl(["disable", "--now", `${SYSTEMD_UNIT}.service`]);
+  const unit = systemdUnitPath();
+  if (existsSync4(unit)) {
+    try {
+      unlinkSync(unit);
+    } catch {
+    }
+  }
+  systemctl(["daemon-reload"]);
+}
+function linuxStatus() {
+  const r = systemctl(["is-active", `${SYSTEMD_UNIT}.service`]);
+  const active = r.out.trim() === "active";
+  return { running: active, hint: active ? "systemd managed" : "not running" };
+}
+function installService() {
+  const p = platform2();
+  if (p === "darwin") {
+    macInstall();
+    return { path: plistPath(), logs: logDir() };
+  }
+  if (p === "linux") {
+    linuxInstall();
+    return { path: systemdUnitPath(), logs: logDir() };
+  }
+  throw new Error(
+    `Service installation isn't supported on ${p}. Run 'modelstat start' manually to keep the agent running.`
+  );
+}
+function uninstallService() {
+  const p = platform2();
+  if (p === "darwin") return macUninstall();
+  if (p === "linux") return linuxUninstall();
+  throw new Error(`Service uninstall isn't supported on ${p}.`);
+}
+function serviceStatus() {
+  const p = platform2();
+  if (p === "darwin") return macStatus();
+  if (p === "linux") return linuxStatus();
+  return { running: false, hint: `unsupported platform (${p})` };
+}
+function logsDir() {
+  return logDir();
+}
+
+// src/cli.ts
 function tryOpenBrowser(url) {
-  const p = platform3();
+  const p = platform4();
   const cmd = p === "darwin" ? "open" : p === "win32" ? "cmd" : "xdg-open";
   const args = p === "win32" ? ["/c", "start", "", url] : [url];
   try {
@@ -5807,7 +6033,7 @@ function tryOpenBrowser(url) {
 }
 var AGENT_VERSION3 = "agent-dev-0.0.1";
 function osFamily() {
-  const p = platform3();
+  const p = platform4();
   if (p === "darwin") return "macos";
   if (p === "linux") return "linux";
   return "other";
@@ -5897,13 +6123,33 @@ async function cmdConnect() {
         }
         state.setDeviceId(enroll.device_id);
         console.log(`\u2713 device registered  (${enroll.device_id.slice(0, 8)}\u2026)`);
-        console.log();
-        console.log("Starting daemon \u2014 the dashboard takes it from here.");
-        console.log("Press Ctrl-C to stop.");
-        console.log();
-        const { runDaemon: runDaemon2 } = await Promise.resolve().then(() => (init_daemon(), daemon_exports));
-        await runDaemon2();
-        return;
+        try {
+          const svc = installService();
+          console.log(`\u2713 service installed    (${svc.path})`);
+          console.log(`  logs:                ${svc.logs}`);
+          console.log();
+          const banner = "\u2501".repeat(60);
+          console.log(banner);
+          console.log("  You're connected. The agent is now running in the");
+          console.log("  background and will start on login.");
+          console.log();
+          console.log("  Manage it from the dashboard:");
+          console.log("    \x1B[1;36mhttps://modelstat.ai/dashboard\x1B[0m");
+          console.log();
+          console.log("  Stop it any time:   \x1B[2mmodelstat stop\x1B[0m");
+          console.log("  Uninstall:          \x1B[2mmodelstat uninstall\x1B[0m");
+          console.log(banner);
+          return;
+        } catch (err) {
+          console.warn();
+          console.warn(`\u26A0 couldn't install as a background service: ${err.message}`);
+          console.warn("  Running in the foreground instead \u2014 press Ctrl-C to stop.");
+          console.warn("  (Install globally with `npm i -g modelstat` then re-run `modelstat connect`.)");
+          console.warn();
+          const { runDaemon: runDaemon2 } = await Promise.resolve().then(() => (init_daemon(), daemon_exports));
+          await runDaemon2();
+          return;
+        }
       }
     } catch (e) {
       if (dots % 15 === 0) console.warn(`
@@ -5947,6 +6193,29 @@ async function cmdStart() {
   const { runDaemon: runDaemon2 } = await Promise.resolve().then(() => (init_daemon(), daemon_exports));
   await runDaemon2();
 }
+async function cmdStop() {
+  try {
+    uninstallService();
+    console.log("\u2713 service stopped and uninstalled");
+    console.log(`  Your device pairing is still in ${state.storePath}`);
+    console.log("  Run `modelstat connect` again to re-enable.");
+  } catch (err) {
+    console.error(`\u2717 ${err.message}`);
+    process.exit(1);
+  }
+}
+async function cmdStatus() {
+  const s = serviceStatus();
+  const paired = !!state.bearer && !!state.deviceId;
+  console.log(`paired:  ${paired ? "yes" : "no"}`);
+  if (paired) {
+    console.log(`  user:    ${state.userEmail ?? "(unknown)"}`);
+    console.log(`  device:  ${state.deviceId}`);
+  }
+  console.log(`service: ${s.running ? "running" : "stopped"}  (${s.hint})`);
+  console.log(`logs:    ${logsDir()}`);
+  console.log(`state:   ${state.storePath}`);
+}
 async function main() {
   const cmd = process.argv[2];
   switch (cmd) {
@@ -5964,16 +6233,25 @@ async function main() {
       return cmdScan();
     case "watch":
       return cmdWatch();
+    case "stop":
+    case "uninstall":
+      return cmdStop();
+    case "status":
+      return cmdStatus();
     default:
-      console.log("usage: modelstat [connect|start|discover|scan|watch|register]");
+      console.log(
+        "usage: modelstat [connect|status|stop|start|discover|scan|watch|register]"
+      );
       console.log();
-      console.log("  (no args)  \u2014 pair if needed, then run the daemon (default)");
-      console.log("  connect    \u2014 pair + start daemon (recommended first run)");
-      console.log("  start      \u2014 run the daemon (requires prior pairing)");
-      console.log("  discover   \u2014 one-shot report of installs/identities");
-      console.log("  scan       \u2014 one-shot parse + upload of local JSONL");
-      console.log("  watch      \u2014 continuous (chokidar) with periodic backstop");
-      console.log("  register   \u2014 DEV shortcut (email-only, no OAuth)");
+      console.log("  (no args)   \u2014 pair if needed, then run the daemon in the foreground");
+      console.log("  connect     \u2014 pair + install as a background service, then exit");
+      console.log("  status      \u2014 show pairing + service state");
+      console.log("  stop        \u2014 stop and uninstall the background service");
+      console.log("  start       \u2014 run the daemon (used by the installed service)");
+      console.log("  discover    \u2014 one-shot report of installs/identities");
+      console.log("  scan        \u2014 one-shot parse + upload of local JSONL");
+      console.log("  watch       \u2014 continuous (chokidar) with periodic backstop");
+      console.log("  register    \u2014 DEV shortcut (email-only, no OAuth)");
       process.exit(1);
   }
 }