@ashx-j/lunr 0.1.0

package/dist/builtin-extensions/pi-mcp-adapter/index.js

@@ -0,0 +1,334 @@
+import { Type } from "typebox";
+import { showStatus, showTools, reconnectServers, authenticateServer, logoutServer, openMcpAuthPanel, openMcpPanel, openMcpSetup } from "./commands.js";
+import { loadMcpConfig } from "./config.js";
+import { buildProxyDescription, createDirectToolExecutor, getMissingConfiguredDirectToolServers, resolveDirectTools } from "./direct-tools.js";
+import { flushMetadataCache, initializeMcp, updateStatusBar } from "./init.js";
+import { loadMetadataCache } from "./metadata-cache.js";
+import { executeAuthComplete, executeAuthStart, executeCall, executeConnect, executeDescribe, executeList, executeSearch, executeStatus, executeUiMessages } from "./proxy-modes.js";
+import { getConfigPathFromArgv, normalizeDirectToolInputSchema, truncateAtWord } from "./utils.js";
+import { initializeOAuth, shutdownOAuth } from "./mcp-auth-flow.js";
+import { createMcpDirectToolCallRenderer, renderMcpProxyToolCall, renderMcpToolResult } from "./tool-result-renderer.js";
+import { toolErrorOverride } from "./error-signal.js";
+export default function mcpAdapter(pi) {
+    let state = null;
+    let initPromise = null;
+    let lifecycleGeneration = 0;
+    async function shutdownState(currentState, reason) {
+        if (!currentState)
+            return;
+        if (currentState.uiServer) {
+            currentState.uiServer.close(reason);
+            currentState.uiServer = null;
+        }
+        let flushError;
+        try {
+            flushMetadataCache(currentState);
+        }
+        catch (error) {
+            flushError = error;
+        }
+        try {
+            await currentState.lifecycle.gracefulShutdown();
+        }
+        catch (error) {
+            if (flushError) {
+                console.error("MCP: graceful shutdown failed after metadata flush error", error);
+            }
+            else {
+                throw error;
+            }
+        }
+        if (flushError) {
+            throw flushError;
+        }
+    }
+    const earlyConfigPath = getConfigPathFromArgv();
+    const earlyConfig = loadMcpConfig(earlyConfigPath);
+    const earlyCache = loadMetadataCache();
+    const prefix = earlyConfig.settings?.toolPrefix ?? "server";
+    const envRaw = process.env.MCP_DIRECT_TOOLS;
+    const directSpecs = envRaw === "__none__"
+        ? []
+        : resolveDirectTools(earlyConfig, earlyCache, prefix, envRaw?.split(",").map(s => s.trim()).filter(Boolean));
+    const missingConfiguredDirectToolServers = getMissingConfiguredDirectToolServers(earlyConfig, earlyCache);
+    const shouldRegisterProxyTool = earlyConfig.settings?.disableProxyTool !== true
+        || directSpecs.length === 0
+        || missingConfiguredDirectToolServers.length > 0;
+    for (const spec of directSpecs) {
+        pi.registerTool({
+            name: spec.prefixedName,
+            label: `MCP: ${spec.originalName}`,
+            description: spec.description || "(no description)",
+            promptSnippet: truncateAtWord(spec.description, 100) || `MCP tool from ${spec.serverName}`,
+            parameters: Type.Unsafe(normalizeDirectToolInputSchema(spec.inputSchema)),
+            execute: createDirectToolExecutor(() => state, () => initPromise, spec),
+            renderCall: createMcpDirectToolCallRenderer(spec.prefixedName),
+            renderResult: renderMcpToolResult,
+        });
+    }
+    const getPiTools = () => pi.getAllTools();
+    pi.registerFlag("mcp-config", {
+        description: "Path to MCP config file",
+        type: "string",
+    });
+    pi.on("session_start", async (_event, ctx) => {
+        const generation = ++lifecycleGeneration;
+        const previousState = state;
+        state = null;
+        initPromise = null;
+        try {
+            await Promise.all([
+                shutdownState(previousState, "session_restart"),
+                shutdownOAuth(),
+            ]);
+        }
+        catch (error) {
+            console.error("MCP: failed to shut down previous session state", error);
+        }
+        if (generation !== lifecycleGeneration) {
+            return;
+        }
+        await initializeOAuth().catch(err => {
+            console.error("MCP OAuth initialization failed:", err);
+        });
+        const promise = initializeMcp(pi, ctx);
+        initPromise = promise;
+        promise.then(async (nextState) => {
+            if (generation !== lifecycleGeneration || initPromise !== promise) {
+                try {
+                    await shutdownState(nextState, "stale_session_start");
+                }
+                catch (error) {
+                    console.error("MCP: failed to clean stale session state", error);
+                }
+                return;
+            }
+            state = nextState;
+            updateStatusBar(nextState);
+            initPromise = null;
+        }).catch(err => {
+            if (generation !== lifecycleGeneration) {
+                return;
+            }
+            if (initPromise !== promise && initPromise !== null) {
+                return;
+            }
+            console.error("MCP initialization failed:", err);
+            initPromise = null;
+        });
+    });
+    pi.on("session_shutdown", async () => {
+        ++lifecycleGeneration;
+        const currentState = state;
+        state = null;
+        initPromise = null;
+        try {
+            await Promise.all([
+                shutdownState(currentState, "session_shutdown"),
+                shutdownOAuth(),
+            ]);
+        }
+        catch (error) {
+            console.error("MCP: session shutdown cleanup failed", error);
+        }
+    });
+    // Re-flag returned MCP tool failures so pi registers them as errors (see toolErrorOverride).
+    pi.on("tool_result", (event) => toolErrorOverride(event.details));
+    pi.registerCommand("mcp", {
+        description: "Show MCP server status",
+        handler: async (args, ctx) => {
+            if (!state && initPromise) {
+                try {
+                    state = await initPromise;
+                }
+                catch (error) {
+                    const message = error instanceof Error ? error.message : String(error);
+                    if (ctx.hasUI)
+                        ctx.ui.notify(`MCP initialization failed: ${message}`, "error");
+                    return;
+                }
+            }
+            if (!state) {
+                if (ctx.hasUI)
+                    ctx.ui.notify("MCP not initialized", "error");
+                return;
+            }
+            const parts = args?.trim()?.split(/\s+/) ?? [];
+            const subcommand = parts[0] ?? "";
+            const targetServer = parts[1];
+            const rest = parts.slice(1).join(" ");
+            switch (subcommand) {
+                case "reconnect":
+                    await reconnectServers(state, ctx, targetServer);
+                    break;
+                case "tools":
+                    await showTools(state, ctx);
+                    break;
+                case "setup": {
+                    const result = await openMcpSetup(state, pi, ctx, earlyConfigPath, "setup");
+                    if (result?.configChanged) {
+                        await ctx.reload();
+                        return;
+                    }
+                    break;
+                }
+                case "logout": {
+                    const serverName = rest;
+                    if (!serverName) {
+                        if (ctx.hasUI)
+                            ctx.ui.notify("Usage: /mcp logout <server>", "error");
+                        return;
+                    }
+                    await logoutServer(serverName, state, ctx);
+                    break;
+                }
+                case "status":
+                case "":
+                default:
+                    if (ctx.hasUI) {
+                        const result = await openMcpPanel(state, pi, ctx, earlyConfigPath);
+                        if (result?.configChanged) {
+                            await ctx.reload();
+                            return;
+                        }
+                    }
+                    else {
+                        await showStatus(state, ctx);
+                    }
+                    break;
+            }
+        },
+    });
+    pi.registerCommand("mcp-auth", {
+        description: "Authenticate with an MCP server (OAuth)",
+        handler: async (args, ctx) => {
+            const serverName = args?.trim();
+            if (!serverName && !ctx.hasUI) {
+                return;
+            }
+            if (!state && initPromise) {
+                try {
+                    state = await initPromise;
+                }
+                catch (error) {
+                    const message = error instanceof Error ? error.message : String(error);
+                    if (ctx.hasUI)
+                        ctx.ui.notify(`MCP initialization failed: ${message}`, "error");
+                    return;
+                }
+            }
+            if (!state) {
+                if (ctx.hasUI)
+                    ctx.ui.notify("MCP not initialized", "error");
+                return;
+            }
+            if (!serverName) {
+                await openMcpAuthPanel(state, pi, ctx, earlyConfigPath);
+                return;
+            }
+            await authenticateServer(serverName, state.config, ctx);
+        },
+    });
+    if (shouldRegisterProxyTool) {
+        pi.registerTool({
+            name: "mcp",
+            label: "MCP",
+            description: buildProxyDescription(earlyConfig, earlyCache, directSpecs),
+            promptSnippet: "MCP gateway - connect to MCP servers and call their tools",
+            renderCall: renderMcpProxyToolCall,
+            parameters: Type.Object({
+                tool: Type.Optional(Type.String({ description: "Tool name to call (e.g., 'xcodebuild_list_sims')" })),
+                args: Type.Optional(Type.String({ description: "Arguments as JSON string (e.g., '{\"key\": \"value\"}')" })),
+                connect: Type.Optional(Type.String({ description: "Server name to connect (lazy connect + metadata refresh)" })),
+                describe: Type.Optional(Type.String({ description: "Tool name to describe (shows parameters)" })),
+                search: Type.Optional(Type.String({ description: "Search tools by name/description" })),
+                regex: Type.Optional(Type.Boolean({ description: "Treat search as regex (default: substring match)" })),
+                includeSchemas: Type.Optional(Type.Boolean({ description: "Include parameter schemas in search results (default: true)" })),
+                server: Type.Optional(Type.String({ description: "Filter to specific server (also disambiguates tool calls)" })),
+                action: Type.Optional(Type.String({ description: "Action: 'ui-messages', 'auth-start', or 'auth-complete'" })),
+            }),
+            renderResult: renderMcpToolResult,
+            async execute(_toolCallId, params, signal, _onUpdate, _ctx) {
+                let parsedArgs;
+                if (params.args) {
+                    try {
+                        parsedArgs = JSON.parse(params.args);
+                        if (typeof parsedArgs !== "object" || parsedArgs === null || Array.isArray(parsedArgs)) {
+                            const gotType = Array.isArray(parsedArgs) ? "array" : parsedArgs === null ? "null" : typeof parsedArgs;
+                            throw new Error(`Invalid args: expected a JSON object, got ${gotType}`);
+                        }
+                    }
+                    catch (error) {
+                        if (error instanceof SyntaxError) {
+                            throw new Error(`Invalid args JSON: ${error.message}`, { cause: error });
+                        }
+                        throw error;
+                    }
+                }
+                if (!state && initPromise) {
+                    try {
+                        state = await initPromise;
+                    }
+                    catch (error) {
+                        const message = error instanceof Error ? error.message : String(error);
+                        return {
+                            content: [{ type: "text", text: `MCP initialization failed: ${message}` }],
+                            details: { error: "init_failed", message },
+                        };
+                    }
+                }
+                if (!state) {
+                    return {
+                        content: [{ type: "text", text: "MCP not initialized" }],
+                        details: { error: "not_initialized" },
+                    };
+                }
+                if (params.action === "ui-messages") {
+                    return executeUiMessages(state);
+                }
+                if (params.action === "auth-start") {
+                    if (!params.server) {
+                        return {
+                            content: [{ type: "text", text: "auth-start requires `server`. Example: mcp({ action: \"auth-start\", server: \"linear-server\" })" }],
+                            details: { mode: "auth-start", error: "missing_server" },
+                        };
+                    }
+                    return executeAuthStart(state, params.server);
+                }
+                if (params.action === "auth-complete") {
+                    if (!params.server) {
+                        return {
+                            content: [{ type: "text", text: "auth-complete requires `server`." }],
+                            details: { mode: "auth-complete", error: "missing_server" },
+                        };
+                    }
+                    const input = parsedArgs?.redirectUrl ?? parsedArgs?.code ?? parsedArgs?.input;
+                    if (typeof input !== "string" || input.trim().length === 0) {
+                        return {
+                            content: [{ type: "text", text: "auth-complete requires args with `redirectUrl`, `code`, or `input`." }],
+                            details: { mode: "auth-complete", error: "missing_input" },
+                        };
+                    }
+                    return executeAuthComplete(state, params.server, input);
+                }
+                if (params.tool) {
+                    return executeCall(state, params.tool, parsedArgs, params.server, getPiTools, signal);
+                }
+                if (params.connect) {
+                    return executeConnect(state, params.connect, signal);
+                }
+                if (params.describe) {
+                    return executeDescribe(state, params.describe);
+                }
+                if (params.search) {
+                    return executeSearch(state, params.search, params.regex, params.server, params.includeSchemas);
+                }
+                if (params.server) {
+                    return executeList(state, params.server);
+                }
+                return executeStatus(state);
+            },
+        });
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/builtin-extensions/pi-mcp-adapter/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/index.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC/B,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,gBAAgB,EAAE,kBAAkB,EAAE,YAAY,EAAE,gBAAgB,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AACxJ,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,qCAAqC,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAC/I,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC;AAC/E,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,mBAAmB,EAAE,gBAAgB,EAAE,WAAW,EAAE,cAAc,EAAE,eAAe,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrL,OAAO,EAAE,qBAAqB,EAAE,8BAA8B,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACnG,OAAO,EAAE,eAAe,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACpE,OAAO,EAAE,+BAA+B,EAAE,sBAAsB,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AACzH,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEtD,MAAM,CAAC,OAAO,UAAU,UAAU,CAAC,EAAgB,EAAE;IACnD,IAAI,KAAK,GAA6B,IAAI,CAAC;IAC3C,IAAI,WAAW,GAAsC,IAAI,CAAC;IAC1D,IAAI,mBAAmB,GAAG,CAAC,CAAC;IAE5B,KAAK,UAAU,aAAa,CAAC,YAAsC,EAAE,MAAc,EAAiB;QAClG,IAAI,CAAC,YAAY;YAAE,OAAO;QAE1B,IAAI,YAAY,CAAC,QAAQ,EAAE,CAAC;YAC1B,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;YACpC,YAAY,CAAC,QAAQ,GAAG,IAAI,CAAC;QAC/B,CAAC;QAED,IAAI,UAAmB,CAAC;QACxB,IAAI,CAAC;YACH,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,UAAU,GAAG,KAAK,CAAC;QACrB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,YAAY,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC;QAClD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,0DAA0D,EAAE,KAAK,CAAC,CAAC;YACnF,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,MAAM,UAAU,CAAC;QACnB,CAAC;IAAA,CACF;IAED,MAAM,eAAe,GAAG,qBAAqB,EAAE,CAAC;IAChD,MAAM,WAAW,GAAG,aAAa,CAAC,eAAe,CAAC,CAAC;IACnD,MAAM,UAAU,GAAG,iBAAiB,EAAE,CAAC;IACvC,MAAM,MAAM,GAAG,WAAW,CAAC,QAAQ,EAAE,UAAU,IAAI,QAAQ,CAAC;IAE5D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC5C,MAAM,WAAW,GAAG,MAAM,KAAK,UAAU;QACvC,CAAC,CAAC,EAAE;QACJ,CAAC,CAAC,kBAAkB,CAChB,WAAW,EACX,UAAU,EACV,MAAM,EACN,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CACtD,CAAC;IACN,MAAM,kCAAkC,GAAG,qCAAqC,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;IAC1G,MAAM,uBAAuB,GAC3B,WAAW,CAAC,QAAQ,EAAE,gBAAgB,KAAK,IAAI;WAC5C,WAAW,CAAC,MAAM,KAAK,CAAC;WACxB,kCAAkC,CAAC,MAAM,GAAG,CAAC,CAAC;IAEnD,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC9B,EAAE,CAAC,YAA2C,CAAC;YAC9C,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,KAAK,EAAE,QAAQ,IAAI,CAAC,YAAY,EAAE;YAClC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,kBAAkB;YACnD,aAAa,EAAE,cAAc,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,IAAI,iBAAiB,IAAI,CAAC,UAAU,EAAE;YAC1F,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,8BAA8B,CAAC,IAAI,CAAC,WAAW,CAAU,CAAC;YAClF,OAAO,EAAE,wBAAwB,CAAC,GAAG,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,CAAC,WAAW,EAAE,IAAI,CAAC;YACvE,UAAU,EAAE,+BAA+B,CAAC,IAAI,CAAC,YAAY,CAAC;YAC9D,YAAY,EAAE,mBAAmB;SAClC,CAAC,CAAC;IACL,CAAC;IAED,MAAM,UAAU,GAAG,GAAe,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;IAEtD,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE;QAC5B,WAAW,EAAE,yBAAyB;QACtC,IAAI,EAAE,QAAQ;KACf,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAC5C,MAAM,UAAU,GAAG,EAAE,mBAAmB,CAAC;QACzC,MAAM,aAAa,GAAG,KAAK,CAAC;QAC5B,KAAK,GAAG,IAAI,CAAC;QACb,WAAW,GAAG,IAAI,CAAC;QAEnB,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,aAAa,CAAC,aAAa,EAAE,iBAAiB,CAAC;gBAC/C,aAAa,EAAE;aAChB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,iDAAiD,EAAE,KAAK,CAAC,CAAC;QAC1E,CAAC;QAED,IAAI,UAAU,KAAK,mBAAmB,EAAE,CAAC;YACvC,OAAO;QACT,CAAC;QAED,MAAM,eAAe,EAAE,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACnC,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;QAAA,CACxD,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,aAAa,CAAC,EAAE,EAAE,GAAG,CAAC,CAAC;QACvC,WAAW,GAAG,OAAO,CAAC;QAEtB,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAAC;YAChC,IAAI,UAAU,KAAK,mBAAmB,IAAI,WAAW,KAAK,OAAO,EAAE,CAAC;gBAClE,IAAI,CAAC;oBACH,MAAM,aAAa,CAAC,SAAS,EAAE,qBAAqB,CAAC,CAAC;gBACxD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,0CAA0C,EAAE,KAAK,CAAC,CAAC;gBACnE,CAAC;gBACD,OAAO;YACT,CAAC;YAED,KAAK,GAAG,SAAS,CAAC;YAClB,eAAe,CAAC,SAAS,CAAC,CAAC;YAC3B,WAAW,GAAG,IAAI,CAAC;QAAA,CACpB,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;YACd,IAAI,UAAU,KAAK,mBAAmB,EAAE,CAAC;gBACvC,OAAO;YACT,CAAC;YACD,IAAI,WAAW,KAAK,OAAO,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;gBACpD,OAAO;YACT,CAAC;YACD,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,GAAG,CAAC,CAAC;YACjD,WAAW,GAAG,IAAI,CAAC;QAAA,CACpB,CAAC,CAAC;IAAA,CACJ,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE,CAAC;QACpC,EAAE,mBAAmB,CAAC;QACtB,MAAM,YAAY,GAAG,KAAK,CAAC;QAC3B,KAAK,GAAG,IAAI,CAAC;QACb,WAAW,GAAG,IAAI,CAAC;QAEnB,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChB,aAAa,CAAC,YAAY,EAAE,kBAAkB,CAAC;gBAC/C,aAAa,EAAE;aAChB,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE,KAAK,CAAC,CAAC;QAC/D,CAAC;IAAA,CACF,CAAC,CAAC;IAEH,6FAA6F;IAC7F,EAAE,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;IAElE,EAAE,CAAC,eAAe,CAAC,KAAK,EAAE;QACxB,WAAW,EAAE,wBAAwB;QACrC,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC;YAC5B,IAAI,CAAC,KAAK,IAAI,WAAW,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,KAAK,GAAG,MAAM,WAAW,CAAC;gBAC5B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACvE,IAAI,GAAG,CAAC,KAAK;wBAAE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,8BAA8B,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;oBAC/E,OAAO;gBACT,CAAC;YACH,CAAC;YACD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,IAAI,GAAG,CAAC,KAAK;oBAAE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;gBAC7D,OAAO;YACT,CAAC;YAED,MAAM,KAAK,GAAG,IAAI,EAAE,IAAI,EAAE,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;YAC/C,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAClC,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;YAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YAEtC,QAAQ,UAAU,EAAE,CAAC;gBACnB,KAAK,WAAW;oBACd,MAAM,gBAAgB,CAAC,KAAK,EAAE,GAAG,EAAE,YAAY,CAAC,CAAC;oBACjD,MAAM;gBACR,KAAK,OAAO;oBACV,MAAM,SAAS,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;oBAC5B,MAAM;gBACR,KAAK,OAAO,EAAE,CAAC;oBACb,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;oBAC5E,IAAI,MAAM,EAAE,aAAa,EAAE,CAAC;wBAC1B,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC;wBACnB,OAAO;oBACT,CAAC;oBACD,MAAM;gBACR,CAAC;gBACD,KAAK,QAAQ,EAAE,CAAC;oBACd,MAAM,UAAU,GAAG,IAAI,CAAC;oBACxB,IAAI,CAAC,UAAU,EAAE,CAAC;wBAChB,IAAI,GAAG,CAAC,KAAK;4BAAE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,6BAA6B,EAAE,OAAO,CAAC,CAAC;wBACrE,OAAO;oBACT,CAAC;oBACD,MAAM,YAAY,CAAC,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC;oBAC3C,MAAM;gBACR,CAAC;gBACD,KAAK,QAAQ,CAAC;gBACd,KAAK,EAAE,CAAC;gBACR;oBACE,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;wBACd,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,eAAe,CAAC,CAAC;wBACnE,IAAI,MAAM,EAAE,aAAa,EAAE,CAAC;4BAC1B,MAAM,GAAG,CAAC,MAAM,EAAE,CAAC;4BACnB,OAAO;wBACT,CAAC;oBACH,CAAC;yBAAM,CAAC;wBACN,MAAM,UAAU,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;oBAC/B,CAAC;oBACD,MAAM;YACV,CAAC;QAAA,CACF;KACF,CAAC,CAAC;IAEH,EAAE,CAAC,eAAe,CAAC,UAAU,EAAE;QAC7B,WAAW,EAAE,yCAAyC;QACtD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC;YAC5B,MAAM,UAAU,GAAG,IAAI,EAAE,IAAI,EAAE,CAAC;YAChC,IAAI,CAAC,UAAU,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;gBAC9B,OAAO;YACT,CAAC;YAED,IAAI,CAAC,KAAK,IAAI,WAAW,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,KAAK,GAAG,MAAM,WAAW,CAAC;gBAC5B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACvE,IAAI,GAAG,CAAC,KAAK;wBAAE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,8BAA8B,OAAO,EAAE,EAAE,OAAO,CAAC,CAAC;oBAC/E,OAAO;gBACT,CAAC;YACH,CAAC;YACD,IAAI,CAAC,KAAK,EAAE,CAAC;gBACX,IAAI,GAAG,CAAC,KAAK;oBAAE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,qBAAqB,EAAE,OAAO,CAAC,CAAC;gBAC7D,OAAO;YACT,CAAC;YAED,IAAI,CAAC,UAAU,EAAE,CAAC;gBAChB,MAAM,gBAAgB,CAAC,KAAK,EAAE,EAAE,EAAE,GAAG,EAAE,eAAe,CAAC,CAAC;gBACxD,OAAO;YACT,CAAC;YAED,MAAM,kBAAkB,CAAC,UAAU,EAAE,KAAK,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAAA,CACzD;KACF,CAAC,CAAC;IAEH,IAAI,uBAAuB,EAAE,CAAC;QAC3B,EAAE,CAAC,YAA2C,CAAC;YAC9C,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK;YACZ,WAAW,EAAE,qBAAqB,CAAC,WAAW,EAAE,UAAU,EAAE,WAAW,CAAC;YACxE,aAAa,EAAE,2DAA2D;YAC1E,UAAU,EAAE,sBAAsB;YAClC,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC;gBACtB,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,kDAAkD,EAAE,CAAC,CAAC;gBACrG,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,yDAAyD,EAAE,CAAC,CAAC;gBAC5G,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,0DAA0D,EAAE,CAAC,CAAC;gBAChH,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,0CAA0C,EAAE,CAAC,CAAC;gBACjG,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,kCAAkC,EAAE,CAAC,CAAC;gBACvF,KAAK,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,kDAAkD,EAAE,CAAC,CAAC;gBACvG,cAAc,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,6DAA6D,EAAE,CAAC,CAAC;gBAC3H,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,2DAA2D,EAAE,CAAC,CAAC;gBAChH,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,yDAAyD,EAAE,CAAC,CAAC;aAC/G,CAAC;YACF,YAAY,EAAE,mBAAmB;YACjC,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,MAU1B,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE;gBAC1B,IAAI,UAA+C,CAAC;gBACpD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChB,IAAI,CAAC;wBACH,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;wBACrC,IAAI,OAAO,UAAU,KAAK,QAAQ,IAAI,UAAU,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;4BACvF,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,UAAU,CAAC;4BACvG,MAAM,IAAI,KAAK,CAAC,6CAA6C,OAAO,EAAE,CAAC,CAAC;wBAC1E,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;4BACjC,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,CAAC,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;wBAC3E,CAAC;wBACD,MAAM,KAAK,CAAC;oBACd,CAAC;gBACH,CAAC;gBAED,IAAI,CAAC,KAAK,IAAI,WAAW,EAAE,CAAC;oBAC1B,IAAI,CAAC;wBACH,KAAK,GAAG,MAAM,WAAW,CAAC;oBAC5B,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;wBACvE,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,8BAA8B,OAAO,EAAE,EAAE,CAAC;4BACnF,OAAO,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,OAAO,EAAE;yBAC3C,CAAC;oBACJ,CAAC;gBACH,CAAC;gBACD,IAAI,CAAC,KAAK,EAAE,CAAC;oBACX,OAAO;wBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,qBAAqB,EAAE,CAAC;wBACjE,OAAO,EAAE,EAAE,KAAK,EAAE,iBAAiB,EAAE;qBACtC,CAAC;gBACJ,CAAC;gBAED,IAAI,MAAM,CAAC,MAAM,KAAK,aAAa,EAAE,CAAC;oBACpC,OAAO,iBAAiB,CAAC,KAAK,CAAC,CAAC;gBAClC,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;oBACnC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;wBACnB,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,mGAAmG,EAAE,CAAC;4BAC/I,OAAO,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,gBAAgB,EAAE;yBACzD,CAAC;oBACJ,CAAC;oBACD,OAAO,gBAAgB,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBAChD,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,KAAK,eAAe,EAAE,CAAC;oBACtC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;wBACnB,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,kCAAkC,EAAE,CAAC;4BAC9E,OAAO,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,gBAAgB,EAAE;yBAC5D,CAAC;oBACJ,CAAC;oBACD,MAAM,KAAK,GAAG,UAAU,EAAE,WAAW,IAAI,UAAU,EAAE,IAAI,IAAI,UAAU,EAAE,KAAK,CAAC;oBAC/E,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC3D,OAAO;4BACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,qEAAqE,EAAE,CAAC;4BACjH,OAAO,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,eAAe,EAAE;yBAC3D,CAAC;oBACJ,CAAC;oBACD,OAAO,mBAAmB,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;gBAC1D,CAAC;gBACD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;oBAChB,OAAO,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;gBACxF,CAAC;gBACD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO,cAAc,CAAC,KAAK,EAAE,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;gBACvD,CAAC;gBACD,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACpB,OAAO,eAAe,CAAC,KAAK,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACjD,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClB,OAAO,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,cAAc,CAAC,CAAC;gBACjG,CAAC;gBACD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;oBAClB,OAAO,WAAW,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;gBAC3C,CAAC;gBACD,OAAO,aAAa,CAAC,KAAK,CAAC,CAAC;YAAA,CAC7B;SACF,CAAC,CAAC;IACL,CAAC;AAAA,CACF","sourcesContent":["// @ts-nocheck\nimport type { ExtensionAPI, ToolInfo } from \"@earendil-works/pi-coding-agent\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport { Type } from \"typebox\";\nimport { showStatus, showTools, reconnectServers, authenticateServer, logoutServer, openMcpAuthPanel, openMcpPanel, openMcpSetup } from \"./commands.ts\";\nimport { loadMcpConfig } from \"./config.ts\";\nimport { buildProxyDescription, createDirectToolExecutor, getMissingConfiguredDirectToolServers, resolveDirectTools } from \"./direct-tools.ts\";\nimport { flushMetadataCache, initializeMcp, updateStatusBar } from \"./init.ts\";\nimport { loadMetadataCache } from \"./metadata-cache.ts\";\nimport { executeAuthComplete, executeAuthStart, executeCall, executeConnect, executeDescribe, executeList, executeSearch, executeStatus, executeUiMessages } from \"./proxy-modes.ts\";\nimport { getConfigPathFromArgv, normalizeDirectToolInputSchema, truncateAtWord } from \"./utils.ts\";\nimport { initializeOAuth, shutdownOAuth } from \"./mcp-auth-flow.ts\";\nimport { createMcpDirectToolCallRenderer, renderMcpProxyToolCall, renderMcpToolResult } from \"./tool-result-renderer.ts\";\nimport { toolErrorOverride } from \"./error-signal.ts\";\n\nexport default function mcpAdapter(pi: ExtensionAPI) {\n  let state: McpExtensionState | null = null;\n  let initPromise: Promise<McpExtensionState> | null = null;\n  let lifecycleGeneration = 0;\n\n  async function shutdownState(currentState: McpExtensionState | null, reason: string): Promise<void> {\n    if (!currentState) return;\n\n    if (currentState.uiServer) {\n      currentState.uiServer.close(reason);\n      currentState.uiServer = null;\n    }\n\n    let flushError: unknown;\n    try {\n      flushMetadataCache(currentState);\n    } catch (error) {\n      flushError = error;\n    }\n\n    try {\n      await currentState.lifecycle.gracefulShutdown();\n    } catch (error) {\n      if (flushError) {\n        console.error(\"MCP: graceful shutdown failed after metadata flush error\", error);\n      } else {\n        throw error;\n      }\n    }\n\n    if (flushError) {\n      throw flushError;\n    }\n  }\n\n  const earlyConfigPath = getConfigPathFromArgv();\n  const earlyConfig = loadMcpConfig(earlyConfigPath);\n  const earlyCache = loadMetadataCache();\n  const prefix = earlyConfig.settings?.toolPrefix ?? \"server\";\n\n  const envRaw = process.env.MCP_DIRECT_TOOLS;\n  const directSpecs = envRaw === \"__none__\"\n    ? []\n    : resolveDirectTools(\n        earlyConfig,\n        earlyCache,\n        prefix,\n        envRaw?.split(\",\").map(s => s.trim()).filter(Boolean),\n      );\n  const missingConfiguredDirectToolServers = getMissingConfiguredDirectToolServers(earlyConfig, earlyCache);\n  const shouldRegisterProxyTool =\n    earlyConfig.settings?.disableProxyTool !== true\n    || directSpecs.length === 0\n    || missingConfiguredDirectToolServers.length > 0;\n\n  for (const spec of directSpecs) {\n    (pi.registerTool as (tool: unknown) => unknown)({\n      name: spec.prefixedName,\n      label: `MCP: ${spec.originalName}`,\n      description: spec.description || \"(no description)\",\n      promptSnippet: truncateAtWord(spec.description, 100) || `MCP tool from ${spec.serverName}`,\n      parameters: Type.Unsafe(normalizeDirectToolInputSchema(spec.inputSchema) as never),\n      execute: createDirectToolExecutor(() => state, () => initPromise, spec),\n      renderCall: createMcpDirectToolCallRenderer(spec.prefixedName),\n      renderResult: renderMcpToolResult,\n    });\n  }\n\n  const getPiTools = (): ToolInfo[] => pi.getAllTools();\n\n  pi.registerFlag(\"mcp-config\", {\n    description: \"Path to MCP config file\",\n    type: \"string\",\n  });\n\n  pi.on(\"session_start\", async (_event, ctx) => {\n    const generation = ++lifecycleGeneration;\n    const previousState = state;\n    state = null;\n    initPromise = null;\n\n    try {\n      await Promise.all([\n        shutdownState(previousState, \"session_restart\"),\n        shutdownOAuth(),\n      ]);\n    } catch (error) {\n      console.error(\"MCP: failed to shut down previous session state\", error);\n    }\n\n    if (generation !== lifecycleGeneration) {\n      return;\n    }\n\n    await initializeOAuth().catch(err => {\n      console.error(\"MCP OAuth initialization failed:\", err);\n    });\n\n    const promise = initializeMcp(pi, ctx);\n    initPromise = promise;\n\n    promise.then(async (nextState) => {\n      if (generation !== lifecycleGeneration || initPromise !== promise) {\n        try {\n          await shutdownState(nextState, \"stale_session_start\");\n        } catch (error) {\n          console.error(\"MCP: failed to clean stale session state\", error);\n        }\n        return;\n      }\n\n      state = nextState;\n      updateStatusBar(nextState);\n      initPromise = null;\n    }).catch(err => {\n      if (generation !== lifecycleGeneration) {\n        return;\n      }\n      if (initPromise !== promise && initPromise !== null) {\n        return;\n      }\n      console.error(\"MCP initialization failed:\", err);\n      initPromise = null;\n    });\n  });\n\n  pi.on(\"session_shutdown\", async () => {\n    ++lifecycleGeneration;\n    const currentState = state;\n    state = null;\n    initPromise = null;\n\n    try {\n      await Promise.all([\n        shutdownState(currentState, \"session_shutdown\"),\n        shutdownOAuth(),\n      ]);\n    } catch (error) {\n      console.error(\"MCP: session shutdown cleanup failed\", error);\n    }\n  });\n\n  // Re-flag returned MCP tool failures so pi registers them as errors (see toolErrorOverride).\n  pi.on(\"tool_result\", (event) => toolErrorOverride(event.details));\n\n  pi.registerCommand(\"mcp\", {\n    description: \"Show MCP server status\",\n    handler: async (args, ctx) => {\n      if (!state && initPromise) {\n        try {\n          state = await initPromise;\n        } catch (error) {\n          const message = error instanceof Error ? error.message : String(error);\n          if (ctx.hasUI) ctx.ui.notify(`MCP initialization failed: ${message}`, \"error\");\n          return;\n        }\n      }\n      if (!state) {\n        if (ctx.hasUI) ctx.ui.notify(\"MCP not initialized\", \"error\");\n        return;\n      }\n\n      const parts = args?.trim()?.split(/\\s+/) ?? [];\n      const subcommand = parts[0] ?? \"\";\n      const targetServer = parts[1];\n      const rest = parts.slice(1).join(\" \");\n\n      switch (subcommand) {\n        case \"reconnect\":\n          await reconnectServers(state, ctx, targetServer);\n          break;\n        case \"tools\":\n          await showTools(state, ctx);\n          break;\n        case \"setup\": {\n          const result = await openMcpSetup(state, pi, ctx, earlyConfigPath, \"setup\");\n          if (result?.configChanged) {\n            await ctx.reload();\n            return;\n          }\n          break;\n        }\n        case \"logout\": {\n          const serverName = rest;\n          if (!serverName) {\n            if (ctx.hasUI) ctx.ui.notify(\"Usage: /mcp logout <server>\", \"error\");\n            return;\n          }\n          await logoutServer(serverName, state, ctx);\n          break;\n        }\n        case \"status\":\n        case \"\":\n        default:\n          if (ctx.hasUI) {\n            const result = await openMcpPanel(state, pi, ctx, earlyConfigPath);\n            if (result?.configChanged) {\n              await ctx.reload();\n              return;\n            }\n          } else {\n            await showStatus(state, ctx);\n          }\n          break;\n      }\n    },\n  });\n\n  pi.registerCommand(\"mcp-auth\", {\n    description: \"Authenticate with an MCP server (OAuth)\",\n    handler: async (args, ctx) => {\n      const serverName = args?.trim();\n      if (!serverName && !ctx.hasUI) {\n        return;\n      }\n\n      if (!state && initPromise) {\n        try {\n          state = await initPromise;\n        } catch (error) {\n          const message = error instanceof Error ? error.message : String(error);\n          if (ctx.hasUI) ctx.ui.notify(`MCP initialization failed: ${message}`, \"error\");\n          return;\n        }\n      }\n      if (!state) {\n        if (ctx.hasUI) ctx.ui.notify(\"MCP not initialized\", \"error\");\n        return;\n      }\n\n      if (!serverName) {\n        await openMcpAuthPanel(state, pi, ctx, earlyConfigPath);\n        return;\n      }\n\n      await authenticateServer(serverName, state.config, ctx);\n    },\n  });\n\n  if (shouldRegisterProxyTool) {\n    (pi.registerTool as (tool: unknown) => unknown)({\n      name: \"mcp\",\n      label: \"MCP\",\n      description: buildProxyDescription(earlyConfig, earlyCache, directSpecs),\n      promptSnippet: \"MCP gateway - connect to MCP servers and call their tools\",\n      renderCall: renderMcpProxyToolCall,\n      parameters: Type.Object({\n        tool: Type.Optional(Type.String({ description: \"Tool name to call (e.g., 'xcodebuild_list_sims')\" })),\n        args: Type.Optional(Type.String({ description: \"Arguments as JSON string (e.g., '{\\\"key\\\": \\\"value\\\"}')\" })),\n        connect: Type.Optional(Type.String({ description: \"Server name to connect (lazy connect + metadata refresh)\" })),\n        describe: Type.Optional(Type.String({ description: \"Tool name to describe (shows parameters)\" })),\n        search: Type.Optional(Type.String({ description: \"Search tools by name/description\" })),\n        regex: Type.Optional(Type.Boolean({ description: \"Treat search as regex (default: substring match)\" })),\n        includeSchemas: Type.Optional(Type.Boolean({ description: \"Include parameter schemas in search results (default: true)\" })),\n        server: Type.Optional(Type.String({ description: \"Filter to specific server (also disambiguates tool calls)\" })),\n        action: Type.Optional(Type.String({ description: \"Action: 'ui-messages', 'auth-start', or 'auth-complete'\" })),\n      }),\n      renderResult: renderMcpToolResult,\n      async execute(_toolCallId, params: {\n        tool?: string;\n        args?: string;\n        connect?: string;\n        describe?: string;\n        search?: string;\n        regex?: boolean;\n        includeSchemas?: boolean;\n        server?: string;\n        action?: string;\n      }, signal, _onUpdate, _ctx) {\n        let parsedArgs: Record<string, unknown> | undefined;\n        if (params.args) {\n          try {\n            parsedArgs = JSON.parse(params.args);\n            if (typeof parsedArgs !== \"object\" || parsedArgs === null || Array.isArray(parsedArgs)) {\n              const gotType = Array.isArray(parsedArgs) ? \"array\" : parsedArgs === null ? \"null\" : typeof parsedArgs;\n              throw new Error(`Invalid args: expected a JSON object, got ${gotType}`);\n            }\n          } catch (error) {\n            if (error instanceof SyntaxError) {\n              throw new Error(`Invalid args JSON: ${error.message}`, { cause: error });\n            }\n            throw error;\n          }\n        }\n\n        if (!state && initPromise) {\n          try {\n            state = await initPromise;\n          } catch (error) {\n            const message = error instanceof Error ? error.message : String(error);\n            return {\n              content: [{ type: \"text\" as const, text: `MCP initialization failed: ${message}` }],\n              details: { error: \"init_failed\", message },\n            };\n          }\n        }\n        if (!state) {\n          return {\n            content: [{ type: \"text\" as const, text: \"MCP not initialized\" }],\n            details: { error: \"not_initialized\" },\n          };\n        }\n\n        if (params.action === \"ui-messages\") {\n          return executeUiMessages(state);\n        }\n        if (params.action === \"auth-start\") {\n          if (!params.server) {\n            return {\n              content: [{ type: \"text\" as const, text: \"auth-start requires `server`. Example: mcp({ action: \\\"auth-start\\\", server: \\\"linear-server\\\" })\" }],\n              details: { mode: \"auth-start\", error: \"missing_server\" },\n            };\n          }\n          return executeAuthStart(state, params.server);\n        }\n        if (params.action === \"auth-complete\") {\n          if (!params.server) {\n            return {\n              content: [{ type: \"text\" as const, text: \"auth-complete requires `server`.\" }],\n              details: { mode: \"auth-complete\", error: \"missing_server\" },\n            };\n          }\n          const input = parsedArgs?.redirectUrl ?? parsedArgs?.code ?? parsedArgs?.input;\n          if (typeof input !== \"string\" || input.trim().length === 0) {\n            return {\n              content: [{ type: \"text\" as const, text: \"auth-complete requires args with `redirectUrl`, `code`, or `input`.\" }],\n              details: { mode: \"auth-complete\", error: \"missing_input\" },\n            };\n          }\n          return executeAuthComplete(state, params.server, input);\n        }\n        if (params.tool) {\n          return executeCall(state, params.tool, parsedArgs, params.server, getPiTools, signal);\n        }\n        if (params.connect) {\n          return executeConnect(state, params.connect, signal);\n        }\n        if (params.describe) {\n          return executeDescribe(state, params.describe);\n        }\n        if (params.search) {\n          return executeSearch(state, params.search, params.regex, params.server, params.includeSchemas);\n        }\n        if (params.server) {\n          return executeList(state, params.server);\n        }\n        return executeStatus(state);\n      },\n    });\n  }\n}\n"]}

package/dist/builtin-extensions/pi-mcp-adapter/init.d.ts

@@ -0,0 +1,11 @@
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { McpExtensionState } from "./state.ts";
+export declare function isTuiMode(ctx: Pick<ExtensionContext, "hasUI" | "mode">): boolean;
+export declare function initializeMcp(pi: ExtensionAPI, ctx: ExtensionContext): Promise<McpExtensionState>;
+export declare function updateServerMetadata(state: McpExtensionState, serverName: string): void;
+export declare function updateMetadataCache(state: McpExtensionState, serverName: string): void;
+export declare function flushMetadataCache(state: McpExtensionState): void;
+export declare function updateStatusBar(state: McpExtensionState): void;
+export declare function getFailureAgeSeconds(state: McpExtensionState, serverName: string): number | null;
+export declare function lazyConnect(state: McpExtensionState, serverName: string, signal?: AbortSignal): Promise<boolean>;
+//# sourceMappingURL=init.d.ts.map

package/dist/builtin-extensions/pi-mcp-adapter/init.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/init.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,YAAY,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACtF,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AA2BpD,wBAAgB,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,gBAAgB,EAAE,OAAO,GAAG,MAAM,CAAC,GAAG,OAAO,CAEhF;AAED,wBAAsB,aAAa,CACjC,EAAE,EAAE,YAAY,EAChB,GAAG,EAAE,gBAAgB,GACpB,OAAO,CAAC,iBAAiB,CAAC,CA8L5B;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CAWvF;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,GAAG,IAAI,CA+BtF;AAED,wBAAgB,kBAAkB,CAAC,KAAK,EAAE,iBAAiB,GAAG,IAAI,CAMjE;AAED,wBAAgB,eAAe,CAAC,KAAK,EAAE,iBAAiB,GAAG,IAAI,CAU9D;AAED,wBAAgB,oBAAoB,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAMhG;AAED,wBAAsB,WAAW,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,CAuCtH","sourcesContent":["// @ts-nocheck\nimport type { ExtensionAPI, ExtensionContext } from \"@earendil-works/pi-coding-agent\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport type { ToolMetadata } from \"./types.ts\";\nimport { existsSync } from \"node:fs\";\nimport { loadMcpConfig } from \"./config.ts\";\nimport { ConsentManager } from \"./consent-manager.ts\";\nimport { McpLifecycleManager } from \"./lifecycle.ts\";\nimport {\n  computeServerHash,\n  getMetadataCachePath,\n  isServerCacheValid,\n  loadMetadataCache,\n  reconstructToolMetadata,\n  saveMetadataCache,\n  serializeResources,\n  serializeTools,\n  type ServerCacheEntry,\n} from \"./metadata-cache.ts\";\nimport { McpServerManager } from \"./server-manager.ts\";\nimport { buildToolMetadata, totalToolCount } from \"./tool-metadata.ts\";\nimport { UiResourceHandler } from \"./ui-resource-handler.ts\";\nimport { openUrl, parallelLimit } from \"./utils.ts\";\nimport { logger } from \"./logger.ts\";\nimport { getMissingConfiguredDirectToolServers } from \"./direct-tools.ts\";\nimport { throwIfAborted } from \"./abort.ts\";\n\nconst FAILURE_BACKOFF_MS = 60 * 1000;\n\nexport function isTuiMode(ctx: Pick<ExtensionContext, \"hasUI\" | \"mode\">): boolean {\n  return ctx.hasUI && ctx.mode === \"tui\";\n}\n\nexport async function initializeMcp(\n  pi: ExtensionAPI,\n  ctx: ExtensionContext\n): Promise<McpExtensionState> {\n  const configPath = pi.getFlag(\"mcp-config\") as string | undefined;\n  const config = loadMcpConfig(configPath, ctx.cwd);\n\n  const manager = new McpServerManager(ctx.cwd);\n  manager.setDefaultRequestTimeoutMs(config.settings?.requestTimeoutMs);\n  const samplingAutoApprove = config.settings?.samplingAutoApprove === true;\n  if (config.settings?.sampling !== false && (ctx.hasUI || samplingAutoApprove)) {\n    manager.setSamplingConfig({\n      autoApprove: samplingAutoApprove,\n      ui: ctx.hasUI ? ctx.ui : undefined,\n      modelRegistry: ctx.modelRegistry,\n      getCurrentModel: () => ctx.model,\n      getSignal: () => ctx.signal,\n    });\n  }\n  const elicitationEnabled = config.settings?.elicitation !== false && ctx.hasUI;\n  if (elicitationEnabled) {\n    manager.setElicitationConfig({\n      ui: ctx.ui,\n      allowUrl: isTuiMode(ctx),\n    });\n  }\n  const lifecycle = new McpLifecycleManager(manager);\n  const toolMetadata = new Map<string, ToolMetadata[]>();\n  const failureTracker = new Map<string, number>();\n  const uiResourceHandler = new UiResourceHandler(manager);\n  const consentManager = new ConsentManager(\"once-per-server\");\n  const ui = ctx.hasUI ? ctx.ui : undefined;\n  const state: McpExtensionState = {\n    manager,\n    lifecycle,\n    toolMetadata,\n    config,\n    failureTracker,\n    uiResourceHandler,\n    consentManager,\n    uiServer: null,\n    completedUiSessions: [],\n    openBrowser: (url: string) => openUrl(pi, url, process.env.BROWSER),\n    ui,\n    sendMessage: (message, options) => pi.sendMessage(message as unknown as Parameters<typeof pi.sendMessage>[0], options),\n  };\n\n  const serverEntries = Object.entries(config.mcpServers);\n  if (serverEntries.length === 0) {\n    return state;\n  }\n\n  const idleSetting = typeof config.settings?.idleTimeout === \"number\" ? config.settings.idleTimeout : 10;\n  lifecycle.setGlobalIdleTimeout(idleSetting);\n\n  const cachePath = getMetadataCachePath();\n  const cacheFileExists = existsSync(cachePath);\n  let cache = loadMetadataCache();\n  let bootstrapAll = false;\n\n  if (!cacheFileExists) {\n    bootstrapAll = true;\n    saveMetadataCache({ version: 1, servers: {} });\n  } else if (!cache) {\n    cache = { version: 1, servers: {} };\n    saveMetadataCache(cache);\n  }\n\n  const prefix = config.settings?.toolPrefix ?? \"server\";\n\n  for (const [name, definition] of serverEntries) {\n    const lifecycleMode = definition.lifecycle ?? \"lazy\";\n    const idleOverride = definition.idleTimeout ?? (lifecycleMode === \"eager\" ? 0 : undefined);\n    lifecycle.registerServer(\n      name,\n      definition,\n      idleOverride !== undefined ? { idleTimeout: idleOverride } : undefined\n    );\n    if (lifecycleMode === \"keep-alive\") {\n      lifecycle.markKeepAlive(name, definition);\n    }\n\n    if (cache?.servers?.[name] && isServerCacheValid(cache.servers[name], definition)) {\n      const metadata = reconstructToolMetadata(name, cache.servers[name], prefix, definition);\n      toolMetadata.set(name, metadata);\n    }\n  }\n\n  const startupServers = bootstrapAll\n    ? serverEntries\n    : serverEntries.filter(([, definition]) => {\n        const mode = definition.lifecycle ?? \"lazy\";\n        return mode === \"keep-alive\" || mode === \"eager\";\n      });\n\n  if (ctx.hasUI && startupServers.length > 0) {\n    ctx.ui.setStatus(\"mcp\", `MCP: connecting to ${startupServers.length} servers...`);\n  }\n\n  const results = await parallelLimit(startupServers, 10, async ([name, definition]) => {\n    try {\n      const connection = await manager.connect(name, definition, ctx.signal);\n      if (connection.status === \"needs-auth\") {\n        return { name, definition, connection: null, error: `OAuth authentication required. Run /mcp-auth ${name}.` };\n      }\n      return { name, definition, connection, error: null };\n    } catch (error) {\n      const message = error instanceof Error ? error.message : String(error);\n      return { name, definition, connection: null, error: message };\n    }\n  });\n\n  for (const { name, definition, connection, error } of results) {\n    if (error || !connection) {\n      if (ctx.hasUI) {\n        ctx.ui.notify(`MCP: Failed to connect to ${name}: ${error}`, \"error\");\n      }\n      console.error(`MCP: Failed to connect to ${name}: ${error}`);\n      continue;\n    }\n\n    const { metadata, failedTools } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix);\n    toolMetadata.set(name, metadata);\n    updateMetadataCache(state, name);\n\n    if (failedTools.length > 0 && ctx.hasUI) {\n      ctx.ui.notify(\n        `MCP: ${name} - ${failedTools.length} tools skipped`,\n        \"warning\"\n      );\n    }\n  }\n\n  const connectedCount = results.filter(r => r.connection).length;\n  const failedCount = results.filter(r => r.error).length;\n  if (ctx.hasUI && connectedCount > 0) {\n    const totalTools = totalToolCount(state);\n    const msg = failedCount > 0\n      ? `MCP: ${connectedCount}/${startupServers.length} servers connected (${totalTools} tools)`\n      : `MCP: ${connectedCount} servers connected (${totalTools} tools)`;\n    ctx.ui.notify(msg, \"info\");\n  }\n\n  const envDirect = process.env.MCP_DIRECT_TOOLS;\n  if (envDirect !== \"__none__\") {\n    const currentCache = loadMetadataCache();\n    const missingCacheServers = getMissingConfiguredDirectToolServers(config, currentCache);\n\n    if (missingCacheServers.length > 0) {\n      const bootstrapResults = await parallelLimit(\n        missingCacheServers.filter(name => !results.some(r => r.name === name && r.connection)),\n        10,\n        async (name) => {\n          const definition = config.mcpServers[name];\n          try {\n            const connection = await manager.connect(name, definition, ctx.signal);\n            if (connection.status === \"needs-auth\") {\n              return { name, ok: false };\n            }\n            const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix);\n            toolMetadata.set(name, metadata);\n            updateMetadataCache(state, name);\n            return { name, ok: true };\n          } catch (error) {\n            const message = error instanceof Error ? error.message : String(error);\n            logger.debug(`MCP: direct-tools bootstrap failed for ${name}: ${message}`);\n            return { name, ok: false };\n          }\n        },\n      );\n      const bootstrapped = bootstrapResults.filter(r => r.ok).map(r => r.name);\n      if (bootstrapped.length > 0 && ctx.hasUI) {\n        ctx.ui.notify(`MCP: direct tools for ${bootstrapped.join(\", \")} will be available after restart`, \"info\");\n      }\n    }\n  }\n\n  lifecycle.setReconnectCallback((serverName) => {\n    updateServerMetadata(state, serverName);\n    updateMetadataCache(state, serverName);\n    state.failureTracker.delete(serverName);\n    updateStatusBar(state);\n  });\n\n  lifecycle.setIdleShutdownCallback((serverName) => {\n    const idleMinutes = getEffectiveIdleTimeoutMinutes(state, serverName);\n    logger.debug(`${serverName} shut down (idle ${idleMinutes}m)`);\n    updateStatusBar(state);\n  });\n\n  lifecycle.startHealthChecks();\n\n  return state;\n}\n\nexport function updateServerMetadata(state: McpExtensionState, serverName: string): void {\n  const connection = state.manager.getConnection(serverName);\n  if (!connection || connection.status !== \"connected\") return;\n\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) return;\n\n  const prefix = state.config.settings?.toolPrefix ?? \"server\";\n\n  const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, serverName, prefix);\n  state.toolMetadata.set(serverName, metadata);\n}\n\nexport function updateMetadataCache(state: McpExtensionState, serverName: string): void {\n  const connection = state.manager.getConnection(serverName);\n  if (!connection || connection.status !== \"connected\") return;\n\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) return;\n\n  const configHash = computeServerHash(definition);\n  const existing = loadMetadataCache();\n  const existingEntry = existing?.servers?.[serverName];\n\n  const tools = serializeTools(connection.tools);\n  let resources = definition.exposeResources === false ? [] : serializeResources(connection.resources);\n\n  if (\n    definition.exposeResources !== false &&\n    resources.length === 0 &&\n    existingEntry?.resources?.length &&\n    existingEntry.configHash === configHash\n  ) {\n    resources = existingEntry.resources;\n  }\n\n  const entry: ServerCacheEntry = {\n    configHash,\n    tools,\n    resources,\n    cachedAt: Date.now(),\n  };\n\n  saveMetadataCache({ version: 1, servers: { [serverName]: entry } });\n}\n\nexport function flushMetadataCache(state: McpExtensionState): void {\n  for (const [name, connection] of state.manager.getAllConnections()) {\n    if (connection.status === \"connected\") {\n      updateMetadataCache(state, name);\n    }\n  }\n}\n\nexport function updateStatusBar(state: McpExtensionState): void {\n  const ui = state.ui;\n  if (!ui) return;\n  const total = Object.keys(state.config.mcpServers).length;\n  if (total === 0) {\n    ui.setStatus(\"mcp\", undefined);\n    return;\n  }\n  const connectedCount = state.manager.getAllConnections().size;\n  ui.setStatus(\"mcp\", ui.theme.fg(\"accent\", `MCP: ${connectedCount}/${total} servers`));\n}\n\nexport function getFailureAgeSeconds(state: McpExtensionState, serverName: string): number | null {\n  const failedAt = state.failureTracker.get(serverName);\n  if (!failedAt) return null;\n  const ageMs = Date.now() - failedAt;\n  if (ageMs > FAILURE_BACKOFF_MS) return null;\n  return Math.round(ageMs / 1000);\n}\n\nexport async function lazyConnect(state: McpExtensionState, serverName: string, signal?: AbortSignal): Promise<boolean> {\n  const connection = state.manager.getConnection(serverName);\n  if (connection?.status === \"needs-auth\") {\n    return false;\n  }\n  if (connection?.status === \"connected\") {\n    updateServerMetadata(state, serverName);\n    return true;\n  }\n\n  const failedAgo = getFailureAgeSeconds(state, serverName);\n  if (failedAgo !== null) return false;\n\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) return false;\n\n  try {\n    if (state.ui) {\n      state.ui.setStatus(\"mcp\", `MCP: connecting to ${serverName}...`);\n    }\n    const newConnection = await state.manager.connect(serverName, definition, signal);\n    if (newConnection.status === \"needs-auth\") {\n      return false;\n    }\n    state.failureTracker.delete(serverName);\n    updateServerMetadata(state, serverName);\n    updateMetadataCache(state, serverName);\n    updateStatusBar(state);\n    return true;\n  } catch (error) {\n    if (signal?.aborted) {\n      throwIfAborted(signal);\n    }\n    state.failureTracker.set(serverName, Date.now());\n    const message = error instanceof Error ? error.message : String(error);\n    logger.debug(`MCP: lazy connect failed for ${serverName}: ${message}`);\n    updateStatusBar(state);\n    return false;\n  }\n}\n\nfunction getEffectiveIdleTimeoutMinutes(state: McpExtensionState, serverName: string): number {\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) {\n    return typeof state.config.settings?.idleTimeout === \"number\" ? state.config.settings.idleTimeout : 10;\n  }\n  if (typeof definition.idleTimeout === \"number\") return definition.idleTimeout;\n  const mode = definition.lifecycle ?? \"lazy\";\n  if (mode === \"eager\") return 0;\n  return typeof state.config.settings?.idleTimeout === \"number\" ? state.config.settings.idleTimeout : 10;\n}\n"]}