agent-bridge-mcp 1.1.0 → 1.2.0

package/README.md

@@ -154,6 +154,50 @@ Pass env vars through your MCP config:
 }
 ```
 
+## Automatic Context Sharing (Stop Hook)
+
+The `--ingest` flag lets you auto-capture every conversation exchange into the context directory — no manual `share_context` calls needed.
+
+**How it works:** Claude Code fires a [Stop hook](https://docs.anthropic.com/en/docs/claude-code/hooks) after every agent response. The hook pipes `session_id` and `last_assistant_message` via stdin. `agent-bridge-mcp --ingest` reads this, pairs it with the user prompt (saved by a `UserPromptSubmit` hook), strips system tags, and appends the exchange to the agent's context file.
+
+**Setup:**
+
+1. Add a `UserPromptSubmit` hook to save user prompts (required for `--ingest` to capture both sides):
+
+Add to `~/.claude/settings.json`:
+```json
+{
+  "hooks": {
+    "UserPromptSubmit": [
+      {
+        "hooks": [
+          {
+            "type": "command",
+            "command": "npx agent-bridge-mcp --capture-prompt",
+            "timeout": 5
+          }
+        ]
+      }
+    ],
+    "Stop": [
+      {
+        "hooks": [
+          {
+            "type": "command",
+            "command": "npx agent-bridge-mcp --ingest",
+            "timeout": 10
+          }
+        ]
+      }
+    ]
+  }
+}
+```
+
+Set `AGENT_BRIDGE_NAME` in your project's `.mcp.json` so the ingest knows which agent name to write as.
+
+Context is stored as the last 20 exchanges per agent, accessible via `get_context`.
+
 ## Requirements
 
 - Node.js >= 18

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-bridge-mcp",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "MCP server enabling communication between multiple Claude Code sessions via shared filesystem message bus",
   "type": "module",
   "main": "server.mjs",

package/server.mjs

@@ -586,7 +586,7 @@ async function handleToolCall(name, args) {
 // ============================================================
 
 const server = new Server(
-  { name: 'agent-bridge', version: '1.1.0' },
+  { name: 'agent-bridge', version: '1.2.0' },
   { capabilities: { tools: {} } }
 );
 
@@ -621,34 +621,179 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
 });
 
 // ============================================================
-// Main
+// CLI: --capture-prompt (UserPromptSubmit hook mode)
 // ============================================================
+// Saves user prompt to temp file for --ingest to pair with.
+// Usage: node server.mjs --capture-prompt
+
+async function runCapturePrompt() {
+  const chunks = [];
+  for await (const chunk of process.stdin) chunks.push(chunk);
+  const raw = Buffer.concat(chunks).toString('utf8');
+
+  let data;
+  try { data = JSON.parse(raw); } catch { process.exit(0); }
+
+  const sessionId = data.session_id || '';
+  const prompt = data.prompt || '';
+  if (!sessionId || !prompt) process.exit(0);
 
-async function main() {
+  // Strip system tags
+  const tagPatterns = [
+    /<system-reminder>[\s\S]*?<\/system-reminder>/g,
+    /<ide_opened_file>[\s\S]*?<\/ide_opened_file>/g,
+    /<ide_selection>[\s\S]*?<\/ide_selection>/g,
+    /<task-notification>[\s\S]*?<\/task-notification>/g,
+  ];
+  let clean = prompt;
+  for (const pat of tagPatterns) clean = clean.replace(pat, '');
+  clean = clean.replace(/\n\s*\n/g, '\n').trim();
+
+  if (clean.length < 5) process.exit(0);
+
+  const tmp = tmpdir();
+  writeFileSync(resolve(tmp, `agent-bridge-prompt-${sessionId}`), clean, 'utf8');
+  process.exit(0);
+}
+
+// ============================================================
+// CLI: --ingest (Stop hook mode)
+// ============================================================
+// Called by Claude Code Stop hook. Reads stdin JSON with
+// session_id + last_assistant_message, writes to context dir.
+// Usage: node server.mjs --ingest
+
+async function runIngest() {
   ensureDirs();
-  cleanOldMessages();
-  writeAgentFile();
-  startHeartbeat();
 
-  // Cleanup on exit
-  const cleanup = () => {
-    if (heartbeatTimer) clearInterval(heartbeatTimer);
-    removeAgentFile();
-  };
-  process.on('exit', cleanup);
-  process.on('SIGTERM', () => { cleanup(); process.exit(0); });
-  process.on('SIGINT', () => { cleanup(); process.exit(0); });
+  // Read stdin
+  const chunks = [];
+  for await (const chunk of process.stdin) chunks.push(chunk);
+  const raw = Buffer.concat(chunks).toString('utf8');
+
+  let data;
+  try {
+    data = JSON.parse(raw);
+  } catch {
+    process.exit(0);
+  }
+
+  const sessionId = data.session_id || '';
+  const assistantMsg = data.last_assistant_message || '';
+  if (!sessionId || !assistantMsg) process.exit(0);
+
+  // Read user prompt saved by --capture-prompt hook
+  const tmp = tmpdir();
+  const promptPath = resolve(tmp, `agent-bridge-prompt-${sessionId}`);
+  let userMsg = '';
+  try {
+    userMsg = readFileSync(promptPath, 'utf8');
+  } catch {}
+
+  // Strip system tags
+  const tagPatterns = [
+    /<system-reminder>[\s\S]*?<\/system-reminder>/g,
+    /<task-notification>[\s\S]*?<\/task-notification>/g,
+    /<ide_opened_file>[\s\S]*?<\/ide_opened_file>/g,
+    /<ide_selection>[\s\S]*?<\/ide_selection>/g,
+  ];
+  let cleanAssistant = assistantMsg;
+  for (const pat of tagPatterns) cleanAssistant = cleanAssistant.replace(pat, '');
+  cleanAssistant = cleanAssistant.replace(/\n\s*\n/g, '\n').trim();
+
+  if (!cleanAssistant || cleanAssistant.length < 50) process.exit(0);
+
+  // Determine agent name from env or from active agent files
+  const name = process.env.AGENT_BRIDGE_NAME || findAgentNameForSession(sessionId);
+  if (!name) process.exit(0);
+
+  // Read existing context and append new exchange
+  const safeName = name.replace(/[^a-zA-Z0-9_-]/g, '_').toLowerCase();
+  const ctxPath = resolve(CONTEXT_DIR, `${safeName}.json`);
+  let existing = { exchanges: [] };
+  try {
+    const prev = JSON.parse(readFileSync(ctxPath, 'utf8'));
+    if (prev.exchanges) existing = prev;
+    else if (prev.content) existing = { exchanges: [{ role: 'context', content: prev.content }] };
+  } catch {}
+
+  // Keep last 20 exchanges to avoid unbounded growth
+  existing.exchanges.push({
+    timestamp: new Date().toISOString(),
+    user: userMsg.slice(0, 2000),
+    assistant: cleanAssistant.slice(0, 5000)
+  });
+  if (existing.exchanges.length > 20) {
+    existing.exchanges = existing.exchanges.slice(-20);
+  }
+
+  const ctxData = JSON.stringify({
+    agentName: name,
+    project: process.cwd().split(/[\\/]/).pop(),
+    updatedAt: new Date().toISOString(),
+    sessionId,
+    exchanges: existing.exchanges
+  }, null, 2);
 
-  // Connect stdio transport
-  const transport = new StdioServerTransport();
-  await server.connect(transport);
+  const tmpPath = ctxPath + '.tmp';
+  try {
+    writeFileSync(tmpPath, ctxData);
+    renameSync(tmpPath, ctxPath);
+  } catch {
+    try { writeFileSync(ctxPath, ctxData); } catch {}
+  }
 
-  console.error(`[AgentBridge] Server running`);
-  console.error(`[AgentBridge] Agent ID: ${agentId}`);
-  console.error(`[AgentBridge] Data dir: ${BRIDGE_DIR}`);
+  process.exit(0);
 }
 
-main().catch(err => {
-  console.error(`[AgentBridge] Fatal: ${err.message}`);
-  process.exit(1);
-});
+function findAgentNameForSession(sessionId) {
+  // Look through agent files to find one from this process's cwd
+  try {
+    const files = readdirSync(AGENTS_DIR).filter(f => f.endsWith('.json'));
+    const cwd = process.cwd();
+    for (const file of files) {
+      const agent = readAgentFile(resolve(AGENTS_DIR, file));
+      if (agent && agent.name && agent.cwd === cwd) return agent.name;
+    }
+  } catch {}
+  return null;
+}
+
+// ============================================================
+// Main
+// ============================================================
+
+if (process.argv.includes('--capture-prompt')) {
+  runCapturePrompt().catch(() => process.exit(0));
+} else if (process.argv.includes('--ingest')) {
+  runIngest().catch(() => process.exit(0));
+} else {
+  async function main() {
+    ensureDirs();
+    cleanOldMessages();
+    writeAgentFile();
+    startHeartbeat();
+
+    // Cleanup on exit
+    const cleanup = () => {
+      if (heartbeatTimer) clearInterval(heartbeatTimer);
+      removeAgentFile();
+    };
+    process.on('exit', cleanup);
+    process.on('SIGTERM', () => { cleanup(); process.exit(0); });
+    process.on('SIGINT', () => { cleanup(); process.exit(0); });
+
+    // Connect stdio transport
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+
+    console.error(`[AgentBridge] Server running`);
+    console.error(`[AgentBridge] Agent ID: ${agentId}`);
+    console.error(`[AgentBridge] Data dir: ${BRIDGE_DIR}`);
+  }
+
+  main().catch(err => {
+    console.error(`[AgentBridge] Fatal: ${err.message}`);
+    process.exit(1);
+  });
+}