temp

src/mcp/agent-server.ts

@@ -0,0 +1,104 @@
+#!/usr/bin/env node
+// =============================================================================
+// vibn-agent-mcp
+// -----------------------------------------------------------------------------
+// Stdio MCP server exposing the vibn-agent-runner sub-agent orchestration API.
+// This lets any MCP-speaking client (Goose, Claude Desktop, Cursor, etc.)
+// spawn Coder / PM / Marketing jobs against the vibn-agent-runner HTTP service
+// and poll their status.
+//
+// Config (env):
+//   AGENT_RUNNER_URL   (default: http://localhost:3333) — URL of the runner
+// =============================================================================
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { CallToolRequestSchema, ListToolsRequestSchema } from '@modelcontextprotocol/sdk/types.js';
+
+import * as api from '../tools/agent-api';
+import type { AgentRunnerConfig } from '../tools/agent-api';
+
+function loadConfig(): AgentRunnerConfig {
+    const runnerUrl = process.env.AGENT_RUNNER_URL?.trim() || 'http://localhost:3333';
+    return { runnerUrl };
+}
+
+const TOOL_DEFINITIONS = [
+    {
+        name: 'spawn_agent',
+        description: 'Dispatch a sub-agent job to run in the background on the vibn-agent-runner. Returns a job ID.',
+        inputSchema: {
+            type: 'object' as const,
+            properties: {
+                agent: { type: 'string', description: '"Coder", "PM", or "Marketing"' },
+                task: { type: 'string', description: 'Detailed task description for the agent' },
+                repo: { type: 'string', description: 'Gitea repo in "owner/name" format' },
+            },
+            required: ['agent', 'task', 'repo'],
+        },
+    },
+    {
+        name: 'get_job_status',
+        description: 'Check the status of a previously spawned agent job.',
+        inputSchema: {
+            type: 'object' as const,
+            properties: {
+                job_id: { type: 'string', description: 'Job ID returned by spawn_agent' },
+            },
+            required: ['job_id'],
+        },
+    },
+];
+
+async function dispatch(cfg: AgentRunnerConfig, name: string, args: Record<string, unknown>): Promise<unknown> {
+    switch (name) {
+        case 'spawn_agent':
+            return api.spawnAgent(cfg, {
+                agent: String(args.agent),
+                task: String(args.task),
+                repo: String(args.repo),
+            });
+        case 'get_job_status':
+            return api.getJobStatus(cfg, String(args.job_id));
+        default:
+            throw new Error(`Unknown tool: ${name}`);
+    }
+}
+
+function buildServer(cfg: AgentRunnerConfig): Server {
+    const server = new Server(
+        { name: 'vibn-agent-mcp', version: '0.1.0' },
+        { capabilities: { tools: {} } },
+    );
+
+    server.setRequestHandler(ListToolsRequestSchema, async () => ({ tools: TOOL_DEFINITIONS }));
+
+    server.setRequestHandler(CallToolRequestSchema, async (request) => {
+        const name = request.params.name;
+        const args = (request.params.arguments ?? {}) as Record<string, unknown>;
+        try {
+            const result = await dispatch(cfg, name, args);
+            return { content: [{ type: 'text', text: typeof result === 'string' ? result : JSON.stringify(result, null, 2) }] };
+        } catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            return { isError: true, content: [{ type: 'text', text: `Error: ${message}` }] };
+        }
+    });
+
+    return server;
+}
+
+async function main(): Promise<void> {
+    const cfg = loadConfig();
+    const server = buildServer(cfg);
+    const transport = new StdioServerTransport();
+    await server.connect(transport);
+    // eslint-disable-next-line no-console
+    console.error(`[vibn-agent-mcp] ready — ${TOOL_DEFINITIONS.length} tools exposed (runner=${cfg.runnerUrl})`);
+}
+
+main().catch((err) => {
+    // eslint-disable-next-line no-console
+    console.error('[vibn-agent-mcp] fatal:', err);
+    process.exit(1);
+});