vibeteam 0.2.2 → 0.2.4

package/mcp/server.js

@@ -0,0 +1,389 @@
+#!/usr/bin/env node
+
+/**
+ * VibeTeam MCP Server
+ *
+ * Exposes VibeTeam's agent orchestration API as MCP tools so that
+ * Claude Code (or any MCP-compatible AI) can control VibeTeam remotely.
+ *
+ * Usage:
+ *   claude mcp add vibeteam -- node /path/to/vibeteam/packages/cli/mcp/server.js
+ *   # or if installed globally via npm:
+ *   claude mcp add vibeteam -- vibeteam-mcp
+ *
+ * Environment variables:
+ *   VIBETEAM_PORT  — VibeTeam backend port (default: 4003)
+ */
+
+import { Server } from '@modelcontextprotocol/sdk/server/index.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+} from '@modelcontextprotocol/sdk/types.js';
+
+const BASE_URL = `http://localhost:${process.env.VIBETEAM_PORT || 4003}`;
+
+// ---------------------------------------------------------------------------
+// HTTP helpers
+// ---------------------------------------------------------------------------
+
+async function apiGet(path) {
+  const res = await fetch(`${BASE_URL}${path}`);
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`GET ${path} failed (${res.status}): ${text}`);
+  }
+  return res.json();
+}
+
+async function apiPost(path, body) {
+  const res = await fetch(`${BASE_URL}${path}`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify(body),
+  });
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`POST ${path} failed (${res.status}): ${text}`);
+  }
+  return res.json();
+}
+
+async function apiDelete(path) {
+  const res = await fetch(`${BASE_URL}${path}`, { method: 'DELETE' });
+  if (!res.ok) {
+    const text = await res.text();
+    throw new Error(`DELETE ${path} failed (${res.status}): ${text}`);
+  }
+  return res.json();
+}
+
+// ---------------------------------------------------------------------------
+// Tool definitions
+// ---------------------------------------------------------------------------
+
+const TOOLS = [
+  {
+    name: 'vibeteam_overview',
+    description:
+      'Get a full overview of the VibeTeam system: all agents with status, projects, active pipelines, and task counts by column. Use this first to understand what is going on.',
+    inputSchema: { type: 'object', properties: {}, required: [] },
+  },
+  {
+    name: 'vibeteam_list_agents',
+    description:
+      'List all VibeTeam agents (Claude Code sessions) with their current status, name, and project.',
+    inputSchema: { type: 'object', properties: {}, required: [] },
+  },
+  {
+    name: 'vibeteam_create_agent',
+    description:
+      'Spawn a new Claude Code agent in a project. Returns the new agent session.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        name: { type: 'string', description: 'Display name for the agent' },
+        projectId: { type: 'string', description: 'Project ID to work in' },
+        prompt: { type: 'string', description: 'Initial prompt to send to the agent after spawn' },
+      },
+      required: ['projectId'],
+    },
+  },
+  {
+    name: 'vibeteam_send_message',
+    description: 'Send a prompt/message to a running agent.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        sessionId: { type: 'string', description: 'Agent session ID' },
+        prompt: { type: 'string', description: 'Message to send' },
+      },
+      required: ['sessionId', 'prompt'],
+    },
+  },
+  {
+    name: 'vibeteam_agent_output',
+    description:
+      'Get the recent terminal output from an agent. Useful for checking what an agent has done.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        sessionId: { type: 'string', description: 'Agent session ID' },
+        lines: { type: 'number', description: 'Number of terminal lines to return (default 100, max 500)' },
+      },
+      required: ['sessionId'],
+    },
+  },
+  {
+    name: 'vibeteam_stop_agent',
+    description: 'Stop and remove an agent.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        sessionId: { type: 'string', description: 'Agent session ID to stop' },
+      },
+      required: ['sessionId'],
+    },
+  },
+  {
+    name: 'vibeteam_list_projects',
+    description: 'List all registered projects.',
+    inputSchema: { type: 'object', properties: {}, required: [] },
+  },
+  {
+    name: 'vibeteam_list_tasks',
+    description:
+      'List kanban tasks. Optionally filter by project ID or column.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        projectId: { type: 'string', description: 'Filter tasks by project ID' },
+        column: {
+          type: 'string',
+          description: 'Filter by column: ideas, planning, in_progress, ai_review, human_review, done',
+        },
+      },
+      required: [],
+    },
+  },
+  {
+    name: 'vibeteam_create_task',
+    description: 'Create a new kanban task.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        projectId: { type: 'string', description: 'Project ID' },
+        title: { type: 'string', description: 'Task title' },
+        description: { type: 'string', description: 'Task description' },
+        columnId: {
+          type: 'string',
+          description: 'Column to place task in (default: planning)',
+        },
+        priority: {
+          type: 'string',
+          description: 'Priority: low, medium, high, urgent (default: medium)',
+        },
+        tags: {
+          type: 'array',
+          items: { type: 'string' },
+          description: 'Tags for the task',
+        },
+      },
+      required: ['projectId', 'title'],
+    },
+  },
+  {
+    name: 'vibeteam_move_task',
+    description: 'Move a kanban task to a different column.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        taskId: { type: 'string', description: 'Task ID to move' },
+        column: {
+          type: 'string',
+          description: 'Target column: ideas, planning, in_progress, ai_review, human_review, done',
+        },
+      },
+      required: ['taskId', 'column'],
+    },
+  },
+  {
+    name: 'vibeteam_start_pipeline',
+    description:
+      'Start a planning pipeline for a kanban task. The pipeline orchestrates an agent through plan → implement → review → ship phases.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        taskId: { type: 'string', description: 'Kanban task ID' },
+        projectId: { type: 'string', description: 'Project ID' },
+      },
+      required: ['taskId', 'projectId'],
+    },
+  },
+  {
+    name: 'vibeteam_run_task',
+    description:
+      'One-shot: create a kanban task AND start a pipeline for it immediately. This is the simplest way to tell VibeTeam "do this thing".',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        projectId: { type: 'string', description: 'Project ID' },
+        title: { type: 'string', description: 'Task title' },
+        description: { type: 'string', description: 'Detailed task description' },
+        priority: { type: 'string', description: 'Priority: low, medium, high, urgent' },
+        tags: { type: 'array', items: { type: 'string' }, description: 'Tags' },
+      },
+      required: ['projectId', 'title'],
+    },
+  },
+  {
+    name: 'vibeteam_pipeline_status',
+    description: 'Get detailed status of a pipeline.',
+    inputSchema: {
+      type: 'object',
+      properties: {
+        pipelineId: { type: 'string', description: 'Pipeline ID' },
+      },
+      required: ['pipelineId'],
+    },
+  },
+  {
+    name: 'vibeteam_restart_server',
+    description:
+      'Gracefully restart the VibeTeam backend server. Sessions, pipelines, and tasks persist. Agents keep running in tmux. Use this after making changes to the backend code.',
+    inputSchema: { type: 'object', properties: {}, required: [] },
+  },
+];
+
+// ---------------------------------------------------------------------------
+// Tool handlers
+// ---------------------------------------------------------------------------
+
+async function handleTool(name, args) {
+  switch (name) {
+    case 'vibeteam_overview': {
+      const data = await apiGet('/api/overview');
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_list_agents': {
+      const data = await apiGet('/sessions');
+      const agents = (data.sessions || []).map((s) => ({
+        id: s.id,
+        name: s.name,
+        status: s.status,
+        currentTool: s.currentTool || null,
+        projectId: s.projectId || null,
+        cwd: s.cwd,
+      }));
+      return JSON.stringify({ agents }, null, 2);
+    }
+
+    case 'vibeteam_create_agent': {
+      const body = { projectId: args.projectId };
+      if (args.name) body.name = args.name;
+      if (args.prompt) body.prompt = args.prompt;
+      const data = await apiPost('/sessions', body);
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_send_message': {
+      const data = await apiPost(`/sessions/${args.sessionId}/prompt`, {
+        prompt: args.prompt,
+      });
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_agent_output': {
+      const lines = args.lines || 100;
+      const data = await apiGet(`/sessions/${args.sessionId}/output?lines=${lines}`);
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_stop_agent': {
+      const data = await apiDelete(`/sessions/${args.sessionId}`);
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_list_projects': {
+      const data = await apiGet('/projects');
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_list_tasks': {
+      let path = '/kanban/tasks';
+      const params = [];
+      if (args.projectId) params.push(`projectId=${encodeURIComponent(args.projectId)}`);
+      if (params.length) path += '?' + params.join('&');
+      const data = await apiGet(path);
+      let tasks = data.tasks || [];
+      if (args.column) {
+        tasks = tasks.filter((t) => t.columnId === args.column);
+      }
+      return JSON.stringify({ tasks }, null, 2);
+    }
+
+    case 'vibeteam_create_task': {
+      const data = await apiPost('/kanban/tasks', {
+        projectId: args.projectId,
+        title: args.title,
+        description: args.description || '',
+        columnId: args.columnId || 'planning',
+        priority: args.priority || 'medium',
+        tags: args.tags || [],
+      });
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_move_task': {
+      const data = await apiPost(`/kanban/tasks/${args.taskId}/move`, {
+        toColumn: args.column,
+      });
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_start_pipeline': {
+      const data = await apiPost('/pipelines', {
+        taskId: args.taskId,
+        projectId: args.projectId,
+      });
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_run_task': {
+      const data = await apiPost('/api/run-task', {
+        projectId: args.projectId,
+        title: args.title,
+        description: args.description || '',
+        priority: args.priority,
+        tags: args.tags,
+      });
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_pipeline_status': {
+      const data = await apiGet(`/pipelines/${args.pipelineId}`);
+      return JSON.stringify(data, null, 2);
+    }
+
+    case 'vibeteam_restart_server': {
+      const data = await apiPost('/api/restart', {});
+      return JSON.stringify(data, null, 2);
+    }
+
+    default:
+      throw new Error(`Unknown tool: ${name}`);
+  }
+}
+
+// ---------------------------------------------------------------------------
+// Server setup
+// ---------------------------------------------------------------------------
+
+const server = new Server(
+  { name: 'vibeteam', version: '0.2.3' },
+  { capabilities: { tools: {} } },
+);
+
+server.setRequestHandler(ListToolsRequestSchema, async () => ({
+  tools: TOOLS,
+}));
+
+server.setRequestHandler(CallToolRequestSchema, async (request) => {
+  const { name, arguments: args } = request.params;
+  try {
+    const result = await handleTool(name, args || {});
+    return { content: [{ type: 'text', text: result }] };
+  } catch (err) {
+    const message = err instanceof Error ? err.message : String(err);
+    return {
+      content: [{ type: 'text', text: `Error: ${message}` }],
+      isError: true,
+    };
+  }
+});
+
+// Start
+const transport = new StdioServerTransport();
+await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vibeteam",
-  "version": "0.2.2",
+  "version": "0.2.4",
   "description": "3D visualization of Claude Code as an interactive workshop",
   "author": "asara",
   "license": "MIT",
@@ -22,12 +22,14 @@
   ],
   "type": "module",
   "bin": {
-    "vibeteam": "./bin/cli.js"
+    "vibeteam": "./bin/cli.js",
+    "vibeteam-mcp": "./mcp/server.js"
   },
   "files": [
     "bin/",
     "hooks/",
     "dist/",
+    "mcp/",
     "public/",
     "!dist/**/*.map",
     "data/.gitkeep"
@@ -39,6 +41,7 @@
   },
   "dependencies": {
     "@deepgram/sdk": "^4.11.3",
+    "@modelcontextprotocol/sdk": "^1.12.1",
     "chokidar": "^4.0.0",
     "ws": "^8.18.0"
   },