claude-code-rehab 1.0.0

package/index.mjs

@@ -0,0 +1,117 @@
+#!/usr/bin/env node
+// Claude Code Rehab — Lightweight data server
+// Run: node server.mjs
+// Then open the app on your phone and enter this computer's IP
+
+import { createServer } from "http";
+import { createReadStream, existsSync, readdirSync, statSync, readFileSync } from "fs";
+import { join, basename } from "path";
+import { homedir, networkInterfaces } from "os";
+import { createInterface } from "readline";
+
+const PORT = 3456;
+const CLAUDE_DIR = join(homedir(), ".claude", "projects");
+
+function mapModel(m) { return m?.toLowerCase().includes("opus") ? "opus" : "sonnet"; }
+function projName(cwd) { return cwd ? basename(cwd) : "unknown"; }
+
+function dayKey(ts) { const d = new Date(ts); d.setHours(0,0,0,0); return d.getTime(); }
+
+async function parseFile(fp, dirName) {
+  return new Promise((resolve) => {
+    const fileId = basename(fp, ".jsonl");
+    let project = dirName.split("-").filter(Boolean).pop() || dirName;
+    const days = {};
+    let hasData = false;
+    const rl = createInterface({ input: createReadStream(fp, { encoding: "utf-8" }), crlfDelay: Infinity });
+    rl.on("line", (line) => {
+      if (!line.trim()) return;
+      try {
+        const obj = JSON.parse(line);
+        const ts = obj.timestamp ? new Date(obj.timestamp).getTime() : null;
+        if (!ts || isNaN(ts)) return;
+        const dk = dayKey(ts);
+        if (!days[dk]) days[dk] = { inp: 0, out: 0, msgs: 0, models: { opus: 0, sonnet: 0 }, minTs: ts, maxTs: ts };
+        const b = days[dk];
+        if (ts < b.minTs) b.minTs = ts; if (ts > b.maxTs) b.maxTs = ts;
+        if (obj.type === "user") { b.msgs++; if (obj.cwd) project = projName(obj.cwd); }
+        if (obj.type === "assistant" && obj.message) {
+          if (obj.message.model) b.models[mapModel(obj.message.model)]++;
+          if (obj.message.usage) { hasData = true; b.inp += obj.message.usage.input_tokens || 0; b.out += obj.message.usage.output_tokens || 0; }
+        }
+      } catch {}
+    });
+    rl.on("close", () => {
+      if (!hasData) return resolve(null);
+      const sessions = [];
+      for (const [dk, b] of Object.entries(days)) {
+        if (!b.inp && !b.out) continue;
+        sessions.push({
+          id: `${fileId}-${dk}`, project,
+          model: b.models.opus >= b.models.sonnet ? "opus" : "sonnet",
+          startTime: b.minTs, endTime: b.maxTs,
+          durationMin: Math.max(1, Math.round((b.maxTs - b.minTs) / 60000)),
+          inputTokens: b.inp, outputTokens: b.out, messages: b.msgs,
+        });
+      }
+      resolve(sessions.length ? sessions : null);
+    });
+    rl.on("error", () => resolve(null));
+  });
+}
+
+async function scan() {
+  if (!existsSync(CLAUDE_DIR)) return [];
+  const sessions = [];
+  const dirs = readdirSync(CLAUDE_DIR).filter(d => statSync(join(CLAUDE_DIR, d)).isDirectory());
+  for (const dir of dirs) {
+    const dp = join(CLAUDE_DIR, dir);
+    let files; try { files = readdirSync(dp).filter(f => f.endsWith(".jsonl")); } catch { continue; }
+    for (const f of files) {
+      const r = await parseFile(join(dp, f), dir);
+      if (r) sessions.push(...r);
+    }
+  }
+  return sessions.sort((a, b) => a.startTime - b.startTime);
+}
+
+function getLocalIPs() {
+  const ifs = networkInterfaces();
+  const ips = [];
+  for (const name of Object.keys(ifs)) {
+    for (const iface of ifs[name]) {
+      if (iface.family === "IPv4" && !iface.internal) ips.push(iface.address);
+    }
+  }
+  return ips;
+}
+
+let cached = null, lastScan = 0;
+
+const server = createServer(async (req, res) => {
+  res.setHeader("Access-Control-Allow-Origin", "*");
+  res.setHeader("Access-Control-Allow-Methods", "GET");
+
+  if (req.url === "/api/sessions") {
+    // Cache for 15 seconds
+    if (!cached || Date.now() - lastScan > 15000) {
+      cached = await scan();
+      lastScan = Date.now();
+    }
+    res.setHeader("Content-Type", "application/json");
+    res.end(JSON.stringify({ sessions: cached, meta: { sessionCount: cached.length, lastRefresh: lastScan, ready: true } }));
+  } else {
+    res.writeHead(200, { "Content-Type": "text/plain" });
+    res.end("Claude Code Rehab data server running. Connect from the app.");
+  }
+});
+
+server.listen(PORT, "0.0.0.0", () => {
+  const ips = getLocalIPs();
+  console.log("\n  Claude Code Rehab — Data Server\n");
+  console.log(`  Local:   http://localhost:${PORT}`);
+  ips.forEach(ip => console.log(`  Network: http://${ip}:${PORT}`));
+  console.log(`\n  Open the app on your phone and enter:`);
+  ips.forEach(ip => console.log(`    ${ip}:${PORT}`));
+  console.log("\n  Ctrl+C to stop\n");
+});

package/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "claude-code-rehab",
+  "version": "1.0.0",
+  "description": "Claude Code usage data server — connects to the Rehab dashboard",
+  "type": "module",
+  "bin": {
+    "claude-code-rehab": "./index.mjs"
+  },
+  "keywords": ["claude", "claude-code", "usage", "monitor", "rehab"],
+  "license": "MIT"
+}