claude-friends 0.3.0 → 0.3.2

package/cli.js

@@ -165,7 +165,11 @@ Token usage is shared automatically.
     const lines = sorted.map((f) => {
       const dot = f.online ? "🟢" : "⚫";
       const status = f.status && f.status !== "offline" && f.status !== "unknown" ? ` — ${f.status}` : "";
-      const tokens = f.tokensUsed ? ` [${(f.tokensUsed / 1000).toFixed(1)}K tokens]` : "";
+      const tokens = f.tokensUsed
+        ? f.tokensUsed >= 1_000_000
+          ? ` [${(f.tokensUsed / 1_000_000).toFixed(1)}M tokens]`
+          : ` [${(f.tokensUsed / 1000).toFixed(1)}K tokens]`
+        : "";
       return `${dot} ${f.name}${status}${tokens}`;
     });
 

package/hooks/update-tokens.js

@@ -1,62 +1,86 @@
 #!/usr/bin/env node
 
-// Hook script: reads token estimate from session, pushes to PartyKit
-// Called by Claude Code's Stop hook after each response
+// Hook script: reads REAL token usage from Claude Code session files
+// and pushes it to PartyKit. Called by the Stop hook after each response.
 
-import { readFileSync, writeFileSync, existsSync } from "fs";
+import { readFileSync, readdirSync, statSync, existsSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
 
 const CONFIG_PATH = join(homedir(), ".claude-friends.json");
-const TOKENS_PATH = join(homedir(), ".claude-friends-tokens.json");
 const PARTY_HOST = "claude-friends-app.nandinitalwar.partykit.dev";
 
-// Read config
 let config;
 try {
   config = JSON.parse(readFileSync(CONFIG_PATH, "utf-8"));
 } catch {
-  process.exit(0); // silently exit if not set up
+  process.exit(0);
 }
 
-// Read hook input from stdin
-let input = "";
-try {
-  input = readFileSync("/dev/stdin", "utf-8");
-} catch {}
+// Find the most recently modified session file
+function getLatestSessionFile() {
+  const projectsDir = join(homedir(), ".claude", "projects");
+  if (!existsSync(projectsDir)) return null;
 
-// Parse the stop event to estimate tokens
-let tokensThisCall = 0;
-try {
-  const event = JSON.parse(input);
-  // Estimate based on content length — rough but functional
-  const content = JSON.stringify(event);
-  tokensThisCall = Math.ceil(content.length / 4); // ~4 chars per token
-} catch {
-  tokensThisCall = 500; // default estimate per response
+  let latestFile = null;
+  let latestMtime = 0;
+
+  try {
+    for (const dir of readdirSync(projectsDir)) {
+      const dirPath = join(projectsDir, dir);
+      try {
+        for (const file of readdirSync(dirPath)) {
+          if (!file.endsWith(".jsonl")) continue;
+          const filePath = join(dirPath, file);
+          const mtime = statSync(filePath).mtimeMs;
+          if (mtime > latestMtime) {
+            latestMtime = mtime;
+            latestFile = filePath;
+          }
+        }
+      } catch {}
+    }
+  } catch {}
+
+  return latestFile;
 }
 
-// Accumulate session tokens
-let sessionTokens = 0;
-try {
-  if (existsSync(TOKENS_PATH)) {
-    const data = JSON.parse(readFileSync(TOKENS_PATH, "utf-8"));
-    // Reset if older than 4 hours (new session)
-    if (Date.now() - data.lastUpdate < 4 * 60 * 60 * 1000) {
-      sessionTokens = data.tokens || 0;
+// Sum token usage from session file
+function getTokensFromSession(filePath) {
+  if (!filePath) return 0;
+
+  try {
+    const content = readFileSync(filePath, "utf-8");
+    const lines = content.trim().split("\n");
+
+    let totalTokens = 0;
+    for (const line of lines) {
+      try {
+        const entry = JSON.parse(line);
+        const usage = entry?.message?.usage;
+        if (usage) {
+          // Count input + output + cache writes (real cost)
+          // Exclude cache_read — those are nearly free and inflate the number
+          totalTokens +=
+            (usage.input_tokens || 0) +
+            (usage.cache_creation_input_tokens || 0) +
+            (usage.output_tokens || 0);
+        }
+      } catch {}
     }
+
+    return totalTokens;
+  } catch {
+    return 0;
   }
-} catch {}
+}
 
-sessionTokens += tokensThisCall;
+const sessionFile = getLatestSessionFile();
+const totalTokens = getTokensFromSession(sessionFile);
 
-writeFileSync(TOKENS_PATH, JSON.stringify({
-  tokens: sessionTokens,
-  lastUpdate: Date.now(),
-}));
+if (totalTokens === 0) process.exit(0);
 
-// Push to PartyKit via HTTP (faster than WebSocket for one-shot)
-// Use the WebSocket approach since PartyKit is WS-only
+// Push to PartyKit
 try {
   const { default: PartySocket } = await import("partysocket");
   const ws = new PartySocket({
@@ -66,13 +90,11 @@ try {
   });
 
   ws.addEventListener("open", () => {
-    ws.send(JSON.stringify({ type: "share-tokens", tokens: sessionTokens }));
+    ws.send(JSON.stringify({ type: "share-tokens", tokens: totalTokens }));
     setTimeout(() => { ws.close(); process.exit(0); }, 500);
   });
 
   ws.addEventListener("error", () => process.exit(0));
-
-  // Don't hang
   setTimeout(() => process.exit(0), 3000);
 } catch {
   process.exit(0);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "claude-friends",
-  "version": "0.3.0",
+  "version": "0.3.2",
   "description": "See who's online in Claude Code. Add friends, share status, nudge each other.",
   "type": "module",
   "bin": {