@ashx-j/lunr 0.1.0

package/dist/builtin-extensions/pi-mcp-adapter/server-manager.js

@@ -0,0 +1,403 @@
+// @ts-nocheck
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
+import { StreamableHTTPClientTransport } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+import { UnauthorizedError } from "@modelcontextprotocol/sdk/client/auth.js";
+import { ElicitationCompleteNotificationSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { serverStreamResultPatchNotificationSchema } from "./types.js";
+import { resolveNpxBinary } from "./npx-resolver.js";
+import { logger } from "./logger.js";
+import { McpOAuthProvider } from "./mcp-oauth-provider.js";
+import { extractOAuthConfig, supportsOAuth } from "./mcp-auth-flow.js";
+import { registerSamplingHandler } from "./sampling-handler.js";
+import { handleUrlElicitation, registerElicitationHandler, } from "./elicitation-handler.js";
+import { interpolateEnvRecord, resolveBearerToken, resolveConfigPath } from "./utils.js";
+import { abortable, throwIfAborted } from "./abort.js";
+export class McpServerManager {
+    defaultCwd;
+    connections = new Map();
+    connectPromises = new Map();
+    uiStreamListeners = new Map();
+    samplingConfig;
+    elicitationConfig;
+    acceptedUrlElicitations = new Map();
+    defaultRequestTimeoutMs;
+    /** Default cwd for stdio servers without an explicit config `cwd`. */
+    constructor(defaultCwd) {
+        this.defaultCwd = defaultCwd;
+    }
+    setSamplingConfig(config) {
+        this.samplingConfig = config;
+    }
+    setElicitationConfig(config) {
+        this.elicitationConfig = config;
+    }
+    setDefaultRequestTimeoutMs(timeoutMs) {
+        this.defaultRequestTimeoutMs = normalizeRequestTimeoutMs(timeoutMs);
+    }
+    getRequestOptions(name, signal) {
+        const connection = this.connections.get(name);
+        return this.buildRequestOptions(connection?.definition, signal);
+    }
+    getResolvedRequestTimeoutMs(definition) {
+        if (definition?.requestTimeoutMs !== undefined) {
+            return normalizeRequestTimeoutMs(definition.requestTimeoutMs);
+        }
+        return this.defaultRequestTimeoutMs;
+    }
+    buildRequestOptions(definition, signal) {
+        const timeout = this.getResolvedRequestTimeoutMs(definition);
+        if (!signal && timeout === undefined) {
+            return undefined;
+        }
+        return {
+            ...(signal ? { signal } : {}),
+            ...(timeout !== undefined ? { timeout } : {}),
+        };
+    }
+    async connect(name, definition, signal) {
+        throwIfAborted(signal);
+        // Dedupe concurrent connection attempts
+        if (this.connectPromises.has(name)) {
+            return abortable(this.connectPromises.get(name), signal);
+        }
+        // Reuse existing connection if healthy
+        const existing = this.connections.get(name);
+        if (existing?.status === "connected") {
+            existing.lastUsedAt = Date.now();
+            return existing;
+        }
+        const promise = this.createConnection(name, definition, signal);
+        this.connectPromises.set(name, promise);
+        try {
+            const connection = await promise;
+            this.connections.set(name, connection);
+            return connection;
+        }
+        finally {
+            this.connectPromises.delete(name);
+        }
+    }
+    async createConnection(name, definition, signal) {
+        throwIfAborted(signal);
+        const client = this.createClient(name);
+        let transport;
+        if (definition.command) {
+            let command = definition.command;
+            let args = definition.args ?? [];
+            if (command === "npx" || command === "npm") {
+                const resolved = await resolveNpxBinary(command, args);
+                if (resolved) {
+                    command = resolved.isJs ? "node" : resolved.binPath;
+                    args = resolved.isJs ? [resolved.binPath, ...resolved.extraArgs] : resolved.extraArgs;
+                    logger.debug(`${name} resolved to ${resolved.binPath} (skipping npm parent)`);
+                }
+            }
+            transport = new StdioClientTransport({
+                command,
+                args,
+                env: resolveEnv(definition.env),
+                cwd: resolveConfigPath(definition.cwd) ?? this.defaultCwd,
+                stderr: definition.debug ? "inherit" : "ignore",
+            });
+        }
+        else if (definition.url) {
+            // HTTP transport with fallback
+            transport = await this.createHttpTransport(definition, name, signal);
+        }
+        else {
+            throw new Error(`Server ${name} has no command or url`);
+        }
+        const requestOptions = this.buildRequestOptions(definition, signal);
+        try {
+            await client.connect(transport, requestOptions);
+            this.attachAdapterNotificationHandlers(name, client);
+            // Discover tools and resources
+            const [tools, resources] = await Promise.all([
+                this.fetchAllTools(client, requestOptions),
+                this.fetchAllResources(client, requestOptions),
+            ]);
+            return {
+                client,
+                transport,
+                definition,
+                tools,
+                resources,
+                lastUsedAt: Date.now(),
+                inFlight: 0,
+                status: "connected",
+            };
+        }
+        catch (error) {
+            // Check for UnauthorizedError - server requires OAuth
+            if (error instanceof UnauthorizedError && supportsOAuth(definition)) {
+                // Clean up both client and transport before reporting needs-auth.
+                await client.close().catch(() => { });
+                await transport.close().catch(() => { });
+                return {
+                    client,
+                    transport,
+                    definition,
+                    tools: [],
+                    resources: [],
+                    lastUsedAt: Date.now(),
+                    inFlight: 0,
+                    status: "needs-auth",
+                };
+            }
+            // Clean up both client and transport on any error
+            await client.close().catch(() => { });
+            await transport.close().catch(() => { });
+            throw error;
+        }
+    }
+    buildClientCapabilities() {
+        return {
+            ...(this.samplingConfig ? { sampling: {} } : {}),
+            ...(this.elicitationConfig
+                ? {
+                    elicitation: {
+                        form: {},
+                        ...(this.elicitationConfig.allowUrl ? { url: {} } : {}),
+                    },
+                }
+                : {}),
+        };
+    }
+    createClient(serverName) {
+        const capabilities = this.buildClientCapabilities();
+        const client = new Client({ name: `pi-mcp-${serverName}`, version: "1.0.0" }, Object.keys(capabilities).length > 0 ? { capabilities } : undefined);
+        if (this.samplingConfig) {
+            registerSamplingHandler(client, { ...this.samplingConfig, serverName });
+        }
+        if (this.elicitationConfig) {
+            registerElicitationHandler(client, {
+                ...this.elicitationConfig,
+                serverName,
+                onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),
+            });
+            if (this.elicitationConfig.allowUrl) {
+                client.setNotificationHandler(ElicitationCompleteNotificationSchema, notification => {
+                    const accepted = this.acceptedUrlElicitations.get(serverName);
+                    if (!accepted?.delete(notification.params.elicitationId))
+                        return;
+                    this.elicitationConfig?.ui.notify(`MCP browser interaction for ${serverName} completed. You can retry the tool now.`, "info");
+                });
+            }
+        }
+        return client;
+    }
+    async handleUrlElicitationRequired(serverName, error) {
+        if (!this.elicitationConfig?.allowUrl)
+            return "cancel";
+        for (const params of error.elicitations) {
+            const result = await handleUrlElicitation({
+                ...this.elicitationConfig,
+                serverName,
+                onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),
+            }, params);
+            if (result.action !== "accept")
+                return result.action;
+        }
+        return "accept";
+    }
+    rememberUrlElicitation(serverName, elicitationId) {
+        let accepted = this.acceptedUrlElicitations.get(serverName);
+        if (!accepted) {
+            accepted = new Set();
+            this.acceptedUrlElicitations.set(serverName, accepted);
+        }
+        accepted.add(elicitationId);
+    }
+    async createHttpTransport(definition, serverName, signal) {
+        throwIfAborted(signal);
+        const url = new URL(definition.url);
+        // Build headers first (including any bearer token)
+        const headers = resolveHeaders(definition.headers) ?? {};
+        // For bearer auth, add the token to headers BEFORE creating requestInit
+        if (definition.auth === "bearer") {
+            const token = resolveBearerToken(definition);
+            if (token) {
+                headers["Authorization"] = `Bearer ${token}`;
+            }
+        }
+        // Create request init with headers (Authorization now included for bearer auth)
+        const requestInit = Object.keys(headers).length > 0 ? { headers } : undefined;
+        // For OAuth servers, create an auth provider
+        let authProvider;
+        if (supportsOAuth(definition)) {
+            const oauthConfig = extractOAuthConfig(definition);
+            authProvider = new McpOAuthProvider(serverName, definition.url, oauthConfig, {
+                onRedirect: async (_authUrl) => {
+                    // URL is captured by startAuth, no need to log
+                },
+            });
+        }
+        // Try StreamableHTTP first (modern MCP servers)
+        const streamableTransport = new StreamableHTTPClientTransport(url, {
+            requestInit,
+            authProvider,
+        });
+        try {
+            // Create a test client to verify the transport works
+            const testClient = new Client({ name: "pi-mcp-probe", version: "2.1.2" });
+            await testClient.connect(streamableTransport, this.buildRequestOptions(definition, signal));
+            await testClient.close().catch(() => { });
+            // Close probe transport before creating fresh one
+            await streamableTransport.close().catch(() => { });
+            // StreamableHTTP works - create fresh transport for actual use
+            return new StreamableHTTPClientTransport(url, { requestInit, authProvider });
+        }
+        catch (error) {
+            // StreamableHTTP failed, close and try SSE fallback
+            await streamableTransport.close().catch(() => { });
+            // Host cancellation is not transport capability evidence; do not fall
+            // through to SSE when the caller is trying to cancel the connect.
+            if (signal?.aborted) {
+                throwIfAborted(signal);
+            }
+            // If this was an UnauthorizedError, don't try SSE - the server needs auth
+            if (error instanceof UnauthorizedError) {
+                throw error;
+            }
+            // SSE is the legacy transport
+            return new SSEClientTransport(url, { requestInit, authProvider });
+        }
+    }
+    async fetchAllTools(client, requestOptions) {
+        const allTools = [];
+        let cursor;
+        do {
+            const result = await client.listTools(cursor ? { cursor } : undefined, requestOptions);
+            allTools.push(...(result.tools ?? []));
+            cursor = result.nextCursor;
+        } while (cursor);
+        return allTools;
+    }
+    async fetchAllResources(client, requestOptions) {
+        try {
+            const allResources = [];
+            let cursor;
+            do {
+                const result = await client.listResources(cursor ? { cursor } : undefined, requestOptions);
+                allResources.push(...(result.resources ?? []));
+                cursor = result.nextCursor;
+            } while (cursor);
+            return allResources;
+        }
+        catch {
+            if (requestOptions?.signal?.aborted) {
+                throwIfAborted(requestOptions.signal);
+            }
+            // Server may not support resources
+            return [];
+        }
+    }
+    attachAdapterNotificationHandlers(serverName, client) {
+        client.setNotificationHandler(serverStreamResultPatchNotificationSchema, (notification) => {
+            const listener = this.uiStreamListeners.get(notification.params.streamToken);
+            if (!listener)
+                return;
+            listener(serverName, notification.params);
+        });
+    }
+    registerUiStreamListener(streamToken, listener) {
+        this.uiStreamListeners.set(streamToken, listener);
+    }
+    removeUiStreamListener(streamToken) {
+        this.uiStreamListeners.delete(streamToken);
+    }
+    async readResource(name, uri, signal) {
+        const connection = this.connections.get(name);
+        if (!connection || connection.status !== "connected") {
+            throw new Error(`Server "${name}" is not connected`);
+        }
+        try {
+            this.touch(name);
+            this.incrementInFlight(name);
+            return await connection.client.readResource({ uri }, this.getRequestOptions(name, signal));
+        }
+        finally {
+            this.decrementInFlight(name);
+            this.touch(name);
+        }
+    }
+    async close(name) {
+        const connection = this.connections.get(name);
+        if (!connection)
+            return;
+        // Delete from map BEFORE async cleanup to prevent a race where a
+        // concurrent connect() creates a new connection that our deferred
+        // delete() would then remove, orphaning the new server process.
+        connection.status = "closed";
+        this.connections.delete(name);
+        this.acceptedUrlElicitations.delete(name);
+        await connection.client.close().catch(() => { });
+        await connection.transport.close().catch(() => { });
+    }
+    async closeAll() {
+        const names = [...this.connections.keys()];
+        await Promise.all(names.map(name => this.close(name)));
+    }
+    getConnection(name) {
+        return this.connections.get(name);
+    }
+    getAllConnections() {
+        return new Map(this.connections);
+    }
+    touch(name) {
+        const connection = this.connections.get(name);
+        if (connection) {
+            connection.lastUsedAt = Date.now();
+        }
+    }
+    incrementInFlight(name) {
+        const connection = this.connections.get(name);
+        if (connection) {
+            connection.inFlight = (connection.inFlight ?? 0) + 1;
+        }
+    }
+    decrementInFlight(name) {
+        const connection = this.connections.get(name);
+        if (connection && connection.inFlight) {
+            connection.inFlight--;
+        }
+    }
+    isIdle(name, timeoutMs) {
+        const connection = this.connections.get(name);
+        if (!connection || connection.status !== "connected")
+            return false;
+        if (connection.inFlight > 0)
+            return false;
+        return (Date.now() - connection.lastUsedAt) > timeoutMs;
+    }
+}
+/**
+ * Resolve environment variables with interpolation.
+ */
+function resolveEnv(env) {
+    // Copy process.env, filtering out undefined values
+    const resolved = {};
+    for (const [key, value] of Object.entries(process.env)) {
+        if (value !== undefined) {
+            resolved[key] = value;
+        }
+    }
+    if (!env)
+        return resolved;
+    const overrides = interpolateEnvRecord(env);
+    return overrides ? { ...resolved, ...overrides } : resolved;
+}
+/**
+ * Resolve headers with environment variable interpolation.
+ */
+function resolveHeaders(headers) {
+    return interpolateEnvRecord(headers);
+}
+function normalizeRequestTimeoutMs(timeoutMs) {
+    return typeof timeoutMs === "number" && Number.isFinite(timeoutMs) && timeoutMs > 0
+        ? timeoutMs
+        : undefined;
+}
+//# sourceMappingURL=server-manager.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"server-manager.js","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/server-manager.ts"],"names":[],"mappings":"AAAA,cAAc;AACd,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,MAAM,oDAAoD,CAAC;AACnG,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAC7E,OAAO,EAAE,iBAAiB,EAAE,MAAM,0CAA0C,CAAC;AAE7E,OAAO,EACL,qCAAqC,GAGtC,MAAM,oCAAoC,CAAC;AAQ5C,OAAO,EAAE,yCAAyC,EAAE,MAAM,YAAY,CAAC;AACvE,OAAO,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAE,uBAAuB,EAA6B,MAAM,uBAAuB,CAAC;AAC3F,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACzF,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAevD,MAAM,OAAO,gBAAgB;IAUE,UAAU;IAT/B,WAAW,GAAG,IAAI,GAAG,EAA4B,CAAC;IAClD,eAAe,GAAG,IAAI,GAAG,EAAqC,CAAC;IAC/D,iBAAiB,GAAG,IAAI,GAAG,EAA4B,CAAC;IACxD,cAAc,CAAmC;IACjD,iBAAiB,CAAsC;IACvD,uBAAuB,GAAG,IAAI,GAAG,EAAuB,CAAC;IACzD,uBAAuB,CAAqB;IAEpD,sEAAsE;IACtE,YAA6B,UAAmB,EAAE;0BAArB,UAAU;IAAY,CAAC;IAEpD,iBAAiB,CAAC,MAAwC,EAAQ;QAChE,IAAI,CAAC,cAAc,GAAG,MAAM,CAAC;IAAA,CAC9B;IAED,oBAAoB,CAAC,MAA2C,EAAQ;QACtE,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC;IAAA,CACjC;IAED,0BAA0B,CAAC,SAA6B,EAAQ;QAC9D,IAAI,CAAC,uBAAuB,GAAG,yBAAyB,CAAC,SAAS,CAAC,CAAC;IAAA,CACrE;IAED,iBAAiB,CAAC,IAAY,EAAE,MAAoB,EAA8B;QAChF,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,OAAO,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAAA,CACjE;IAEO,2BAA2B,CAAC,UAA6B,EAAsB;QACrF,IAAI,UAAU,EAAE,gBAAgB,KAAK,SAAS,EAAE,CAAC;YAC/C,OAAO,yBAAyB,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;QAChE,CAAC;QACD,OAAO,IAAI,CAAC,uBAAuB,CAAC;IAAA,CACrC;IAEO,mBAAmB,CACzB,UAA6B,EAC7B,MAAoB,EACQ;QAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,2BAA2B,CAAC,UAAU,CAAC,CAAC;QAE7D,IAAI,CAAC,MAAM,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACrC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7B,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC9C,CAAC;IAAA,CACH;IAED,KAAK,CAAC,OAAO,CAAC,IAAY,EAAE,UAA4B,EAAE,MAAoB,EAA6B;QACzG,cAAc,CAAC,MAAM,CAAC,CAAC;QACvB,wCAAwC;QACxC,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,OAAO,SAAS,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAE,EAAE,MAAM,CAAC,CAAC;QAC5D,CAAC;QAED,uCAAuC;QACvC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,QAAQ,EAAE,MAAM,KAAK,WAAW,EAAE,CAAC;YACrC,QAAQ,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACjC,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAChE,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QAExC,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC;YACjC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;YACvC,OAAO,UAAU,CAAC;QACpB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;IAAA,CACF;IAEO,KAAK,CAAC,gBAAgB,CAC5B,IAAY,EACZ,UAA4B,EAC5B,MAAoB,EACO;QAC3B,cAAc,CAAC,MAAM,CAAC,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC;QAEvC,IAAI,SAAoB,CAAC;QAEzB,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;YACvB,IAAI,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC;YACjC,IAAI,IAAI,GAAG,UAAU,CAAC,IAAI,IAAI,EAAE,CAAC;YAEjC,IAAI,OAAO,KAAK,KAAK,IAAI,OAAO,KAAK,KAAK,EAAE,CAAC;gBAC3C,MAAM,QAAQ,GAAG,MAAM,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;gBACvD,IAAI,QAAQ,EAAE,CAAC;oBACb,OAAO,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC;oBACpD,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,EAAE,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;oBACtF,MAAM,CAAC,KAAK,CAAC,GAAG,IAAI,gBAAgB,QAAQ,CAAC,OAAO,wBAAwB,CAAC,CAAC;gBAChF,CAAC;YACH,CAAC;YAED,SAAS,GAAG,IAAI,oBAAoB,CAAC;gBACnC,OAAO;gBACP,IAAI;gBACJ,GAAG,EAAE,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC;gBAC/B,GAAG,EAAE,iBAAiB,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,IAAI,CAAC,UAAU;gBACzD,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;aAChD,CAAC,CAAC;QACL,CAAC;aAAM,IAAI,UAAU,CAAC,GAAG,EAAE,CAAC;YAC1B,+BAA+B;YAC/B,SAAS,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACvE,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,UAAU,IAAI,wBAAwB,CAAC,CAAC;QAC1D,CAAC;QAED,MAAM,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAEpE,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YAChD,IAAI,CAAC,iCAAiC,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YAErD,+BAA+B;YAC/B,MAAM,CAAC,KAAK,EAAE,SAAS,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC3C,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,cAAc,CAAC;gBAC1C,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,cAAc,CAAC;aAC/C,CAAC,CAAC;YAEH,OAAO;gBACL,MAAM;gBACN,SAAS;gBACT,UAAU;gBACV,KAAK;gBACL,SAAS;gBACT,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;gBACtB,QAAQ,EAAE,CAAC;gBACX,MAAM,EAAE,WAAW;aACpB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,sDAAsD;YACtD,IAAI,KAAK,YAAY,iBAAiB,IAAI,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC;gBACpE,kEAAkE;gBAClE,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;gBACrC,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;gBAExC,OAAO;oBACL,MAAM;oBACN,SAAS;oBACT,UAAU;oBACV,KAAK,EAAE,EAAE;oBACT,SAAS,EAAE,EAAE;oBACb,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE;oBACtB,QAAQ,EAAE,CAAC;oBACX,MAAM,EAAE,YAAY;iBACrB,CAAC;YACJ,CAAC;YAED,kDAAkD;YAClD,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;YACrC,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;YACxC,MAAM,KAAK,CAAC;QACd,CAAC;IAAA,CACF;IAEO,uBAAuB,GAAG;QAChC,OAAO;YACL,GAAG,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAChD,GAAG,CAAC,IAAI,CAAC,iBAAiB;gBACxB,CAAC,CAAC;oBACE,WAAW,EAAE;wBACX,IAAI,EAAE,EAAE;wBACR,GAAG,CAAC,IAAI,CAAC,iBAAiB,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;qBACxD;iBACF;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;IAAA,CACH;IAEO,YAAY,CAAC,UAAkB,EAAU;QAC/C,MAAM,YAAY,GAAG,IAAI,CAAC,uBAAuB,EAAE,CAAC;QACpD,MAAM,MAAM,GAAG,IAAI,MAAM,CACvB,EAAE,IAAI,EAAE,UAAU,UAAU,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAClD,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CACpE,CAAC;QACF,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,uBAAuB,CAAC,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,cAAc,EAAE,UAAU,EAAE,CAAC,CAAC;QAC1E,CAAC;QACD,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,0BAA0B,CAAC,MAAM,EAAE;gBACjC,GAAG,IAAI,CAAC,iBAAiB;gBACzB,UAAU;gBACV,aAAa,EAAE,aAAa,CAAC,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,aAAa,CAAC;aACvF,CAAC,CAAC;YACH,IAAI,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;gBACpC,MAAM,CAAC,sBAAsB,CAAC,qCAAqC,EAAE,YAAY,CAAC,EAAE,CAAC;oBACnF,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;oBAC9D,IAAI,CAAC,QAAQ,EAAE,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,aAAa,CAAC;wBAAE,OAAO;oBACjE,IAAI,CAAC,iBAAiB,EAAE,EAAE,CAAC,MAAM,CAC/B,+BAA+B,UAAU,yCAAyC,EAClF,MAAM,CACP,CAAC;gBAAA,CACH,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,OAAO,MAAM,CAAC;IAAA,CACf;IAED,KAAK,CAAC,4BAA4B,CAChC,UAAkB,EAClB,KAAkC,EACQ;QAC1C,IAAI,CAAC,IAAI,CAAC,iBAAiB,EAAE,QAAQ;YAAE,OAAO,QAAQ,CAAC;QACvD,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YACxC,MAAM,MAAM,GAAG,MAAM,oBAAoB,CAAC;gBACxC,GAAG,IAAI,CAAC,iBAAiB;gBACzB,UAAU;gBACV,aAAa,EAAE,aAAa,CAAC,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,UAAU,EAAE,aAAa,CAAC;aACvF,EAAE,MAAM,CAAC,CAAC;YACX,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ;gBAAE,OAAO,MAAM,CAAC,MAAM,CAAC;QACvD,CAAC;QACD,OAAO,QAAQ,CAAC;IAAA,CACjB;IAEO,sBAAsB,CAAC,UAAkB,EAAE,aAAqB,EAAQ;QAC9E,IAAI,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5D,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,QAAQ,GAAG,IAAI,GAAG,EAAE,CAAC;YACrB,IAAI,CAAC,uBAAuB,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;QACzD,CAAC;QACD,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAAA,CAC7B;IAEO,KAAK,CAAC,mBAAmB,CAC/B,UAA4B,EAC5B,UAAkB,EAClB,MAAoB,EACA;QACpB,cAAc,CAAC,MAAM,CAAC,CAAC;QACvB,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,GAAI,CAAC,CAAC;QAErC,mDAAmD;QACnD,MAAM,OAAO,GAAG,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAEzD,wEAAwE;QACxE,IAAI,UAAU,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACjC,MAAM,KAAK,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAC7C,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,KAAK,EAAE,CAAC;YAC/C,CAAC;QACH,CAAC;QAED,gFAAgF;QAChF,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;QAE9E,6CAA6C;QAC7C,IAAI,YAA0C,CAAC;QAC/C,IAAI,aAAa,CAAC,UAAU,CAAC,EAAE,CAAC;YAC9B,MAAM,WAAW,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YACnD,YAAY,GAAG,IAAI,gBAAgB,CACjC,UAAU,EACV,UAAU,CAAC,GAAI,EACf,WAAW,EACX;gBACE,UAAU,EAAE,KAAK,EAAE,QAAQ,EAAE,EAAE,CAAC;oBAC9B,+CAA+C;gBADhB,CAEhC;aACF,CACF,CAAC;QACJ,CAAC;QAED,gDAAgD;QAChD,MAAM,mBAAmB,GAAG,IAAI,6BAA6B,CAAC,GAAG,EAAE;YACjE,WAAW;YACX,YAAY;SACb,CAAC,CAAC;QAEH,IAAI,CAAC;YACH,qDAAqD;YACrD,MAAM,UAAU,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YAC1E,MAAM,UAAU,CAAC,OAAO,CAAC,mBAAmB,EAAE,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC;YAC5F,MAAM,UAAU,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;YACzC,kDAAkD;YAClD,MAAM,mBAAmB,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;YAElD,+DAA+D;YAC/D,OAAO,IAAI,6BAA6B,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC;QAC/E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,oDAAoD;YACpD,MAAM,mBAAmB,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;YAElD,sEAAsE;YACtE,kEAAkE;YAClE,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,cAAc,CAAC,MAAM,CAAC,CAAC;YACzB,CAAC;YAED,0EAA0E;YAC1E,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;gBACvC,MAAM,KAAK,CAAC;YACd,CAAC;YAED,8BAA8B;YAC9B,OAAO,IAAI,kBAAkB,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC;QACpE,CAAC;IAAA,CACF;IAEO,KAAK,CAAC,aAAa,CAAC,MAAc,EAAE,cAA+B,EAAsB;QAC/F,MAAM,QAAQ,GAAc,EAAE,CAAC;QAC/B,IAAI,MAA0B,CAAC;QAE/B,GAAG,CAAC;YACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;YACvF,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC;YACvC,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC;QAC7B,CAAC,QAAQ,MAAM,EAAE;QAEjB,OAAO,QAAQ,CAAC;IAAA,CACjB;IAEO,KAAK,CAAC,iBAAiB,CAAC,MAAc,EAAE,cAA+B,EAA0B;QACvG,IAAI,CAAC;YACH,MAAM,YAAY,GAAkB,EAAE,CAAC;YACvC,IAAI,MAA0B,CAAC;YAE/B,GAAG,CAAC;gBACF,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,cAAc,CAAC,CAAC;gBAC3F,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC/C,MAAM,GAAG,MAAM,CAAC,UAAU,CAAC;YAC7B,CAAC,QAAQ,MAAM,EAAE;YAEjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACP,IAAI,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpC,cAAc,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YACxC,CAAC;YACD,mCAAmC;YACnC,OAAO,EAAE,CAAC;QACZ,CAAC;IAAA,CACF;IAEO,iCAAiC,CAAC,UAAkB,EAAE,MAAc,EAAQ;QAClF,MAAM,CAAC,sBAAsB,CAAC,yCAAyC,EAAE,CAAC,YAAY,EAAE,EAAE,CAAC;YACzF,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;YAC7E,IAAI,CAAC,QAAQ;gBAAE,OAAO;YACtB,QAAQ,CAAC,UAAU,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;QAAA,CAC3C,CAAC,CAAC;IAAA,CACJ;IAED,wBAAwB,CAAC,WAAmB,EAAE,QAA0B,EAAQ;QAC9E,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;IAAA,CACnD;IAED,sBAAsB,CAAC,WAAmB,EAAQ;QAChD,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAAA,CAC5C;IAED,KAAK,CAAC,YAAY,CAAC,IAAY,EAAE,GAAW,EAAE,MAAoB,EAA+B;QAC/F,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,WAAW,IAAI,oBAAoB,CAAC,CAAC;QACvD,CAAC;QAED,IAAI,CAAC;YACH,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACjB,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC7B,OAAO,MAAM,UAAU,CAAC,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;QAC7F,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;YAC7B,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACnB,CAAC;IAAA,CACF;IAED,KAAK,CAAC,KAAK,CAAC,IAAY,EAAiB;QACvC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,UAAU;YAAE,OAAO;QAExB,iEAAiE;QACjE,kEAAkE;QAClE,gEAAgE;QAChE,UAAU,CAAC,MAAM,GAAG,QAAQ,CAAC;QAC7B,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,uBAAuB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;QAChD,MAAM,UAAU,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC,CAAC;IAAA,CACpD;IAED,KAAK,CAAC,QAAQ,GAAkB;QAC9B,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;QAC3C,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAAA,CACxD;IAED,aAAa,CAAC,IAAY,EAAgC;QACxD,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAAA,CACnC;IAED,iBAAiB,GAAkC;QACjD,OAAO,IAAI,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAAA,CAClC;IAED,KAAK,CAAC,IAAY,EAAQ;QACxB,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACrC,CAAC;IAAA,CACF;IAED,iBAAiB,CAAC,IAAY,EAAQ;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,UAAU,EAAE,CAAC;YACf,UAAU,CAAC,QAAQ,GAAG,CAAC,UAAU,CAAC,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACvD,CAAC;IAAA,CACF;IAED,iBAAiB,CAAC,IAAY,EAAQ;QACpC,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,UAAU,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;YACtC,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,CAAC;IAAA,CACF;IAED,MAAM,CAAC,IAAY,EAAE,SAAiB,EAAW;QAC/C,MAAM,UAAU,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW;YAAE,OAAO,KAAK,CAAC;QACnE,IAAI,UAAU,CAAC,QAAQ,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QAC1C,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,UAAU,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;IAAA,CACzD;CACF;AAED;;GAEG;AACH,SAAS,UAAU,CAAC,GAA4B,EAA0B;IACxE,mDAAmD;IACnD,MAAM,QAAQ,GAA2B,EAAE,CAAC;IAC5C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACvD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,QAAQ,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;QACxB,CAAC;IACH,CAAC;IAED,IAAI,CAAC,GAAG;QAAE,OAAO,QAAQ,CAAC;IAE1B,MAAM,SAAS,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAC5C,OAAO,SAAS,CAAC,CAAC,CAAC,EAAE,GAAG,QAAQ,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;AAAA,CAC7D;AAED;;GAEG;AACH,SAAS,cAAc,CAAC,OAAgC,EAAsC;IAC5F,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;AAAA,CACtC;AAED,SAAS,yBAAyB,CAAC,SAA6B,EAAsB;IACpF,OAAO,OAAO,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,GAAG,CAAC;QACjF,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,SAAS,CAAC;AAAA,CACf","sourcesContent":["// @ts-nocheck\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { UnauthorizedError } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type { RequestOptions } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\nimport {\n  ElicitationCompleteNotificationSchema,\n  type ReadResourceResult,\n  type UrlElicitationRequiredError,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type {\n  McpTool,\n  McpResource,\n  ServerDefinition,\n  ServerStreamResultPatchNotification,\n  Transport,\n} from \"./types.ts\";\nimport { serverStreamResultPatchNotificationSchema } from \"./types.ts\";\nimport { resolveNpxBinary } from \"./npx-resolver.ts\";\nimport { logger } from \"./logger.ts\";\nimport { McpOAuthProvider } from \"./mcp-oauth-provider.ts\";\nimport { extractOAuthConfig, supportsOAuth } from \"./mcp-auth-flow.ts\";\nimport { registerSamplingHandler, type ServerSamplingConfig } from \"./sampling-handler.ts\";\nimport {\n  handleUrlElicitation,\n  registerElicitationHandler,\n  type ServerElicitationConfig,\n} from \"./elicitation-handler.ts\";\nimport { interpolateEnvRecord, resolveBearerToken, resolveConfigPath } from \"./utils.ts\";\nimport { abortable, throwIfAborted } from \"./abort.ts\";\n\ninterface ServerConnection {\n  client: Client;\n  transport: Transport;\n  definition: ServerDefinition;\n  tools: McpTool[];\n  resources: McpResource[];\n  lastUsedAt: number;\n  inFlight: number;\n  status: \"connected\" | \"closed\" | \"needs-auth\";\n}\n\ntype UiStreamListener = (serverName: string, notification: ServerStreamResultPatchNotification[\"params\"]) => void;\n\nexport class McpServerManager {\n  private connections = new Map<string, ServerConnection>();\n  private connectPromises = new Map<string, Promise<ServerConnection>>();\n  private uiStreamListeners = new Map<string, UiStreamListener>();\n  private samplingConfig: ServerSamplingConfig | undefined;\n  private elicitationConfig: ServerElicitationConfig | undefined;\n  private acceptedUrlElicitations = new Map<string, Set<string>>();\n  private defaultRequestTimeoutMs: number | undefined;\n\n  /** Default cwd for stdio servers without an explicit config `cwd`. */\n  constructor(private readonly defaultCwd?: string) {}\n\n  setSamplingConfig(config: ServerSamplingConfig | undefined): void {\n    this.samplingConfig = config;\n  }\n\n  setElicitationConfig(config: ServerElicitationConfig | undefined): void {\n    this.elicitationConfig = config;\n  }\n\n  setDefaultRequestTimeoutMs(timeoutMs: number | undefined): void {\n    this.defaultRequestTimeoutMs = normalizeRequestTimeoutMs(timeoutMs);\n  }\n\n  getRequestOptions(name: string, signal?: AbortSignal): RequestOptions | undefined {\n    const connection = this.connections.get(name);\n    return this.buildRequestOptions(connection?.definition, signal);\n  }\n\n  private getResolvedRequestTimeoutMs(definition?: ServerDefinition): number | undefined {\n    if (definition?.requestTimeoutMs !== undefined) {\n      return normalizeRequestTimeoutMs(definition.requestTimeoutMs);\n    }\n    return this.defaultRequestTimeoutMs;\n  }\n\n  private buildRequestOptions(\n    definition?: ServerDefinition,\n    signal?: AbortSignal,\n  ): RequestOptions | undefined {\n    const timeout = this.getResolvedRequestTimeoutMs(definition);\n\n    if (!signal && timeout === undefined) {\n      return undefined;\n    }\n\n    return {\n      ...(signal ? { signal } : {}),\n      ...(timeout !== undefined ? { timeout } : {}),\n    };\n  }\n\n  async connect(name: string, definition: ServerDefinition, signal?: AbortSignal): Promise<ServerConnection> {\n    throwIfAborted(signal);\n    // Dedupe concurrent connection attempts\n    if (this.connectPromises.has(name)) {\n      return abortable(this.connectPromises.get(name)!, signal);\n    }\n\n    // Reuse existing connection if healthy\n    const existing = this.connections.get(name);\n    if (existing?.status === \"connected\") {\n      existing.lastUsedAt = Date.now();\n      return existing;\n    }\n\n    const promise = this.createConnection(name, definition, signal);\n    this.connectPromises.set(name, promise);\n\n    try {\n      const connection = await promise;\n      this.connections.set(name, connection);\n      return connection;\n    } finally {\n      this.connectPromises.delete(name);\n    }\n  }\n\n  private async createConnection(\n    name: string,\n    definition: ServerDefinition,\n    signal?: AbortSignal,\n  ): Promise<ServerConnection> {\n    throwIfAborted(signal);\n    const client = this.createClient(name);\n\n    let transport: Transport;\n\n    if (definition.command) {\n      let command = definition.command;\n      let args = definition.args ?? [];\n\n      if (command === \"npx\" || command === \"npm\") {\n        const resolved = await resolveNpxBinary(command, args);\n        if (resolved) {\n          command = resolved.isJs ? \"node\" : resolved.binPath;\n          args = resolved.isJs ? [resolved.binPath, ...resolved.extraArgs] : resolved.extraArgs;\n          logger.debug(`${name} resolved to ${resolved.binPath} (skipping npm parent)`);\n        }\n      }\n\n      transport = new StdioClientTransport({\n        command,\n        args,\n        env: resolveEnv(definition.env),\n        cwd: resolveConfigPath(definition.cwd) ?? this.defaultCwd,\n        stderr: definition.debug ? \"inherit\" : \"ignore\",\n      });\n    } else if (definition.url) {\n      // HTTP transport with fallback\n      transport = await this.createHttpTransport(definition, name, signal);\n    } else {\n      throw new Error(`Server ${name} has no command or url`);\n    }\n\n    const requestOptions = this.buildRequestOptions(definition, signal);\n\n    try {\n      await client.connect(transport, requestOptions);\n      this.attachAdapterNotificationHandlers(name, client);\n\n      // Discover tools and resources\n      const [tools, resources] = await Promise.all([\n        this.fetchAllTools(client, requestOptions),\n        this.fetchAllResources(client, requestOptions),\n      ]);\n\n      return {\n        client,\n        transport,\n        definition,\n        tools,\n        resources,\n        lastUsedAt: Date.now(),\n        inFlight: 0,\n        status: \"connected\",\n      };\n    } catch (error) {\n      // Check for UnauthorizedError - server requires OAuth\n      if (error instanceof UnauthorizedError && supportsOAuth(definition)) {\n        // Clean up both client and transport before reporting needs-auth.\n        await client.close().catch(() => {});\n        await transport.close().catch(() => {});\n\n        return {\n          client,\n          transport,\n          definition,\n          tools: [],\n          resources: [],\n          lastUsedAt: Date.now(),\n          inFlight: 0,\n          status: \"needs-auth\",\n        };\n      }\n\n      // Clean up both client and transport on any error\n      await client.close().catch(() => {});\n      await transport.close().catch(() => {});\n      throw error;\n    }\n  }\n\n  private buildClientCapabilities() {\n    return {\n      ...(this.samplingConfig ? { sampling: {} } : {}),\n      ...(this.elicitationConfig\n        ? {\n            elicitation: {\n              form: {},\n              ...(this.elicitationConfig.allowUrl ? { url: {} } : {}),\n            },\n          }\n        : {}),\n    };\n  }\n\n  private createClient(serverName: string): Client {\n    const capabilities = this.buildClientCapabilities();\n    const client = new Client(\n      { name: `pi-mcp-${serverName}`, version: \"1.0.0\" },\n      Object.keys(capabilities).length > 0 ? { capabilities } : undefined,\n    );\n    if (this.samplingConfig) {\n      registerSamplingHandler(client, { ...this.samplingConfig, serverName });\n    }\n    if (this.elicitationConfig) {\n      registerElicitationHandler(client, {\n        ...this.elicitationConfig,\n        serverName,\n        onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),\n      });\n      if (this.elicitationConfig.allowUrl) {\n        client.setNotificationHandler(ElicitationCompleteNotificationSchema, notification => {\n          const accepted = this.acceptedUrlElicitations.get(serverName);\n          if (!accepted?.delete(notification.params.elicitationId)) return;\n          this.elicitationConfig?.ui.notify(\n            `MCP browser interaction for ${serverName} completed. You can retry the tool now.`,\n            \"info\",\n          );\n        });\n      }\n    }\n    return client;\n  }\n\n  async handleUrlElicitationRequired(\n    serverName: string,\n    error: UrlElicitationRequiredError,\n  ): Promise<\"accept\" | \"decline\" | \"cancel\"> {\n    if (!this.elicitationConfig?.allowUrl) return \"cancel\";\n    for (const params of error.elicitations) {\n      const result = await handleUrlElicitation({\n        ...this.elicitationConfig,\n        serverName,\n        onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),\n      }, params);\n      if (result.action !== \"accept\") return result.action;\n    }\n    return \"accept\";\n  }\n\n  private rememberUrlElicitation(serverName: string, elicitationId: string): void {\n    let accepted = this.acceptedUrlElicitations.get(serverName);\n    if (!accepted) {\n      accepted = new Set();\n      this.acceptedUrlElicitations.set(serverName, accepted);\n    }\n    accepted.add(elicitationId);\n  }\n\n  private async createHttpTransport(\n    definition: ServerDefinition,\n    serverName: string,\n    signal?: AbortSignal,\n  ): Promise<Transport> {\n    throwIfAborted(signal);\n    const url = new URL(definition.url!);\n\n    // Build headers first (including any bearer token)\n    const headers = resolveHeaders(definition.headers) ?? {};\n\n    // For bearer auth, add the token to headers BEFORE creating requestInit\n    if (definition.auth === \"bearer\") {\n      const token = resolveBearerToken(definition);\n      if (token) {\n        headers[\"Authorization\"] = `Bearer ${token}`;\n      }\n    }\n\n    // Create request init with headers (Authorization now included for bearer auth)\n    const requestInit = Object.keys(headers).length > 0 ? { headers } : undefined;\n\n    // For OAuth servers, create an auth provider\n    let authProvider: McpOAuthProvider | undefined;\n    if (supportsOAuth(definition)) {\n      const oauthConfig = extractOAuthConfig(definition);\n      authProvider = new McpOAuthProvider(\n        serverName,\n        definition.url!,\n        oauthConfig,\n        {\n          onRedirect: async (_authUrl) => {\n            // URL is captured by startAuth, no need to log\n          },\n        }\n      );\n    }\n\n    // Try StreamableHTTP first (modern MCP servers)\n    const streamableTransport = new StreamableHTTPClientTransport(url, {\n      requestInit,\n      authProvider,\n    });\n\n    try {\n      // Create a test client to verify the transport works\n      const testClient = new Client({ name: \"pi-mcp-probe\", version: \"2.1.2\" });\n      await testClient.connect(streamableTransport, this.buildRequestOptions(definition, signal));\n      await testClient.close().catch(() => {});\n      // Close probe transport before creating fresh one\n      await streamableTransport.close().catch(() => {});\n\n      // StreamableHTTP works - create fresh transport for actual use\n      return new StreamableHTTPClientTransport(url, { requestInit, authProvider });\n    } catch (error) {\n      // StreamableHTTP failed, close and try SSE fallback\n      await streamableTransport.close().catch(() => {});\n\n      // Host cancellation is not transport capability evidence; do not fall\n      // through to SSE when the caller is trying to cancel the connect.\n      if (signal?.aborted) {\n        throwIfAborted(signal);\n      }\n\n      // If this was an UnauthorizedError, don't try SSE - the server needs auth\n      if (error instanceof UnauthorizedError) {\n        throw error;\n      }\n\n      // SSE is the legacy transport\n      return new SSEClientTransport(url, { requestInit, authProvider });\n    }\n  }\n\n  private async fetchAllTools(client: Client, requestOptions?: RequestOptions): Promise<McpTool[]> {\n    const allTools: McpTool[] = [];\n    let cursor: string | undefined;\n\n    do {\n      const result = await client.listTools(cursor ? { cursor } : undefined, requestOptions);\n      allTools.push(...(result.tools ?? []));\n      cursor = result.nextCursor;\n    } while (cursor);\n\n    return allTools;\n  }\n\n  private async fetchAllResources(client: Client, requestOptions?: RequestOptions): Promise<McpResource[]> {\n    try {\n      const allResources: McpResource[] = [];\n      let cursor: string | undefined;\n\n      do {\n        const result = await client.listResources(cursor ? { cursor } : undefined, requestOptions);\n        allResources.push(...(result.resources ?? []));\n        cursor = result.nextCursor;\n      } while (cursor);\n\n      return allResources;\n    } catch {\n      if (requestOptions?.signal?.aborted) {\n        throwIfAborted(requestOptions.signal);\n      }\n      // Server may not support resources\n      return [];\n    }\n  }\n\n  private attachAdapterNotificationHandlers(serverName: string, client: Client): void {\n    client.setNotificationHandler(serverStreamResultPatchNotificationSchema, (notification) => {\n      const listener = this.uiStreamListeners.get(notification.params.streamToken);\n      if (!listener) return;\n      listener(serverName, notification.params);\n    });\n  }\n\n  registerUiStreamListener(streamToken: string, listener: UiStreamListener): void {\n    this.uiStreamListeners.set(streamToken, listener);\n  }\n\n  removeUiStreamListener(streamToken: string): void {\n    this.uiStreamListeners.delete(streamToken);\n  }\n\n  async readResource(name: string, uri: string, signal?: AbortSignal): Promise<ReadResourceResult> {\n    const connection = this.connections.get(name);\n    if (!connection || connection.status !== \"connected\") {\n      throw new Error(`Server \"${name}\" is not connected`);\n    }\n\n    try {\n      this.touch(name);\n      this.incrementInFlight(name);\n      return await connection.client.readResource({ uri }, this.getRequestOptions(name, signal));\n    } finally {\n      this.decrementInFlight(name);\n      this.touch(name);\n    }\n  }\n\n  async close(name: string): Promise<void> {\n    const connection = this.connections.get(name);\n    if (!connection) return;\n\n    // Delete from map BEFORE async cleanup to prevent a race where a\n    // concurrent connect() creates a new connection that our deferred\n    // delete() would then remove, orphaning the new server process.\n    connection.status = \"closed\";\n    this.connections.delete(name);\n    this.acceptedUrlElicitations.delete(name);\n    await connection.client.close().catch(() => {});\n    await connection.transport.close().catch(() => {});\n  }\n\n  async closeAll(): Promise<void> {\n    const names = [...this.connections.keys()];\n    await Promise.all(names.map(name => this.close(name)));\n  }\n\n  getConnection(name: string): ServerConnection | undefined {\n    return this.connections.get(name);\n  }\n\n  getAllConnections(): Map<string, ServerConnection> {\n    return new Map(this.connections);\n  }\n\n  touch(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection) {\n      connection.lastUsedAt = Date.now();\n    }\n  }\n\n  incrementInFlight(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection) {\n      connection.inFlight = (connection.inFlight ?? 0) + 1;\n    }\n  }\n\n  decrementInFlight(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection && connection.inFlight) {\n      connection.inFlight--;\n    }\n  }\n\n  isIdle(name: string, timeoutMs: number): boolean {\n    const connection = this.connections.get(name);\n    if (!connection || connection.status !== \"connected\") return false;\n    if (connection.inFlight > 0) return false;\n    return (Date.now() - connection.lastUsedAt) > timeoutMs;\n  }\n}\n\n/**\n * Resolve environment variables with interpolation.\n */\nfunction resolveEnv(env?: Record<string, string>): Record<string, string> {\n  // Copy process.env, filtering out undefined values\n  const resolved: Record<string, string> = {};\n  for (const [key, value] of Object.entries(process.env)) {\n    if (value !== undefined) {\n      resolved[key] = value;\n    }\n  }\n\n  if (!env) return resolved;\n\n  const overrides = interpolateEnvRecord(env);\n  return overrides ? { ...resolved, ...overrides } : resolved;\n}\n\n/**\n * Resolve headers with environment variable interpolation.\n */\nfunction resolveHeaders(headers?: Record<string, string>): Record<string, string> | undefined {\n  return interpolateEnvRecord(headers);\n}\n\nfunction normalizeRequestTimeoutMs(timeoutMs: number | undefined): number | undefined {\n  return typeof timeoutMs === \"number\" && Number.isFinite(timeoutMs) && timeoutMs > 0\n    ? timeoutMs\n    : undefined;\n}\n"]}

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

@@ -0,0 +1,41 @@
+import type { ExtensionContext } from "@earendil-works/pi-coding-agent";
+import type { ConsentManager } from "./consent-manager.ts";
+import type { McpLifecycleManager } from "./lifecycle.ts";
+import type { McpServerManager } from "./server-manager.ts";
+import type { ToolMetadata, McpConfig, UiSessionMessages, UiStreamSummary } from "./types.ts";
+import type { UiResourceHandler } from "./ui-resource-handler.ts";
+import type { UiServerHandle } from "./ui-server.ts";
+export interface CompletedUiSession {
+    serverName: string;
+    toolName: string;
+    completedAt: Date;
+    reason: string;
+    messages: UiSessionMessages;
+    stream?: UiStreamSummary;
+}
+export type SendMessageFn = (message: {
+    customType: string;
+    content: Array<{
+        type: "text";
+        text: string;
+    }>;
+    display?: string;
+    details?: unknown;
+}, options?: {
+    triggerTurn?: boolean;
+}) => void;
+export interface McpExtensionState {
+    manager: McpServerManager;
+    lifecycle: McpLifecycleManager;
+    toolMetadata: Map<string, ToolMetadata[]>;
+    config: McpConfig;
+    failureTracker: Map<string, number>;
+    uiResourceHandler: UiResourceHandler;
+    consentManager: ConsentManager;
+    uiServer: UiServerHandle | null;
+    completedUiSessions: CompletedUiSession[];
+    openBrowser: (url: string) => Promise<void>;
+    ui?: ExtensionContext["ui"];
+    sendMessage?: SendMessageFn;
+}
+//# sourceMappingURL=state.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"state.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/state.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AACxE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAC5D,OAAO,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,iBAAiB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAC9F,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AAErD,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,EAAE,IAAI,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,iBAAiB,CAAC;IAC5B,MAAM,CAAC,EAAE,eAAe,CAAC;CAC1B;AAED,MAAM,MAAM,aAAa,GAAG,CAC1B,OAAO,EAAE;IACP,UAAU,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,CAAC;IAC/C,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,EACD,OAAO,CAAC,EAAE;IAAE,WAAW,CAAC,EAAE,OAAO,CAAA;CAAE,KAChC,IAAI,CAAC;AAEV,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,gBAAgB,CAAC;IAC1B,SAAS,EAAE,mBAAmB,CAAC;IAC/B,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,YAAY,EAAE,CAAC,CAAC;IAC1C,MAAM,EAAE,SAAS,CAAC;IAClB,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpC,iBAAiB,EAAE,iBAAiB,CAAC;IACrC,cAAc,EAAE,cAAc,CAAC;IAC/B,QAAQ,EAAE,cAAc,GAAG,IAAI,CAAC;IAChC,mBAAmB,EAAE,kBAAkB,EAAE,CAAC;IAC1C,WAAW,EAAE,CAAC,GAAG,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAC5C,EAAE,CAAC,EAAE,gBAAgB,CAAC,IAAI,CAAC,CAAC;IAC5B,WAAW,CAAC,EAAE,aAAa,CAAC;CAC7B","sourcesContent":["// @ts-nocheck\nimport type { ExtensionContext } from \"@earendil-works/pi-coding-agent\";\nimport type { ConsentManager } from \"./consent-manager.ts\";\nimport type { McpLifecycleManager } from \"./lifecycle.ts\";\nimport type { McpServerManager } from \"./server-manager.ts\";\nimport type { ToolMetadata, McpConfig, UiSessionMessages, UiStreamSummary } from \"./types.ts\";\nimport type { UiResourceHandler } from \"./ui-resource-handler.ts\";\nimport type { UiServerHandle } from \"./ui-server.ts\";\n\nexport interface CompletedUiSession {\n  serverName: string;\n  toolName: string;\n  completedAt: Date;\n  reason: string;\n  messages: UiSessionMessages;\n  stream?: UiStreamSummary;\n}\n\nexport type SendMessageFn = (\n  message: {\n    customType: string;\n    content: Array<{ type: \"text\"; text: string }>;\n    display?: string;\n    details?: unknown;\n  },\n  options?: { triggerTurn?: boolean }\n) => void;\n\nexport interface McpExtensionState {\n  manager: McpServerManager;\n  lifecycle: McpLifecycleManager;\n  toolMetadata: Map<string, ToolMetadata[]>;\n  config: McpConfig;\n  failureTracker: Map<string, number>;\n  uiResourceHandler: UiResourceHandler;\n  consentManager: ConsentManager;\n  uiServer: UiServerHandle | null;\n  completedUiSessions: CompletedUiSession[];\n  openBrowser: (url: string) => Promise<void>;\n  ui?: ExtensionContext[\"ui\"];\n  sendMessage?: SendMessageFn;\n}\n"]}

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

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=state.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"state.js","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/state.ts"],"names":[],"mappings":"","sourcesContent":["// @ts-nocheck\nimport type { ExtensionContext } from \"@earendil-works/pi-coding-agent\";\nimport type { ConsentManager } from \"./consent-manager.ts\";\nimport type { McpLifecycleManager } from \"./lifecycle.ts\";\nimport type { McpServerManager } from \"./server-manager.ts\";\nimport type { ToolMetadata, McpConfig, UiSessionMessages, UiStreamSummary } from \"./types.ts\";\nimport type { UiResourceHandler } from \"./ui-resource-handler.ts\";\nimport type { UiServerHandle } from \"./ui-server.ts\";\n\nexport interface CompletedUiSession {\n  serverName: string;\n  toolName: string;\n  completedAt: Date;\n  reason: string;\n  messages: UiSessionMessages;\n  stream?: UiStreamSummary;\n}\n\nexport type SendMessageFn = (\n  message: {\n    customType: string;\n    content: Array<{ type: \"text\"; text: string }>;\n    display?: string;\n    details?: unknown;\n  },\n  options?: { triggerTurn?: boolean }\n) => void;\n\nexport interface McpExtensionState {\n  manager: McpServerManager;\n  lifecycle: McpLifecycleManager;\n  toolMetadata: Map<string, ToolMetadata[]>;\n  config: McpConfig;\n  failureTracker: Map<string, number>;\n  uiResourceHandler: UiResourceHandler;\n  consentManager: ConsentManager;\n  uiServer: UiServerHandle | null;\n  completedUiSessions: CompletedUiSession[];\n  openBrowser: (url: string) => Promise<void>;\n  ui?: ExtensionContext[\"ui\"];\n  sendMessage?: SendMessageFn;\n}\n"]}

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

@@ -0,0 +1,11 @@
+import type { McpExtensionState } from "./state.ts";
+import type { ToolMetadata, McpTool, McpResource, ServerEntry } from "./types.ts";
+export declare function buildToolMetadata(tools: McpTool[], resources: McpResource[], definition: ServerEntry, serverName: string, prefix: "server" | "none" | "short"): {
+    metadata: ToolMetadata[];
+    failedTools: string[];
+};
+export declare function getToolNames(state: McpExtensionState, serverName: string): string[];
+export declare function totalToolCount(state: McpExtensionState): number;
+export declare function findToolByName(metadata: ToolMetadata[] | undefined, toolName: string): ToolMetadata | undefined;
+export declare function formatSchema(schema: unknown, indent?: string): string;
+//# sourceMappingURL=tool-metadata.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tool-metadata.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/tool-metadata.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAKlF,wBAAgB,iBAAiB,CAC/B,KAAK,EAAE,OAAO,EAAE,EAChB,SAAS,EAAE,WAAW,EAAE,EACxB,UAAU,EAAE,WAAW,EACvB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,QAAQ,GAAG,MAAM,GAAG,OAAO,GAClC;IAAE,QAAQ,EAAE,YAAY,EAAE,CAAC;IAAC,WAAW,EAAE,MAAM,EAAE,CAAA;CAAE,CA8CrD;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CAEnF;AAED,wBAAgB,cAAc,CAAC,KAAK,EAAE,iBAAiB,GAAG,MAAM,CAM/D;AAED,wBAAgB,cAAc,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,YAAY,GAAG,SAAS,CAM/G;AAED,wBAAgB,YAAY,CAAC,MAAM,EAAE,OAAO,EAAE,MAAM,SAAO,GAAG,MAAM,CAiCnE","sourcesContent":["// @ts-nocheck\nimport { getToolUiResourceUri } from \"@modelcontextprotocol/ext-apps/app-bridge\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport type { ToolMetadata, McpTool, McpResource, ServerEntry } from \"./types.ts\";\nimport { formatToolName, isToolExcluded } from \"./types.ts\";\nimport { resourceNameToToolName } from \"./resource-tools.ts\";\nimport { extractToolUiStreamMode } from \"./utils.ts\";\n\nexport function buildToolMetadata(\n  tools: McpTool[],\n  resources: McpResource[],\n  definition: ServerEntry,\n  serverName: string,\n  prefix: \"server\" | \"none\" | \"short\"\n): { metadata: ToolMetadata[]; failedTools: string[] } {\n  const metadata: ToolMetadata[] = [];\n  const failedTools: string[] = [];\n\n  for (const tool of tools) {\n    if (!tool?.name) {\n      failedTools.push(\"(unnamed)\");\n      continue;\n    }\n    if (isToolExcluded(tool.name, serverName, prefix, definition.excludeTools)) {\n      continue;\n    }\n\n    let uiResourceUri: string | undefined;\n    try {\n      uiResourceUri = getToolUiResourceUri({ _meta: tool._meta });\n    } catch {\n      failedTools.push(tool.name);\n    }\n    metadata.push({\n      name: formatToolName(tool.name, serverName, prefix),\n      originalName: tool.name,\n      description: tool.description ?? \"\",\n      inputSchema: tool.inputSchema,\n      uiResourceUri,\n      uiStreamMode: extractToolUiStreamMode(tool._meta),\n    });\n  }\n\n  if (definition.exposeResources !== false) {\n    for (const resource of resources) {\n      const baseName = `get_${resourceNameToToolName(resource.name)}`;\n      if (isToolExcluded(baseName, serverName, prefix, definition.excludeTools)) {\n        continue;\n      }\n\n      metadata.push({\n        name: formatToolName(baseName, serverName, prefix),\n        originalName: baseName,\n        description: resource.description ?? `Read resource: ${resource.uri}`,\n        resourceUri: resource.uri,\n      });\n    }\n  }\n\n  return { metadata, failedTools };\n}\n\nexport function getToolNames(state: McpExtensionState, serverName: string): string[] {\n  return state.toolMetadata.get(serverName)?.map(m => m.name) ?? [];\n}\n\nexport function totalToolCount(state: McpExtensionState): number {\n  let count = 0;\n  for (const metadata of state.toolMetadata.values()) {\n    count += metadata.length;\n  }\n  return count;\n}\n\nexport function findToolByName(metadata: ToolMetadata[] | undefined, toolName: string): ToolMetadata | undefined {\n  if (!metadata) return undefined;\n  const exact = metadata.find(m => m.name === toolName);\n  if (exact) return exact;\n  const normalized = toolName.replace(/-/g, \"_\");\n  return metadata.find(m => m.name.replace(/-/g, \"_\") === normalized);\n}\n\nexport function formatSchema(schema: unknown, indent = \"  \"): string {\n  if (!schema || typeof schema !== \"object\" || Array.isArray(schema)) {\n    return `${indent}(no schema)`;\n  }\n\n  const s = schema as Record<string, unknown>;\n\n  if (s.type === \"object\" && s.properties && typeof s.properties === \"object\" && !Array.isArray(s.properties)) {\n    const props = s.properties as Record<string, unknown>;\n    const required = Array.isArray(s.required) ? s.required.filter((name): name is string => typeof name === \"string\") : [];\n\n    if (Object.keys(props).length === 0) {\n      return `${indent}(no parameters)`;\n    }\n\n    const lines: string[] = [];\n    for (const [name, propSchema] of Object.entries(props)) {\n      lines.push(...formatProperty(name, propSchema, required.includes(name), indent));\n    }\n    return lines.join(\"\\n\");\n  }\n\n  const lines = formatNestedSchema(s, indent);\n  if (lines.length > 0) {\n    return lines.join(\"\\n\");\n  }\n\n  const typeStr = formatType(s);\n  if (typeStr) {\n    return `${indent}(${typeStr})`;\n  }\n\n  return `${indent}(complex schema)`;\n}\n\nfunction formatProperty(name: string, schema: unknown, required: boolean, indent: string): string[] {\n  if (!schema || typeof schema !== \"object\" || Array.isArray(schema)) {\n    return [`${indent}${name}${required ? \" *required*\" : \"\"}`];\n  }\n\n  const s = schema as Record<string, unknown>;\n  const parts = [`${indent}${name}`];\n  const typeStr = formatType(s);\n  if (typeStr) parts.push(`(${typeStr})`);\n  if (required) parts.push(\"*required*\");\n  appendSchemaAnnotations(parts, s);\n\n  return [parts.join(\" \"), ...formatNestedSchema(s, `${indent}  `)];\n}\n\nfunction formatNestedSchema(schema: Record<string, unknown>, indent: string): string[] {\n  const lines: string[] = [];\n\n  if (Array.isArray(schema.anyOf)) {\n    lines.push(...formatVariants(\"anyOf\", schema.anyOf, indent));\n  }\n  if (Array.isArray(schema.oneOf)) {\n    lines.push(...formatVariants(\"oneOf\", schema.oneOf, indent));\n  }\n  if (schema.items !== undefined) {\n    lines.push(...formatProperty(\"items\", schema.items, false, indent));\n  }\n  if (schema.properties && typeof schema.properties === \"object\" && !Array.isArray(schema.properties)) {\n    const required = Array.isArray(schema.required) ? schema.required.filter((name): name is string => typeof name === \"string\") : [];\n    for (const [name, propSchema] of Object.entries(schema.properties as Record<string, unknown>)) {\n      lines.push(...formatProperty(name, propSchema, required.includes(name), indent));\n    }\n  }\n\n  return lines;\n}\n\nfunction formatVariants(keyword: \"anyOf\" | \"oneOf\", variants: unknown[], indent: string): string[] {\n  const lines = [`${indent}${keyword}:`];\n\n  for (const variant of variants) {\n    if (!variant || typeof variant !== \"object\" || Array.isArray(variant)) {\n      lines.push(`${indent}  - ${JSON.stringify(variant)}`);\n      continue;\n    }\n\n    const s = variant as Record<string, unknown>;\n    const typeStr = formatType(s) || \"schema\";\n    const parts = [`${indent}  - ${typeStr}`];\n    appendSchemaAnnotations(parts, s);\n    lines.push(parts.join(\" \"));\n    lines.push(...formatNestedSchema(s, `${indent}    `));\n  }\n\n  return lines;\n}\n\nfunction formatType(schema: Record<string, unknown>): string {\n  if (Object.hasOwn(schema, \"const\")) {\n    return `const ${JSON.stringify(schema.const)}`;\n  }\n\n  if (Array.isArray(schema.enum)) {\n    return `enum: ${schema.enum.map(v => JSON.stringify(v)).join(\", \")}`;\n  }\n\n  if (Array.isArray(schema.type)) {\n    return schema.type.map(type => String(type)).join(\" | \");\n  }\n\n  if (schema.type) {\n    return String(schema.type);\n  }\n\n  if (schema.properties && typeof schema.properties === \"object\" && !Array.isArray(schema.properties)) {\n    return \"object\";\n  }\n\n  if (schema.items !== undefined) {\n    return \"array\";\n  }\n\n  return \"\";\n}\n\nfunction appendSchemaAnnotations(parts: string[], schema: Record<string, unknown>): void {\n  if (schema.description && typeof schema.description === \"string\") {\n    parts.push(`- ${schema.description}`);\n  }\n\n  for (const key of [\"minLength\", \"maxLength\", \"minimum\", \"maximum\", \"minItems\", \"maxItems\", \"format\", \"pattern\"] as const) {\n    if (schema[key] !== undefined) {\n      parts.push(`[${key}: ${JSON.stringify(schema[key])}]`);\n    }\n  }\n\n  if (schema.default !== undefined) {\n    parts.push(`[default: ${JSON.stringify(schema.default)}]`);\n  }\n}\n"]}