claude-friends 0.2.0 → 0.2.1

package/cli.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 
-import { writeFileSync, existsSync, mkdirSync, copyFileSync, readdirSync } from "fs";
+import { writeFileSync, readFileSync, existsSync, mkdirSync, copyFileSync, readdirSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
 import { createInterface } from "readline";
@@ -53,6 +53,37 @@ if (command === "setup") {
     console.log(`\nInstalled slash commands: ${files.map((f) => "/" + f.replace(".md", "")).join(", ")}`);
   }
 
+  // Install token-sharing hook to ~/.claude/settings.json
+  const settingsPath = join(homedir(), ".claude", "settings.json");
+  const hookCommand = `node ${join(__dirname, "hooks", "update-tokens.js")}`;
+  try {
+    let settings = {};
+    if (existsSync(settingsPath)) {
+      settings = JSON.parse(readFileSync(settingsPath, "utf-8"));
+    }
+    if (!settings.hooks) settings.hooks = {};
+    if (!settings.hooks.Stop) settings.hooks.Stop = [];
+
+    // Check if hook already installed
+    const alreadyInstalled = settings.hooks.Stop.some((h) =>
+      h.hooks?.some((hk) => hk.command?.includes("update-tokens"))
+    );
+
+    if (!alreadyInstalled) {
+      settings.hooks.Stop.push({
+        hooks: [{
+          type: "command",
+          command: hookCommand,
+          async: true,
+        }],
+      });
+      writeFileSync(settingsPath, JSON.stringify(settings, null, 2));
+      console.log("Installed auto token-sharing hook.");
+    }
+  } catch (err) {
+    console.log("Could not install token hook (non-critical):", err.message);
+  }
+
   console.log(`
 Done! You're "${username.trim()}".
 
@@ -66,6 +97,8 @@ Then in Claude Code:
   /nudge bob          Nudge someone
   /status debugging   Set your status
   /unfriend alice     Remove a friend
+
+Token usage is shared automatically with friends.
 `);
 
   rl.close();

package/hooks/update-tokens.js

@@ -0,0 +1,79 @@
+#!/usr/bin/env node
+
+// Hook script: reads token estimate from session, pushes to PartyKit
+// Called by Claude Code's Stop hook after each response
+
+import { readFileSync, writeFileSync, 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
+}
+
+// Read hook input from stdin
+let input = "";
+try {
+  input = readFileSync("/dev/stdin", "utf-8");
+} catch {}
+
+// 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
+}
+
+// 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;
+    }
+  }
+} catch {}
+
+sessionTokens += tokensThisCall;
+
+writeFileSync(TOKENS_PATH, JSON.stringify({
+  tokens: sessionTokens,
+  lastUpdate: Date.now(),
+}));
+
+// Push to PartyKit via HTTP (faster than WebSocket for one-shot)
+// Use the WebSocket approach since PartyKit is WS-only
+try {
+  const { default: PartySocket } = await import("partysocket");
+  const ws = new PartySocket({
+    host: PARTY_HOST,
+    room: "lobby",
+    query: { username: config.username },
+  });
+
+  ws.addEventListener("open", () => {
+    ws.send(JSON.stringify({ type: "share-tokens", tokens: sessionTokens }));
+    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.2.0",
+  "version": "0.2.1",
   "description": "See who's online in Claude Code. Add friends, share status, nudge each other.",
   "type": "module",
   "bin": {
@@ -16,7 +16,8 @@
     "mcp-server.js",
     "statusline.js",
     "client.js",
-    "commands/"
+    "commands/",
+    "hooks/"
   ],
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.12.1",