@akon-labs/nexus-cli 0.1.9 → 0.1.11-rc.2

package/dist/extensions/mcp/lifecycle.js

@@ -0,0 +1,395 @@
+import { SdkError, SdkErrorCode } from "@modelcontextprotocol/client";
+import { logger } from "./logger.js";
+import { hasPendingAuth } from "./mcp-auth-flow.js";
+import { isTerminatedSession } from "./session-recovery.js";
+import { isServerDisabled } from "./types.js";
+import { formatTerminalError, parallelLimit, sanitizeTerminalText } from "./utils.js";
+const KEEP_ALIVE_RETRY_BASE_MS = 30_000;
+const KEEP_ALIVE_RETRY_MAX_MS = 5 * 60_000;
+const KEEP_ALIVE_CHECK_CONCURRENCY = 10;
+export class McpLifecycleManager {
+    keepAliveServers = new Map();
+    allServers = new Map();
+    serverSettings = new Map();
+    globalIdleTimeout = 10 * 60 * 1000;
+    healthCheckInterval;
+    onReconnect;
+    onReconnectFailure;
+    onHealthRestored;
+    onAuthRequired;
+    onIdleShutdown;
+    activeHealthCheck;
+    activeConvergence;
+    shutdownPromise;
+    retryStates = new Map();
+    pendingMetadataPublications = new Set();
+    stopped = false;
+    removeHealthAbortListener;
+    manager;
+    hasPendingAuthForServer;
+    constructor(manager, hasPendingAuthForServer = hasPendingAuth) {
+        this.manager = manager;
+        this.hasPendingAuthForServer = hasPendingAuthForServer;
+    }
+    setReconnectCallback(callback) {
+        this.onReconnect = callback;
+    }
+    setReconnectFailureCallback(callback) {
+        this.onReconnectFailure = callback;
+    }
+    setHealthRestoredCallback(callback) {
+        this.onHealthRestored = callback;
+    }
+    setAuthRequiredCallback(callback) {
+        this.onAuthRequired = callback;
+    }
+    markKeepAlive(name, definition) {
+        if (isServerDisabled(definition))
+            return;
+        this.keepAliveServers.set(name, definition);
+    }
+    registerServer(name, definition, settings) {
+        if (isServerDisabled(definition))
+            return;
+        this.allServers.set(name, definition);
+        if (settings?.idleTimeout !== undefined)
+            this.serverSettings.set(name, settings);
+    }
+    unregisterServer(name) {
+        this.allServers.delete(name);
+        this.keepAliveServers.delete(name);
+        this.serverSettings.delete(name);
+        this.retryStates.delete(name);
+        this.pendingMetadataPublications.delete(name);
+    }
+    setGlobalIdleTimeout(minutes) {
+        this.globalIdleTimeout = minutes * 60 * 1000;
+    }
+    setIdleShutdownCallback(callback) {
+        this.onIdleShutdown = callback;
+    }
+    startHealthChecks(signalOrInterval, maybeIntervalMs = 30000) {
+        const signal = typeof signalOrInterval === "number" ? undefined : signalOrInterval;
+        const intervalMs = typeof signalOrInterval === "number" ? signalOrInterval : maybeIntervalMs;
+        this.stopped = false;
+        if (signal?.aborted) {
+            this.stopped = true;
+            return;
+        }
+        const stop = () => {
+            this.stopped = true;
+            if (this.healthCheckInterval)
+                clearInterval(this.healthCheckInterval);
+            this.healthCheckInterval = undefined;
+        };
+        signal?.addEventListener("abort", stop, { once: true });
+        this.removeHealthAbortListener = () => signal?.removeEventListener("abort", stop);
+        this.healthCheckInterval = setInterval(() => {
+            if (this.stopped || signal?.aborted || this.activeHealthCheck)
+                return;
+            const check = this.checkConnections(signal)
+                .catch((error) => {
+                console.error(`MCP: Health check failed: ${formatTerminalError(error)}`);
+            })
+                .finally(() => {
+                if (this.activeHealthCheck === check)
+                    this.activeHealthCheck = undefined;
+            });
+            this.activeHealthCheck = check;
+        }, intervalMs);
+        this.healthCheckInterval.unref();
+    }
+    async ensureConverged(signal) {
+        if (this.stopped || signal?.aborted)
+            return;
+        if (this.activeConvergence)
+            return this.activeConvergence;
+        const check = this.checkKeepAliveConnections(signal);
+        this.activeConvergence = check;
+        try {
+            await check;
+        }
+        finally {
+            if (this.activeConvergence === check)
+                this.activeConvergence = undefined;
+        }
+    }
+    async checkConnections(signal) {
+        if (this.stopped || signal?.aborted)
+            return;
+        await this.ensureConverged(signal);
+        if (this.stopped || signal?.aborted)
+            return;
+        for (const [name] of this.allServers) {
+            if (this.keepAliveServers.has(name))
+                continue;
+            const timeout = this.getIdleTimeout(name);
+            const connection = this.manager.getConnection(name);
+            if (connection && timeout > 0 && this.manager.isIdle(name, timeout)) {
+                const closed = await this.manager.closeIfCurrent(name, connection);
+                if (this.stopped || signal?.aborted)
+                    return;
+                if (closed)
+                    this.onIdleShutdown?.(name);
+            }
+        }
+    }
+    async checkKeepAliveConnections(signal) {
+        if (this.stopped || signal?.aborted)
+            return;
+        await parallelLimit([...this.keepAliveServers], KEEP_ALIVE_CHECK_CONCURRENCY, ([name, definition]) => this.checkKeepAliveConnection(name, definition, signal));
+    }
+    async checkKeepAliveConnection(name, definition, signal, retrySuperseded = true) {
+        if (isServerDisabled(definition) || this.stopped || signal?.aborted)
+            return;
+        // Fence against unregisterServer racing an in-flight convergence pass.
+        // Identity comparison also rejects stale passes after a same-name
+        // replacement registration, which a name check would wrongly accept.
+        if (this.keepAliveServers.get(name) !== definition)
+            return;
+        const connection = this.manager.getConnection(name);
+        if (connection?.status === "needs-auth") {
+            this.pendingMetadataPublications.delete(name);
+            return;
+        }
+        if (!this.shouldAttemptConnection(name, connection))
+            return;
+        if (!connection || connection.status !== "connected") {
+            if (this.hasPendingAuthForServer(name)) {
+                logger.debug(`Skipping reconnect for ${name} while OAuth authorization is pending`);
+                return;
+            }
+            let freshConnection;
+            try {
+                freshConnection = await this.manager.connect(name, definition, signal);
+            }
+            catch (error) {
+                if (this.stopped || signal?.aborted)
+                    return;
+                this.reportConnectionFailure(name, definition, error, "reconnect", this.manager.getConnection(name));
+                return;
+            }
+            if (this.stopped || signal?.aborted)
+                return;
+            if (freshConnection.status === "needs-auth") {
+                await this.notifyAuthRequired(name, definition, freshConnection);
+                return;
+            }
+            if (freshConnection.status !== "connected") {
+                this.reportConnectionFailure(name, definition, new Error(`MCP server ${name} did not return a connected session`), "reconnect", freshConnection);
+                return;
+            }
+            logger.debug(`Reconnected to ${name}`);
+            await this.publishConnectedMetadata(name, definition, freshConnection);
+            return;
+        }
+        if (this.pendingMetadataPublications.has(name)) {
+            await this.publishConnectedMetadata(name, definition, connection);
+            return;
+        }
+        const hadSessionId = connection.transport?.sessionId != null;
+        let refreshResult;
+        try {
+            refreshResult = await this.manager.refreshTools(name, connection, signal);
+        }
+        catch (error) {
+            if (this.stopped || signal?.aborted)
+                return;
+            const current = this.manager.getConnection(name);
+            if (current !== connection || connection.status !== "connected") {
+                await this.handleSupersededConnection(name, definition, connection, signal, retrySuperseded);
+                return;
+            }
+            if (!shouldReconnectAfterRefresh(error, hadSessionId)) {
+                this.reportConnectionFailure(name, definition, error, "refresh", connection);
+                return;
+            }
+            if (this.hasPendingAuthForServer(name)) {
+                logger.debug(`Skipping reconnect for ${name} while OAuth authorization is pending`);
+                return;
+            }
+            let freshConnection;
+            try {
+                freshConnection = await this.manager.reconnect(name, definition, connection, signal);
+            }
+            catch (reconnectError) {
+                if (this.stopped || signal?.aborted)
+                    return;
+                this.reportConnectionFailure(name, definition, reconnectError, "reconnect", this.manager.getConnection(name));
+                return;
+            }
+            if (this.stopped || signal?.aborted)
+                return;
+            if (freshConnection.status === "needs-auth") {
+                await this.notifyAuthRequired(name, definition, freshConnection);
+                return;
+            }
+            if (freshConnection.status !== "connected") {
+                this.reportConnectionFailure(name, definition, new Error(`MCP server ${name} did not return a connected session`), "reconnect", freshConnection);
+                return;
+            }
+            logger.debug(`Reconnected stale MCP session for ${name}`);
+            await this.publishConnectedMetadata(name, definition, freshConnection);
+            return;
+        }
+        if (refreshResult === "superseded") {
+            await this.handleSupersededConnection(name, definition, connection, signal, retrySuperseded);
+            return;
+        }
+        if (refreshResult === "refresh-timeout") {
+            this.deferRefreshTimeout(name, definition, connection);
+            return;
+        }
+        if (this.keepAliveServers.get(name) !== definition)
+            return;
+        if (this.retryStates.delete(name)) {
+            await this.onHealthRestored?.(name);
+        }
+    }
+    async handleSupersededConnection(name, definition, staleConnection, signal, retrySuperseded) {
+        const current = this.manager.getConnection(name);
+        if (this.keepAliveServers.get(name) !== definition)
+            return;
+        if (current === staleConnection && current.status === "connected") {
+            if (this.retryStates.delete(name))
+                await this.onHealthRestored?.(name);
+            return;
+        }
+        if (current?.status === "connected") {
+            await this.publishConnectedMetadata(name, definition, current);
+            return;
+        }
+        if (current?.status === "needs-auth") {
+            await this.notifyAuthRequired(name, definition, current);
+            return;
+        }
+        if (retrySuperseded) {
+            await this.checkKeepAliveConnection(name, definition, signal, false);
+        }
+    }
+    async publishConnectedMetadata(name, definition, connection) {
+        // Fence stale convergence passes by definition identity so disposal, and
+        // disposal followed by a same-name replacement, never re-track this
+        // server. Close the pass's own connection only when it is still current,
+        // so a replacement's connection is never touched.
+        if (this.keepAliveServers.get(name) !== definition) {
+            if (this.manager.getConnection(name) === connection)
+                await this.manager.close(name);
+            return;
+        }
+        this.pendingMetadataPublications.add(name);
+        try {
+            await this.onReconnect?.(name);
+            this.pendingMetadataPublications.delete(name);
+            this.retryStates.delete(name);
+        }
+        catch (error) {
+            if (this.stopped)
+                return;
+            this.reportConnectionFailure(name, definition, error, "publish", connection);
+        }
+    }
+    async notifyAuthRequired(name, definition, connection) {
+        // Fence stale convergence passes by definition identity. Close the
+        // leftover needs-auth connection only while it is still the manager's
+        // current entry, so a replacement's connection is never touched and the
+        // stale client/transport cannot leak when a replacement connects later.
+        if (this.keepAliveServers.get(name) !== definition) {
+            if (this.manager.getConnection(name) === connection)
+                await this.manager.close(name);
+            return;
+        }
+        this.pendingMetadataPublications.delete(name);
+        this.retryStates.delete(name);
+        try {
+            await this.onAuthRequired?.(name);
+        }
+        catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            logger.debug(`MCP: auth-required callback failed for ${name}: ${sanitizeTerminalText(message)}`);
+        }
+    }
+    shouldAttemptConnection(name, connection) {
+        const retry = this.retryStates.get(name);
+        if (!retry)
+            return true;
+        if (retry.connection !== connection || retry.status !== connection?.status) {
+            this.retryStates.delete(name);
+            return true;
+        }
+        return Date.now() >= retry.nextAttemptAt;
+    }
+    reportConnectionFailure(name, definition, error, action, connection) {
+        if (!this.recordRetry(name, definition, connection))
+            return;
+        this.onReconnectFailure?.(name, error);
+        const message = error instanceof Error ? error.message : String(error);
+        const target = action === "reconnect"
+            ? `reconnect to ${name}`
+            : action === "publish"
+                ? `publish metadata for ${name}`
+                : `refresh ${name}`;
+        console.error(`MCP: Failed to ${target}: ${sanitizeTerminalText(message)}`);
+    }
+    deferRefreshTimeout(name, definition, connection) {
+        if (!this.recordRetry(name, definition, connection))
+            return;
+        logger.debug(`MCP: keep-alive tools/list refresh timed out for ${name}; retrying after backoff`);
+    }
+    recordRetry(name, definition, connection) {
+        // Do not recreate retry/failure state from a stale convergence pass after
+        // disposal or a same-name replacement registration.
+        if (this.keepAliveServers.get(name) !== definition)
+            return false;
+        const attempts = (this.retryStates.get(name)?.attempts ?? 0) + 1;
+        const delay = Math.min(KEEP_ALIVE_RETRY_BASE_MS * 2 ** Math.min(attempts - 1, 10), KEEP_ALIVE_RETRY_MAX_MS);
+        this.retryStates.set(name, {
+            attempts,
+            nextAttemptAt: Date.now() + delay,
+            connection,
+            status: connection?.status,
+        });
+        return true;
+    }
+    getIdleTimeout(name) {
+        const perServer = this.serverSettings.get(name)?.idleTimeout;
+        if (perServer !== undefined)
+            return perServer * 60 * 1000;
+        return this.globalIdleTimeout;
+    }
+    async gracefulShutdown() {
+        if (this.shutdownPromise)
+            return this.shutdownPromise;
+        this.shutdownPromise = this.shutdownOnce();
+        return this.shutdownPromise;
+    }
+    async shutdownOnce() {
+        this.stopped = true;
+        if (this.healthCheckInterval)
+            clearInterval(this.healthCheckInterval);
+        this.healthCheckInterval = undefined;
+        this.removeHealthAbortListener?.();
+        this.removeHealthAbortListener = undefined;
+        await this.activeHealthCheck;
+        this.activeHealthCheck = undefined;
+        await this.activeConvergence;
+        this.activeConvergence = undefined;
+        this.onReconnect = undefined;
+        this.onReconnectFailure = undefined;
+        this.onHealthRestored = undefined;
+        this.onAuthRequired = undefined;
+        this.onIdleShutdown = undefined;
+        this.retryStates.clear();
+        this.pendingMetadataPublications.clear();
+        if (typeof this.manager.closeAll === "function") {
+            await this.manager.closeAll();
+        }
+    }
+}
+function shouldReconnectAfterRefresh(error, hadSessionId) {
+    if (isTerminatedSession(error, hadSessionId))
+        return true;
+    return (error instanceof SdkError &&
+        (error.code === SdkErrorCode.NotConnected || error.code === SdkErrorCode.ConnectionClosed));
+}
+//# sourceMappingURL=lifecycle.js.map

package/dist/extensions/mcp/lifecycle.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lifecycle.js","sourceRoot":"","sources":["../../../src/extensions/mcp/lifecycle.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAEpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAyB,MAAM,YAAY,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,aAAa,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAOtF,MAAM,wBAAwB,GAAG,MAAM,CAAC;AACxC,MAAM,uBAAuB,GAAG,CAAC,GAAG,MAAM,CAAC;AAC3C,MAAM,4BAA4B,GAAG,EAAE,CAAC;AASxC,MAAM,OAAO,mBAAmB;IACvB,gBAAgB,GAAG,IAAI,GAAG,EAA4B,CAAC;IACvD,UAAU,GAAG,IAAI,GAAG,EAA4B,CAAC;IACjD,cAAc,GAAG,IAAI,GAAG,EAAoC,CAAC;IAC7D,iBAAiB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IACnC,mBAAmB,CAA6B;IAChD,WAAW,CAAgC;IAC3C,kBAAkB,CAAuC;IACzD,gBAAgB,CAAqC;IACrD,cAAc,CAAmC;IACjD,cAAc,CAA6C;IAC3D,iBAAiB,CAA4B;IAC7C,iBAAiB,CAA4B;IAC7C,eAAe,CAA4B;IAC3C,WAAW,GAAG,IAAI,GAAG,EAAsB,CAAC;IAC5C,2BAA2B,GAAG,IAAI,GAAG,EAAU,CAAC;IAChD,OAAO,GAAG,KAAK,CAAC;IAChB,yBAAyB,CAA2B;IAC3C,OAAO,CAAmB;IAC1B,uBAAuB,CAAwB;IAEhE,YAAY,OAAyB,EAAE,uBAAuB,GAAG,cAAc,EAAE;QAChF,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;IAAA,CACvD;IAED,oBAAoB,CAAC,QAA2B,EAAQ;QACvD,IAAI,CAAC,WAAW,GAAG,QAAQ,CAAC;IAAA,CAC5B;IAED,2BAA2B,CAAC,QAAkC,EAAQ;QACrE,IAAI,CAAC,kBAAkB,GAAG,QAAQ,CAAC;IAAA,CACnC;IAED,yBAAyB,CAAC,QAAgC,EAAQ;QACjE,IAAI,CAAC,gBAAgB,GAAG,QAAQ,CAAC;IAAA,CACjC;IAED,uBAAuB,CAAC,QAA8B,EAAQ;QAC7D,IAAI,CAAC,cAAc,GAAG,QAAQ,CAAC;IAAA,CAC/B;IAED,aAAa,CAAC,IAAY,EAAE,UAA4B,EAAQ;QAC/D,IAAI,gBAAgB,CAAC,UAAU,CAAC;YAAE,OAAO;QACzC,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAAA,CAC5C;IAED,cAAc,CAAC,IAAY,EAAE,UAA4B,EAAE,QAAmC,EAAQ;QACrG,IAAI,gBAAgB,CAAC,UAAU,CAAC;YAAE,OAAO;QACzC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACtC,IAAI,QAAQ,EAAE,WAAW,KAAK,SAAS;YAAE,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAAA,CACjF;IAED,gBAAgB,CAAC,IAAY,EAAQ;QACpC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QACjC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAAA,CAC9C;IAED,oBAAoB,CAAC,OAAe,EAAQ;QAC3C,IAAI,CAAC,iBAAiB,GAAG,OAAO,GAAG,EAAE,GAAG,IAAI,CAAC;IAAA,CAC7C;IAED,uBAAuB,CAAC,QAAsC,EAAQ;QACrE,IAAI,CAAC,cAAc,GAAG,QAAQ,CAAC;IAAA,CAC/B;IAED,iBAAiB,CAAC,gBAAuC,EAAE,eAAe,GAAG,KAAK,EAAQ;QACzF,MAAM,MAAM,GAAG,OAAO,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,gBAAgB,CAAC;QACnF,MAAM,UAAU,GAAG,OAAO,gBAAgB,KAAK,QAAQ,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,eAAe,CAAC;QAC7F,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACrB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,OAAO;QACR,CAAC;QACD,MAAM,IAAI,GAAG,GAAG,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,IAAI,CAAC,mBAAmB;gBAAE,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;YACtE,IAAI,CAAC,mBAAmB,GAAG,SAAS,CAAC;QAAA,CACrC,CAAC;QACF,MAAM,EAAE,gBAAgB,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,yBAAyB,GAAG,GAAG,EAAE,CAAC,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAClF,IAAI,CAAC,mBAAmB,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO,IAAI,IAAI,CAAC,iBAAiB;gBAAE,OAAO;YACtE,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;iBACzC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,6BAA6B,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAAA,CACzE,CAAC;iBACD,OAAO,CAAC,GAAG,EAAE,CAAC;gBACd,IAAI,IAAI,CAAC,iBAAiB,KAAK,KAAK;oBAAE,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;YAAA,CACzE,CAAC,CAAC;YACJ,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;QAAA,CAC/B,EAAE,UAAU,CAAC,CAAC;QACf,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAAA,CACjC;IAED,KAAK,CAAC,eAAe,CAAC,MAAoB,EAAiB;QAC1D,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO;QAC5C,IAAI,IAAI,CAAC,iBAAiB;YAAE,OAAO,IAAI,CAAC,iBAAiB,CAAC;QAE1D,MAAM,KAAK,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC;QAC/B,IAAI,CAAC;YACJ,MAAM,KAAK,CAAC;QACb,CAAC;gBAAS,CAAC;YACV,IAAI,IAAI,CAAC,iBAAiB,KAAK,KAAK;gBAAE,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QAC1E,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,gBAAgB,CAAC,MAAoB,EAAiB;QACnE,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO;QAC5C,MAAM,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QACnC,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO;QAE5C,KAAK,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACtC,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC;gBAAE,SAAS;YAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAC1C,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACpD,IAAI,UAAU,IAAI,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC;gBACrE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBACnE,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;oBAAE,OAAO;gBAC5C,IAAI,MAAM;oBAAE,IAAI,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,CAAC;YACzC,CAAC;QACF,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,yBAAyB,CAAC,MAAoB,EAAiB;QAC5E,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO;QAC5C,MAAM,aAAa,CAAC,CAAC,GAAG,IAAI,CAAC,gBAAgB,CAAC,EAAE,4BAA4B,EAAE,CAAC,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,CACpG,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CACvD,CAAC;IAAA,CACF;IAEO,KAAK,CAAC,wBAAwB,CACrC,IAAY,EACZ,UAA4B,EAC5B,MAAoB,EACpB,eAAe,GAAG,IAAI,EACN;QAChB,IAAI,gBAAgB,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO;QAC5E,uEAAuE;QACvE,kEAAkE;QAClE,qEAAqE;QACrE,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,UAAU;YAAE,OAAO;QAC3D,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACpD,IAAI,UAAU,EAAE,MAAM,KAAK,YAAY,EAAE,CAAC;YACzC,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC9C,OAAO;QACR,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,UAAU,CAAC;YAAE,OAAO;QAC5D,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACtD,IAAI,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,MAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,uCAAuC,CAAC,CAAC;gBACpF,OAAO;YACR,CAAC;YACD,IAAI,eAAiC,CAAC;YACtC,IAAI,CAAC;gBACJ,eAAe,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YACxE,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAChB,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;oBAAE,OAAO;gBAC5C,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;gBACrG,OAAO;YACR,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;gBAAE,OAAO;YAC5C,IAAI,eAAe,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;gBAC7C,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;gBACjE,OAAO;YACR,CAAC;YACD,IAAI,eAAe,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAC5C,IAAI,CAAC,uBAAuB,CAC3B,IAAI,EACJ,UAAU,EACV,IAAI,KAAK,CAAC,cAAc,IAAI,qCAAqC,CAAC,EAClE,WAAW,EACX,eAAe,CACf,CAAC;gBACF,OAAO;YACR,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAC;YACvC,MAAM,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;YACvE,OAAO;QACR,CAAC;QAED,IAAI,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC;YAChD,MAAM,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;YAClE,OAAO;QACR,CAAC;QACD,MAAM,YAAY,GAAI,UAAU,CAAC,SAAgD,EAAE,SAAS,IAAI,IAAI,CAAC;QACrG,IAAI,aAAoE,CAAC;QACzE,IAAI,CAAC;YACJ,aAAa,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAC3E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;gBAAE,OAAO;YAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,OAAO,KAAK,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACjE,MAAM,IAAI,CAAC,0BAA0B,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,eAAe,CAAC,CAAC;gBAC7F,OAAO;YACR,CAAC;YACD,IAAI,CAAC,2BAA2B,CAAC,KAAK,EAAE,YAAY,CAAC,EAAE,CAAC;gBACvD,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;gBAC7E,OAAO;YACR,CAAC;YACD,IAAI,IAAI,CAAC,uBAAuB,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxC,MAAM,CAAC,KAAK,CAAC,0BAA0B,IAAI,uCAAuC,CAAC,CAAC;gBACpF,OAAO;YACR,CAAC;YACD,IAAI,eAAiC,CAAC;YACtC,IAAI,CAAC;gBACJ,eAAe,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;YACtF,CAAC;YAAC,OAAO,cAAc,EAAE,CAAC;gBACzB,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;oBAAE,OAAO;gBAC5C,IAAI,CAAC,uBAAuB,CAC3B,IAAI,EACJ,UAAU,EACV,cAAc,EACd,WAAW,EACX,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAChC,CAAC;gBACF,OAAO;YACR,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,EAAE,OAAO;gBAAE,OAAO;YAC5C,IAAI,eAAe,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;gBAC7C,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;gBACjE,OAAO;YACR,CAAC;YACD,IAAI,eAAe,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBAC5C,IAAI,CAAC,uBAAuB,CAC3B,IAAI,EACJ,UAAU,EACV,IAAI,KAAK,CAAC,cAAc,IAAI,qCAAqC,CAAC,EAClE,WAAW,EACX,eAAe,CACf,CAAC;gBACF,OAAO;YACR,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,qCAAqC,IAAI,EAAE,CAAC,CAAC;YAC1D,MAAM,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,eAAe,CAAC,CAAC;YACvE,OAAO;QACR,CAAC;QAED,IAAI,aAAa,KAAK,YAAY,EAAE,CAAC;YACpC,MAAM,IAAI,CAAC,0BAA0B,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,EAAE,eAAe,CAAC,CAAC;YAC7F,OAAO;QACR,CAAC;QACD,IAAI,aAAa,KAAK,iBAAiB,EAAE,CAAC;YACzC,IAAI,CAAC,mBAAmB,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;YACvD,OAAO;QACR,CAAC;QACD,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,UAAU;YAAE,OAAO;QAC3D,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;YACnC,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,0BAA0B,CACvC,IAAY,EACZ,UAA4B,EAC5B,eAAiC,EACjC,MAA+B,EAC/B,eAAwB,EACR;QAChB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACjD,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,UAAU;YAAE,OAAO;QAC3D,IAAI,OAAO,KAAK,eAAe,IAAI,OAAO,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACnE,IAAI,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC;gBAAE,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC,IAAI,CAAC,CAAC;YACvE,OAAO;QACR,CAAC;QACD,IAAI,OAAO,EAAE,MAAM,KAAK,WAAW,EAAE,CAAC;YACrC,MAAM,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YAC/D,OAAO;QACR,CAAC;QACD,IAAI,OAAO,EAAE,MAAM,KAAK,YAAY,EAAE,CAAC;YACtC,MAAM,IAAI,CAAC,kBAAkB,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YACzD,OAAO;QACR,CAAC;QACD,IAAI,eAAe,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;QACtE,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,wBAAwB,CACrC,IAAY,EACZ,UAA4B,EAC5B,UAA4B,EACZ;QAChB,yEAAyE;QACzE,oEAAoE;QACpE,yEAAyE;QACzE,kDAAkD;QAClD,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACpD,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,UAAU;gBAAE,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpF,OAAO;QACR,CAAC;QACD,IAAI,CAAC,2BAA2B,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC3C,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC;YAC/B,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC9C,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,IAAI,CAAC,OAAO;gBAAE,OAAO;YACzB,IAAI,CAAC,uBAAuB,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QAC9E,CAAC;IAAA,CACD;IAEO,KAAK,CAAC,kBAAkB,CAC/B,IAAY,EACZ,UAA4B,EAC5B,UAA4B,EACZ;QAChB,mEAAmE;QACnE,sEAAsE;QACtE,wEAAwE;QACxE,wEAAwE;QACxE,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACpD,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,UAAU;gBAAE,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACpF,OAAO;QACR,CAAC;QACD,IAAI,CAAC,2BAA2B,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC;YACJ,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,MAAM,CAAC,KAAK,CAAC,0CAA0C,IAAI,KAAK,oBAAoB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAClG,CAAC;IAAA,CACD;IAEO,uBAAuB,CAAC,IAAY,EAAE,UAAwC,EAAW;QAChG,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK;YAAE,OAAO,IAAI,CAAC;QACxB,IAAI,KAAK,CAAC,UAAU,KAAK,UAAU,IAAI,KAAK,CAAC,MAAM,KAAK,UAAU,EAAE,MAAM,EAAE,CAAC;YAC5E,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC9B,OAAO,IAAI,CAAC;QACb,CAAC;QACD,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,KAAK,CAAC,aAAa,CAAC;IAAA,CACzC;IAEO,uBAAuB,CAC9B,IAAY,EACZ,UAA4B,EAC5B,KAAc,EACd,MAA2C,EAC3C,UAAwC,EACjC;QACP,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;YAAE,OAAO;QAC5D,IAAI,CAAC,kBAAkB,EAAE,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,MAAM,GACX,MAAM,KAAK,WAAW;YACrB,CAAC,CAAC,gBAAgB,IAAI,EAAE;YACxB,CAAC,CAAC,MAAM,KAAK,SAAS;gBACrB,CAAC,CAAC,wBAAwB,IAAI,EAAE;gBAChC,CAAC,CAAC,WAAW,IAAI,EAAE,CAAC;QACvB,OAAO,CAAC,KAAK,CAAC,kBAAkB,MAAM,KAAK,oBAAoB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IAAA,CAC5E;IAEO,mBAAmB,CAC1B,IAAY,EACZ,UAA4B,EAC5B,UAAwC,EACjC;QACP,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,UAAU,EAAE,UAAU,CAAC;YAAE,OAAO;QAC5D,MAAM,CAAC,KAAK,CAAC,oDAAoD,IAAI,0BAA0B,CAAC,CAAC;IAAA,CACjG;IAEO,WAAW,CAAC,IAAY,EAAE,UAA4B,EAAE,UAAwC,EAAW;QAClH,0EAA0E;QAC1E,oDAAoD;QACpD,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,UAAU;YAAE,OAAO,KAAK,CAAC;QACjE,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,QAAQ,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,wBAAwB,GAAG,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,GAAG,CAAC,EAAE,EAAE,CAAC,EAAE,uBAAuB,CAAC,CAAC;QAC5G,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,EAAE;YAC1B,QAAQ;YACR,aAAa,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK;YACjC,UAAU;YACV,MAAM,EAAE,UAAU,EAAE,MAAM;SAC1B,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IAAA,CACZ;IAEO,cAAc,CAAC,IAAY,EAAU;QAC5C,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,WAAW,CAAC;QAC7D,IAAI,SAAS,KAAK,SAAS;YAAE,OAAO,SAAS,GAAG,EAAE,GAAG,IAAI,CAAC;QAC1D,OAAO,IAAI,CAAC,iBAAiB,CAAC;IAAA,CAC9B;IAED,KAAK,CAAC,gBAAgB,GAAkB;QACvC,IAAI,IAAI,CAAC,eAAe;YAAE,OAAO,IAAI,CAAC,eAAe,CAAC;QACtD,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAC3C,OAAO,IAAI,CAAC,eAAe,CAAC;IAAA,CAC5B;IAEO,KAAK,CAAC,YAAY,GAAkB;QAC3C,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,IAAI,CAAC,mBAAmB;YAAE,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACtE,IAAI,CAAC,mBAAmB,GAAG,SAAS,CAAC;QACrC,IAAI,CAAC,yBAAyB,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,yBAAyB,GAAG,SAAS,CAAC;QAC3C,MAAM,IAAI,CAAC,iBAAiB,CAAC;QAC7B,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACnC,MAAM,IAAI,CAAC,iBAAiB,CAAC;QAC7B,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;QACnC,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,CAAC,kBAAkB,GAAG,SAAS,CAAC;QACpC,IAAI,CAAC,gBAAgB,GAAG,SAAS,CAAC;QAClC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,IAAI,CAAC,cAAc,GAAG,SAAS,CAAC;QAChC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;QACzB,IAAI,CAAC,2BAA2B,CAAC,KAAK,EAAE,CAAC;QACzC,IAAI,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,KAAK,UAAU,EAAE,CAAC;YACjD,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC/B,CAAC;IAAA,CACD;CACD;AAED,SAAS,2BAA2B,CAAC,KAAc,EAAE,YAAqB,EAAW;IACpF,IAAI,mBAAmB,CAAC,KAAK,EAAE,YAAY,CAAC;QAAE,OAAO,IAAI,CAAC;IAC1D,OAAO,CACN,KAAK,YAAY,QAAQ;QACzB,CAAC,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC,gBAAgB,CAAC,CAC1F,CAAC;AAAA,CACF","sourcesContent":["import { SdkError, SdkErrorCode } from \"@modelcontextprotocol/client\";\nimport { logger } from \"./logger.ts\";\nimport { hasPendingAuth } from \"./mcp-auth-flow.ts\";\nimport type { McpServerManager, ServerConnection } from \"./server-manager.ts\";\nimport { isTerminatedSession } from \"./session-recovery.ts\";\nimport { isServerDisabled, type ServerDefinition } from \"./types.ts\";\nimport { formatTerminalError, parallelLimit, sanitizeTerminalText } from \"./utils.ts\";\n\nexport type ReconnectCallback = (serverName: string) => void | Promise<void>;\nexport type ReconnectFailureCallback = (serverName: string, error: unknown) => void;\nexport type HealthRestoredCallback = (serverName: string) => void | Promise<void>;\nexport type AuthRequiredCallback = (serverName: string) => void | Promise<void>;\n\nconst KEEP_ALIVE_RETRY_BASE_MS = 30_000;\nconst KEEP_ALIVE_RETRY_MAX_MS = 5 * 60_000;\nconst KEEP_ALIVE_CHECK_CONCURRENCY = 10;\n\ninterface RetryState {\n\tattempts: number;\n\tnextAttemptAt: number;\n\tconnection: ServerConnection | undefined;\n\tstatus: ServerConnection[\"status\"] | undefined;\n}\n\nexport class McpLifecycleManager {\n\tprivate keepAliveServers = new Map<string, ServerDefinition>();\n\tprivate allServers = new Map<string, ServerDefinition>();\n\tprivate serverSettings = new Map<string, { idleTimeout?: number }>();\n\tprivate globalIdleTimeout = 10 * 60 * 1000;\n\tprivate healthCheckInterval: NodeJS.Timeout | undefined;\n\tprivate onReconnect: ReconnectCallback | undefined;\n\tprivate onReconnectFailure: ReconnectFailureCallback | undefined;\n\tprivate onHealthRestored: HealthRestoredCallback | undefined;\n\tprivate onAuthRequired: AuthRequiredCallback | undefined;\n\tprivate onIdleShutdown: ((serverName: string) => void) | undefined;\n\tprivate activeHealthCheck: Promise<void> | undefined;\n\tprivate activeConvergence: Promise<void> | undefined;\n\tprivate shutdownPromise: Promise<void> | undefined;\n\tprivate retryStates = new Map<string, RetryState>();\n\tprivate pendingMetadataPublications = new Set<string>();\n\tprivate stopped = false;\n\tprivate removeHealthAbortListener: (() => void) | undefined;\n\tprivate readonly manager: McpServerManager;\n\tprivate readonly hasPendingAuthForServer: typeof hasPendingAuth;\n\n\tconstructor(manager: McpServerManager, hasPendingAuthForServer = hasPendingAuth) {\n\t\tthis.manager = manager;\n\t\tthis.hasPendingAuthForServer = hasPendingAuthForServer;\n\t}\n\n\tsetReconnectCallback(callback: ReconnectCallback): void {\n\t\tthis.onReconnect = callback;\n\t}\n\n\tsetReconnectFailureCallback(callback: ReconnectFailureCallback): void {\n\t\tthis.onReconnectFailure = callback;\n\t}\n\n\tsetHealthRestoredCallback(callback: HealthRestoredCallback): void {\n\t\tthis.onHealthRestored = callback;\n\t}\n\n\tsetAuthRequiredCallback(callback: AuthRequiredCallback): void {\n\t\tthis.onAuthRequired = callback;\n\t}\n\n\tmarkKeepAlive(name: string, definition: ServerDefinition): void {\n\t\tif (isServerDisabled(definition)) return;\n\t\tthis.keepAliveServers.set(name, definition);\n\t}\n\n\tregisterServer(name: string, definition: ServerDefinition, settings?: { idleTimeout?: number }): void {\n\t\tif (isServerDisabled(definition)) return;\n\t\tthis.allServers.set(name, definition);\n\t\tif (settings?.idleTimeout !== undefined) this.serverSettings.set(name, settings);\n\t}\n\n\tunregisterServer(name: string): void {\n\t\tthis.allServers.delete(name);\n\t\tthis.keepAliveServers.delete(name);\n\t\tthis.serverSettings.delete(name);\n\t\tthis.retryStates.delete(name);\n\t\tthis.pendingMetadataPublications.delete(name);\n\t}\n\n\tsetGlobalIdleTimeout(minutes: number): void {\n\t\tthis.globalIdleTimeout = minutes * 60 * 1000;\n\t}\n\n\tsetIdleShutdownCallback(callback: (serverName: string) => void): void {\n\t\tthis.onIdleShutdown = callback;\n\t}\n\n\tstartHealthChecks(signalOrInterval?: AbortSignal | number, maybeIntervalMs = 30000): void {\n\t\tconst signal = typeof signalOrInterval === \"number\" ? undefined : signalOrInterval;\n\t\tconst intervalMs = typeof signalOrInterval === \"number\" ? signalOrInterval : maybeIntervalMs;\n\t\tthis.stopped = false;\n\t\tif (signal?.aborted) {\n\t\t\tthis.stopped = true;\n\t\t\treturn;\n\t\t}\n\t\tconst stop = () => {\n\t\t\tthis.stopped = true;\n\t\t\tif (this.healthCheckInterval) clearInterval(this.healthCheckInterval);\n\t\t\tthis.healthCheckInterval = undefined;\n\t\t};\n\t\tsignal?.addEventListener(\"abort\", stop, { once: true });\n\t\tthis.removeHealthAbortListener = () => signal?.removeEventListener(\"abort\", stop);\n\t\tthis.healthCheckInterval = setInterval(() => {\n\t\t\tif (this.stopped || signal?.aborted || this.activeHealthCheck) return;\n\t\t\tconst check = this.checkConnections(signal)\n\t\t\t\t.catch((error) => {\n\t\t\t\t\tconsole.error(`MCP: Health check failed: ${formatTerminalError(error)}`);\n\t\t\t\t})\n\t\t\t\t.finally(() => {\n\t\t\t\t\tif (this.activeHealthCheck === check) this.activeHealthCheck = undefined;\n\t\t\t\t});\n\t\t\tthis.activeHealthCheck = check;\n\t\t}, intervalMs);\n\t\tthis.healthCheckInterval.unref();\n\t}\n\n\tasync ensureConverged(signal?: AbortSignal): Promise<void> {\n\t\tif (this.stopped || signal?.aborted) return;\n\t\tif (this.activeConvergence) return this.activeConvergence;\n\n\t\tconst check = this.checkKeepAliveConnections(signal);\n\t\tthis.activeConvergence = check;\n\t\ttry {\n\t\t\tawait check;\n\t\t} finally {\n\t\t\tif (this.activeConvergence === check) this.activeConvergence = undefined;\n\t\t}\n\t}\n\n\tprivate async checkConnections(signal?: AbortSignal): Promise<void> {\n\t\tif (this.stopped || signal?.aborted) return;\n\t\tawait this.ensureConverged(signal);\n\t\tif (this.stopped || signal?.aborted) return;\n\n\t\tfor (const [name] of this.allServers) {\n\t\t\tif (this.keepAliveServers.has(name)) continue;\n\t\t\tconst timeout = this.getIdleTimeout(name);\n\t\t\tconst connection = this.manager.getConnection(name);\n\t\t\tif (connection && timeout > 0 && this.manager.isIdle(name, timeout)) {\n\t\t\t\tconst closed = await this.manager.closeIfCurrent(name, connection);\n\t\t\t\tif (this.stopped || signal?.aborted) return;\n\t\t\t\tif (closed) this.onIdleShutdown?.(name);\n\t\t\t}\n\t\t}\n\t}\n\n\tprivate async checkKeepAliveConnections(signal?: AbortSignal): Promise<void> {\n\t\tif (this.stopped || signal?.aborted) return;\n\t\tawait parallelLimit([...this.keepAliveServers], KEEP_ALIVE_CHECK_CONCURRENCY, ([name, definition]) =>\n\t\t\tthis.checkKeepAliveConnection(name, definition, signal),\n\t\t);\n\t}\n\n\tprivate async checkKeepAliveConnection(\n\t\tname: string,\n\t\tdefinition: ServerDefinition,\n\t\tsignal?: AbortSignal,\n\t\tretrySuperseded = true,\n\t): Promise<void> {\n\t\tif (isServerDisabled(definition) || this.stopped || signal?.aborted) return;\n\t\t// Fence against unregisterServer racing an in-flight convergence pass.\n\t\t// Identity comparison also rejects stale passes after a same-name\n\t\t// replacement registration, which a name check would wrongly accept.\n\t\tif (this.keepAliveServers.get(name) !== definition) return;\n\t\tconst connection = this.manager.getConnection(name);\n\t\tif (connection?.status === \"needs-auth\") {\n\t\t\tthis.pendingMetadataPublications.delete(name);\n\t\t\treturn;\n\t\t}\n\t\tif (!this.shouldAttemptConnection(name, connection)) return;\n\t\tif (!connection || connection.status !== \"connected\") {\n\t\t\tif (this.hasPendingAuthForServer(name)) {\n\t\t\t\tlogger.debug(`Skipping reconnect for ${name} while OAuth authorization is pending`);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tlet freshConnection: ServerConnection;\n\t\t\ttry {\n\t\t\t\tfreshConnection = await this.manager.connect(name, definition, signal);\n\t\t\t} catch (error) {\n\t\t\t\tif (this.stopped || signal?.aborted) return;\n\t\t\t\tthis.reportConnectionFailure(name, definition, error, \"reconnect\", this.manager.getConnection(name));\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tif (this.stopped || signal?.aborted) return;\n\t\t\tif (freshConnection.status === \"needs-auth\") {\n\t\t\t\tawait this.notifyAuthRequired(name, definition, freshConnection);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tif (freshConnection.status !== \"connected\") {\n\t\t\t\tthis.reportConnectionFailure(\n\t\t\t\t\tname,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tnew Error(`MCP server ${name} did not return a connected session`),\n\t\t\t\t\t\"reconnect\",\n\t\t\t\t\tfreshConnection,\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tlogger.debug(`Reconnected to ${name}`);\n\t\t\tawait this.publishConnectedMetadata(name, definition, freshConnection);\n\t\t\treturn;\n\t\t}\n\n\t\tif (this.pendingMetadataPublications.has(name)) {\n\t\t\tawait this.publishConnectedMetadata(name, definition, connection);\n\t\t\treturn;\n\t\t}\n\t\tconst hadSessionId = (connection.transport as { sessionId?: string } | undefined)?.sessionId != null;\n\t\tlet refreshResult: Awaited<ReturnType<McpServerManager[\"refreshTools\"]>>;\n\t\ttry {\n\t\t\trefreshResult = await this.manager.refreshTools(name, connection, signal);\n\t\t} catch (error) {\n\t\t\tif (this.stopped || signal?.aborted) return;\n\t\t\tconst current = this.manager.getConnection(name);\n\t\t\tif (current !== connection || connection.status !== \"connected\") {\n\t\t\t\tawait this.handleSupersededConnection(name, definition, connection, signal, retrySuperseded);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tif (!shouldReconnectAfterRefresh(error, hadSessionId)) {\n\t\t\t\tthis.reportConnectionFailure(name, definition, error, \"refresh\", connection);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tif (this.hasPendingAuthForServer(name)) {\n\t\t\t\tlogger.debug(`Skipping reconnect for ${name} while OAuth authorization is pending`);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tlet freshConnection: ServerConnection;\n\t\t\ttry {\n\t\t\t\tfreshConnection = await this.manager.reconnect(name, definition, connection, signal);\n\t\t\t} catch (reconnectError) {\n\t\t\t\tif (this.stopped || signal?.aborted) return;\n\t\t\t\tthis.reportConnectionFailure(\n\t\t\t\t\tname,\n\t\t\t\t\tdefinition,\n\t\t\t\t\treconnectError,\n\t\t\t\t\t\"reconnect\",\n\t\t\t\t\tthis.manager.getConnection(name),\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tif (this.stopped || signal?.aborted) return;\n\t\t\tif (freshConnection.status === \"needs-auth\") {\n\t\t\t\tawait this.notifyAuthRequired(name, definition, freshConnection);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tif (freshConnection.status !== \"connected\") {\n\t\t\t\tthis.reportConnectionFailure(\n\t\t\t\t\tname,\n\t\t\t\t\tdefinition,\n\t\t\t\t\tnew Error(`MCP server ${name} did not return a connected session`),\n\t\t\t\t\t\"reconnect\",\n\t\t\t\t\tfreshConnection,\n\t\t\t\t);\n\t\t\t\treturn;\n\t\t\t}\n\t\t\tlogger.debug(`Reconnected stale MCP session for ${name}`);\n\t\t\tawait this.publishConnectedMetadata(name, definition, freshConnection);\n\t\t\treturn;\n\t\t}\n\n\t\tif (refreshResult === \"superseded\") {\n\t\t\tawait this.handleSupersededConnection(name, definition, connection, signal, retrySuperseded);\n\t\t\treturn;\n\t\t}\n\t\tif (refreshResult === \"refresh-timeout\") {\n\t\t\tthis.deferRefreshTimeout(name, definition, connection);\n\t\t\treturn;\n\t\t}\n\t\tif (this.keepAliveServers.get(name) !== definition) return;\n\t\tif (this.retryStates.delete(name)) {\n\t\t\tawait this.onHealthRestored?.(name);\n\t\t}\n\t}\n\n\tprivate async handleSupersededConnection(\n\t\tname: string,\n\t\tdefinition: ServerDefinition,\n\t\tstaleConnection: ServerConnection,\n\t\tsignal: AbortSignal | undefined,\n\t\tretrySuperseded: boolean,\n\t): Promise<void> {\n\t\tconst current = this.manager.getConnection(name);\n\t\tif (this.keepAliveServers.get(name) !== definition) return;\n\t\tif (current === staleConnection && current.status === \"connected\") {\n\t\t\tif (this.retryStates.delete(name)) await this.onHealthRestored?.(name);\n\t\t\treturn;\n\t\t}\n\t\tif (current?.status === \"connected\") {\n\t\t\tawait this.publishConnectedMetadata(name, definition, current);\n\t\t\treturn;\n\t\t}\n\t\tif (current?.status === \"needs-auth\") {\n\t\t\tawait this.notifyAuthRequired(name, definition, current);\n\t\t\treturn;\n\t\t}\n\t\tif (retrySuperseded) {\n\t\t\tawait this.checkKeepAliveConnection(name, definition, signal, false);\n\t\t}\n\t}\n\n\tprivate async publishConnectedMetadata(\n\t\tname: string,\n\t\tdefinition: ServerDefinition,\n\t\tconnection: ServerConnection,\n\t): Promise<void> {\n\t\t// Fence stale convergence passes by definition identity so disposal, and\n\t\t// disposal followed by a same-name replacement, never re-track this\n\t\t// server. Close the pass's own connection only when it is still current,\n\t\t// so a replacement's connection is never touched.\n\t\tif (this.keepAliveServers.get(name) !== definition) {\n\t\t\tif (this.manager.getConnection(name) === connection) await this.manager.close(name);\n\t\t\treturn;\n\t\t}\n\t\tthis.pendingMetadataPublications.add(name);\n\t\ttry {\n\t\t\tawait this.onReconnect?.(name);\n\t\t\tthis.pendingMetadataPublications.delete(name);\n\t\t\tthis.retryStates.delete(name);\n\t\t} catch (error) {\n\t\t\tif (this.stopped) return;\n\t\t\tthis.reportConnectionFailure(name, definition, error, \"publish\", connection);\n\t\t}\n\t}\n\n\tprivate async notifyAuthRequired(\n\t\tname: string,\n\t\tdefinition: ServerDefinition,\n\t\tconnection: ServerConnection,\n\t): Promise<void> {\n\t\t// Fence stale convergence passes by definition identity. Close the\n\t\t// leftover needs-auth connection only while it is still the manager's\n\t\t// current entry, so a replacement's connection is never touched and the\n\t\t// stale client/transport cannot leak when a replacement connects later.\n\t\tif (this.keepAliveServers.get(name) !== definition) {\n\t\t\tif (this.manager.getConnection(name) === connection) await this.manager.close(name);\n\t\t\treturn;\n\t\t}\n\t\tthis.pendingMetadataPublications.delete(name);\n\t\tthis.retryStates.delete(name);\n\t\ttry {\n\t\t\tawait this.onAuthRequired?.(name);\n\t\t} catch (error) {\n\t\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\t\tlogger.debug(`MCP: auth-required callback failed for ${name}: ${sanitizeTerminalText(message)}`);\n\t\t}\n\t}\n\n\tprivate shouldAttemptConnection(name: string, connection: ServerConnection | undefined): boolean {\n\t\tconst retry = this.retryStates.get(name);\n\t\tif (!retry) return true;\n\t\tif (retry.connection !== connection || retry.status !== connection?.status) {\n\t\t\tthis.retryStates.delete(name);\n\t\t\treturn true;\n\t\t}\n\t\treturn Date.now() >= retry.nextAttemptAt;\n\t}\n\n\tprivate reportConnectionFailure(\n\t\tname: string,\n\t\tdefinition: ServerDefinition,\n\t\terror: unknown,\n\t\taction: \"refresh\" | \"reconnect\" | \"publish\",\n\t\tconnection: ServerConnection | undefined,\n\t): void {\n\t\tif (!this.recordRetry(name, definition, connection)) return;\n\t\tthis.onReconnectFailure?.(name, error);\n\t\tconst message = error instanceof Error ? error.message : String(error);\n\t\tconst target =\n\t\t\taction === \"reconnect\"\n\t\t\t\t? `reconnect to ${name}`\n\t\t\t\t: action === \"publish\"\n\t\t\t\t\t? `publish metadata for ${name}`\n\t\t\t\t\t: `refresh ${name}`;\n\t\tconsole.error(`MCP: Failed to ${target}: ${sanitizeTerminalText(message)}`);\n\t}\n\n\tprivate deferRefreshTimeout(\n\t\tname: string,\n\t\tdefinition: ServerDefinition,\n\t\tconnection: ServerConnection | undefined,\n\t): void {\n\t\tif (!this.recordRetry(name, definition, connection)) return;\n\t\tlogger.debug(`MCP: keep-alive tools/list refresh timed out for ${name}; retrying after backoff`);\n\t}\n\n\tprivate recordRetry(name: string, definition: ServerDefinition, connection: ServerConnection | undefined): boolean {\n\t\t// Do not recreate retry/failure state from a stale convergence pass after\n\t\t// disposal or a same-name replacement registration.\n\t\tif (this.keepAliveServers.get(name) !== definition) return false;\n\t\tconst attempts = (this.retryStates.get(name)?.attempts ?? 0) + 1;\n\t\tconst delay = Math.min(KEEP_ALIVE_RETRY_BASE_MS * 2 ** Math.min(attempts - 1, 10), KEEP_ALIVE_RETRY_MAX_MS);\n\t\tthis.retryStates.set(name, {\n\t\t\tattempts,\n\t\t\tnextAttemptAt: Date.now() + delay,\n\t\t\tconnection,\n\t\t\tstatus: connection?.status,\n\t\t});\n\t\treturn true;\n\t}\n\n\tprivate getIdleTimeout(name: string): number {\n\t\tconst perServer = this.serverSettings.get(name)?.idleTimeout;\n\t\tif (perServer !== undefined) return perServer * 60 * 1000;\n\t\treturn this.globalIdleTimeout;\n\t}\n\n\tasync gracefulShutdown(): Promise<void> {\n\t\tif (this.shutdownPromise) return this.shutdownPromise;\n\t\tthis.shutdownPromise = this.shutdownOnce();\n\t\treturn this.shutdownPromise;\n\t}\n\n\tprivate async shutdownOnce(): Promise<void> {\n\t\tthis.stopped = true;\n\t\tif (this.healthCheckInterval) clearInterval(this.healthCheckInterval);\n\t\tthis.healthCheckInterval = undefined;\n\t\tthis.removeHealthAbortListener?.();\n\t\tthis.removeHealthAbortListener = undefined;\n\t\tawait this.activeHealthCheck;\n\t\tthis.activeHealthCheck = undefined;\n\t\tawait this.activeConvergence;\n\t\tthis.activeConvergence = undefined;\n\t\tthis.onReconnect = undefined;\n\t\tthis.onReconnectFailure = undefined;\n\t\tthis.onHealthRestored = undefined;\n\t\tthis.onAuthRequired = undefined;\n\t\tthis.onIdleShutdown = undefined;\n\t\tthis.retryStates.clear();\n\t\tthis.pendingMetadataPublications.clear();\n\t\tif (typeof this.manager.closeAll === \"function\") {\n\t\t\tawait this.manager.closeAll();\n\t\t}\n\t}\n}\n\nfunction shouldReconnectAfterRefresh(error: unknown, hadSessionId: boolean): boolean {\n\tif (isTerminatedSession(error, hadSessionId)) return true;\n\treturn (\n\t\terror instanceof SdkError &&\n\t\t(error.code === SdkErrorCode.NotConnected || error.code === SdkErrorCode.ConnectionClosed)\n\t);\n}\n"]}

package/dist/extensions/mcp/logger.d.ts

@@ -0,0 +1,52 @@
+/**
+ * Centralized logging for MCP UI operations.
+ * Provides structured, contextual logs with levels.
+ */
+export type LogLevel = "debug" | "info" | "warn" | "error";
+export interface LogContext {
+    server?: string;
+    session?: string;
+    tool?: string;
+    uri?: string;
+    [key: string]: unknown;
+}
+export interface LogEntry {
+    level: LogLevel;
+    message: string;
+    context?: LogContext;
+    error?: Error;
+    timestamp: Date;
+}
+type LogHandler = (entry: LogEntry) => void;
+declare class Logger {
+    private minLevel;
+    private handlers;
+    private defaultContext;
+    setLevel(level: LogLevel): void;
+    setDefaultContext(context: LogContext): void;
+    addHandler(handler: LogHandler): void;
+    clearHandlers(): void;
+    private shouldLog;
+    private emit;
+    debug(message: string, context?: LogContext): void;
+    info(message: string, context?: LogContext): void;
+    warn(message: string, context?: LogContext): void;
+    error(message: string, error?: Error, context?: LogContext): void;
+    /**
+     * Create a child logger with additional default context.
+     */
+    child(context: LogContext): ChildLogger;
+}
+declare class ChildLogger {
+    private parent;
+    private context;
+    constructor(parent: Logger, context: LogContext);
+    debug(message: string, context?: LogContext): void;
+    info(message: string, context?: LogContext): void;
+    warn(message: string, context?: LogContext): void;
+    error(message: string, error?: Error, context?: LogContext): void;
+    child(context: LogContext): ChildLogger;
+}
+export declare const logger: Logger;
+export {};
+//# sourceMappingURL=logger.d.ts.map

package/dist/extensions/mcp/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/extensions/mcp/logger.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAIH,MAAM,MAAM,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;AAE3D,MAAM,WAAW,UAAU;IAC1B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,QAAQ;IACxB,KAAK,EAAE,QAAQ,CAAC;IAChB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,UAAU,CAAC;IACrB,KAAK,CAAC,EAAE,KAAK,CAAC;IACd,SAAS,EAAE,IAAI,CAAC;CAChB;AAED,KAAK,UAAU,GAAG,CAAC,KAAK,EAAE,QAAQ,KAAK,IAAI,CAAC;AAgB5C,cAAM,MAAM;IACX,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,QAAQ,CAAoB;IACpC,OAAO,CAAC,cAAc,CAAkB;IAExC,QAAQ,CAAC,KAAK,EAAE,QAAQ,GAAG,IAAI,CAE9B;IAED,iBAAiB,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,CAE3C;IAED,UAAU,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,CAEpC;IAED,aAAa,IAAI,IAAI,CAEpB;IAED,OAAO,CAAC,SAAS;IAIjB,OAAO,CAAC,IAAI;IAqCZ,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEjD;IAED,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEhD;IAED,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEhD;IAED,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEhE;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,EAAE,UAAU,GAAG,WAAW,CAEtC;CACD;AAED,cAAM,WAAW;IAChB,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,OAAO,CAAa;IAE5B,YAAY,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAG9C;IAED,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEjD;IAED,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEhD;IAED,IAAI,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEhD;IAED,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,UAAU,GAAG,IAAI,CAEhE;IAED,KAAK,CAAC,OAAO,EAAE,UAAU,GAAG,WAAW,CAEtC;CACD;AAeD,eAAO,MAAM,MAAM,QAAe,CAAC","sourcesContent":["/**\n * Centralized logging for MCP UI operations.\n * Provides structured, contextual logs with levels.\n */\n\nimport { sanitizeTerminalText } from \"./utils.ts\";\n\nexport type LogLevel = \"debug\" | \"info\" | \"warn\" | \"error\";\n\nexport interface LogContext {\n\tserver?: string;\n\tsession?: string;\n\ttool?: string;\n\turi?: string;\n\t[key: string]: unknown;\n}\n\nexport interface LogEntry {\n\tlevel: LogLevel;\n\tmessage: string;\n\tcontext?: LogContext;\n\terror?: Error;\n\ttimestamp: Date;\n}\n\ntype LogHandler = (entry: LogEntry) => void;\n\nconst LEVEL_PRIORITY: Record<LogLevel, number> = {\n\tdebug: 0,\n\tinfo: 1,\n\twarn: 2,\n\terror: 3,\n};\n\nconst LEVEL_PREFIX: Record<LogLevel, string> = {\n\tdebug: \"[MCP-UI:DEBUG]\",\n\tinfo: \"[MCP-UI]\",\n\twarn: \"[MCP-UI:WARN]\",\n\terror: \"[MCP-UI:ERROR]\",\n};\n\nclass Logger {\n\tprivate minLevel: LogLevel = \"info\";\n\tprivate handlers: LogHandler[] = [];\n\tprivate defaultContext: LogContext = {};\n\n\tsetLevel(level: LogLevel): void {\n\t\tthis.minLevel = level;\n\t}\n\n\tsetDefaultContext(context: LogContext): void {\n\t\tthis.defaultContext = context;\n\t}\n\n\taddHandler(handler: LogHandler): void {\n\t\tthis.handlers.push(handler);\n\t}\n\n\tclearHandlers(): void {\n\t\tthis.handlers = [];\n\t}\n\n\tprivate shouldLog(level: LogLevel): boolean {\n\t\treturn LEVEL_PRIORITY[level] >= LEVEL_PRIORITY[this.minLevel];\n\t}\n\n\tprivate emit(level: LogLevel, message: string, context?: LogContext, error?: Error): void {\n\t\tif (!this.shouldLog(level)) return;\n\n\t\tconst entry: LogEntry = {\n\t\t\tlevel,\n\t\t\tmessage,\n\t\t\tcontext: { ...this.defaultContext, ...context },\n\t\t\t...(error !== undefined ? { error } : {}),\n\t\t\ttimestamp: new Date(),\n\t\t};\n\n\t\t// Default console output\n\t\tconst prefix = LEVEL_PREFIX[level];\n\t\tconst contextStr = formatContext(entry.context);\n\t\tconst terminalMessage = sanitizeTerminalText(message);\n\t\tconst fullMessage = contextStr ? `${prefix} ${terminalMessage} ${contextStr}` : `${prefix} ${terminalMessage}`;\n\n\t\tif (level === \"error\") {\n\t\t\tconsole.error(fullMessage, error ? sanitizeTerminalText(error.stack ?? error.message) : \"\");\n\t\t} else if (level === \"warn\") {\n\t\t\tconsole.warn(fullMessage);\n\t\t} else if (level === \"debug\") {\n\t\t\tconsole.debug(fullMessage);\n\t\t} else {\n\t\t\tconsole.log(fullMessage);\n\t\t}\n\n\t\t// Custom handlers\n\t\tfor (const handler of this.handlers) {\n\t\t\ttry {\n\t\t\t\thandler(entry);\n\t\t\t} catch {\n\t\t\t\t// Ignore handler errors\n\t\t\t}\n\t\t}\n\t}\n\n\tdebug(message: string, context?: LogContext): void {\n\t\tthis.emit(\"debug\", message, context);\n\t}\n\n\tinfo(message: string, context?: LogContext): void {\n\t\tthis.emit(\"info\", message, context);\n\t}\n\n\twarn(message: string, context?: LogContext): void {\n\t\tthis.emit(\"warn\", message, context);\n\t}\n\n\terror(message: string, error?: Error, context?: LogContext): void {\n\t\tthis.emit(\"error\", message, context, error);\n\t}\n\n\t/**\n\t * Create a child logger with additional default context.\n\t */\n\tchild(context: LogContext): ChildLogger {\n\t\treturn new ChildLogger(this, context);\n\t}\n}\n\nclass ChildLogger {\n\tprivate parent: Logger;\n\tprivate context: LogContext;\n\n\tconstructor(parent: Logger, context: LogContext) {\n\t\tthis.parent = parent;\n\t\tthis.context = context;\n\t}\n\n\tdebug(message: string, context?: LogContext): void {\n\t\tthis.parent.debug(message, { ...this.context, ...context });\n\t}\n\n\tinfo(message: string, context?: LogContext): void {\n\t\tthis.parent.info(message, { ...this.context, ...context });\n\t}\n\n\twarn(message: string, context?: LogContext): void {\n\t\tthis.parent.warn(message, { ...this.context, ...context });\n\t}\n\n\terror(message: string, error?: Error, context?: LogContext): void {\n\t\tthis.parent.error(message, error, { ...this.context, ...context });\n\t}\n\n\tchild(context: LogContext): ChildLogger {\n\t\treturn new ChildLogger(this.parent, { ...this.context, ...context });\n\t}\n}\n\nfunction formatContext(context?: LogContext): string {\n\tif (!context || Object.keys(context).length === 0) return \"\";\n\tconst parts: string[] = [];\n\tfor (const [key, value] of Object.entries(context)) {\n\t\tif (value !== undefined && value !== null) {\n\t\t\tconst rendered = typeof value === \"string\" ? value : (JSON.stringify(value) ?? String(value));\n\t\t\tparts.push(`${sanitizeTerminalText(key)}=${sanitizeTerminalText(rendered)}`);\n\t\t}\n\t}\n\treturn parts.length > 0 ? `(${parts.join(\", \")})` : \"\";\n}\n\n// Singleton instance\nexport const logger = new Logger();\n\n// Enable debug mode via environment variable\nif (process.env.MCP_UI_DEBUG === \"1\" || process.env.MCP_UI_DEBUG === \"true\") {\n\tlogger.setLevel(\"debug\");\n}\n"]}

package/dist/extensions/mcp/logger.js

@@ -0,0 +1,134 @@
+/**
+ * Centralized logging for MCP UI operations.
+ * Provides structured, contextual logs with levels.
+ */
+import { sanitizeTerminalText } from "./utils.js";
+const LEVEL_PRIORITY = {
+    debug: 0,
+    info: 1,
+    warn: 2,
+    error: 3,
+};
+const LEVEL_PREFIX = {
+    debug: "[MCP-UI:DEBUG]",
+    info: "[MCP-UI]",
+    warn: "[MCP-UI:WARN]",
+    error: "[MCP-UI:ERROR]",
+};
+class Logger {
+    minLevel = "info";
+    handlers = [];
+    defaultContext = {};
+    setLevel(level) {
+        this.minLevel = level;
+    }
+    setDefaultContext(context) {
+        this.defaultContext = context;
+    }
+    addHandler(handler) {
+        this.handlers.push(handler);
+    }
+    clearHandlers() {
+        this.handlers = [];
+    }
+    shouldLog(level) {
+        return LEVEL_PRIORITY[level] >= LEVEL_PRIORITY[this.minLevel];
+    }
+    emit(level, message, context, error) {
+        if (!this.shouldLog(level))
+            return;
+        const entry = {
+            level,
+            message,
+            context: { ...this.defaultContext, ...context },
+            ...(error !== undefined ? { error } : {}),
+            timestamp: new Date(),
+        };
+        // Default console output
+        const prefix = LEVEL_PREFIX[level];
+        const contextStr = formatContext(entry.context);
+        const terminalMessage = sanitizeTerminalText(message);
+        const fullMessage = contextStr ? `${prefix} ${terminalMessage} ${contextStr}` : `${prefix} ${terminalMessage}`;
+        if (level === "error") {
+            console.error(fullMessage, error ? sanitizeTerminalText(error.stack ?? error.message) : "");
+        }
+        else if (level === "warn") {
+            console.warn(fullMessage);
+        }
+        else if (level === "debug") {
+            console.debug(fullMessage);
+        }
+        else {
+            console.log(fullMessage);
+        }
+        // Custom handlers
+        for (const handler of this.handlers) {
+            try {
+                handler(entry);
+            }
+            catch {
+                // Ignore handler errors
+            }
+        }
+    }
+    debug(message, context) {
+        this.emit("debug", message, context);
+    }
+    info(message, context) {
+        this.emit("info", message, context);
+    }
+    warn(message, context) {
+        this.emit("warn", message, context);
+    }
+    error(message, error, context) {
+        this.emit("error", message, context, error);
+    }
+    /**
+     * Create a child logger with additional default context.
+     */
+    child(context) {
+        return new ChildLogger(this, context);
+    }
+}
+class ChildLogger {
+    parent;
+    context;
+    constructor(parent, context) {
+        this.parent = parent;
+        this.context = context;
+    }
+    debug(message, context) {
+        this.parent.debug(message, { ...this.context, ...context });
+    }
+    info(message, context) {
+        this.parent.info(message, { ...this.context, ...context });
+    }
+    warn(message, context) {
+        this.parent.warn(message, { ...this.context, ...context });
+    }
+    error(message, error, context) {
+        this.parent.error(message, error, { ...this.context, ...context });
+    }
+    child(context) {
+        return new ChildLogger(this.parent, { ...this.context, ...context });
+    }
+}
+function formatContext(context) {
+    if (!context || Object.keys(context).length === 0)
+        return "";
+    const parts = [];
+    for (const [key, value] of Object.entries(context)) {
+        if (value !== undefined && value !== null) {
+            const rendered = typeof value === "string" ? value : (JSON.stringify(value) ?? String(value));
+            parts.push(`${sanitizeTerminalText(key)}=${sanitizeTerminalText(rendered)}`);
+        }
+    }
+    return parts.length > 0 ? `(${parts.join(", ")})` : "";
+}
+// Singleton instance
+export const logger = new Logger();
+// Enable debug mode via environment variable
+if (process.env.MCP_UI_DEBUG === "1" || process.env.MCP_UI_DEBUG === "true") {
+    logger.setLevel("debug");
+}
+//# sourceMappingURL=logger.js.map