claude-friends 0.2.0 → 0.3.0

package/README.md

@@ -11,55 +11,47 @@ See who's online in Claude Code. Add friends, share what you're working on, nudg
 ```bash
 npm install -g claude-friends
 claude-friends setup
-claude mcp add claude-friends -- claude-friends serve
 ```
 
-That's it. No database, no API keys.
+That's it. No database, no API keys, no MCP server to configure.
 
-## What you get
+## Usage
 
-A status line in Claude Code showing online friends, plus these tools:
+### Slash commands (inside Claude Code)
 
-| Command | What it does |
-|---|---|
-| "who's online?" | See friends with 🟢/⚫ indicators |
-| "add friend alice" | Add someone by username |
-| "set my status to debugging auth" | Share what you're working on |
-| "nudge bob" | Poke a friend with a message |
-| "share my token usage: 45000" | Let friends see your token count |
-| "check nudges" | See if anyone poked you |
-
-## Status line
+```
+/friend alice       Add a friend
+/friends            See who's online
+/nudge bob hey!     Nudge someone
+/status debugging   Set your status
+/unfriend alice     Remove a friend
+```
 
-Add to your Claude Code settings (`~/.claude/settings.json`):
+### CLI (from any terminal)
 
-```json
-{
-  "statusLine": {
-    "type": "command",
-    "command": "node /path/to/claude-friends/statusline.js"
-  }
-}
+```bash
+claude-friends add alice
+claude-friends online
+claude-friends nudge bob "ship it!"
+claude-friends status "pair programming"
+claude-friends remove alice
 ```
 
-Or after global install:
+## Features
 
-```json
-{
-  "statusLine": {
-    "type": "command",
-    "command": "claude-friends statusline"
-  }
-}
-```
+- **Online presence** — see who's in Claude Code right now
+- **Status messages** — share what you're working on
+- **Nudges** — poke a friend with a message
+- **Token usage** — automatically shared with friends (opt-in via hook)
+- **Status line** — friend count shown in Claude Code's bottom bar
 
 ## How it works
 
 - **PartyKit** handles real-time presence via WebSockets
-- When you open Claude Code → you go online
-- When you close it → PartyKit detects the disconnect → you go offline
-- Friend lists and nudges are stored in-memory on the server
+- When you use Claude Code → you appear online
+- Friend lists and nudges are stored on the server
 - No accounts, no passwords — just a username
+- Setup auto-installs slash commands, status line, and token-sharing hook
 
 ## Self-hosting
 

package/cli.js

@@ -1,10 +1,10 @@
 #!/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";
-import { getConfig } from "./client.js";
+import { getConfig, createConnection } from "./client.js";
 import { fileURLToPath } from "url";
 import { dirname } from "path";
 
@@ -12,6 +12,37 @@ const __dirname = dirname(fileURLToPath(import.meta.url));
 const CONFIG_PATH = join(homedir(), ".claude-friends.json");
 
 const command = process.argv[2];
+const args = process.argv.slice(3).join(" ").trim();
+
+// Helper: connect, send a message, wait for response, print, exit
+function run(messageType, payload, responseType, formatter) {
+  const config = getConfig();
+  if (!config) {
+    console.log("Not set up. Run: claude-friends setup");
+    process.exit(1);
+  }
+
+  const ws = createConnection(config.username);
+
+  ws.addEventListener("open", () => {
+    ws.send(JSON.stringify({ type: messageType, ...payload }));
+  });
+
+  ws.addEventListener("message", (event) => {
+    const msg = JSON.parse(event.data);
+    if (msg.type === responseType || msg.type === "error") {
+      if (msg.type === "error") {
+        console.log("Error:", msg.message);
+      } else {
+        console.log(formatter(msg));
+      }
+      ws.close();
+      process.exit(0);
+    }
+  });
+
+  setTimeout(() => { console.log("Timeout connecting to server."); process.exit(1); }, 5000);
+}
 
 if (command === "setup") {
   const existing = getConfig();
@@ -53,43 +84,142 @@ if (command === "setup") {
     console.log(`\nInstalled slash commands: ${files.map((f) => "/" + f.replace(".md", "")).join(", ")}`);
   }
 
-  console.log(`
-Done! You're "${username.trim()}".
+  // 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 = [];
+
+    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);
+  }
 
-Now add the MCP server to Claude Code:
+  // Install statusline
+  try {
+    const settingsPath2 = join(homedir(), ".claude", "settings.json");
+    let settings = {};
+    if (existsSync(settingsPath2)) {
+      settings = JSON.parse(readFileSync(settingsPath2, "utf-8"));
+    }
+    if (!settings.statusLine) {
+      settings.statusLine = {
+        type: "command",
+        command: `node ${join(__dirname, "statusline.js")}`,
+      };
+      writeFileSync(settingsPath2, JSON.stringify(settings, null, 2));
+      console.log("Installed status line.");
+    }
+  } catch {}
 
-  claude mcp add claude-friends -- claude-friends serve
+  console.log(`
+Done! You're "${username.trim()}".
 
-Then in Claude Code:
+In Claude Code:
   /friend alice       Add a friend
   /friends            See who's online
-  /nudge bob          Nudge someone
+  /nudge bob hey!     Nudge someone
   /status debugging   Set your status
   /unfriend alice     Remove a friend
+
+Token usage is shared automatically.
 `);
 
   rl.close();
-} else if (command === "serve") {
-  // Start the MCP server directly
-  await import("./mcp-server.js");
+
+} else if (command === "add") {
+  if (!args) { console.log("Usage: claude-friends add <username>"); process.exit(1); }
+  run("add-friend", { friend: args }, "friend-added", () => `Added ${args} as a friend!`);
+
+} else if (command === "remove") {
+  if (!args) { console.log("Usage: claude-friends remove <username>"); process.exit(1); }
+  run("remove-friend", { friend: args }, "friend-removed", () => `Removed ${args}.`);
+
+} else if (command === "online" || command === "list") {
+  run("get-friends", {}, "friends-list", (msg) => {
+    const friends = msg.friends || [];
+    if (friends.length === 0) return "No friends yet. Run: claude-friends add <username>";
+
+    const sorted = [...friends].sort((a, b) => (b.online ? 1 : 0) - (a.online ? 1 : 0));
+    const onlineCount = sorted.filter((f) => f.online).length;
+
+    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]` : "";
+      return `${dot} ${f.name}${status}${tokens}`;
+    });
+
+    return `Friends (${onlineCount}/${friends.length} online):\n${lines.join("\n")}`;
+  });
+
+} else if (command === "status") {
+  if (!args) { console.log("Usage: claude-friends status <message>"); process.exit(1); }
+  const config = getConfig();
+  if (!config) { console.log("Not set up. Run: claude-friends setup"); process.exit(1); }
+  const ws = createConnection(config.username);
+  ws.addEventListener("open", () => {
+    ws.send(JSON.stringify({ type: "set-status", status: args }));
+    console.log(`Status set: "${args}"`);
+    setTimeout(() => { ws.close(); process.exit(0); }, 500);
+  });
+  setTimeout(() => process.exit(1), 5000);
+
+} else if (command === "nudge") {
+  const parts = args.split(" ");
+  const friend = parts[0];
+  const message = parts.slice(1).join(" ") || undefined;
+  if (!friend) { console.log("Usage: claude-friends nudge <username> [message]"); process.exit(1); }
+  run("nudge", { friend, message }, "nudge-sent", () => `Nudge sent to ${friend}!`);
+
 } else if (command === "whoami") {
   const config = getConfig();
-  if (!config) {
-    console.log("Not set up yet. Run: claude-friends setup");
-  } else {
-    console.log(config.username);
-  }
+  if (!config) { console.log("Not set up. Run: claude-friends setup"); process.exit(1); }
+  console.log(config.username);
+
+} else if (command === "serve") {
+  // Keep for backwards compat with anyone who set up MCP
+  await import("./mcp-server.js");
+
 } else {
   console.log(`
 claude-friends — social presence for Claude Code
 
 Commands:
-  setup     Pick a username (one-time)
-  serve     Start the MCP server (used by Claude Code)
-  whoami    Show your username
+  setup               Pick a username (one-time)
+  add <username>      Add a friend
+  remove <username>   Remove a friend
+  online              See who's online
+  status <message>    Set your status
+  nudge <user> [msg]  Nudge a friend
+  whoami              Show your username
 
 Quick start:
   claude-friends setup
-  claude mcp add claude-friends -- claude-friends serve
+  claude-friends add alice
+
+In Claude Code:
+  /friend alice
+  /friends
+  /nudge bob hey!
 `);
 }

package/commands/friend.md

@@ -1 +1 @@
-Use the add-friend MCP tool to add "$ARGUMENTS" as a friend. If no username is provided, ask for one.
+Run `claude-friends add $ARGUMENTS` and show the output. If no username is provided, ask for one first.

package/commands/friends.md

@@ -1 +1 @@
-Use the friends-online MCP tool to show who's currently online.
+Run `claude-friends online` and show the output.

package/commands/nudge.md

@@ -1 +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.
+Run `claude-friends nudge $ARGUMENTS` and show the output. If no username is provided, ask for one first.

package/commands/status.md

@@ -1 +1 @@
-Use the set-status MCP tool to set your status to "$ARGUMENTS". If no status is provided, ask what to set it to.
+Run `claude-friends status $ARGUMENTS` and show the output. If no status message is provided, ask for one first.

package/commands/unfriend.md

@@ -1 +1 @@
-Use the remove-friend MCP tool to remove "$ARGUMENTS" from your friends list. If no username is provided, ask for one.
+Run `claude-friends remove $ARGUMENTS` and show the output. If no username is provided, ask for one first.

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.3.0",
   "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",