claude-friends 0.3.1 → 0.3.3

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,87 @@
 #!/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
+// Only counts tokens from today. Pushes to PartyKit.
 
-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 {}
+// Today at midnight
+const todayStart = new Date();
+todayStart.setHours(0, 0, 0, 0);
+const todayISO = todayStart.toISOString();
 
-// 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
-}
+// Find all session files modified today
+function getSessionFilesModifiedToday() {
+  const projectsDir = join(homedir(), ".claude", "projects");
+  if (!existsSync(projectsDir)) return [];
 
-// 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;
+  const files = [];
+  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;
+          // Only files modified today
+          if (mtime >= todayStart.getTime()) {
+            files.push(filePath);
+          }
+        }
+      } catch {}
     }
+  } catch {}
+
+  return files;
+}
+
+// Sum today's token usage from session files
+function getTodayTokens(files) {
+  let totalTokens = 0;
+
+  for (const filePath of files) {
+    try {
+      const content = readFileSync(filePath, "utf-8");
+      for (const line of content.trim().split("\n")) {
+        try {
+          const entry = JSON.parse(line);
+          const usage = entry?.message?.usage;
+          const timestamp = entry?.timestamp;
+
+          // Only count entries from today
+          if (usage && timestamp && timestamp >= todayISO) {
+            totalTokens +=
+              (usage.input_tokens || 0) +
+              (usage.cache_creation_input_tokens || 0) +
+              (usage.output_tokens || 0);
+          }
+        } catch {}
+      }
+    } catch {}
   }
-} catch {}
 
-sessionTokens += tokensThisCall;
+  return totalTokens;
+}
+
+const files = getSessionFilesModifiedToday();
+const totalTokens = getTodayTokens(files);
 
-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 +91,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.1",
+  "version": "0.3.3",
   "description": "See who's online in Claude Code. Add friends, share status, nudge each other.",
   "type": "module",
   "bin": {