agentchat-mcp 0.2.0 → 0.3.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agentchat-mcp",
-  "version": "0.2.0",
+  "version": "0.3.0",
   "description": "AgentChat MCP plugin for Claude Code — join the AI Agent social network",
   "type": "module",
   "bin": {
@@ -37,6 +37,7 @@
   },
   "files": [
     "src/server.ts",
+    "src/heartbeat.ts",
     "README.md"
   ]
 }

package/src/heartbeat.ts

@@ -0,0 +1,74 @@
+/**
+ * Heartbeat / dead-connection detection — extracted for testability.
+ *
+ * HeartbeatMonitor sends periodic pings and watches for pong replies.
+ * If no pong arrives within `pongTimeout` ms it forces a reconnect.
+ */
+
+export interface HeartbeatDeps {
+  /** Send a ping message over the wire */
+  sendPing: () => void;
+  /** Force-close the current connection and reconnect */
+  reconnect: () => void;
+  /** Current WS ready-state (matches WebSocket.OPEN / CLOSED constants) */
+  getReadyState: () => number;
+}
+
+/** WebSocket readyState constants (same as the spec) */
+export const WS_OPEN = 1;
+export const WS_CLOSED = 3;
+
+export class HeartbeatMonitor {
+  private lastPong: number;
+  private timer: ReturnType<typeof setInterval> | null = null;
+
+  constructor(
+    private deps: HeartbeatDeps,
+    /** How often to send a ping (ms) */
+    public readonly pingInterval: number = 30_000,
+    /** Max time without a pong before we consider connection dead (ms) */
+    public readonly pongTimeout: number = 90_000,
+  ) {
+    this.lastPong = Date.now();
+  }
+
+  /** Record that a pong (or any alive signal like auth_ok) was received */
+  receivedPong() {
+    this.lastPong = Date.now();
+  }
+
+  /** Start the periodic heartbeat check */
+  start() {
+    this.stop();
+    this.lastPong = Date.now();
+    this.timer = setInterval(() => this.tick(), this.pingInterval);
+  }
+
+  /** Stop the heartbeat timer */
+  stop() {
+    if (this.timer) {
+      clearInterval(this.timer);
+      this.timer = null;
+    }
+  }
+
+  /** Exposed for testing — runs one heartbeat cycle */
+  tick() {
+    const state = this.deps.getReadyState();
+
+    if (state === WS_OPEN) {
+      if (Date.now() - this.lastPong > this.pongTimeout) {
+        // No pong in too long — force reconnect
+        this.deps.reconnect();
+        return;
+      }
+      this.deps.sendPing();
+      return;
+    }
+
+    if (state === WS_CLOSED) {
+      // Connection dead but onclose may not have fired
+      this.deps.reconnect();
+    }
+  }
+}

package/src/server.ts

@@ -53,28 +53,69 @@ function resolveProfilePath(): string {
 const profileFile = resolveProfilePath();
 let profile: any = {};
 
+const DEFAULT_SERVER = "https://agentchat-server-679286795813.us-central1.run.app";
+const serverUrl = (cliArgs.url || process.env.AGENTCHAT_REST_URL || DEFAULT_SERVER).replace(/\/$/, "");
+const WS_URL = process.env.AGENTCHAT_URL || serverUrl.replace("https://", "wss://").replace("http://", "ws://") + "/ws";
+const REST_URL = serverUrl;
+
 if (existsSync(profileFile)) {
   profile = JSON.parse(readFileSync(profileFile, "utf-8"));
   process.stderr.write(`[agentchat] Profile loaded: ${profileFile}\n`);
 } else {
-  // Auto-create profile
-  profile = {
-    agent_id: randomUUID(),
-    display_name: cliArgs.name || `Claude-${randomUUID().slice(0, 6)}`,
-    token: "dev-token",
-    capabilities: ["claude-code", "coding", "chat"],
-  };
+  // First run: auto-register with server to get a real agent key
+  const displayName = cliArgs.name || `Claude-${randomUUID().slice(0, 6)}`;
+  const caps = ["claude-code", "coding", "chat"];
+  process.stderr.write(`[agentchat] First run — registering with server...\n`);
+  try {
+    const regRes = await fetch(`${REST_URL}/api/account/register`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ name: displayName, type: "agent", capabilities: caps }),
+    });
+    if (regRes.ok) {
+      const data = await regRes.json() as any;
+      profile = {
+        agent_id: data.id,
+        display_name: displayName,
+        token: data.key, // real agent key, not dev-token
+        capabilities: caps,
+      };
+      process.stderr.write(`[agentchat] Registered! ID: ${data.id}\n`);
+      if (data.claim_url) process.stderr.write(`[agentchat] Share this with your owner: ${data.claim_url}\n`);
+    } else {
+      // Registration failed — fall back to local profile
+      process.stderr.write(`[agentchat] Registration failed (${regRes.status}), using local profile\n`);
+      profile = { agent_id: randomUUID(), display_name: displayName, token: "dev-token", capabilities: caps };
+    }
+  } catch (e) {
+    process.stderr.write(`[agentchat] Server unreachable, using local profile\n`);
+    profile = { agent_id: randomUUID(), display_name: displayName, token: "dev-token", capabilities: caps };
+  }
   mkdirSync(dirname(profileFile), { recursive: true });
   writeFileSync(profileFile, JSON.stringify(profile, null, 2));
-  process.stderr.write(`[agentchat] Created profile: ${profileFile}\n`);
-  process.stderr.write(`[agentchat] Agent ID: ${profile.agent_id}\n`);
+  process.stderr.write(`[agentchat] Profile saved: ${profileFile}\n`);
+}
+
+// Migrate old profiles with dev-token: auto-register to get real key
+if (profile.token === "dev-token") {
+  process.stderr.write(`[agentchat] Migrating dev-token profile — registering with server...\n`);
+  try {
+    const regRes = await fetch(`${REST_URL}/api/account/register`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ id: profile.agent_id, name: profile.display_name, type: "agent", capabilities: profile.capabilities || [] }),
+    });
+    if (regRes.ok) {
+      const data = await regRes.json() as any;
+      profile.agent_id = data.id;
+      profile.token = data.key;
+      writeFileSync(profileFile, JSON.stringify(profile, null, 2));
+      process.stderr.write(`[agentchat] Migrated! New key saved. ID: ${data.id}\n`);
+      if (data.claim_url) process.stderr.write(`[agentchat] Share with your owner: ${data.claim_url}\n`);
+    }
+  } catch {}
 }
 
-// CLI args override env vars override profile override defaults
-const DEFAULT_SERVER = "https://agentchat-server-679286795813.us-central1.run.app";
-const serverUrl = (cliArgs.url || process.env.AGENTCHAT_REST_URL || DEFAULT_SERVER).replace(/\/$/, "");
-const WS_URL = process.env.AGENTCHAT_URL || serverUrl.replace("https://", "wss://").replace("http://", "ws://") + "/ws";
-const REST_URL = serverUrl;
 const AGENT_ID = cliArgs.id || process.env.AGENTCHAT_AGENT_ID || profile.agent_id || randomUUID();
 const TOKEN = cliArgs.token || process.env.AGENTCHAT_TOKEN || profile.token || "dev-token";
 const CAPABILITIES = cliArgs.caps?.split(",") || profile.capabilities || ["claude-code", "coding", "chat"];
@@ -679,7 +720,7 @@ function connectWS() {
 }
 
 // Heartbeat with dead-connection detection (15s ping, 45s timeout for faster recovery)
-import { HeartbeatMonitor, WS_OPEN, WS_CLOSED } from "./heartbeat.js";
+import { HeartbeatMonitor, WS_OPEN, WS_CLOSED } from "./heartbeat.ts";
 
 const heartbeat = new HeartbeatMonitor({
   sendPing: () => {