claude-stats 0.1.0 → 0.2.0

package/dist/index.js

@@ -1042,10 +1042,157 @@ function collectData() {
   };
 }
 
+// src/daemon.ts
+var fs3 = __toESM(require("fs"));
+var path3 = __toESM(require("path"));
+var os4 = __toESM(require("os"));
+var import_child_process3 = require("child_process");
+var PID_DIR = path3.join(os4.homedir(), ".claude-stats");
+var PID_FILE = path3.join(PID_DIR, "daemon.pid");
+var LOG_FILE = path3.join(PID_DIR, "daemon.log");
+function ensurePidDir() {
+  if (!fs3.existsSync(PID_DIR)) {
+    fs3.mkdirSync(PID_DIR, { recursive: true });
+  }
+}
+function writePid(pid) {
+  ensurePidDir();
+  fs3.writeFileSync(PID_FILE, pid.toString());
+}
+function readPid() {
+  try {
+    if (!fs3.existsSync(PID_FILE)) {
+      return null;
+    }
+    const pid = parseInt(fs3.readFileSync(PID_FILE, "utf-8").trim(), 10);
+    return isNaN(pid) ? null : pid;
+  } catch {
+    return null;
+  }
+}
+function removePid() {
+  try {
+    if (fs3.existsSync(PID_FILE)) {
+      fs3.unlinkSync(PID_FILE);
+    }
+  } catch {
+  }
+}
+function isProcessRunning(pid) {
+  try {
+    process.kill(pid, 0);
+    return true;
+  } catch {
+    return false;
+  }
+}
+function getDaemonStatus() {
+  const pid = readPid();
+  if (pid === null) {
+    return { running: false, pid: null };
+  }
+  const running = isProcessRunning(pid);
+  if (!running) {
+    removePid();
+    return { running: false, pid: null };
+  }
+  return { running: true, pid };
+}
+function startDaemon(options) {
+  const status = getDaemonStatus();
+  if (status.running) {
+    return { success: false, error: `Daemon is already running (PID: ${status.pid})` };
+  }
+  ensurePidDir();
+  const args = [
+    "start",
+    "--foreground",
+    "--server",
+    options.server,
+    "--token",
+    options.token,
+    "--interval",
+    String(options.interval || 1e4)
+  ];
+  if (options.name) {
+    args.push("--name", options.name);
+  }
+  let scriptPath = process.argv[1];
+  const logFd = fs3.openSync(LOG_FILE, "a");
+  const child = (0, import_child_process3.spawn)(process.execPath, [scriptPath, ...args], {
+    detached: true,
+    stdio: ["ignore", logFd, logFd],
+    env: {
+      ...process.env,
+      CLAUDE_STATS_DAEMON: "1"
+    }
+  });
+  child.unref();
+  if (child.pid) {
+    writePid(child.pid);
+    fs3.closeSync(logFd);
+    return { success: true, pid: child.pid };
+  }
+  fs3.closeSync(logFd);
+  return { success: false, error: "Failed to start daemon process" };
+}
+function stopDaemon() {
+  const status = getDaemonStatus();
+  if (!status.running || status.pid === null) {
+    return { success: false, error: "Daemon is not running" };
+  }
+  try {
+    process.kill(status.pid, "SIGTERM");
+    let attempts = 0;
+    while (attempts < 10 && isProcessRunning(status.pid)) {
+      (0, import_child_process3.execSync)("sleep 0.1");
+      attempts++;
+    }
+    if (isProcessRunning(status.pid)) {
+      process.kill(status.pid, "SIGKILL");
+    }
+    removePid();
+    return { success: true };
+  } catch (error) {
+    removePid();
+    return { success: false, error: `Failed to stop daemon: ${error instanceof Error ? error.message : error}` };
+  }
+}
+function getLogFile() {
+  return LOG_FILE;
+}
+function getRecentLogs(lines = 20) {
+  try {
+    if (!fs3.existsSync(LOG_FILE)) {
+      return [];
+    }
+    const content = fs3.readFileSync(LOG_FILE, "utf-8");
+    const allLines = content.split("\n").filter((l) => l.trim());
+    return allLines.slice(-lines);
+  } catch {
+    return [];
+  }
+}
+function isDaemonProcess() {
+  return process.env.CLAUDE_STATS_DAEMON === "1";
+}
+function writeDaemonPid() {
+  writePid(process.pid);
+}
+function setupDaemonCleanup() {
+  const cleanup = () => {
+    removePid();
+    process.exit(0);
+  };
+  process.on("SIGINT", cleanup);
+  process.on("SIGTERM", cleanup);
+  process.on("exit", () => removePid());
+}
+
 // src/index.ts
 var DEFAULT_SERVER = "https://cm.cban.top";
 var program = new import_commander.Command();
-program.name("claude-stats").description("Monitor and stream Claude Code usage statistics").version("0.1.0");
+program.name("claude-stats").description("Monitor and stream Claude Code usage statistics").version("0.2.0");
 function promptForInput(question, hidden = false) {
   return new Promise((resolve) => {
     const rl = readline.createInterface({
@@ -1083,12 +1230,28 @@ function promptForInput(question, hidden = false) {
     }
   });
 }
-program.command("start", { isDefault: true }).description("Start streaming usage data to the monitor server").option("-s, --server <url>", "Server URL", process.env.MONITOR_SERVER || DEFAULT_SERVER).option("-t, --token <token>", "Authentication token", process.env.MONITOR_TOKEN).option("-n, --name <name>", "Machine name (defaults to hostname)", process.env.MONITOR_NAME).option("-i, --interval <ms>", "Polling interval in milliseconds", "10000").action(async (options) => {
+program.command("start", { isDefault: true }).description("Start streaming usage data to the monitor server").option("-s, --server <url>", "Server URL", process.env.MONITOR_SERVER || DEFAULT_SERVER).option("-t, --token <token>", "Authentication token", process.env.MONITOR_TOKEN).option("-n, --name <name>", "Machine name (defaults to hostname)", process.env.MONITOR_NAME).option("-i, --interval <ms>", "Polling interval in milliseconds", "10000").option("-f, --foreground", "Run in foreground instead of background").action(async (options) => {
   let server = options.server;
   let token = options.token;
   if (server && !server.startsWith("http://") && !server.startsWith("https://")) {
     server = `https://${server}`;
   }
+  if (isDaemonProcess()) {
+    setupDaemonCleanup();
+    writeDaemonPid();
+    console.log(`[${(/* @__PURE__ */ new Date()).toISOString()}] Daemon started`);
+    console.log(`Server: ${server}`);
+    console.log(`Machine name: ${options.name || "default"}`);
+    const streamer = new Streamer({
+      server,
+      token,
+      name: options.name,
+      interval: parseInt(options.interval, 10),
+      getAccountId
+    });
+    streamer.start(collectData);
+    return;
+  }
   if (!token) {
     console.log("Claude Stats Monitor");
     console.log("====================");
@@ -1100,21 +1263,74 @@ program.command("start", { isDefault: true }).description("Start streaming usage
       process.exit(1);
     }
   }
-  console.log("Claude Stats Monitor");
-  console.log("====================");
-  fetchUsageLimits().then((limits) => {
-    if (limits) {
-      console.log(`Usage limits loaded (Session: ${limits.session?.percent_used ?? "?"}% used)`);
+  const status = getDaemonStatus();
+  if (status.running) {
+    console.log(`Daemon is already running (PID: ${status.pid})`);
+    console.log('Use "claude-stats stop" to stop it first.');
+    process.exit(1);
+  }
+  if (options.foreground) {
+    console.log("Claude Stats Monitor");
+    console.log("====================");
+    fetchUsageLimits().then((limits) => {
+      if (limits) {
+        console.log(`Usage limits loaded (Session: ${limits.session?.percent_used ?? "?"}% used)`);
+      }
+    });
+    const streamer = new Streamer({
+      server,
+      token,
+      name: options.name,
+      interval: parseInt(options.interval, 10),
+      getAccountId
+    });
+    streamer.start(collectData);
+  } else {
+    console.log("Starting Claude Stats Monitor in background...");
+    const result = startDaemon({
+      server,
+      token,
+      name: options.name,
+      interval: parseInt(options.interval, 10)
+    });
+    if (result.success) {
+      console.log(`Daemon started successfully (PID: ${result.pid})`);
+      console.log(`Logs: ${getLogFile()}`);
+      console.log("");
+      console.log('Use "claude-stats status" to check status');
+      console.log('Use "claude-stats stop" to stop the daemon');
+      console.log('Use "claude-stats logs" to view recent logs');
+    } else {
+      console.error(`Failed to start daemon: ${result.error}`);
+      process.exit(1);
     }
-  });
-  const streamer = new Streamer({
-    server,
-    token,
-    name: options.name,
-    interval: parseInt(options.interval, 10),
-    getAccountId
-  });
-  streamer.start(collectData);
+  }
+});
+program.command("stop").description("Stop the background daemon").action(() => {
+  const result = stopDaemon();
+  if (result.success) {
+    console.log("Daemon stopped successfully");
+  } else {
+    console.error(result.error);
+    process.exit(1);
+  }
+});
+program.command("status").description("Check if the daemon is running").action(() => {
+  const status = getDaemonStatus();
+  if (status.running) {
+    console.log(`Daemon is running (PID: ${status.pid})`);
+    console.log(`Logs: ${getLogFile()}`);
+  } else {
+    console.log("Daemon is not running");
+  }
+});
+program.command("logs").description("View recent daemon logs").option("-n, --lines <number>", "Number of lines to show", "20").action((options) => {
+  const lines = getRecentLogs(parseInt(options.lines, 10));
+  if (lines.length === 0) {
+    console.log("No logs found");
+  } else {
+    console.log(lines.join("\n"));
+  }
 });
 program.command("test").description("Collect and display current data without streaming").action(() => {
   console.log("Collecting data...\n");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-stats",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Monitor and stream Claude Code usage statistics to a remote dashboard",
   "keywords": [
     "claude",