@ouro.bot/cli 0.1.0-alpha.56 → 0.1.0-alpha.561

package/dist/heart/mcp/mcp-server.js

@@ -0,0 +1,683 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports._senseTurnCommandTimeoutMs = exports._senseTurnRetryDelays = exports.SENSE_TURN_COMMAND_TIMEOUT_MS = exports.SENSE_TURN_RETRY_DELAYS_MS = exports.SENSE_TURN_MAX_RETRIES = void 0;
+exports._setSenseTurnRetryDelays = _setSenseTurnRetryDelays;
+exports._setSenseTurnCommandTimeoutMs = _setSenseTurnCommandTimeoutMs;
+exports.createMcpServer = createMcpServer;
+exports.getToolSchemas = getToolSchemas;
+const socket_client_1 = require("../daemon/socket-client");
+const agentService = __importStar(require("../daemon/agent-service"));
+const runtime_1 = require("../../nerves/runtime");
+const session_id_resolver_1 = require("../daemon/session-id-resolver");
+const pending_1 = require("../../mind/pending");
+exports.SENSE_TURN_MAX_RETRIES = 3;
+exports.SENSE_TURN_RETRY_DELAYS_MS = [1000, 2000, 4000];
+exports.SENSE_TURN_COMMAND_TIMEOUT_MS = 10 * 60 * 1000;
+// Allow test override
+exports._senseTurnRetryDelays = exports.SENSE_TURN_RETRY_DELAYS_MS;
+function _setSenseTurnRetryDelays(delays) { exports._senseTurnRetryDelays = delays; }
+exports._senseTurnCommandTimeoutMs = exports.SENSE_TURN_COMMAND_TIMEOUT_MS;
+function _setSenseTurnCommandTimeoutMs(timeoutMs) { exports._senseTurnCommandTimeoutMs = timeoutMs; }
+async function withSenseTurnTimeout(promise, timeoutMs, command) {
+    let timer = null;
+    try {
+        return await Promise.race([
+            promise,
+            new Promise((_, reject) => {
+                timer = setTimeout(() => {
+                    const error = new Error(`MCP send_message to ${command.agent} timed out after ${timeoutMs}ms waiting for daemon response; command status is unknown.`);
+                    (0, runtime_1.emitNervesEvent)({
+                        level: "error",
+                        component: "daemon",
+                        event: "daemon.mcp_sense_turn_timeout",
+                        message: "MCP senseTurn timed out waiting for daemon response",
+                        meta: { agent: command.agent, friendId: command.friendId, sessionKey: command.sessionKey, timeoutMs },
+                    });
+                    reject(error);
+                }, timeoutMs);
+            }),
+        ]);
+    }
+    finally {
+        /* v8 ignore next -- Promise.race installs the timer synchronously; null is only a defensive cleanup guard @preserve */
+        if (timer)
+            clearTimeout(timer);
+    }
+}
+/**
+ * Send a senseTurn command to the daemon with retry logic.
+ * Retries on transient failures: empty response (daemon mid-restart),
+ * ECONNREFUSED (daemon not yet listening), ENOENT (socket not yet created).
+ */
+async function sendSenseTurnWithRetry(socketPath, command) {
+    let lastError = null;
+    /* v8 ignore start -- retry loop: functionally tested via mcp-send-message retry tests @preserve */
+    for (let attempt = 0; attempt <= exports.SENSE_TURN_MAX_RETRIES; attempt++) {
+        try {
+            const response = await withSenseTurnTimeout((0, socket_client_1.sendDaemonCommand)(socketPath, command), exports._senseTurnCommandTimeoutMs, command);
+            return response;
+        }
+        catch (error) {
+            lastError = error instanceof Error ? error : new Error(String(error));
+            const msg = lastError.message;
+            const isTransient = msg.includes("ECONNREFUSED")
+                || msg.includes("ENOENT")
+                || msg.includes("empty response")
+                || msg.includes("Empty response");
+            if (!isTransient || attempt >= exports.SENSE_TURN_MAX_RETRIES) {
+                throw lastError;
+            }
+            const delay = exports._senseTurnRetryDelays[attempt] ?? 4000;
+            (0, runtime_1.emitNervesEvent)({
+                component: "daemon",
+                event: "daemon.mcp_sense_turn_retry",
+                message: `senseTurn attempt ${attempt + 1} failed, retrying in ${delay}ms`,
+                meta: { attempt: attempt + 1, error: msg, delay },
+            });
+            await new Promise((resolve) => setTimeout(resolve, delay));
+        }
+    }
+    throw lastError ?? new Error("senseTurn failed after retries");
+    /* v8 ignore stop */
+}
+/**
+ * Maps MCP tool names to daemon command kinds.
+ */
+const TOOL_TO_COMMAND = {
+    ask: "agent.ask",
+    status: "agent.status",
+    catchup: "agent.catchup",
+    delegate: "agent.delegate",
+    get_context: "agent.getContext",
+    search_notes: "agent.searchNotes",
+    get_task: "agent.getTask",
+    check_scope: "agent.checkScope",
+    request_decision: "agent.requestDecision",
+    check_guidance: "agent.checkGuidance",
+    report_progress: "agent.reportProgress",
+    report_blocker: "agent.reportBlocker",
+    report_complete: "agent.reportComplete",
+};
+/**
+ * Create an MCP server that speaks JSON-RPC 2.0 over stdio.
+ * Handles initialize, initialized, tools/list, and tools/call.
+ * Forwards tool calls to the daemon via Unix socket.
+ */
+function createMcpServer(options) {
+    const { agent, friendId, socketPath, stdin, stdout } = options;
+    let buffer = "";
+    let running = false;
+    let useContentLengthFraming = true; // default to Content-Length, auto-detect from first message
+    // Resolve session ID once per MCP server instance for conversation continuity
+    const sessionId = (0, session_id_resolver_1.resolveSessionId)();
+    function writeResponse(response) {
+        const body = JSON.stringify(response);
+        if (useContentLengthFraming) {
+            const header = `Content-Length: ${Buffer.byteLength(body)}\r\n\r\n`;
+            stdout.write(header + body);
+        }
+        else {
+            stdout.write(body + "\n");
+        }
+    }
+    function tryParseContentLength() {
+        const headerEnd = buffer.indexOf("\r\n\r\n");
+        /* v8 ignore start -- partial header delivery only in real I/O */
+        if (headerEnd === -1)
+            return false;
+        /* v8 ignore stop */
+        const headerSection = buffer.slice(0, headerEnd);
+        const contentLengthMatch = headerSection.match(/Content-Length:\s*(\d+)/i);
+        if (!contentLengthMatch) {
+            buffer = buffer.slice(headerEnd + 4);
+            return true; // consumed invalid header, try again
+        }
+        const contentLength = parseInt(contentLengthMatch[1], 10);
+        const bodyStart = headerEnd + 4;
+        /* v8 ignore start -- partial body delivery only in real I/O */
+        if (buffer.length < bodyStart + contentLength)
+            return false;
+        /* v8 ignore stop */
+        const body = buffer.slice(bodyStart, bodyStart + contentLength);
+        buffer = buffer.slice(bodyStart + contentLength);
+        parseAndDispatch(body);
+        return true;
+    }
+    function tryParseNewlineDelimited() {
+        const newlineIdx = buffer.indexOf("\n");
+        /* v8 ignore start -- partial line delivery only in real I/O */
+        if (newlineIdx === -1)
+            return false;
+        /* v8 ignore stop */
+        const line = buffer.slice(0, newlineIdx).trim();
+        buffer = buffer.slice(newlineIdx + 1);
+        if (line.length === 0)
+            return true; // skip blank lines
+        parseAndDispatch(line);
+        return true;
+    }
+    function parseAndDispatch(body) {
+        let request;
+        try {
+            request = JSON.parse(body);
+        }
+        catch {
+            writeResponse({
+                jsonrpc: "2.0",
+                id: null,
+                error: { code: -32700, message: "Parse error" },
+            });
+            return;
+        }
+        void handleRequest(request);
+    }
+    let framingDetected = false;
+    function handleData(chunk) {
+        buffer += chunk.toString("utf-8");
+        // Auto-detect framing from first message and mirror it in responses
+        if (!framingDetected && buffer.length > 0) {
+            useContentLengthFraming = buffer.startsWith("Content-Length:");
+            framingDetected = true;
+        }
+        // Support both Content-Length framing (Claude Code) and newline-delimited JSON (Codex)
+        while (buffer.length > 0) {
+            const hasContentLength = buffer.startsWith("Content-Length:");
+            const parsed = hasContentLength ? tryParseContentLength() : tryParseNewlineDelimited();
+            /* v8 ignore start -- break on partial message only in real I/O */
+            if (!parsed)
+                break;
+            /* v8 ignore stop */
+        }
+    }
+    async function handleRequest(request) {
+        (0, runtime_1.emitNervesEvent)({
+            component: "daemon",
+            event: "daemon.mcp_request_start",
+            message: "handling MCP request",
+            meta: { method: request.method, agent },
+        });
+        // Notifications (no id) don't get responses
+        if (request.id === undefined) {
+            (0, runtime_1.emitNervesEvent)({
+                component: "daemon",
+                event: "daemon.mcp_request_end",
+                message: "handled MCP notification",
+                meta: { method: request.method, agent },
+            });
+            return;
+        }
+        switch (request.method) {
+            case "initialize":
+                await handleInitialize(request);
+                break;
+            case "tools/list":
+                handleToolsList(request);
+                break;
+            case "tools/call":
+                await handleToolsCall(request);
+                break;
+            default:
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    error: {
+                        code: -32601,
+                        message: `Method not found: ${request.method}`,
+                    },
+                });
+                break;
+        }
+        (0, runtime_1.emitNervesEvent)({
+            component: "daemon",
+            event: "daemon.mcp_request_end",
+            message: "completed MCP request",
+            meta: { method: request.method, agent },
+        });
+    }
+    async function handleInitialize(request) {
+        // MCP server works standalone (agent-service reads filesystem directly)
+        // Daemon is optional — only needed for commands without a direct service handler
+        writeResponse({
+            jsonrpc: "2.0",
+            id: request.id,
+            result: {
+                protocolVersion: "2024-11-05",
+                serverInfo: {
+                    name: "ouro-mcp-server",
+                    version: "0.1.0",
+                },
+                capabilities: {
+                    tools: { listChanged: false },
+                },
+            },
+        });
+    }
+    function handleToolsList(request) {
+        const tools = getToolSchemas();
+        writeResponse({
+            jsonrpc: "2.0",
+            id: request.id,
+            result: { tools },
+        });
+    }
+    /** Map tool name → agent-service handler function name */
+    const TOOL_TO_SERVICE = {
+        ask: "handleAgentAsk",
+        status: "handleAgentStatus",
+        catchup: "handleAgentCatchup",
+        delegate: "handleAgentDelegate",
+        get_context: "handleAgentGetContext",
+        search_notes: "handleAgentSearchNotes",
+        get_task: "handleAgentGetTask",
+        check_scope: "handleAgentCheckScope",
+        request_decision: "handleAgentRequestDecision",
+        check_guidance: "handleAgentCheckGuidance",
+        report_progress: "handleAgentReportProgress",
+        report_blocker: "handleAgentReportBlocker",
+        report_complete: "handleAgentReportComplete",
+    };
+    async function handleToolsCall(request) {
+        /* v8 ignore start — ?? fallbacks are defensive; MCP clients always send params */
+        const params = request.params ?? {};
+        const toolName = params.name;
+        const toolArgs = (params.arguments ?? {});
+        /* v8 ignore stop */
+        // ── Conversation tools: send_message, check_response ──
+        if (toolName === "send_message") {
+            /* v8 ignore start — ?? fallback defensive; MCP clients always send message */
+            const message = toolArgs.message ?? "";
+            /* v8 ignore stop */
+            try {
+                const response = await sendSenseTurnWithRetry(socketPath, {
+                    kind: "agent.senseTurn",
+                    agent,
+                    friendId,
+                    channel: "mcp",
+                    sessionKey: sessionId,
+                    message,
+                });
+                /* v8 ignore next -- branch: ?? fallback for empty daemon response @preserve */
+                const text = response.message ?? "(empty response)";
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    result: {
+                        content: [{ type: "text", text }],
+                        isError: !response.ok,
+                    },
+                });
+            }
+            catch (error) {
+                /* v8 ignore start — instanceof guard defensive; thrown errors are always Error */
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                /* v8 ignore stop */
+                /* v8 ignore start -- daemon-down detection: only triggers with real socket I/O @preserve */
+                const isDaemonDown = errorMessage.includes("ECONNREFUSED") || errorMessage.includes("ENOENT");
+                const userMessage = isDaemonDown
+                    ? "The daemon is not running. Start it with `ouro up` (production) or `ouro dev` (development), then retry."
+                    : `Error: ${errorMessage}`;
+                /* v8 ignore stop */
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    result: {
+                        content: [{ type: "text", text: userMessage }],
+                        isError: true,
+                    },
+                });
+            }
+            return;
+        }
+        if (toolName === "check_response") {
+            const pendingDir = (0, pending_1.getPendingDir)(agent, friendId, "mcp", sessionId);
+            const pending = (0, pending_1.drainPending)(pendingDir);
+            if (pending.length === 0) {
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    result: {
+                        content: [{ type: "text", text: "no pending messages" }],
+                        isError: false,
+                    },
+                });
+            }
+            else {
+                const text = pending.map((m) => m.content).join("\n\n---\n\n");
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    result: {
+                        content: [{ type: "text", text }],
+                        isError: false,
+                    },
+                });
+            }
+            return;
+        }
+        // ── delegate: full conversation turn via daemon ──
+        if (toolName === "delegate") {
+            /* v8 ignore start — ?? fallback defensive; MCP clients always send task */
+            const task = toolArgs.task ?? "";
+            /* v8 ignore stop */
+            const context = toolArgs.context;
+            const delegateMessage = context ? `[delegate] ${task}\n\ncontext: ${context}` : `[delegate] ${task}`;
+            try {
+                const response = await sendSenseTurnWithRetry(socketPath, {
+                    kind: "agent.senseTurn",
+                    agent,
+                    friendId,
+                    channel: "mcp",
+                    sessionKey: sessionId,
+                    message: delegateMessage,
+                });
+                /* v8 ignore next -- branch: ?? fallback for empty daemon response @preserve */
+                const text = response.message ?? "(empty response)";
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    result: {
+                        content: [{ type: "text", text }],
+                        isError: !response.ok,
+                    },
+                });
+            }
+            catch (error) {
+                /* v8 ignore start — instanceof guard defensive; thrown errors are always Error */
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                /* v8 ignore stop */
+                /* v8 ignore start -- daemon-down detection: only triggers with real socket I/O @preserve */
+                const isDaemonDown = errorMessage.includes("ECONNREFUSED") || errorMessage.includes("ENOENT");
+                const userMessage = isDaemonDown
+                    ? "The daemon is not running. Start it with `ouro up` (production) or `ouro dev` (development), then retry."
+                    : `Error: ${errorMessage}`;
+                /* v8 ignore stop */
+                writeResponse({
+                    jsonrpc: "2.0",
+                    id: request.id,
+                    result: {
+                        content: [{ type: "text", text: userMessage }],
+                        isError: true,
+                    },
+                });
+            }
+            return;
+        }
+        // ── Legacy daemon/service tools ──
+        const commandKind = TOOL_TO_COMMAND[toolName];
+        if (!commandKind) {
+            writeResponse({
+                jsonrpc: "2.0",
+                id: request.id,
+                result: {
+                    content: [{ type: "text", text: `Unknown tool: ${toolName}` }],
+                    isError: true,
+                },
+            });
+            return;
+        }
+        // Call agent-service directly (no daemon roundtrip needed for read-only ops)
+        const serviceHandler = TOOL_TO_SERVICE[toolName];
+        let response;
+        /* v8 ignore start — typeof guard always true; instanceof check defensive; else branch unreachable for known tools */
+        if (serviceHandler && typeof agentService[serviceHandler] === "function") {
+            const handlerFn = agentService[serviceHandler];
+            try {
+                response = await handlerFn({ agent, friendId, socketPath, ...toolArgs });
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                response = { ok: false, error: `Service error: ${errorMessage}` };
+            }
+        }
+        else {
+            try {
+                response = await (0, socket_client_1.sendDaemonCommand)(socketPath, {
+                    kind: commandKind, agent, friendId, ...toolArgs,
+                });
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                response = { ok: false, error: `Daemon error: ${errorMessage}` };
+            }
+        }
+        /* v8 ignore stop */
+        const text = response.message
+            ?? response.summary
+            ?? JSON.stringify(response.data ?? { ok: response.ok });
+        writeResponse({
+            jsonrpc: "2.0",
+            id: request.id,
+            result: {
+                content: [{ type: "text", text }],
+                isError: !response.ok,
+            },
+        });
+    }
+    function onData(chunk) {
+        handleData(chunk);
+    }
+    return {
+        agent,
+        friendId,
+        start() {
+            if (running)
+                return;
+            running = true;
+            stdin.on("data", onData);
+            (0, runtime_1.emitNervesEvent)({
+                component: "daemon",
+                event: "daemon.mcp_server_start",
+                message: "MCP server started",
+                meta: { agent, friendId, socketPath },
+            });
+        },
+        stop() {
+            if (!running)
+                return;
+            running = false;
+            stdin.removeListener("data", onData);
+            // `_end` (not `_stop`) to satisfy the nerves audit's start/end
+            // pairing rule — counterpart to `daemon.mcp_server_start`.
+            (0, runtime_1.emitNervesEvent)({
+                component: "daemon",
+                event: "daemon.mcp_server_end",
+                message: "MCP server stopped",
+                meta: { agent, friendId },
+            });
+        },
+    };
+}
+/**
+ * Returns the list of MCP tool schemas for all 15 agent tools.
+ * Each schema follows JSON Schema for inputSchema as required by MCP.
+ */
+function getToolSchemas() {
+    return [
+        {
+            name: "ask",
+            description: "Ask the agent a question. The agent uses its diary, journal, and recent session context to provide a useful answer.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    question: { type: "string", description: "The question to ask the agent" },
+                },
+                required: ["question"],
+            },
+        },
+        {
+            name: "status",
+            description: "Get the agent's current status including active sessions, diary and journal state, and activity level.",
+            inputSchema: {
+                type: "object",
+                properties: {},
+            },
+        },
+        {
+            name: "catchup",
+            description: "Get a summary of the agent's recent activity including recent sessions and what it has been working on.",
+            inputSchema: {
+                type: "object",
+                properties: {},
+            },
+        },
+        {
+            name: "delegate",
+            description: "Request the agent to handle a task. The agent queues the task and will work on it when available.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    task: { type: "string", description: "Description of the task to delegate" },
+                    context: { type: "string", description: "Additional context about the task" },
+                },
+                required: ["task"],
+            },
+        },
+        {
+            name: "get_context",
+            description: "Get the agent's current working context including note summary, active tasks, and relevant state.",
+            inputSchema: {
+                type: "object",
+                properties: {},
+            },
+        },
+        {
+            name: "search_notes",
+            description: "Search the agent's diary for information about a specific topic. Returns matching diary lines.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    query: { type: "string", description: "Search term to look for in agent notes" },
+                },
+                required: ["query"],
+            },
+        },
+        {
+            name: "get_task",
+            description: "Get details about the agent's current task or list of active tasks.",
+            inputSchema: {
+                type: "object",
+                properties: {},
+            },
+        },
+        {
+            name: "check_scope",
+            description: "Check whether a proposed item or change is in scope for the agent's current work.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    item: { type: "string", description: "The item or change to check scope for" },
+                },
+                required: ["item"],
+            },
+        },
+        {
+            name: "request_decision",
+            description: "Ask the agent to make a decision about a topic. Optionally provide a list of options to choose from.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    topic: { type: "string", description: "The topic requiring a decision" },
+                    options: { type: "string", description: "Comma-separated list of options to consider" },
+                },
+                required: ["topic"],
+            },
+        },
+        {
+            name: "check_guidance",
+            description: "Get guidance from the agent on how to approach a topic. The agent searches its diary, journal, and session context for relevant guidance.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    topic: { type: "string", description: "The topic to get guidance on" },
+                },
+                required: ["topic"],
+            },
+        },
+        {
+            name: "report_progress",
+            description: "Report progress on delegated work back to the agent. The agent records the update.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    summary: { type: "string", description: "Summary of progress made" },
+                },
+                required: ["summary"],
+            },
+        },
+        {
+            name: "report_blocker",
+            description: "Report a blocker on delegated work to the agent. The agent records the blocker for review.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    blocker: { type: "string", description: "Description of the blocker" },
+                },
+                required: ["blocker"],
+            },
+        },
+        {
+            name: "report_complete",
+            description: "Report completion of delegated work to the agent. The agent records the completion.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    summary: { type: "string", description: "Summary of what was completed" },
+                },
+                required: ["summary"],
+            },
+        },
+        {
+            name: "send_message",
+            description: "Send a message to the agent and get a synchronous response. This runs a full agent turn — the agent can use tools, think, and respond. For multi-turn conversations, call repeatedly — the agent keeps prior turns in this session.",
+            inputSchema: {
+                type: "object",
+                properties: {
+                    message: { type: "string", description: "The message to send to the agent" },
+                },
+                required: ["message"],
+            },
+        },
+        {
+            name: "check_response",
+            description: "Check for pending messages from the agent. Use this after send_message returns a ponder deferral, or to pick up proactive messages the agent has surfaced to you.",
+            inputSchema: {
+                type: "object",
+                properties: {},
+            },
+        },
+    ];
+}
+// MCP server v0.1.0-alpha.140