claude-friends 0.1.3 → 0.2.1

package/cli.js

@@ -1,6 +1,6 @@
 #!/usr/bin/env node
 
-import { writeFileSync, existsSync } from "fs";
+import { writeFileSync, readFileSync, existsSync, mkdirSync, copyFileSync, readdirSync } from "fs";
 import { join } from "path";
 import { homedir } from "os";
 import { createInterface } from "readline";
@@ -40,18 +40,65 @@ if (command === "setup") {
 
   writeFileSync(CONFIG_PATH, JSON.stringify({ username: username.trim() }, null, 2));
 
+  // Install slash commands to ~/.claude/commands/
+  const commandsDir = join(homedir(), ".claude", "commands");
+  mkdirSync(commandsDir, { recursive: true });
+
+  const srcCommands = join(__dirname, "commands");
+  if (existsSync(srcCommands)) {
+    const files = readdirSync(srcCommands).filter((f) => f.endsWith(".md"));
+    for (const file of files) {
+      copyFileSync(join(srcCommands, file), join(commandsDir, file));
+    }
+    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()}".
 
 Now add the MCP server to Claude Code:
 
-  claude mcp add claude-friends -- node ${join(__dirname, "mcp-server.js")}
+  claude mcp add claude-friends -- claude-friends serve
+
+Then in Claude Code:
+  /friend alice       Add a friend
+  /friends            See who's online
+  /nudge bob          Nudge someone
+  /status debugging   Set your status
+  /unfriend alice     Remove a friend
 
-Then in Claude Code, try:
-  "who's online?"
-  "add friend alice"
-  "set my status to debugging auth"
-  "nudge bob"
+Token usage is shared automatically with friends.
 `);
 
   rl.close();

package/commands/friend.md

@@ -0,0 +1 @@
+Use the add-friend MCP tool to add "$ARGUMENTS" as a friend. If no username is provided, ask for one.

package/commands/friends.md

@@ -0,0 +1 @@
+Use the friends-online MCP tool to show who's currently online.

package/commands/nudge.md

@@ -0,0 +1 @@
+Use the nudge MCP tool to nudge "$ARGUMENTS". If the input includes a message after the username, pass it as the message parameter. If no username is provided, ask for one.

package/commands/status.md

@@ -0,0 +1 @@
+Use the set-status MCP tool to set your status to "$ARGUMENTS". If no status is provided, ask what to set it to.

package/commands/unfriend.md

@@ -0,0 +1 @@
+Use the remove-friend MCP tool to remove "$ARGUMENTS" from your friends list. If no username is provided, ask for one.

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.1.3",
+  "version": "0.2.1",
   "description": "See who's online in Claude Code. Add friends, share status, nudge each other.",
   "type": "module",
   "bin": {
@@ -15,7 +15,9 @@
     "cli.js",
     "mcp-server.js",
     "statusline.js",
-    "client.js"
+    "client.js",
+    "commands/",
+    "hooks/"
   ],
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.12.1",