@nodetool-ai/websocket 0.7.0-rc.11

package/dist/agent/mcp-tool-server.js

@@ -0,0 +1,292 @@
+/**
+ * HTTP MCP server that exposes NodeTool UI tools.
+ *
+ * Implements the MCP Streamable HTTP transport so any agent SDK
+ * (Claude, Codex, OpenCode) can connect to it as an MCP server.
+ *
+ * Tool calls are forwarded to the renderer whose session originated the
+ * connection via the `AgentTransport` for execution against the live
+ * workflow graph. The session identity is encoded in the URL path as
+ * `/mcp/<sessionId>` so simultaneous sessions from different renderers
+ * can't clobber each other's tool routing.
+ */
+import { createServer, } from "node:http";
+import { createLogger } from "@nodetool-ai/config";
+import { uiToolSchemas } from "@nodetool-ai/protocol";
+import { z, toJSONSchema } from "zod";
+const log = createLogger("nodetool.websocket.agent.mcp");
+// ---------------------------------------------------------------------------
+// Tool schema conversion
+// ---------------------------------------------------------------------------
+function zodShapeToJsonSchema(zodShape) {
+    const schema = z.object(zodShape);
+    const jsonSchema = toJSONSchema(schema);
+    // Remove $schema key — MCP inputSchema doesn't need it
+    delete jsonSchema.$schema;
+    return {
+        type: "object",
+        properties: (jsonSchema.properties ?? {}),
+        required: jsonSchema.required,
+    };
+}
+function buildToolDefinitions() {
+    return Object.entries(uiToolSchemas).map(([name, schema]) => ({
+        name,
+        description: schema.description,
+        inputSchema: zodShapeToJsonSchema(schema.parameters),
+    }));
+}
+// ---------------------------------------------------------------------------
+// MCP HTTP Server (singleton)
+// ---------------------------------------------------------------------------
+let serverInstance = null;
+let serverPort = null;
+/**
+ * Session ID → transport mapping. Each agent session has its own entry so
+ * tool calls are routed to the renderer that originated that session. This
+ * replaces the earlier "active transport" global, which was unsafe when
+ * multiple renderers were connected simultaneously.
+ */
+const sessionTransports = new Map();
+/** Default CORS origins allowed to hit the MCP HTTP server. */
+const DEFAULT_ALLOWED_ORIGIN_PATTERNS = [
+    /^https?:\/\/localhost(?::\d+)?$/,
+    /^https?:\/\/127\.0\.0\.1(?::\d+)?$/,
+    // Electron renderer
+    /^file:\/\//,
+];
+function isOriginAllowed(origin) {
+    // MCP SDK clients (Claude/Codex) typically don't send an Origin header.
+    // In that case there's no browser enforcing CORS, so allow the request.
+    if (!origin)
+        return true;
+    return DEFAULT_ALLOWED_ORIGIN_PATTERNS.some((re) => re.test(origin));
+}
+function readBody(req) {
+    return new Promise((resolve, reject) => {
+        const chunks = [];
+        req.on("data", (chunk) => chunks.push(chunk));
+        req.on("end", () => resolve(Buffer.concat(chunks).toString("utf8")));
+        req.on("error", reject);
+    });
+}
+function sendJson(res, status, body) {
+    const payload = JSON.stringify(body);
+    res.writeHead(status, {
+        "Content-Type": "application/json",
+        "Content-Length": Buffer.byteLength(payload),
+    });
+    res.end(payload);
+}
+function extractSessionIdFromPath(path) {
+    if (!path)
+        return null;
+    // URL is `/mcp/<sessionId>` (optionally with a trailing slash or query).
+    const match = /^\/mcp\/([^/?#]+)\/?/.exec(path);
+    return match ? decodeURIComponent(match[1]) : null;
+}
+async function handleMcpRequest(req, res) {
+    const origin = req.headers.origin;
+    if (isOriginAllowed(origin)) {
+        res.setHeader("Access-Control-Allow-Origin", origin ?? "*");
+    }
+    res.setHeader("Access-Control-Allow-Methods", "POST, OPTIONS");
+    res.setHeader("Access-Control-Allow-Headers", "Content-Type");
+    res.setHeader("Vary", "Origin");
+    if (req.method === "OPTIONS") {
+        res.writeHead(204);
+        res.end();
+        return;
+    }
+    if (req.method !== "POST") {
+        sendJson(res, 405, { error: "Method not allowed" });
+        return;
+    }
+    const sessionId = extractSessionIdFromPath(req.url);
+    if (!sessionId) {
+        sendJson(res, 404, {
+            jsonrpc: "2.0",
+            id: null,
+            error: { code: -32601, message: "Session ID missing from MCP URL path" },
+        });
+        return;
+    }
+    let rpcRequest;
+    try {
+        const body = await readBody(req);
+        rpcRequest = JSON.parse(body);
+    }
+    catch {
+        sendJson(res, 400, {
+            jsonrpc: "2.0",
+            id: null,
+            error: { code: -32700, message: "Parse error" },
+        });
+        return;
+    }
+    const response = await handleRpcMethod(rpcRequest, sessionId);
+    sendJson(res, 200, response);
+}
+async function handleRpcMethod(request, sessionId) {
+    const { id, method, params } = request;
+    switch (method) {
+        case "initialize":
+            return {
+                jsonrpc: "2.0",
+                id,
+                result: {
+                    protocolVersion: "2024-11-05",
+                    capabilities: { tools: {} },
+                    serverInfo: { name: "nodetool-ui", version: "1.0.0" },
+                },
+            };
+        case "tools/list":
+            return {
+                jsonrpc: "2.0",
+                id,
+                result: { tools: buildToolDefinitions() },
+            };
+        case "tools/call": {
+            const toolName = params?.name;
+            const toolArgs = params?.arguments;
+            if (!toolName || !uiToolSchemas[toolName]) {
+                return {
+                    jsonrpc: "2.0",
+                    id,
+                    error: { code: -32602, message: `Unknown tool: ${toolName}` },
+                };
+            }
+            const transport = sessionTransports.get(sessionId);
+            if (!transport || !transport.isAlive) {
+                return {
+                    jsonrpc: "2.0",
+                    id,
+                    result: {
+                        content: [
+                            {
+                                type: "text",
+                                text: "Error: No active renderer for this session — open a workflow in NodeTool first.",
+                            },
+                        ],
+                        isError: true,
+                    },
+                };
+            }
+            try {
+                const result = await transport.executeTool(sessionId, `mcp-${id}`, toolName, toolArgs ?? {});
+                const text = typeof result === "string"
+                    ? result
+                    : JSON.stringify(result ?? null);
+                return {
+                    jsonrpc: "2.0",
+                    id,
+                    result: {
+                        content: [{ type: "text", text }],
+                        isError: false,
+                    },
+                };
+            }
+            catch (error) {
+                const message = error instanceof Error ? error.message : String(error);
+                return {
+                    jsonrpc: "2.0",
+                    id,
+                    result: {
+                        content: [{ type: "text", text: `Error: ${message}` }],
+                        isError: true,
+                    },
+                };
+            }
+        }
+        case "notifications/initialized":
+        case "ping":
+            return { jsonrpc: "2.0", id, result: {} };
+        default:
+            return {
+                jsonrpc: "2.0",
+                id,
+                error: { code: -32601, message: `Method not found: ${method}` },
+            };
+    }
+}
+// ---------------------------------------------------------------------------
+// Public API
+// ---------------------------------------------------------------------------
+async function ensureServerStarted() {
+    if (serverInstance && serverPort)
+        return;
+    await new Promise((resolve, reject) => {
+        const server = createServer((req, res) => {
+            handleMcpRequest(req, res).catch((err) => {
+                log.error("MCP server error", err instanceof Error ? err : new Error(String(err)));
+                sendJson(res, 500, {
+                    jsonrpc: "2.0",
+                    id: null,
+                    error: { code: -32603, message: "Internal error" },
+                });
+            });
+        });
+        server.listen(0, "127.0.0.1", () => {
+            const addr = server.address();
+            if (!addr || typeof addr === "string") {
+                reject(new Error("Failed to get MCP server address"));
+                return;
+            }
+            serverPort = addr.port;
+            serverInstance = server;
+            log.info(`MCP tool server listening at http://127.0.0.1:${serverPort}`);
+            resolve();
+        });
+        server.on("error", (err) => {
+            log.error("MCP tool server error", err);
+            reject(err);
+        });
+    });
+}
+/**
+ * Start (if needed) the MCP HTTP server and register `transport` as the
+ * executor for `sessionId`. Returns the session-scoped MCP URL to hand to
+ * the agent SDK.
+ */
+export async function startMcpToolServer(transport, sessionId) {
+    await ensureServerStarted();
+    sessionTransports.set(sessionId, transport);
+    return `http://127.0.0.1:${serverPort}/mcp/${encodeURIComponent(sessionId)}`;
+}
+/** Update the transport associated with an existing session. */
+export function setMcpToolServerTransport(transport, sessionId) {
+    sessionTransports.set(sessionId, transport);
+}
+/**
+ * Remove all session→transport mappings pointing at the given transport.
+ * Called when a renderer disconnects so we don't keep dead references.
+ */
+export function clearMcpToolServerTransport(transport) {
+    for (const [sessionId, t] of sessionTransports.entries()) {
+        if (t === transport) {
+            sessionTransports.delete(sessionId);
+        }
+    }
+}
+/** Drop a single session's transport mapping. */
+export function clearMcpToolServerSession(sessionId) {
+    sessionTransports.delete(sessionId);
+}
+/** Get the MCP server URL for an already-registered session, or null. */
+export function getMcpToolServerUrl(sessionId) {
+    if (serverInstance && serverPort && sessionTransports.has(sessionId)) {
+        return `http://127.0.0.1:${serverPort}/mcp/${encodeURIComponent(sessionId)}`;
+    }
+    return null;
+}
+/** Stop the MCP tool server. */
+export function stopMcpToolServer() {
+    if (serverInstance) {
+        serverInstance.close();
+        serverInstance = null;
+        serverPort = null;
+        sessionTransports.clear();
+        log.info("MCP tool server stopped");
+    }
+}
+//# sourceMappingURL=mcp-tool-server.js.map

package/dist/agent/mcp-tool-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-tool-server.js","sourceRoot":"","sources":["../../src/agent/mcp-tool-server.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EACL,YAAY,GAIb,MAAM,WAAW,CAAC;AACnB,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,uBAAuB,CAAC;AACtD,OAAO,EAAE,CAAC,EAAE,YAAY,EAAE,MAAM,KAAK,CAAC;AAGtC,MAAM,GAAG,GAAG,YAAY,CAAC,8BAA8B,CAAC,CAAC;AA8BzD,8EAA8E;AAC9E,yBAAyB;AACzB,8EAA8E;AAE9E,SAAS,oBAAoB,CAAC,QAAsC;IAKlE,MAAM,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAClC,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAA4B,CAAC;IACnE,uDAAuD;IACvD,OAAO,UAAU,CAAC,OAAO,CAAC;IAC1B,OAAO;QACL,IAAI,EAAE,QAAQ;QACd,UAAU,EAAE,CAAC,UAAU,CAAC,UAAU,IAAI,EAAE,CAA4B;QACpE,QAAQ,EAAE,UAAU,CAAC,QAAgC;KACtD,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB;IAC3B,OAAO,MAAM,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC;QAC5D,IAAI;QACJ,WAAW,EAAE,MAAM,CAAC,WAAW;QAC/B,WAAW,EAAE,oBAAoB,CAC/B,MAAM,CAAC,UAA0C,CAClD;KACF,CAAC,CAAC,CAAC;AACN,CAAC;AAED,8EAA8E;AAC9E,8BAA8B;AAC9B,8EAA8E;AAE9E,IAAI,cAAc,GAAkB,IAAI,CAAC;AACzC,IAAI,UAAU,GAAkB,IAAI,CAAC;AAErC;;;;;GAKG;AACH,MAAM,iBAAiB,GAAG,IAAI,GAAG,EAA0B,CAAC;AAE5D,+DAA+D;AAC/D,MAAM,+BAA+B,GAAa;IAChD,iCAAiC;IACjC,oCAAoC;IACpC,oBAAoB;IACpB,YAAY;CACb,CAAC;AAEF,SAAS,eAAe,CAAC,MAA0B;IACjD,wEAAwE;IACxE,wEAAwE;IACxE,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IACzB,OAAO,+BAA+B,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,QAAQ,CAAC,GAAoB;IACpC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,MAAM,MAAM,GAAa,EAAE,CAAC;QAC5B,GAAG,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACtD,GAAG,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QACrE,GAAG,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC1B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,QAAQ,CAAC,GAAmB,EAAE,MAAc,EAAE,IAAa;IAClE,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;IACrC,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE;QACpB,cAAc,EAAE,kBAAkB;QAClC,gBAAgB,EAAE,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC;KAC7C,CAAC,CAAC;IACH,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;AACnB,CAAC;AAED,SAAS,wBAAwB,CAAC,IAAwB;IACxD,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IACvB,yEAAyE;IACzE,MAAM,KAAK,GAAG,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChD,OAAO,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AACrD,CAAC;AAED,KAAK,UAAU,gBAAgB,CAC7B,GAAoB,EACpB,GAAmB;IAEnB,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,MAA4B,CAAC;IACxD,IAAI,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC5B,GAAG,CAAC,SAAS,CAAC,6BAA6B,EAAE,MAAM,IAAI,GAAG,CAAC,CAAC;IAC9D,CAAC;IACD,GAAG,CAAC,SAAS,CAAC,8BAA8B,EAAE,eAAe,CAAC,CAAC;IAC/D,GAAG,CAAC,SAAS,CAAC,8BAA8B,EAAE,cAAc,CAAC,CAAC;IAC9D,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IAEhC,IAAI,GAAG,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC7B,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QACnB,GAAG,CAAC,GAAG,EAAE,CAAC;QACV,OAAO;IACT,CAAC;IAED,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;QAC1B,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,KAAK,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACpD,OAAO;IACT,CAAC;IAED,MAAM,SAAS,GAAG,wBAAwB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IACpD,IAAI,CAAC,SAAS,EAAE,CAAC;QACf,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE;YACjB,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,IAAI;YACR,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,sCAAsC,EAAE;SACzE,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,IAAI,UAA6B,CAAC;IAClC,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,GAAG,CAAC,CAAC;QACjC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAsB,CAAC;IACrD,CAAC;IAAC,MAAM,CAAC;QACP,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE;YACjB,OAAO,EAAE,KAAK;YACd,EAAE,EAAE,IAAI;YACR,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE;SAChD,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,eAAe,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IAC9D,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE,QAAQ,CAAC,CAAC;AAC/B,CAAC;AAED,KAAK,UAAU,eAAe,CAC5B,OAA0B,EAC1B,SAAiB;IAEjB,MAAM,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAEvC,QAAQ,MAAM,EAAE,CAAC;QACf,KAAK,YAAY;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,EAAE;gBACF,MAAM,EAAE;oBACN,eAAe,EAAE,YAAY;oBAC7B,YAAY,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE;oBAC3B,UAAU,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,OAAO,EAAE;iBACtD;aACF,CAAC;QAEJ,KAAK,YAAY;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,EAAE;gBACF,MAAM,EAAE,EAAE,KAAK,EAAE,oBAAoB,EAAE,EAAE;aAC1C,CAAC;QAEJ,KAAK,YAAY,CAAC,CAAC,CAAC;YAClB,MAAM,QAAQ,GAAG,MAAM,EAAE,IAAc,CAAC;YACxC,MAAM,QAAQ,GAAG,MAAM,EAAE,SAAgD,CAAC;YAE1E,IAAI,CAAC,QAAQ,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAC1C,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,EAAE;oBACF,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,iBAAiB,QAAQ,EAAE,EAAE;iBAC9D,CAAC;YACJ,CAAC;YAED,MAAM,SAAS,GAAG,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YACnD,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;gBACrC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,EAAE;oBACF,MAAM,EAAE;wBACN,OAAO,EAAE;4BACP;gCACE,IAAI,EAAE,MAAM;gCACZ,IAAI,EACF,iFAAiF;6BACpF;yBACF;wBACD,OAAO,EAAE,IAAI;qBACd;iBACF,CAAC;YACJ,CAAC;YAED,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC,WAAW,CACxC,SAAS,EACT,OAAO,EAAE,EAAE,EACX,QAAQ,EACR,QAAQ,IAAI,EAAE,CACf,CAAC;gBACF,MAAM,IAAI,GACR,OAAO,MAAM,KAAK,QAAQ;oBACxB,CAAC,CAAC,MAAM;oBACR,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;gBACrC,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,EAAE;oBACF,MAAM,EAAE;wBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;wBACjC,OAAO,EAAE,KAAK;qBACf;iBACF,CAAC;YACJ,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,OAAO,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACzD,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,EAAE;oBACF,MAAM,EAAE;wBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,OAAO,EAAE,EAAE,CAAC;wBACtD,OAAO,EAAE,IAAI;qBACd;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,KAAK,2BAA2B,CAAC;QACjC,KAAK,MAAM;YACT,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;QAE5C;YACE,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,EAAE;gBACF,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,qBAAqB,MAAM,EAAE,EAAE;aAChE,CAAC;IACN,CAAC;AACH,CAAC;AAED,8EAA8E;AAC9E,aAAa;AACb,8EAA8E;AAE9E,KAAK,UAAU,mBAAmB;IAChC,IAAI,cAAc,IAAI,UAAU;QAAE,OAAO;IAEzC,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACvC,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;gBACvC,GAAG,CAAC,KAAK,CACP,kBAAkB,EAClB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CACpD,CAAC;gBACF,QAAQ,CAAC,GAAG,EAAE,GAAG,EAAE;oBACjB,OAAO,EAAE,KAAK;oBACd,EAAE,EAAE,IAAI;oBACR,KAAK,EAAE,EAAE,IAAI,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,gBAAgB,EAAE;iBACnD,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE;YACjC,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YAC9B,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;gBACtC,MAAM,CAAC,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC,CAAC;gBACtD,OAAO;YACT,CAAC;YACD,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC;YACvB,cAAc,GAAG,MAAM,CAAC;YACxB,GAAG,CAAC,IAAI,CAAC,iDAAiD,UAAU,EAAE,CAAC,CAAC;YACxE,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YACzB,GAAG,CAAC,KAAK,CAAC,uBAAuB,EAAE,GAAG,CAAC,CAAC;YACxC,MAAM,CAAC,GAAG,CAAC,CAAC;QACd,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,SAAyB,EACzB,SAAiB;IAEjB,MAAM,mBAAmB,EAAE,CAAC;IAC5B,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;IAC5C,OAAO,oBAAoB,UAAU,QAAQ,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC;AAC/E,CAAC;AAED,gEAAgE;AAChE,MAAM,UAAU,yBAAyB,CACvC,SAAyB,EACzB,SAAiB;IAEjB,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;AAC9C,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,2BAA2B,CAAC,SAAyB;IACnE,KAAK,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,IAAI,iBAAiB,CAAC,OAAO,EAAE,EAAE,CAAC;QACzD,IAAI,CAAC,KAAK,SAAS,EAAE,CAAC;YACpB,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;AACH,CAAC;AAED,iDAAiD;AACjD,MAAM,UAAU,yBAAyB,CAAC,SAAiB;IACzD,iBAAiB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;AACtC,CAAC;AAED,yEAAyE;AACzE,MAAM,UAAU,mBAAmB,CAAC,SAAiB;IACnD,IAAI,cAAc,IAAI,UAAU,IAAI,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;QACrE,OAAO,oBAAoB,UAAU,QAAQ,kBAAkB,CAAC,SAAS,CAAC,EAAE,CAAC;IAC/E,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,gCAAgC;AAChC,MAAM,UAAU,iBAAiB;IAC/B,IAAI,cAAc,EAAE,CAAC;QACnB,cAAc,CAAC,KAAK,EAAE,CAAC;QACvB,cAAc,GAAG,IAAI,CAAC;QACtB,UAAU,GAAG,IAAI,CAAC;QAClB,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAC1B,GAAG,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACtC,CAAC;AACH,CAAC"}

package/dist/agent/opencode-agent.d.ts

@@ -0,0 +1,32 @@
+/**
+ * OpenCode SDK agent integration.
+ *
+ * Uses @opencode-ai/sdk to manage sessions and send prompts. OpenCode runs
+ * as a local server; we spawn it and connect via the SDK client.
+ */
+import type { AgentTransport } from "./transport.js";
+import type { AgentGetSessionMessagesRequest, AgentListSessionsRequest, AgentMessage, AgentModelDescriptor, AgentSessionInfoEntry, AgentTranscriptMessage, FrontendToolManifest } from "./types.js";
+export declare function closeOpenCodeServer(): void;
+export declare function listOpenCodeModels(workspacePath?: string): Promise<AgentModelDescriptor[]>;
+export declare function listOpenCodeSessions(options: AgentListSessionsRequest): Promise<AgentSessionInfoEntry[]>;
+export declare function getOpenCodeSessionMessages(options: AgentGetSessionMessagesRequest): Promise<AgentTranscriptMessage[]>;
+export declare class OpenCodeQuerySession {
+    private closed;
+    private readonly providerID;
+    private readonly modelID;
+    private readonly workspacePath;
+    private readonly systemPrompt;
+    private sessionId;
+    private inFlight;
+    constructor(options: {
+        model: string;
+        workspacePath: string;
+        resumeSessionId?: string;
+        systemPrompt?: string;
+    });
+    private ensureSession;
+    send(message: string, _transport: AgentTransport | null, sessionId: string, _manifest: FrontendToolManifest[], onMessage?: (message: AgentMessage) => void, _mcpServerUrl?: string | null): Promise<AgentMessage[]>;
+    interrupt(): Promise<void>;
+    close(): void;
+}
+//# sourceMappingURL=opencode-agent.d.ts.map

package/dist/agent/opencode-agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"opencode-agent.d.ts","sourceRoot":"","sources":["../../src/agent/opencode-agent.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EACV,8BAA8B,EAC9B,wBAAwB,EACxB,YAAY,EACZ,oBAAoB,EACpB,qBAAqB,EACrB,sBAAsB,EACtB,oBAAoB,EACrB,MAAM,YAAY,CAAC;AAkHpB,wBAAgB,mBAAmB,IAAI,IAAI,CAM1C;AAED,wBAAsB,kBAAkB,CACtC,aAAa,CAAC,EAAE,MAAM,GACrB,OAAO,CAAC,oBAAoB,EAAE,CAAC,CA6FjC;AAED,wBAAsB,oBAAoB,CACxC,OAAO,EAAE,wBAAwB,GAChC,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAyBlC;AAED,wBAAsB,0BAA0B,CAC9C,OAAO,EAAE,8BAA8B,GACtC,OAAO,CAAC,sBAAsB,EAAE,CAAC,CAyCnC;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAS;IACvC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IACtC,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,QAAQ,CAAS;gBAEb,OAAO,EAAE;QACnB,KAAK,EAAE,MAAM,CAAC;QACd,aAAa,EAAE,MAAM,CAAC;QACtB,eAAe,CAAC,EAAE,MAAM,CAAC;QACzB,YAAY,CAAC,EAAE,MAAM,CAAC;KACvB;YAca,aAAa;IA8BrB,IAAI,CACR,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,cAAc,GAAG,IAAI,EACjC,SAAS,EAAE,MAAM,EACjB,SAAS,EAAE,oBAAoB,EAAE,EACjC,SAAS,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,KAAK,IAAI,EAC3C,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,GAC5B,OAAO,CAAC,YAAY,EAAE,CAAC;IAiGpB,SAAS,IAAI,OAAO,CAAC,IAAI,CAAC;IAehC,KAAK,IAAI,IAAI;CAId"}

package/dist/agent/opencode-agent.js

@@ -0,0 +1,332 @@
+/**
+ * OpenCode SDK agent integration.
+ *
+ * Uses @opencode-ai/sdk to manage sessions and send prompts. OpenCode runs
+ * as a local server; we spawn it and connect via the SDK client.
+ */
+import { randomUUID } from "node:crypto";
+import { createLogger } from "@nodetool-ai/config";
+const log = createLogger("nodetool.websocket.agent.opencode");
+let clientInstance = null;
+let serverClose = null;
+async function getClient() {
+    if (clientInstance) {
+        return clientInstance;
+    }
+    const sdk = (await import("@opencode-ai/sdk"));
+    const { client, server } = await sdk.createOpencode();
+    clientInstance = client;
+    serverClose = server.close;
+    log.info(`OpenCode server started at ${server.url}`);
+    return clientInstance;
+}
+function getClientIfRunning() {
+    return clientInstance;
+}
+export function closeOpenCodeServer() {
+    if (serverClose) {
+        serverClose();
+        serverClose = null;
+        clientInstance = null;
+    }
+}
+export async function listOpenCodeModels(workspacePath) {
+    const models = [];
+    const seen = new Set();
+    const query = workspacePath ? { query: { directory: workspacePath } } : {};
+    let client;
+    try {
+        client = await getClient();
+    }
+    catch (error) {
+        log.error("Failed to start OpenCode server for model listing", error instanceof Error ? error : new Error(String(error)));
+        return [];
+    }
+    // Primary source: providers the user has actually configured (env vars,
+    // opencode.json, oauth). These are ready to use without further setup.
+    try {
+        const configured = await client.config.providers(query);
+        if (configured.data?.providers) {
+            const defaultModelIds = new Set(Object.values(configured.data.default ?? {}));
+            for (const provider of configured.data.providers) {
+                for (const [modelId, modelInfo] of Object.entries(provider.models)) {
+                    if (!modelInfo.capabilities?.toolcall)
+                        continue;
+                    const compositeId = `${provider.id}/${modelId}`;
+                    if (seen.has(compositeId))
+                        continue;
+                    seen.add(compositeId);
+                    models.push({
+                        id: compositeId,
+                        label: `${modelInfo.name} (${provider.name})`,
+                        provider: "opencode",
+                        isDefault: defaultModelIds.has(modelId),
+                        supportsMaxTurns: false,
+                        supportsReasoningEffort: modelInfo.capabilities?.reasoning ?? false,
+                    });
+                }
+            }
+        }
+    }
+    catch (error) {
+        log.warn(`config.providers failed: ${error instanceof Error ? error.message : String(error)}`);
+    }
+    // Supplemental source: the full catalog from provider.list. Appends models
+    // that weren't in the configured set (labelled "not connected") so the
+    // user can discover and configure them. Also used as the sole source if
+    // config.providers returned nothing usable.
+    try {
+        const all = await client.provider.list(query);
+        if (all.data) {
+            const connectedSet = new Set(all.data.connected ?? []);
+            const defaultModelIds = new Set(Object.values(all.data.default ?? {}));
+            for (const provider of all.data.all ?? []) {
+                const isConnected = connectedSet.has(provider.id);
+                for (const [modelId, modelInfo] of Object.entries(provider.models)) {
+                    if (modelInfo.status === "deprecated")
+                        continue;
+                    if (!modelInfo.tool_call)
+                        continue;
+                    const compositeId = `${provider.id}/${modelId}`;
+                    if (seen.has(compositeId))
+                        continue;
+                    seen.add(compositeId);
+                    const label = isConnected
+                        ? `${modelInfo.name} (${provider.name})`
+                        : `${modelInfo.name} (${provider.name}, not connected)`;
+                    models.push({
+                        id: compositeId,
+                        label,
+                        provider: "opencode",
+                        isDefault: isConnected && defaultModelIds.has(modelId),
+                        supportsMaxTurns: false,
+                        supportsReasoningEffort: modelInfo.reasoning,
+                    });
+                }
+            }
+        }
+    }
+    catch (error) {
+        log.error("provider.list failed", error instanceof Error ? error : new Error(String(error)));
+    }
+    if (models.length > 0 && !models.some((m) => m.isDefault)) {
+        models[0].isDefault = true;
+    }
+    log.info(`Listed ${models.length} OpenCode model(s)`);
+    return models;
+}
+export async function listOpenCodeSessions(options) {
+    try {
+        const client = getClientIfRunning();
+        if (!client)
+            return [];
+        const result = await client.session.list({
+            query: { directory: options.dir },
+        });
+        if (!result.data)
+            return [];
+        return result.data.map((s) => ({
+            sessionId: s.id,
+            summary: s.title,
+            lastModified: s.time.updated,
+            cwd: s.directory,
+            createdAt: s.time.created,
+            provider: "opencode",
+        }));
+    }
+    catch (error) {
+        log.error("Failed to list OpenCode sessions", error instanceof Error ? error : new Error(String(error)));
+        return [];
+    }
+}
+export async function getOpenCodeSessionMessages(options) {
+    try {
+        const client = getClientIfRunning();
+        if (!client)
+            return [];
+        const result = await client.session.messages({
+            path: { id: options.sessionId },
+        });
+        if (!result.data)
+            return [];
+        return result.data
+            .map((entry) => {
+            const role = entry.info.role === "user" ? "user" : "assistant";
+            const textParts = entry.parts
+                .filter((p) => p.type === "text" && "text" in p)
+                .map((p) => p.text);
+            return {
+                type: role,
+                uuid: entry.info.id,
+                session_id: entry.info.sessionID,
+                text: textParts.join("\n"),
+            };
+        })
+            .filter((m) => m.text.length > 0);
+    }
+    catch (error) {
+        log.error("Failed to get OpenCode session messages", error instanceof Error ? error : new Error(String(error)));
+        return [];
+    }
+}
+export class OpenCodeQuerySession {
+    closed = false;
+    providerID;
+    modelID;
+    workspacePath;
+    systemPrompt;
+    sessionId;
+    inFlight = false;
+    constructor(options) {
+        const parts = options.model.split("/");
+        if (parts.length >= 2) {
+            this.providerID = parts[0];
+            this.modelID = parts.slice(1).join("/");
+        }
+        else {
+            this.providerID = "anthropic";
+            this.modelID = options.model;
+        }
+        this.workspacePath = options.workspacePath;
+        this.systemPrompt = options.systemPrompt ?? "";
+        this.sessionId = options.resumeSessionId ?? null;
+    }
+    async ensureSession() {
+        if (this.sessionId)
+            return this.sessionId;
+        const client = await getClient();
+        const result = await client.session.create({
+            body: { title: "NodeTool Agent Session" },
+            query: { directory: this.workspacePath },
+        });
+        if (!result.data) {
+            throw new Error("Failed to create OpenCode session");
+        }
+        this.sessionId = result.data.id;
+        log.info(`OpenCode session created: ${this.sessionId}`);
+        if (this.systemPrompt.trim().length > 0) {
+            await client.session.prompt({
+                path: { id: this.sessionId },
+                body: {
+                    noReply: true,
+                    parts: [{ type: "text", text: this.systemPrompt }],
+                    model: { providerID: this.providerID, modelID: this.modelID },
+                },
+            });
+        }
+        return this.sessionId;
+    }
+    async send(message, _transport, sessionId, _manifest, onMessage, _mcpServerUrl) {
+        if (this.closed) {
+            throw new Error("Cannot send to a closed session");
+        }
+        if (this.inFlight) {
+            throw new Error("A request is already in progress for this session");
+        }
+        this.inFlight = true;
+        try {
+            const ocSessionId = await this.ensureSession();
+            const client = await getClient();
+            const result = await client.session.prompt({
+                path: { id: ocSessionId },
+                body: {
+                    parts: [{ type: "text", text: message }],
+                    model: { providerID: this.providerID, modelID: this.modelID },
+                },
+            });
+            if (!result.data) {
+                throw new Error("OpenCode prompt returned no data");
+            }
+            const resolvedSessionId = this.sessionId ?? sessionId;
+            const outputMessages = [];
+            for (const part of result.data.parts) {
+                if (part.type === "text" && part.text && part.text.trim().length > 0) {
+                    const msg = {
+                        type: "assistant",
+                        uuid: part.id ?? randomUUID(),
+                        session_id: resolvedSessionId,
+                        content: [{ type: "text", text: part.text }],
+                    };
+                    outputMessages.push(msg);
+                    if (onMessage)
+                        onMessage(msg);
+                }
+                else if (part.type === "tool" && "tool" in part) {
+                    const toolInput = "state" in part &&
+                        part.state &&
+                        typeof part.state === "object" &&
+                        "input" in part.state
+                        ? part.state.input
+                        : {};
+                    const msg = {
+                        type: "assistant",
+                        uuid: part.id ?? randomUUID(),
+                        session_id: resolvedSessionId,
+                        content: [],
+                        tool_calls: [
+                            {
+                                id: part.id ?? randomUUID(),
+                                type: "function",
+                                function: {
+                                    name: part.tool ?? "unknown",
+                                    arguments: JSON.stringify(toolInput ?? {}),
+                                },
+                            },
+                        ],
+                    };
+                    outputMessages.push(msg);
+                    if (onMessage)
+                        onMessage(msg);
+                }
+            }
+            if (outputMessages.length === 0) {
+                const msg = {
+                    type: "result",
+                    uuid: randomUUID(),
+                    session_id: resolvedSessionId,
+                    subtype: "success",
+                    text: "Completed",
+                };
+                outputMessages.push(msg);
+                if (onMessage)
+                    onMessage(msg);
+            }
+            return outputMessages;
+        }
+        catch (error) {
+            const errorMsg = {
+                type: "result",
+                uuid: randomUUID(),
+                session_id: this.sessionId ?? sessionId,
+                subtype: "error",
+                is_error: true,
+                errors: [error instanceof Error ? error.message : String(error)],
+            };
+            if (onMessage)
+                onMessage(errorMsg);
+            return [errorMsg];
+        }
+        finally {
+            this.inFlight = false;
+        }
+    }
+    async interrupt() {
+        if (this.sessionId) {
+            try {
+                const client = await getClient();
+                await client.session.abort({
+                    path: { id: this.sessionId },
+                });
+            }
+            catch (error) {
+                log.warn(`Failed to abort OpenCode session: ${error instanceof Error ? error.message : String(error)}`);
+            }
+        }
+    }
+    close() {
+        if (this.closed)
+            return;
+        this.closed = true;
+    }
+}
+//# sourceMappingURL=opencode-agent.js.map