open-multi-agent-kit 0.95.0 → 0.95.2

package/dist/core/mcp/client.js

@@ -0,0 +1,198 @@
+/**
+ * Minimal MCP client: initialize handshake, `tools/list`, `tools/call`.
+ *
+ * Deliberately dependency-free. The upstream SDK carries a full server
+ * implementation, resource/prompt/sampling surfaces, and its own transport
+ * stack; OMK needs a client for three methods, so the client is ~200 lines of
+ * request correlation over a transport this repo already owns.
+ *
+ * Fail-closed contract:
+ * - Every request has a deadline. A hung server rejects, it does not stall a turn.
+ * - A transport exit rejects every in-flight request with the server's stderr tail.
+ * - Calling before a completed handshake throws instead of guessing.
+ */
+import { formatJsonRpcError, isJsonRpcResponse, } from "./protocol.js";
+import { McpStdioTransport } from "./stdio-transport.js";
+/** Protocol revision this client implements. */
+export const MCP_PROTOCOL_VERSION = "2025-06-18";
+/** Default per-request deadline. */
+export const DEFAULT_REQUEST_TIMEOUT_MS = 30_000;
+/** Default handshake deadline. Servers that install on first run need more room than a normal call. */
+export const DEFAULT_HANDSHAKE_TIMEOUT_MS = 60_000;
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+export class McpClient {
+    transport;
+    pending = new Map();
+    requestTimeoutMs;
+    handshakeTimeoutMs;
+    nextId = 1;
+    initialized = false;
+    closed = false;
+    exitReason;
+    info = {};
+    options;
+    constructor(options) {
+        this.options = options;
+        this.requestTimeoutMs = options.requestTimeoutMs ?? DEFAULT_REQUEST_TIMEOUT_MS;
+        this.handshakeTimeoutMs = options.handshakeTimeoutMs ?? DEFAULT_HANDSHAKE_TIMEOUT_MS;
+        this.transport = new McpStdioTransport(options.transport, {
+            onMessage: (message) => this.handleMessage(message),
+            onExit: ({ code, signal }) => this.handleExit(code, signal),
+        });
+    }
+    get name() {
+        return this.options.name;
+    }
+    get serverInfo() {
+        return this.info;
+    }
+    get ready() {
+        return this.initialized && !this.closed;
+    }
+    /** Why the server is unusable, when it is. */
+    get failure() {
+        return this.exitReason;
+    }
+    /**
+     * Spawn the server and complete the MCP handshake. Safe to await once; a
+     * second call after a failure re-throws the recorded reason rather than
+     * silently reusing a dead process.
+     */
+    async connect() {
+        if (this.closed)
+            throw new Error(this.exitReason ?? `MCP server "${this.options.name}" is closed`);
+        if (this.initialized)
+            return;
+        this.transport.start();
+        const result = await this.request("initialize", {
+            protocolVersion: MCP_PROTOCOL_VERSION,
+            capabilities: {},
+            clientInfo: this.options.clientInfo ?? { name: "omk", version: "0.0.0" },
+        }, this.handshakeTimeoutMs);
+        if (isRecord(result) && isRecord(result.serverInfo)) {
+            const { name, version } = result.serverInfo;
+            this.info = {
+                name: typeof name === "string" ? name : undefined,
+                version: typeof version === "string" ? version : undefined,
+            };
+        }
+        this.initialized = true;
+        this.transport.send({ jsonrpc: "2.0", method: "notifications/initialized" });
+    }
+    /**
+     * Protocol-level liveness probe. Rejects when the server is closed, dead,
+     * or fails to answer within `timeoutMs`, so callers can tell a silently
+     * killed process apart from a genuinely connected one.
+     */
+    async ping(timeoutMs) {
+        this.assertReady();
+        await this.request("ping", {}, timeoutMs);
+    }
+    /** List the server's tools. Requires a completed handshake. */
+    async listTools() {
+        this.assertReady();
+        const tools = [];
+        let cursor;
+        do {
+            const result = await this.request("tools/list", cursor ? { cursor } : {});
+            if (!isRecord(result) || !Array.isArray(result.tools))
+                return tools;
+            for (const raw of result.tools) {
+                if (!isRecord(raw) || typeof raw.name !== "string" || raw.name.length === 0)
+                    continue;
+                tools.push({
+                    name: raw.name,
+                    description: typeof raw.description === "string" ? raw.description : undefined,
+                    title: typeof raw.title === "string" ? raw.title : undefined,
+                    inputSchema: isRecord(raw.inputSchema) ? raw.inputSchema : undefined,
+                });
+            }
+            cursor = typeof result.nextCursor === "string" && result.nextCursor.length > 0 ? result.nextCursor : undefined;
+        } while (cursor);
+        return tools;
+    }
+    /**
+     * Invoke a tool. A protocol-level failure rejects; a tool-level failure
+     * resolves with `isError: true`, matching MCP semantics so the model sees
+     * the server's own error text instead of a harness exception.
+     */
+    async callTool(name, args, timeoutMs) {
+        this.assertReady();
+        const result = await this.request("tools/call", { name, arguments: args ?? {} }, timeoutMs);
+        if (!isRecord(result))
+            return { content: [], isError: false };
+        const content = Array.isArray(result.content) ? result.content : [];
+        return {
+            content,
+            isError: result.isError === true,
+            structuredContent: result.structuredContent,
+        };
+    }
+    /** Terminate the server and reject anything still in flight. */
+    close() {
+        if (this.closed)
+            return;
+        this.transport.close();
+        this.handleExit(null, null);
+    }
+    assertReady() {
+        if (this.closed)
+            throw new Error(this.exitReason ?? `MCP server "${this.options.name}" is closed`);
+        if (!this.initialized)
+            throw new Error(`MCP server "${this.options.name}" is not initialized`);
+    }
+    request(method, params, timeoutMs) {
+        if (this.closed) {
+            return Promise.reject(new Error(this.exitReason ?? `MCP server "${this.options.name}" is closed`));
+        }
+        const id = this.nextId++;
+        const effectiveTimeout = timeoutMs ?? this.requestTimeoutMs;
+        return new Promise((resolve, reject) => {
+            const timer = setTimeout(() => {
+                this.pending.delete(id);
+                reject(new Error(`MCP server "${this.options.name}" timed out after ${effectiveTimeout}ms on ${method}`));
+            }, effectiveTimeout);
+            timer.unref?.();
+            this.pending.set(id, { resolve, reject, timer, method });
+            const sent = this.transport.send({ jsonrpc: "2.0", id, method, params });
+            if (!sent) {
+                this.pending.delete(id);
+                clearTimeout(timer);
+                reject(new Error(`MCP server "${this.options.name}" is not writable (${method})`));
+            }
+        });
+    }
+    handleMessage(message) {
+        if (!isJsonRpcResponse(message))
+            return; // Server-initiated requests/notifications are not used yet.
+        const response = message;
+        const pending = this.pending.get(response.id);
+        if (!pending)
+            return;
+        this.pending.delete(response.id);
+        clearTimeout(pending.timer);
+        if (response.error) {
+            pending.reject(new Error(`MCP server "${this.options.name}": ${formatJsonRpcError(response.error)}`));
+            return;
+        }
+        pending.resolve(response.result);
+    }
+    handleExit(code, signal) {
+        if (this.closed)
+            return;
+        this.closed = true;
+        this.initialized = false;
+        const detail = signal ? `signal ${signal}` : code === null ? "spawn failure" : `exit code ${code}`;
+        const stderr = this.transport.stderr.trim();
+        this.exitReason = `MCP server "${this.options.name}" stopped (${detail})${stderr ? `: ${stderr}` : ""}`;
+        const reason = new Error(this.exitReason);
+        for (const [id, pending] of this.pending) {
+            clearTimeout(pending.timer);
+            this.pending.delete(id);
+            pending.reject(reason);
+        }
+    }
+}
+//# sourceMappingURL=client.js.map

package/dist/core/mcp/client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"client.js","sourceRoot":"","sources":["../../../src/core/mcp/client.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EACN,kBAAkB,EAClB,iBAAiB,GAIjB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,iBAAiB,EAA8B,MAAM,sBAAsB,CAAC;AAErF,gDAAgD;AAChD,MAAM,CAAC,MAAM,oBAAoB,GAAG,YAAY,CAAC;AACjD,oCAAoC;AACpC,MAAM,CAAC,MAAM,0BAA0B,GAAG,MAAM,CAAC;AACjD,uGAAuG;AACvG,MAAM,CAAC,MAAM,4BAA4B,GAAG,MAAM,CAAC;AAkDnD,SAAS,QAAQ,CAAC,KAAc,EAAoC;IACnE,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAAA,CAC5E;AAED,MAAM,OAAO,SAAS;IACJ,SAAS,CAAoB;IAC7B,OAAO,GAAG,IAAI,GAAG,EAA6B,CAAC;IAC/C,gBAAgB,CAAS;IACzB,kBAAkB,CAAS;IACpC,MAAM,GAAG,CAAC,CAAC;IACX,WAAW,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,KAAK,CAAC;IACf,UAAU,CAAqB;IAC/B,IAAI,GAAkB,EAAE,CAAC;IAChB,OAAO,CAAmB;IAE3C,YAAY,OAAyB,EAAE;QACtC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,0BAA0B,CAAC;QAC/E,IAAI,CAAC,kBAAkB,GAAG,OAAO,CAAC,kBAAkB,IAAI,4BAA4B,CAAC;QACrF,IAAI,CAAC,SAAS,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,SAAS,EAAE;YACzD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC;YACnD,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC;SAC3D,CAAC,CAAC;IAAA,CACH;IAED,IAAI,IAAI,GAAW;QAClB,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC;IAAA,CACzB;IAED,IAAI,UAAU,GAAkB;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC;IAAA,CACjB;IAED,IAAI,KAAK,GAAY;QACpB,OAAO,IAAI,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IAAA,CACxC;IAED,8CAA8C;IAC9C,IAAI,OAAO,GAAuB;QACjC,OAAO,IAAI,CAAC,UAAU,CAAC;IAAA,CACvB;IAED;;;;OAIG;IACH,KAAK,CAAC,OAAO,GAAkB;QAC9B,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,aAAa,CAAC,CAAC;QACnG,IAAI,IAAI,CAAC,WAAW;YAAE,OAAO;QAC7B,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QAEvB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAChC,YAAY,EACZ;YACC,eAAe,EAAE,oBAAoB;YACrC,YAAY,EAAE,EAAE;YAChB,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE;SACxE,EACD,IAAI,CAAC,kBAAkB,CACvB,CAAC;QACF,IAAI,QAAQ,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YACrD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC;YAC5C,IAAI,CAAC,IAAI,GAAG;gBACX,IAAI,EAAE,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;gBACjD,OAAO,EAAE,OAAO,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS;aAC1D,CAAC;QACH,CAAC;QACD,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,2BAA2B,EAAE,CAAC,CAAC;IAAA,CAC7E;IAED;;;;OAIG;IACH,KAAK,CAAC,IAAI,CAAC,SAAkB,EAAiB;QAC7C,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;IAAA,CAC1C;IAED,+DAA+D;IAC/D,KAAK,CAAC,SAAS,GAA6B;QAC3C,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,MAAM,KAAK,GAAoB,EAAE,CAAC;QAClC,IAAI,MAA0B,CAAC;QAC/B,GAAG,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAC1E,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YACpE,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAChC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,OAAO,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;oBAAE,SAAS;gBACtF,KAAK,CAAC,IAAI,CAAC;oBACV,IAAI,EAAE,GAAG,CAAC,IAAI;oBACd,WAAW,EAAE,OAAO,GAAG,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;oBAC9E,KAAK,EAAE,OAAO,GAAG,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;oBAC5D,WAAW,EAAE,QAAQ,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;iBACpE,CAAC,CAAC;YACJ,CAAC;YACD,MAAM,GAAG,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,IAAI,MAAM,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAChH,CAAC,QAAQ,MAAM,EAAE;QACjB,OAAO,KAAK,CAAC;IAAA,CACb;IAED;;;;OAIG;IACH,KAAK,CAAC,QAAQ,CAAC,IAAY,EAAE,IAAa,EAAE,SAAkB,EAA8B;QAC3F,IAAI,CAAC,WAAW,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,EAAE,EAAE,SAAS,CAAC,CAAC;QAC5F,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC9D,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAE,MAAM,CAAC,OAA6B,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3F,OAAO;YACN,OAAO;YACP,OAAO,EAAE,MAAM,CAAC,OAAO,KAAK,IAAI;YAChC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;SAC3C,CAAC;IAAA,CACF;IAED,gEAAgE;IAChE,KAAK,GAAS;QACb,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;QACvB,IAAI,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAAA,CAC5B;IAEO,WAAW,GAAS;QAC3B,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,aAAa,CAAC,CAAC;QACnG,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,MAAM,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,sBAAsB,CAAC,CAAC;IAAA,CAC/F;IAEO,OAAO,CAAC,MAAc,EAAE,MAAe,EAAE,SAAkB,EAAoB;QACtF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,IAAI,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,aAAa,CAAC,CAAC,CAAC;QACpG,CAAC;QACD,MAAM,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QACzB,MAAM,gBAAgB,GAAG,SAAS,IAAI,IAAI,CAAC,gBAAgB,CAAC;QAC5D,OAAO,IAAI,OAAO,CAAU,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;YAChD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC;gBAC9B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACxB,MAAM,CAAC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,qBAAqB,gBAAgB,SAAS,MAAM,EAAE,CAAC,CAAC,CAAC;YAAA,CAC1G,EAAE,gBAAgB,CAAC,CAAC;YACrB,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;YAChB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;YAEzD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YACzE,IAAI,CAAC,IAAI,EAAE,CAAC;gBACX,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;gBACxB,YAAY,CAAC,KAAK,CAAC,CAAC;gBACpB,MAAM,CAAC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,sBAAsB,MAAM,GAAG,CAAC,CAAC,CAAC;YACpF,CAAC;QAAA,CACD,CAAC,CAAC;IAAA,CACH;IAEO,aAAa,CAAC,OAAuB,EAAQ;QACpD,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC;YAAE,OAAO,CAAC,4DAA4D;QACrG,MAAM,QAAQ,GAAoB,OAAO,CAAC;QAC1C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACjC,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC5B,IAAI,QAAQ,CAAC,KAAK,EAAE,CAAC;YACpB,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,MAAM,kBAAkB,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC;YACtG,OAAO;QACR,CAAC;QACD,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;IAAA,CACjC;IAEO,UAAU,CAAC,IAAmB,EAAE,MAAqB,EAAQ;QACpE,IAAI,IAAI,CAAC,MAAM;YAAE,OAAO;QACxB,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,CAAC,CAAC,CAAC,UAAU,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,aAAa,IAAI,EAAE,CAAC;QACnG,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC5C,IAAI,CAAC,UAAU,GAAG,eAAe,IAAI,CAAC,OAAO,CAAC,IAAI,cAAc,MAAM,IAAI,MAAM,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACxG,MAAM,MAAM,GAAG,IAAI,KAAK,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC1C,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC1C,YAAY,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAC5B,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACxB,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACxB,CAAC;IAAA,CACD;CACD","sourcesContent":["/**\n * Minimal MCP client: initialize handshake, `tools/list`, `tools/call`.\n *\n * Deliberately dependency-free. The upstream SDK carries a full server\n * implementation, resource/prompt/sampling surfaces, and its own transport\n * stack; OMK needs a client for three methods, so the client is ~200 lines of\n * request correlation over a transport this repo already owns.\n *\n * Fail-closed contract:\n * - Every request has a deadline. A hung server rejects, it does not stall a turn.\n * - A transport exit rejects every in-flight request with the server's stderr tail.\n * - Calling before a completed handshake throws instead of guessing.\n */\n\nimport {\n\tformatJsonRpcError,\n\tisJsonRpcResponse,\n\ttype JsonRpcId,\n\ttype JsonRpcMessage,\n\ttype JsonRpcResponse,\n} from \"./protocol.ts\";\nimport { McpStdioTransport, type StdioTransportOptions } from \"./stdio-transport.ts\";\n\n/** Protocol revision this client implements. */\nexport const MCP_PROTOCOL_VERSION = \"2025-06-18\";\n/** Default per-request deadline. */\nexport const DEFAULT_REQUEST_TIMEOUT_MS = 30_000;\n/** Default handshake deadline. Servers that install on first run need more room than a normal call. */\nexport const DEFAULT_HANDSHAKE_TIMEOUT_MS = 60_000;\n\nexport interface McpToolSchema {\n\treadonly name: string;\n\treadonly description?: string;\n\treadonly inputSchema?: Record<string, unknown>;\n\treadonly title?: string;\n}\n\nexport interface McpTextBlock {\n\treadonly type: \"text\";\n\treadonly text: string;\n}\n\nexport interface McpImageBlock {\n\treadonly type: \"image\";\n\treadonly data: string;\n\treadonly mimeType: string;\n}\n\nexport type McpContentBlock = McpTextBlock | McpImageBlock | { readonly type: string; readonly [key: string]: unknown };\n\nexport interface McpToolCallResult {\n\treadonly content: readonly McpContentBlock[];\n\treadonly isError: boolean;\n\treadonly structuredContent?: unknown;\n}\n\nexport interface McpServerInfo {\n\treadonly name?: string;\n\treadonly version?: string;\n}\n\nexport interface McpClientOptions {\n\t/** Server label used in error messages. */\n\treadonly name: string;\n\treadonly transport: StdioTransportOptions;\n\treadonly requestTimeoutMs?: number;\n\treadonly handshakeTimeoutMs?: number;\n\t/** Client identity reported during the handshake. */\n\treadonly clientInfo?: { readonly name: string; readonly version: string };\n}\n\ninterface PendingRequest {\n\treadonly resolve: (result: unknown) => void;\n\treadonly reject: (error: Error) => void;\n\treadonly timer: ReturnType<typeof setTimeout>;\n\treadonly method: string;\n}\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nexport class McpClient {\n\tprivate readonly transport: McpStdioTransport;\n\tprivate readonly pending = new Map<JsonRpcId, PendingRequest>();\n\tprivate readonly requestTimeoutMs: number;\n\tprivate readonly handshakeTimeoutMs: number;\n\tprivate nextId = 1;\n\tprivate initialized = false;\n\tprivate closed = false;\n\tprivate exitReason: string | undefined;\n\tprivate info: McpServerInfo = {};\n\tprivate readonly options: McpClientOptions;\n\n\tconstructor(options: McpClientOptions) {\n\t\tthis.options = options;\n\t\tthis.requestTimeoutMs = options.requestTimeoutMs ?? DEFAULT_REQUEST_TIMEOUT_MS;\n\t\tthis.handshakeTimeoutMs = options.handshakeTimeoutMs ?? DEFAULT_HANDSHAKE_TIMEOUT_MS;\n\t\tthis.transport = new McpStdioTransport(options.transport, {\n\t\t\tonMessage: (message) => this.handleMessage(message),\n\t\t\tonExit: ({ code, signal }) => this.handleExit(code, signal),\n\t\t});\n\t}\n\n\tget name(): string {\n\t\treturn this.options.name;\n\t}\n\n\tget serverInfo(): McpServerInfo {\n\t\treturn this.info;\n\t}\n\n\tget ready(): boolean {\n\t\treturn this.initialized && !this.closed;\n\t}\n\n\t/** Why the server is unusable, when it is. */\n\tget failure(): string | undefined {\n\t\treturn this.exitReason;\n\t}\n\n\t/**\n\t * Spawn the server and complete the MCP handshake. Safe to await once; a\n\t * second call after a failure re-throws the recorded reason rather than\n\t * silently reusing a dead process.\n\t */\n\tasync connect(): Promise<void> {\n\t\tif (this.closed) throw new Error(this.exitReason ?? `MCP server \"${this.options.name}\" is closed`);\n\t\tif (this.initialized) return;\n\t\tthis.transport.start();\n\n\t\tconst result = await this.request(\n\t\t\t\"initialize\",\n\t\t\t{\n\t\t\t\tprotocolVersion: MCP_PROTOCOL_VERSION,\n\t\t\t\tcapabilities: {},\n\t\t\t\tclientInfo: this.options.clientInfo ?? { name: \"omk\", version: \"0.0.0\" },\n\t\t\t},\n\t\t\tthis.handshakeTimeoutMs,\n\t\t);\n\t\tif (isRecord(result) && isRecord(result.serverInfo)) {\n\t\t\tconst { name, version } = result.serverInfo;\n\t\t\tthis.info = {\n\t\t\t\tname: typeof name === \"string\" ? name : undefined,\n\t\t\t\tversion: typeof version === \"string\" ? version : undefined,\n\t\t\t};\n\t\t}\n\t\tthis.initialized = true;\n\t\tthis.transport.send({ jsonrpc: \"2.0\", method: \"notifications/initialized\" });\n\t}\n\n\t/**\n\t * Protocol-level liveness probe. Rejects when the server is closed, dead,\n\t * or fails to answer within `timeoutMs`, so callers can tell a silently\n\t * killed process apart from a genuinely connected one.\n\t */\n\tasync ping(timeoutMs?: number): Promise<void> {\n\t\tthis.assertReady();\n\t\tawait this.request(\"ping\", {}, timeoutMs);\n\t}\n\n\t/** List the server's tools. Requires a completed handshake. */\n\tasync listTools(): Promise<McpToolSchema[]> {\n\t\tthis.assertReady();\n\t\tconst tools: McpToolSchema[] = [];\n\t\tlet cursor: string | undefined;\n\t\tdo {\n\t\t\tconst result = await this.request(\"tools/list\", cursor ? { cursor } : {});\n\t\t\tif (!isRecord(result) || !Array.isArray(result.tools)) return tools;\n\t\t\tfor (const raw of result.tools) {\n\t\t\t\tif (!isRecord(raw) || typeof raw.name !== \"string\" || raw.name.length === 0) continue;\n\t\t\t\ttools.push({\n\t\t\t\t\tname: raw.name,\n\t\t\t\t\tdescription: typeof raw.description === \"string\" ? raw.description : undefined,\n\t\t\t\t\ttitle: typeof raw.title === \"string\" ? raw.title : undefined,\n\t\t\t\t\tinputSchema: isRecord(raw.inputSchema) ? raw.inputSchema : undefined,\n\t\t\t\t});\n\t\t\t}\n\t\t\tcursor = typeof result.nextCursor === \"string\" && result.nextCursor.length > 0 ? result.nextCursor : undefined;\n\t\t} while (cursor);\n\t\treturn tools;\n\t}\n\n\t/**\n\t * Invoke a tool. A protocol-level failure rejects; a tool-level failure\n\t * resolves with `isError: true`, matching MCP semantics so the model sees\n\t * the server's own error text instead of a harness exception.\n\t */\n\tasync callTool(name: string, args: unknown, timeoutMs?: number): Promise<McpToolCallResult> {\n\t\tthis.assertReady();\n\t\tconst result = await this.request(\"tools/call\", { name, arguments: args ?? {} }, timeoutMs);\n\t\tif (!isRecord(result)) return { content: [], isError: false };\n\t\tconst content = Array.isArray(result.content) ? (result.content as McpContentBlock[]) : [];\n\t\treturn {\n\t\t\tcontent,\n\t\t\tisError: result.isError === true,\n\t\t\tstructuredContent: result.structuredContent,\n\t\t};\n\t}\n\n\t/** Terminate the server and reject anything still in flight. */\n\tclose(): void {\n\t\tif (this.closed) return;\n\t\tthis.transport.close();\n\t\tthis.handleExit(null, null);\n\t}\n\n\tprivate assertReady(): void {\n\t\tif (this.closed) throw new Error(this.exitReason ?? `MCP server \"${this.options.name}\" is closed`);\n\t\tif (!this.initialized) throw new Error(`MCP server \"${this.options.name}\" is not initialized`);\n\t}\n\n\tprivate request(method: string, params: unknown, timeoutMs?: number): Promise<unknown> {\n\t\tif (this.closed) {\n\t\t\treturn Promise.reject(new Error(this.exitReason ?? `MCP server \"${this.options.name}\" is closed`));\n\t\t}\n\t\tconst id = this.nextId++;\n\t\tconst effectiveTimeout = timeoutMs ?? this.requestTimeoutMs;\n\t\treturn new Promise<unknown>((resolve, reject) => {\n\t\t\tconst timer = setTimeout(() => {\n\t\t\t\tthis.pending.delete(id);\n\t\t\t\treject(new Error(`MCP server \"${this.options.name}\" timed out after ${effectiveTimeout}ms on ${method}`));\n\t\t\t}, effectiveTimeout);\n\t\t\ttimer.unref?.();\n\t\t\tthis.pending.set(id, { resolve, reject, timer, method });\n\n\t\t\tconst sent = this.transport.send({ jsonrpc: \"2.0\", id, method, params });\n\t\t\tif (!sent) {\n\t\t\t\tthis.pending.delete(id);\n\t\t\t\tclearTimeout(timer);\n\t\t\t\treject(new Error(`MCP server \"${this.options.name}\" is not writable (${method})`));\n\t\t\t}\n\t\t});\n\t}\n\n\tprivate handleMessage(message: JsonRpcMessage): void {\n\t\tif (!isJsonRpcResponse(message)) return; // Server-initiated requests/notifications are not used yet.\n\t\tconst response: JsonRpcResponse = message;\n\t\tconst pending = this.pending.get(response.id);\n\t\tif (!pending) return;\n\t\tthis.pending.delete(response.id);\n\t\tclearTimeout(pending.timer);\n\t\tif (response.error) {\n\t\t\tpending.reject(new Error(`MCP server \"${this.options.name}\": ${formatJsonRpcError(response.error)}`));\n\t\t\treturn;\n\t\t}\n\t\tpending.resolve(response.result);\n\t}\n\n\tprivate handleExit(code: number | null, signal: string | null): void {\n\t\tif (this.closed) return;\n\t\tthis.closed = true;\n\t\tthis.initialized = false;\n\t\tconst detail = signal ? `signal ${signal}` : code === null ? \"spawn failure\" : `exit code ${code}`;\n\t\tconst stderr = this.transport.stderr.trim();\n\t\tthis.exitReason = `MCP server \"${this.options.name}\" stopped (${detail})${stderr ? `: ${stderr}` : \"\"}`;\n\t\tconst reason = new Error(this.exitReason);\n\t\tfor (const [id, pending] of this.pending) {\n\t\t\tclearTimeout(pending.timer);\n\t\t\tthis.pending.delete(id);\n\t\t\tpending.reject(reason);\n\t\t}\n\t}\n}\n"]}

package/dist/core/mcp/config.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Runtime MCP server configuration loader.
+ *
+ * `mcp-inventory.ts` deliberately strips env *values* so its output is safe to
+ * render; a client that has to spawn the server needs those values, so loading
+ * is split rather than weakening the inventory's redaction guarantee. The
+ * objects returned here are runtime-only — never log or render them directly.
+ *
+ * Source precedence matches the inventory exactly (later wins):
+ *   1. ~/.kimi/mcp.json
+ *   2. ~/.omk/mcp.json
+ *   3. <cwd>/.omk/mcp.json
+ */
+import type { McpServerConfig } from "./manager.ts";
+/** Config file paths consulted, in precedence order. */
+export declare function mcpConfigPaths(cwd?: string, home?: string): string[];
+/**
+ * Load spawnable stdio server configs. Entries without a `command` (HTTP/SSE
+ * servers) are skipped rather than failing the load, so an unsupported
+ * transport in the config cannot disable every other server.
+ */
+export declare function loadMcpServerConfigs(cwd?: string, home?: string): McpServerConfig[];
+//# sourceMappingURL=config.d.ts.map

package/dist/core/mcp/config.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../../src/core/mcp/config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAKH,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAgDpD,wDAAwD;AACxD,wBAAgB,cAAc,CAAC,GAAG,GAAE,MAAsB,EAAE,IAAI,GAAE,MAAqB,GAAG,MAAM,EAAE,CAMjG;AAED;;;;GAIG;AACH,wBAAgB,oBAAoB,CAAC,GAAG,GAAE,MAAsB,EAAE,IAAI,GAAE,MAAqB,GAAG,eAAe,EAAE,CAUhH","sourcesContent":["/**\n * Runtime MCP server configuration loader.\n *\n * `mcp-inventory.ts` deliberately strips env *values* so its output is safe to\n * render; a client that has to spawn the server needs those values, so loading\n * is split rather than weakening the inventory's redaction guarantee. The\n * objects returned here are runtime-only — never log or render them directly.\n *\n * Source precedence matches the inventory exactly (later wins):\n *   1. ~/.kimi/mcp.json\n *   2. ~/.omk/mcp.json\n *   3. <cwd>/.omk/mcp.json\n */\n\nimport fs from \"node:fs\";\nimport os from \"node:os\";\nimport path from \"node:path\";\nimport type { McpServerConfig } from \"./manager.ts\";\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nfunction readJson(filePath: string): unknown {\n\ttry {\n\t\treturn JSON.parse(fs.readFileSync(filePath, \"utf8\"));\n\t} catch {\n\t\t// A missing or malformed config file yields no servers, never a crash.\n\t\treturn undefined;\n\t}\n}\n\nfunction extractServers(raw: unknown): Record<string, unknown> {\n\tif (!isRecord(raw)) return {};\n\tconst candidate = raw.mcpServers ?? raw.servers ?? raw.mcp_servers;\n\treturn isRecord(candidate) ? candidate : {};\n}\n\nfunction toStringRecord(value: unknown): Record<string, string> | undefined {\n\tif (!isRecord(value)) return undefined;\n\tconst out: Record<string, string> = {};\n\tfor (const [key, item] of Object.entries(value)) {\n\t\tif (typeof item === \"string\") out[key] = item;\n\t}\n\treturn Object.keys(out).length > 0 ? out : undefined;\n}\n\nfunction toServerConfig(name: string, raw: unknown): McpServerConfig | undefined {\n\tif (!isRecord(raw)) return undefined;\n\tconst command = raw.command;\n\t// Only stdio servers are supported; an entry with a `url` is a different transport.\n\tif (typeof command !== \"string\" || command.length === 0) return undefined;\n\tconst args = Array.isArray(raw.args) ? raw.args.filter((arg): arg is string => typeof arg === \"string\") : undefined;\n\tconst startupTimeoutSec = typeof raw.startup_timeout_sec === \"number\" ? raw.startup_timeout_sec : undefined;\n\treturn {\n\t\tname,\n\t\tcommand,\n\t\targs,\n\t\tenv: toStringRecord(raw.env),\n\t\tcwd: typeof raw.cwd === \"string\" ? raw.cwd : undefined,\n\t\tdisabled: raw.disabled === true || raw.enabled === false,\n\t\thandshakeTimeoutMs: startupTimeoutSec !== undefined ? Math.max(1, startupTimeoutSec) * 1000 : undefined,\n\t};\n}\n\n/** Config file paths consulted, in precedence order. */\nexport function mcpConfigPaths(cwd: string = process.cwd(), home: string = os.homedir()): string[] {\n\treturn [\n\t\tpath.join(home, \".kimi\", \"mcp.json\"),\n\t\tpath.join(home, \".omk\", \"mcp.json\"),\n\t\tpath.join(cwd, \".omk\", \"mcp.json\"),\n\t];\n}\n\n/**\n * Load spawnable stdio server configs. Entries without a `command` (HTTP/SSE\n * servers) are skipped rather than failing the load, so an unsupported\n * transport in the config cannot disable every other server.\n */\nexport function loadMcpServerConfigs(cwd: string = process.cwd(), home: string = os.homedir()): McpServerConfig[] {\n\tconst merged = new Map<string, McpServerConfig>();\n\tfor (const filePath of mcpConfigPaths(cwd, home)) {\n\t\tconst servers = extractServers(readJson(filePath));\n\t\tfor (const name of Object.keys(servers).sort()) {\n\t\t\tconst config = toServerConfig(name, servers[name]);\n\t\t\tif (config) merged.set(name, config);\n\t\t}\n\t}\n\treturn [...merged.values()].sort((a, b) => a.name.localeCompare(b.name));\n}\n"]}

package/dist/core/mcp/config.js

@@ -0,0 +1,89 @@
+/**
+ * Runtime MCP server configuration loader.
+ *
+ * `mcp-inventory.ts` deliberately strips env *values* so its output is safe to
+ * render; a client that has to spawn the server needs those values, so loading
+ * is split rather than weakening the inventory's redaction guarantee. The
+ * objects returned here are runtime-only — never log or render them directly.
+ *
+ * Source precedence matches the inventory exactly (later wins):
+ *   1. ~/.kimi/mcp.json
+ *   2. ~/.omk/mcp.json
+ *   3. <cwd>/.omk/mcp.json
+ */
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function readJson(filePath) {
+    try {
+        return JSON.parse(fs.readFileSync(filePath, "utf8"));
+    }
+    catch {
+        // A missing or malformed config file yields no servers, never a crash.
+        return undefined;
+    }
+}
+function extractServers(raw) {
+    if (!isRecord(raw))
+        return {};
+    const candidate = raw.mcpServers ?? raw.servers ?? raw.mcp_servers;
+    return isRecord(candidate) ? candidate : {};
+}
+function toStringRecord(value) {
+    if (!isRecord(value))
+        return undefined;
+    const out = {};
+    for (const [key, item] of Object.entries(value)) {
+        if (typeof item === "string")
+            out[key] = item;
+    }
+    return Object.keys(out).length > 0 ? out : undefined;
+}
+function toServerConfig(name, raw) {
+    if (!isRecord(raw))
+        return undefined;
+    const command = raw.command;
+    // Only stdio servers are supported; an entry with a `url` is a different transport.
+    if (typeof command !== "string" || command.length === 0)
+        return undefined;
+    const args = Array.isArray(raw.args) ? raw.args.filter((arg) => typeof arg === "string") : undefined;
+    const startupTimeoutSec = typeof raw.startup_timeout_sec === "number" ? raw.startup_timeout_sec : undefined;
+    return {
+        name,
+        command,
+        args,
+        env: toStringRecord(raw.env),
+        cwd: typeof raw.cwd === "string" ? raw.cwd : undefined,
+        disabled: raw.disabled === true || raw.enabled === false,
+        handshakeTimeoutMs: startupTimeoutSec !== undefined ? Math.max(1, startupTimeoutSec) * 1000 : undefined,
+    };
+}
+/** Config file paths consulted, in precedence order. */
+export function mcpConfigPaths(cwd = process.cwd(), home = os.homedir()) {
+    return [
+        path.join(home, ".kimi", "mcp.json"),
+        path.join(home, ".omk", "mcp.json"),
+        path.join(cwd, ".omk", "mcp.json"),
+    ];
+}
+/**
+ * Load spawnable stdio server configs. Entries without a `command` (HTTP/SSE
+ * servers) are skipped rather than failing the load, so an unsupported
+ * transport in the config cannot disable every other server.
+ */
+export function loadMcpServerConfigs(cwd = process.cwd(), home = os.homedir()) {
+    const merged = new Map();
+    for (const filePath of mcpConfigPaths(cwd, home)) {
+        const servers = extractServers(readJson(filePath));
+        for (const name of Object.keys(servers).sort()) {
+            const config = toServerConfig(name, servers[name]);
+            if (config)
+                merged.set(name, config);
+        }
+    }
+    return [...merged.values()].sort((a, b) => a.name.localeCompare(b.name));
+}
+//# sourceMappingURL=config.js.map

package/dist/core/mcp/config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../../src/core/mcp/config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,IAAI,MAAM,WAAW,CAAC;AAG7B,SAAS,QAAQ,CAAC,KAAc,EAAoC;IACnE,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAAA,CAC5E;AAED,SAAS,QAAQ,CAAC,QAAgB,EAAW;IAC5C,IAAI,CAAC;QACJ,OAAO,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IACtD,CAAC;IAAC,MAAM,CAAC;QACR,uEAAuE;QACvE,OAAO,SAAS,CAAC;IAClB,CAAC;AAAA,CACD;AAED,SAAS,cAAc,CAAC,GAAY,EAA2B;IAC9D,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IAC9B,MAAM,SAAS,GAAG,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,OAAO,IAAI,GAAG,CAAC,WAAW,CAAC;IACnE,OAAO,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;AAAA,CAC5C;AAED,SAAS,cAAc,CAAC,KAAc,EAAsC;IAC3E,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IACvC,MAAM,GAAG,GAA2B,EAAE,CAAC;IACvC,KAAK,MAAM,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACjD,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,GAAG,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC;IAC/C,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC;AAAA,CACrD;AAED,SAAS,cAAc,CAAC,IAAY,EAAE,GAAY,EAA+B;IAChF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,SAAS,CAAC;IACrC,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;IAC5B,oFAAoF;IACpF,IAAI,OAAO,OAAO,KAAK,QAAQ,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC1E,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAiB,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACpH,MAAM,iBAAiB,GAAG,OAAO,GAAG,CAAC,mBAAmB,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5G,OAAO;QACN,IAAI;QACJ,OAAO;QACP,IAAI;QACJ,GAAG,EAAE,cAAc,CAAC,GAAG,CAAC,GAAG,CAAC;QAC5B,GAAG,EAAE,OAAO,GAAG,CAAC,GAAG,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS;QACtD,QAAQ,EAAE,GAAG,CAAC,QAAQ,KAAK,IAAI,IAAI,GAAG,CAAC,OAAO,KAAK,KAAK;QACxD,kBAAkB,EAAE,iBAAiB,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,iBAAiB,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,SAAS;KACvG,CAAC;AAAA,CACF;AAED,wDAAwD;AACxD,MAAM,UAAU,cAAc,CAAC,GAAG,GAAW,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,GAAW,EAAE,CAAC,OAAO,EAAE,EAAY;IAClG,OAAO;QACN,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,UAAU,CAAC;QACpC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,CAAC;QACnC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,UAAU,CAAC;KAClC,CAAC;AAAA,CACF;AAED;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CAAC,GAAG,GAAW,OAAO,CAAC,GAAG,EAAE,EAAE,IAAI,GAAW,EAAE,CAAC,OAAO,EAAE,EAAqB;IACjH,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;IAClD,KAAK,MAAM,QAAQ,IAAI,cAAc,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,CAAC;QAClD,MAAM,OAAO,GAAG,cAAc,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACnD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAChD,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC;YACnD,IAAI,MAAM;gBAAE,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QACtC,CAAC;IACF,CAAC;IACD,OAAO,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;AAAA,CACzE","sourcesContent":["/**\n * Runtime MCP server configuration loader.\n *\n * `mcp-inventory.ts` deliberately strips env *values* so its output is safe to\n * render; a client that has to spawn the server needs those values, so loading\n * is split rather than weakening the inventory's redaction guarantee. The\n * objects returned here are runtime-only — never log or render them directly.\n *\n * Source precedence matches the inventory exactly (later wins):\n *   1. ~/.kimi/mcp.json\n *   2. ~/.omk/mcp.json\n *   3. <cwd>/.omk/mcp.json\n */\n\nimport fs from \"node:fs\";\nimport os from \"node:os\";\nimport path from \"node:path\";\nimport type { McpServerConfig } from \"./manager.ts\";\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\nfunction readJson(filePath: string): unknown {\n\ttry {\n\t\treturn JSON.parse(fs.readFileSync(filePath, \"utf8\"));\n\t} catch {\n\t\t// A missing or malformed config file yields no servers, never a crash.\n\t\treturn undefined;\n\t}\n}\n\nfunction extractServers(raw: unknown): Record<string, unknown> {\n\tif (!isRecord(raw)) return {};\n\tconst candidate = raw.mcpServers ?? raw.servers ?? raw.mcp_servers;\n\treturn isRecord(candidate) ? candidate : {};\n}\n\nfunction toStringRecord(value: unknown): Record<string, string> | undefined {\n\tif (!isRecord(value)) return undefined;\n\tconst out: Record<string, string> = {};\n\tfor (const [key, item] of Object.entries(value)) {\n\t\tif (typeof item === \"string\") out[key] = item;\n\t}\n\treturn Object.keys(out).length > 0 ? out : undefined;\n}\n\nfunction toServerConfig(name: string, raw: unknown): McpServerConfig | undefined {\n\tif (!isRecord(raw)) return undefined;\n\tconst command = raw.command;\n\t// Only stdio servers are supported; an entry with a `url` is a different transport.\n\tif (typeof command !== \"string\" || command.length === 0) return undefined;\n\tconst args = Array.isArray(raw.args) ? raw.args.filter((arg): arg is string => typeof arg === \"string\") : undefined;\n\tconst startupTimeoutSec = typeof raw.startup_timeout_sec === \"number\" ? raw.startup_timeout_sec : undefined;\n\treturn {\n\t\tname,\n\t\tcommand,\n\t\targs,\n\t\tenv: toStringRecord(raw.env),\n\t\tcwd: typeof raw.cwd === \"string\" ? raw.cwd : undefined,\n\t\tdisabled: raw.disabled === true || raw.enabled === false,\n\t\thandshakeTimeoutMs: startupTimeoutSec !== undefined ? Math.max(1, startupTimeoutSec) * 1000 : undefined,\n\t};\n}\n\n/** Config file paths consulted, in precedence order. */\nexport function mcpConfigPaths(cwd: string = process.cwd(), home: string = os.homedir()): string[] {\n\treturn [\n\t\tpath.join(home, \".kimi\", \"mcp.json\"),\n\t\tpath.join(home, \".omk\", \"mcp.json\"),\n\t\tpath.join(cwd, \".omk\", \"mcp.json\"),\n\t];\n}\n\n/**\n * Load spawnable stdio server configs. Entries without a `command` (HTTP/SSE\n * servers) are skipped rather than failing the load, so an unsupported\n * transport in the config cannot disable every other server.\n */\nexport function loadMcpServerConfigs(cwd: string = process.cwd(), home: string = os.homedir()): McpServerConfig[] {\n\tconst merged = new Map<string, McpServerConfig>();\n\tfor (const filePath of mcpConfigPaths(cwd, home)) {\n\t\tconst servers = extractServers(readJson(filePath));\n\t\tfor (const name of Object.keys(servers).sort()) {\n\t\t\tconst config = toServerConfig(name, servers[name]);\n\t\t\tif (config) merged.set(name, config);\n\t\t}\n\t}\n\treturn [...merged.values()].sort((a, b) => a.name.localeCompare(b.name));\n}\n"]}

package/dist/core/mcp/index.d.ts

@@ -0,0 +1,15 @@
+/**
+ * MCP client surface for the OMK harness.
+ *
+ * Before this module, `mcp-inventory.ts` could read MCP configuration and the
+ * doctor could report on it, but the runtime had no way to speak the protocol:
+ * configured servers contributed zero tools to a session. This package closes
+ * that gap with a dependency-free stdio client.
+ */
+export { DEFAULT_HANDSHAKE_TIMEOUT_MS, DEFAULT_REQUEST_TIMEOUT_MS, MCP_PROTOCOL_VERSION, McpClient, type McpClientOptions, type McpContentBlock, type McpServerInfo, type McpToolCallResult, type McpToolSchema, } from "./client.ts";
+export { loadMcpServerConfigs, mcpConfigPaths } from "./config.ts";
+export { McpManager, type McpManagerOptions, type McpServerConfig, type McpServerState, type McpServerStatus, } from "./manager.ts";
+export { createLineDecoder, encodeMessage, formatJsonRpcError, isJsonRpcNotification, isJsonRpcRequest, isJsonRpcResponse, type JsonRpcMessage, MAX_MESSAGE_LINE_BYTES, } from "./protocol.ts";
+export { DEFAULT_KILL_GRACE_MS, MAX_STDERR_TAIL_CHARS, McpStdioTransport, type StdioTransportOptions, } from "./stdio-transport.ts";
+export { buildMcpToolName, createMcpToolDefinition, MAX_TOOL_NAME_LENGTH, MCP_TOOL_NAME_SEPARATOR, type McpToolDetails, mapMcpContent, sanitizeToolNameSegment, } from "./tools.ts";
+//# sourceMappingURL=index.d.ts.map

package/dist/core/mcp/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/mcp/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EACN,4BAA4B,EAC5B,0BAA0B,EAC1B,oBAAoB,EACpB,SAAS,EACT,KAAK,gBAAgB,EACrB,KAAK,eAAe,EACpB,KAAK,aAAa,EAClB,KAAK,iBAAiB,EACtB,KAAK,aAAa,GAClB,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,oBAAoB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACnE,OAAO,EACN,UAAU,EACV,KAAK,iBAAiB,EACtB,KAAK,eAAe,EACpB,KAAK,cAAc,EACnB,KAAK,eAAe,GACpB,MAAM,cAAc,CAAC;AACtB,OAAO,EACN,iBAAiB,EACjB,aAAa,EACb,kBAAkB,EAClB,qBAAqB,EACrB,gBAAgB,EAChB,iBAAiB,EACjB,KAAK,cAAc,EACnB,sBAAsB,GACtB,MAAM,eAAe,CAAC;AACvB,OAAO,EACN,qBAAqB,EACrB,qBAAqB,EACrB,iBAAiB,EACjB,KAAK,qBAAqB,GAC1B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACN,gBAAgB,EAChB,uBAAuB,EACvB,oBAAoB,EACpB,uBAAuB,EACvB,KAAK,cAAc,EACnB,aAAa,EACb,uBAAuB,GACvB,MAAM,YAAY,CAAC","sourcesContent":["/**\n * MCP client surface for the OMK harness.\n *\n * Before this module, `mcp-inventory.ts` could read MCP configuration and the\n * doctor could report on it, but the runtime had no way to speak the protocol:\n * configured servers contributed zero tools to a session. This package closes\n * that gap with a dependency-free stdio client.\n */\n\nexport {\n\tDEFAULT_HANDSHAKE_TIMEOUT_MS,\n\tDEFAULT_REQUEST_TIMEOUT_MS,\n\tMCP_PROTOCOL_VERSION,\n\tMcpClient,\n\ttype McpClientOptions,\n\ttype McpContentBlock,\n\ttype McpServerInfo,\n\ttype McpToolCallResult,\n\ttype McpToolSchema,\n} from \"./client.ts\";\nexport { loadMcpServerConfigs, mcpConfigPaths } from \"./config.ts\";\nexport {\n\tMcpManager,\n\ttype McpManagerOptions,\n\ttype McpServerConfig,\n\ttype McpServerState,\n\ttype McpServerStatus,\n} from \"./manager.ts\";\nexport {\n\tcreateLineDecoder,\n\tencodeMessage,\n\tformatJsonRpcError,\n\tisJsonRpcNotification,\n\tisJsonRpcRequest,\n\tisJsonRpcResponse,\n\ttype JsonRpcMessage,\n\tMAX_MESSAGE_LINE_BYTES,\n} from \"./protocol.ts\";\nexport {\n\tDEFAULT_KILL_GRACE_MS,\n\tMAX_STDERR_TAIL_CHARS,\n\tMcpStdioTransport,\n\ttype StdioTransportOptions,\n} from \"./stdio-transport.ts\";\nexport {\n\tbuildMcpToolName,\n\tcreateMcpToolDefinition,\n\tMAX_TOOL_NAME_LENGTH,\n\tMCP_TOOL_NAME_SEPARATOR,\n\ttype McpToolDetails,\n\tmapMcpContent,\n\tsanitizeToolNameSegment,\n} from \"./tools.ts\";\n"]}

package/dist/core/mcp/index.js

@@ -0,0 +1,15 @@
+/**
+ * MCP client surface for the OMK harness.
+ *
+ * Before this module, `mcp-inventory.ts` could read MCP configuration and the
+ * doctor could report on it, but the runtime had no way to speak the protocol:
+ * configured servers contributed zero tools to a session. This package closes
+ * that gap with a dependency-free stdio client.
+ */
+export { DEFAULT_HANDSHAKE_TIMEOUT_MS, DEFAULT_REQUEST_TIMEOUT_MS, MCP_PROTOCOL_VERSION, McpClient, } from "./client.js";
+export { loadMcpServerConfigs, mcpConfigPaths } from "./config.js";
+export { McpManager, } from "./manager.js";
+export { createLineDecoder, encodeMessage, formatJsonRpcError, isJsonRpcNotification, isJsonRpcRequest, isJsonRpcResponse, MAX_MESSAGE_LINE_BYTES, } from "./protocol.js";
+export { DEFAULT_KILL_GRACE_MS, MAX_STDERR_TAIL_CHARS, McpStdioTransport, } from "./stdio-transport.js";
+export { buildMcpToolName, createMcpToolDefinition, MAX_TOOL_NAME_LENGTH, MCP_TOOL_NAME_SEPARATOR, mapMcpContent, sanitizeToolNameSegment, } from "./tools.js";
+//# sourceMappingURL=index.js.map

package/dist/core/mcp/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/mcp/index.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EACN,4BAA4B,EAC5B,0BAA0B,EAC1B,oBAAoB,EACpB,SAAS,GAMT,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,oBAAoB,EAAE,cAAc,EAAE,MAAM,aAAa,CAAC;AACnE,OAAO,EACN,UAAU,GAKV,MAAM,cAAc,CAAC;AACtB,OAAO,EACN,iBAAiB,EACjB,aAAa,EACb,kBAAkB,EAClB,qBAAqB,EACrB,gBAAgB,EAChB,iBAAiB,EAEjB,sBAAsB,GACtB,MAAM,eAAe,CAAC;AACvB,OAAO,EACN,qBAAqB,EACrB,qBAAqB,EACrB,iBAAiB,GAEjB,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EACN,gBAAgB,EAChB,uBAAuB,EACvB,oBAAoB,EACpB,uBAAuB,EAEvB,aAAa,EACb,uBAAuB,GACvB,MAAM,YAAY,CAAC","sourcesContent":["/**\n * MCP client surface for the OMK harness.\n *\n * Before this module, `mcp-inventory.ts` could read MCP configuration and the\n * doctor could report on it, but the runtime had no way to speak the protocol:\n * configured servers contributed zero tools to a session. This package closes\n * that gap with a dependency-free stdio client.\n */\n\nexport {\n\tDEFAULT_HANDSHAKE_TIMEOUT_MS,\n\tDEFAULT_REQUEST_TIMEOUT_MS,\n\tMCP_PROTOCOL_VERSION,\n\tMcpClient,\n\ttype McpClientOptions,\n\ttype McpContentBlock,\n\ttype McpServerInfo,\n\ttype McpToolCallResult,\n\ttype McpToolSchema,\n} from \"./client.ts\";\nexport { loadMcpServerConfigs, mcpConfigPaths } from \"./config.ts\";\nexport {\n\tMcpManager,\n\ttype McpManagerOptions,\n\ttype McpServerConfig,\n\ttype McpServerState,\n\ttype McpServerStatus,\n} from \"./manager.ts\";\nexport {\n\tcreateLineDecoder,\n\tencodeMessage,\n\tformatJsonRpcError,\n\tisJsonRpcNotification,\n\tisJsonRpcRequest,\n\tisJsonRpcResponse,\n\ttype JsonRpcMessage,\n\tMAX_MESSAGE_LINE_BYTES,\n} from \"./protocol.ts\";\nexport {\n\tDEFAULT_KILL_GRACE_MS,\n\tMAX_STDERR_TAIL_CHARS,\n\tMcpStdioTransport,\n\ttype StdioTransportOptions,\n} from \"./stdio-transport.ts\";\nexport {\n\tbuildMcpToolName,\n\tcreateMcpToolDefinition,\n\tMAX_TOOL_NAME_LENGTH,\n\tMCP_TOOL_NAME_SEPARATOR,\n\ttype McpToolDetails,\n\tmapMcpContent,\n\tsanitizeToolNameSegment,\n} from \"./tools.ts\";\n"]}

package/dist/core/mcp/manager.d.ts

@@ -0,0 +1,84 @@
+/**
+ * Lazy, crash-isolated MCP server manager.
+ *
+ * Design constraints that shaped this:
+ * - A workspace can configure 25 servers. Spawning them all at startup would
+ *   trade a 0.6s cold start for a multi-second one, so connection is lazy and
+ *   only happens when tools are actually requested.
+ * - One broken server must never take a session down. Every connect and every
+ *   listing is isolated; a failure is recorded as server status and the other
+ *   servers still contribute tools.
+ * - Connect work is deduplicated: concurrent `listToolDefinitions()` calls
+ *   share one in-flight connect per server.
+ */
+import type { TSchema } from "typebox";
+import type { ToolDefinition } from "../extensions/types.ts";
+import { McpClient, type McpClientOptions } from "./client.ts";
+import { type McpToolDetails } from "./tools.ts";
+export type McpServerState = "idle" | "connecting" | "ready" | "failed";
+export interface McpServerConfig {
+    readonly name: string;
+    readonly command: string;
+    readonly args?: readonly string[];
+    readonly env?: Readonly<Record<string, string>>;
+    readonly cwd?: string;
+    /** Skip this server without removing it from configuration. */
+    readonly disabled?: boolean;
+    readonly requestTimeoutMs?: number;
+    readonly handshakeTimeoutMs?: number;
+}
+export interface McpServerStatus {
+    readonly name: string;
+    readonly state: McpServerState;
+    readonly toolCount: number;
+    /** Failure reason when `state` is `failed`. Never contains configured env values. */
+    readonly error?: string;
+    readonly serverVersion?: string;
+}
+export interface McpManagerOptions {
+    readonly servers: readonly McpServerConfig[];
+    readonly cwd?: string;
+    readonly clientInfo?: {
+        readonly name: string;
+        readonly version: string;
+    };
+    readonly callTimeoutMs?: number;
+    /** Injected for tests so the manager can be exercised without spawning processes. */
+    readonly createClient?: (options: McpClientOptions) => McpClient;
+}
+export declare class McpManager {
+    private readonly runtimes;
+    private readonly options;
+    constructor(options: McpManagerOptions);
+    /** Configured server names, in configuration order. */
+    get serverNames(): string[];
+    /** Current status per server. Safe to call before any connection attempt. */
+    status(): McpServerStatus[];
+    /**
+     * Connect every enabled server that has not been attempted yet, then return
+     * the union of their tools. Servers are connected concurrently and failures
+     * are isolated, so a partial result is the normal outcome, not an error.
+     */
+    listToolDefinitions(): Promise<ToolDefinition<TSchema, McpToolDetails>[]>;
+    /** Connect a single server by name. Returns its status either way. */
+    connect(name: string): Promise<McpServerStatus>;
+    /** Terminate every connected server. Idempotent. */
+    close(): void;
+    /**
+     * Verify that `ready` servers are actually alive with a protocol ping. A
+     * failed ping closes the client and marks the server `failed`, so a
+     * silently killed process stops masquerading as connected. Servers already
+     * `failed` are re-attempted only when `reconnectFailed` is set — callers
+     * should gate that behind a slow cadence so a permanently broken server is
+     * not respawned every probe. Idle servers stay idle: connection is lazy by
+     * design, and a status probe must not spawn processes.
+     */
+    checkHealth(options?: {
+        pingTimeoutMs?: number;
+        reconnectFailed?: boolean;
+    }): Promise<McpServerStatus[]>;
+    private pingRuntime;
+    private ensureConnected;
+    private connectRuntime;
+}
+//# sourceMappingURL=manager.d.ts.map

package/dist/core/mcp/manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"manager.d.ts","sourceRoot":"","sources":["../../../src/core/mcp/manager.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,SAAS,EAAE,KAAK,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAA2B,KAAK,cAAc,EAAE,MAAM,YAAY,CAAC;AAE1E,MAAM,MAAM,cAAc,GAAG,MAAM,GAAG,YAAY,GAAG,OAAO,GAAG,QAAQ,CAAC;AAExE,MAAM,WAAW,eAAe;IAC/B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IACzB,QAAQ,CAAC,IAAI,CAAC,EAAE,SAAS,MAAM,EAAE,CAAC;IAClC,QAAQ,CAAC,GAAG,CAAC,EAAE,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAChD,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IACtB,+DAA+D;IAC/D,QAAQ,CAAC,QAAQ,CAAC,EAAE,OAAO,CAAC;IAC5B,QAAQ,CAAC,gBAAgB,CAAC,EAAE,MAAM,CAAC;IACnC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;CACrC;AAED,MAAM,WAAW,eAAe;IAC/B,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,KAAK,EAAE,cAAc,CAAC;IAC/B,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC;IAC3B,qFAAqF;IACrF,QAAQ,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC;CAChC;AAED,MAAM,WAAW,iBAAiB;IACjC,QAAQ,CAAC,OAAO,EAAE,SAAS,eAAe,EAAE,CAAC;IAC7C,QAAQ,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,UAAU,CAAC,EAAE;QAAE,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;KAAE,CAAC;IAC1E,QAAQ,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC;IAChC,qFAAqF;IACrF,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC,OAAO,EAAE,gBAAgB,KAAK,SAAS,CAAC;CACjE;AAWD,qBAAa,UAAU;IACtB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAoC;IAC7D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoB;IAE5C,YAAY,OAAO,EAAE,iBAAiB,EAMrC;IAED,uDAAuD;IACvD,IAAI,WAAW,IAAI,MAAM,EAAE,CAE1B;IAED,6EAA6E;IAC7E,MAAM,IAAI,eAAe,EAAE,CAQ1B;IAED;;;;OAIG;IACG,mBAAmB,IAAI,OAAO,CAAC,cAAc,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,CAAC,CAY9E;IAED,sEAAsE;IAChE,OAAO,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC,CAapD;IAED,oDAAoD;IACpD,KAAK,IAAI,IAAI,CAOZ;IAED;;;;;;;;OAQG;IACG,WAAW,CAAC,OAAO,CAAC,EAAE;QAAE,aAAa,CAAC,EAAE,MAAM,CAAC;QAAC,eAAe,CAAC,EAAE,OAAO,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC,CAa7G;YAEa,WAAW;IAazB,OAAO,CAAC,eAAe;YAiBT,cAAc;CAkC5B","sourcesContent":["/**\n * Lazy, crash-isolated MCP server manager.\n *\n * Design constraints that shaped this:\n * - A workspace can configure 25 servers. Spawning them all at startup would\n *   trade a 0.6s cold start for a multi-second one, so connection is lazy and\n *   only happens when tools are actually requested.\n * - One broken server must never take a session down. Every connect and every\n *   listing is isolated; a failure is recorded as server status and the other\n *   servers still contribute tools.\n * - Connect work is deduplicated: concurrent `listToolDefinitions()` calls\n *   share one in-flight connect per server.\n */\n\nimport type { TSchema } from \"typebox\";\nimport type { ToolDefinition } from \"../extensions/types.ts\";\nimport { McpClient, type McpClientOptions } from \"./client.ts\";\nimport { createMcpToolDefinition, type McpToolDetails } from \"./tools.ts\";\n\nexport type McpServerState = \"idle\" | \"connecting\" | \"ready\" | \"failed\";\n\nexport interface McpServerConfig {\n\treadonly name: string;\n\treadonly command: string;\n\treadonly args?: readonly string[];\n\treadonly env?: Readonly<Record<string, string>>;\n\treadonly cwd?: string;\n\t/** Skip this server without removing it from configuration. */\n\treadonly disabled?: boolean;\n\treadonly requestTimeoutMs?: number;\n\treadonly handshakeTimeoutMs?: number;\n}\n\nexport interface McpServerStatus {\n\treadonly name: string;\n\treadonly state: McpServerState;\n\treadonly toolCount: number;\n\t/** Failure reason when `state` is `failed`. Never contains configured env values. */\n\treadonly error?: string;\n\treadonly serverVersion?: string;\n}\n\nexport interface McpManagerOptions {\n\treadonly servers: readonly McpServerConfig[];\n\treadonly cwd?: string;\n\treadonly clientInfo?: { readonly name: string; readonly version: string };\n\treadonly callTimeoutMs?: number;\n\t/** Injected for tests so the manager can be exercised without spawning processes. */\n\treadonly createClient?: (options: McpClientOptions) => McpClient;\n}\n\ninterface ServerRuntime {\n\treadonly config: McpServerConfig;\n\tstate: McpServerState;\n\tclient?: McpClient;\n\ttools: ToolDefinition<TSchema, McpToolDetails>[];\n\terror?: string;\n\tconnecting?: Promise<void>;\n}\n\nexport class McpManager {\n\tprivate readonly runtimes = new Map<string, ServerRuntime>();\n\tprivate readonly options: McpManagerOptions;\n\n\tconstructor(options: McpManagerOptions) {\n\t\tthis.options = options;\n\t\tfor (const config of options.servers) {\n\t\t\tif (this.runtimes.has(config.name)) continue; // First definition wins; inventory already applies precedence.\n\t\t\tthis.runtimes.set(config.name, { config, state: \"idle\", tools: [] });\n\t\t}\n\t}\n\n\t/** Configured server names, in configuration order. */\n\tget serverNames(): string[] {\n\t\treturn [...this.runtimes.keys()];\n\t}\n\n\t/** Current status per server. Safe to call before any connection attempt. */\n\tstatus(): McpServerStatus[] {\n\t\treturn [...this.runtimes.values()].map((runtime) => ({\n\t\t\tname: runtime.config.name,\n\t\t\tstate: runtime.state,\n\t\t\ttoolCount: runtime.tools.length,\n\t\t\terror: runtime.error,\n\t\t\tserverVersion: runtime.client?.serverInfo.version,\n\t\t}));\n\t}\n\n\t/**\n\t * Connect every enabled server that has not been attempted yet, then return\n\t * the union of their tools. Servers are connected concurrently and failures\n\t * are isolated, so a partial result is the normal outcome, not an error.\n\t */\n\tasync listToolDefinitions(): Promise<ToolDefinition<TSchema, McpToolDetails>[]> {\n\t\tawait Promise.all([...this.runtimes.values()].map((runtime) => this.ensureConnected(runtime)));\n\t\tconst tools: ToolDefinition<TSchema, McpToolDetails>[] = [];\n\t\tconst seen = new Set<string>();\n\t\tfor (const runtime of this.runtimes.values()) {\n\t\t\tfor (const tool of runtime.tools) {\n\t\t\t\tif (seen.has(tool.name)) continue; // Namespacing makes this rare; keep it deterministic anyway.\n\t\t\t\tseen.add(tool.name);\n\t\t\t\ttools.push(tool);\n\t\t\t}\n\t\t}\n\t\treturn tools;\n\t}\n\n\t/** Connect a single server by name. Returns its status either way. */\n\tasync connect(name: string): Promise<McpServerStatus> {\n\t\tconst runtime = this.runtimes.get(name);\n\t\tif (!runtime) {\n\t\t\treturn { name, state: \"failed\", toolCount: 0, error: `Unknown MCP server \"${name}\"` };\n\t\t}\n\t\tawait this.ensureConnected(runtime);\n\t\treturn {\n\t\t\tname,\n\t\t\tstate: runtime.state,\n\t\t\ttoolCount: runtime.tools.length,\n\t\t\terror: runtime.error,\n\t\t\tserverVersion: runtime.client?.serverInfo.version,\n\t\t};\n\t}\n\n\t/** Terminate every connected server. Idempotent. */\n\tclose(): void {\n\t\tfor (const runtime of this.runtimes.values()) {\n\t\t\truntime.client?.close();\n\t\t\truntime.client = undefined;\n\t\t\truntime.tools = [];\n\t\t\tif (runtime.state === \"ready\" || runtime.state === \"connecting\") runtime.state = \"idle\";\n\t\t}\n\t}\n\n\t/**\n\t * Verify that `ready` servers are actually alive with a protocol ping. A\n\t * failed ping closes the client and marks the server `failed`, so a\n\t * silently killed process stops masquerading as connected. Servers already\n\t * `failed` are re-attempted only when `reconnectFailed` is set — callers\n\t * should gate that behind a slow cadence so a permanently broken server is\n\t * not respawned every probe. Idle servers stay idle: connection is lazy by\n\t * design, and a status probe must not spawn processes.\n\t */\n\tasync checkHealth(options?: { pingTimeoutMs?: number; reconnectFailed?: boolean }): Promise<McpServerStatus[]> {\n\t\tconst work: Promise<void>[] = [];\n\t\tfor (const runtime of this.runtimes.values()) {\n\t\t\tif (runtime.state === \"ready\" && runtime.client) {\n\t\t\t\twork.push(this.pingRuntime(runtime, options?.pingTimeoutMs));\n\t\t\t} else if (runtime.state === \"failed\" && options?.reconnectFailed && !runtime.config.disabled) {\n\t\t\t\truntime.state = \"idle\";\n\t\t\t\truntime.error = undefined;\n\t\t\t\twork.push(this.ensureConnected(runtime));\n\t\t\t}\n\t\t}\n\t\tawait Promise.all(work);\n\t\treturn this.status();\n\t}\n\n\tprivate async pingRuntime(runtime: ServerRuntime, timeoutMs?: number): Promise<void> {\n\t\ttry {\n\t\t\tawait runtime.client?.ping(timeoutMs);\n\t\t} catch (error) {\n\t\t\t// Isolation point: only this server transitions; the rest stay ready.\n\t\t\truntime.client?.close();\n\t\t\truntime.client = undefined;\n\t\t\truntime.tools = [];\n\t\t\truntime.state = \"failed\";\n\t\t\truntime.error = `health check failed: ${error instanceof Error ? error.message : String(error)}`;\n\t\t}\n\t}\n\n\tprivate ensureConnected(runtime: ServerRuntime): Promise<void> {\n\t\tif (runtime.config.disabled) {\n\t\t\truntime.state = \"failed\";\n\t\t\truntime.error = \"disabled by configuration\";\n\t\t\treturn Promise.resolve();\n\t\t}\n\t\tif (runtime.state === \"ready\" || runtime.state === \"failed\") return Promise.resolve();\n\t\tif (runtime.connecting) return runtime.connecting;\n\n\t\truntime.state = \"connecting\";\n\t\tconst attempt = this.connectRuntime(runtime).finally(() => {\n\t\t\truntime.connecting = undefined;\n\t\t});\n\t\truntime.connecting = attempt;\n\t\treturn attempt;\n\t}\n\n\tprivate async connectRuntime(runtime: ServerRuntime): Promise<void> {\n\t\tconst clientOptions: McpClientOptions = {\n\t\t\tname: runtime.config.name,\n\t\t\tclientInfo: this.options.clientInfo,\n\t\t\trequestTimeoutMs: runtime.config.requestTimeoutMs,\n\t\t\thandshakeTimeoutMs: runtime.config.handshakeTimeoutMs,\n\t\t\ttransport: {\n\t\t\t\tcommand: runtime.config.command,\n\t\t\t\targs: runtime.config.args,\n\t\t\t\tenv: runtime.config.env,\n\t\t\t\tcwd: runtime.config.cwd ?? this.options.cwd,\n\t\t\t},\n\t\t};\n\t\tconst client = this.options.createClient?.(clientOptions) ?? new McpClient(clientOptions);\n\t\ttry {\n\t\t\tawait client.connect();\n\t\t\tconst schemas = await client.listTools();\n\t\t\truntime.client = client;\n\t\t\truntime.tools = schemas.map((schema) =>\n\t\t\t\tcreateMcpToolDefinition(runtime.config.name, client, schema, {\n\t\t\t\t\tcallTimeoutMs: this.options.callTimeoutMs,\n\t\t\t\t}),\n\t\t\t);\n\t\t\truntime.state = \"ready\";\n\t\t\truntime.error = undefined;\n\t\t} catch (error) {\n\t\t\t// Isolation point: this server is out, every other server is unaffected.\n\t\t\tclient.close();\n\t\t\truntime.client = undefined;\n\t\t\truntime.tools = [];\n\t\t\truntime.state = \"failed\";\n\t\t\truntime.error = error instanceof Error ? error.message : String(error);\n\t\t}\n\t}\n}\n"]}