@akon-labs/nexus-cli 0.1.8 → 0.1.10

package/dist/extensions/mcp/init.js

@@ -0,0 +1,631 @@
+import { existsSync } from "node:fs";
+import { throwIfAborted } from "./abort.js";
+import { cloneMcpConfig, loadMcpConfig } from "./config.js";
+import { ConsentManager } from "./consent-manager.js";
+import { McpLifecycleManager } from "./lifecycle.js";
+import { logger } from "./logger.js";
+import { getAuthStorageOptions } from "./mcp-auth.js";
+import { createOAuthRuntime, hasPendingAuth, shutdownOAuth } from "./mcp-auth-flow.js";
+import { publishMcpStatusSnapshot } from "./mcp-status.js";
+import { computeServerHash, createCachedToolSelectorCandidateIndex, getMetadataCachePath, getMissingConfiguredDirectToolServers, isServerCacheValid, loadMetadataCache, reconstructPromptMetadata, reconstructToolMetadata, saveMetadataCache, serializePrompts, serializeResources, serializeTools, } from "./metadata-cache.js";
+import { resourceNameToToolName } from "./resource-tools.js";
+import { combineAbortSignals, createMcpRuntimeOwner, createOwnedUi, isAbortError, } from "./runtime-owner.js";
+import { McpServerManager } from "./server-manager.js";
+import { buildToolMetadata, totalToolCount } from "./tool-metadata.js";
+import { formatToolName, isServerDisabled, resolveToolPrefix, } from "./types.js";
+import { UiResourceHandler } from "./ui-resource-handler.js";
+import { formatMcpStatus, openUrl, parallelLimit, sanitizeTerminalText } from "./utils.js";
+const FAILURE_BACKOFF_MS = 60 * 1000;
+const MAX_FAILURE_MESSAGE_CHARS = 8 * 1024;
+const failureExpiryTimers = new WeakMap();
+function getFailureExpiryTimers(state) {
+    let timers = failureExpiryTimers.get(state);
+    if (!timers) {
+        timers = new Map();
+        failureExpiryTimers.set(state, timers);
+    }
+    return timers;
+}
+export function clearFailure(state, serverName) {
+    state.failureTracker.delete(serverName);
+    state.failureMessages?.delete(serverName);
+    const timers = failureExpiryTimers.get(state);
+    const timer = timers?.get(serverName);
+    if (timer)
+        clearTimeout(timer);
+    timers?.delete(serverName);
+}
+export function recordFailure(state, serverName, message) {
+    clearFailure(state, serverName);
+    const failedAt = Date.now();
+    state.failureTracker.set(serverName, failedAt);
+    state.failureMessages?.set(serverName, message.slice(0, MAX_FAILURE_MESSAGE_CHARS));
+    const timer = setTimeout(() => {
+        if (!state.owner.isActive()) {
+            getFailureExpiryTimers(state).delete(serverName);
+            return;
+        }
+        if (state.failureTracker.get(serverName) === failedAt) {
+            state.failureTracker.delete(serverName);
+            state.failureMessages?.delete(serverName);
+            publishMcpStatusSnapshot(state);
+        }
+        getFailureExpiryTimers(state).delete(serverName);
+    }, FAILURE_BACKOFF_MS);
+    timer.unref?.();
+    getFailureExpiryTimers(state).set(serverName, timer);
+}
+export function isTuiMode(ctx) {
+    return ctx.hasUI && ctx.mode === "tui";
+}
+export async function initializeMcp(nexus, ctx, owner = createMcpRuntimeOwner(), options = {}) {
+    // Nexus guards ExtensionContext getters after reload. Snapshot all values that
+    // can be used by asynchronous work before the first await.
+    const configPath = options.config !== undefined
+        ? undefined
+        : (options.configPath ?? nexus.getFlag("mcp-config"));
+    const cwd = ctx.cwd;
+    const hasUI = ctx.hasUI;
+    const mode = ctx.mode;
+    const rawUi = hasUI ? ctx.ui : undefined;
+    const modelRegistry = ctx.modelRegistry;
+    const initialModel = ctx.model;
+    const initialSignal = ctx.signal;
+    const ui = rawUi ? createOwnedUi(rawUi, owner) : undefined;
+    const runtimeSignal = combineAbortSignals(owner.signal, initialSignal);
+    const config = options.config !== undefined
+        ? cloneMcpConfig(options.config)
+        : loadMcpConfig(configPath, cwd, ctx.isProjectTrusted());
+    const authStorageOptions = getAuthStorageOptions(config.settings?.oauthDir, cwd);
+    const ownsOAuthRuntime = options.oauthRuntime === undefined;
+    const oauthRuntime = options.oauthRuntime ?? createOAuthRuntime(owner.signal);
+    const manager = new McpServerManager(cwd);
+    manager.setRuntimeSignal?.(owner.signal);
+    manager.setOAuthRuntime?.(oauthRuntime);
+    manager.setDefaultRequestTimeoutMs(config.settings?.requestTimeoutMs);
+    manager.setTraceConfig?.(config.settings?.trace);
+    manager.setAuthStorageOptions(authStorageOptions);
+    const samplingAutoApprove = config.settings?.samplingAutoApprove === true;
+    if (config.settings?.sampling !== false && (hasUI || samplingAutoApprove)) {
+        manager.setSamplingConfig({
+            autoApprove: samplingAutoApprove,
+            ...(ui !== undefined ? { ui } : {}),
+            modelRegistry,
+            getCurrentModel: options.getCurrentModel ?? (() => initialModel),
+            getSignal: () => owner.signal,
+        });
+    }
+    const elicitationEnabled = config.settings?.elicitation !== false && hasUI;
+    if (elicitationEnabled && ui) {
+        manager.setElicitationConfig({
+            ui,
+            allowUrl: mode === "tui",
+        });
+    }
+    const lifecycle = new McpLifecycleManager(manager, (serverName) => hasPendingAuth(serverName, undefined, oauthRuntime));
+    const toolMetadata = new Map();
+    const resourceCounts = new Map();
+    const promptMetadata = new Map();
+    const promptMetadataLive = new Set();
+    const serverInstructions = new Map();
+    const failureTracker = new Map();
+    const failureMessages = new Map();
+    const approvedToolCalls = new Map();
+    const uiResourceHandler = new UiResourceHandler(manager, config);
+    const consentManager = new ConsentManager("once-per-server");
+    const state = {
+        owner,
+        manager,
+        lifecycle,
+        toolMetadata,
+        resourceCounts,
+        promptMetadata,
+        promptMetadataLive,
+        serverInstructions,
+        config,
+        programmaticConfig: options.config !== undefined,
+        oauthRuntime,
+        authStorageOptions,
+        failureTracker,
+        failureMessages,
+        approvedToolCalls,
+        approvalEvents: nexus.events,
+        uiResourceHandler,
+        consentManager,
+        uiServer: null,
+        completedUiSessions: [],
+        openBrowser: async (url) => {
+            owner.throwIfInactive();
+            await openUrl(nexus, url, process.env.BROWSER, owner.signal);
+            owner.throwIfInactive();
+        },
+        ...(ui !== undefined ? { ui } : {}),
+        sendMessage: (message, options) => {
+            const deliver = () => {
+                if (!owner.isActive())
+                    return;
+                nexus.sendMessage(message, options);
+            };
+            if (!options?.triggerTurn) {
+                deliver();
+                return;
+            }
+            void lifecycle.ensureConverged(owner.signal).then(deliver, (error) => {
+                if (!owner.isActive() || isAbortError(error, owner.signal))
+                    return;
+                const detail = error instanceof Error ? error.message : String(error);
+                logger.debug(`MCP: pre-turn keep-alive convergence failed: ${sanitizeTerminalText(detail)}`);
+                deliver();
+            });
+        },
+        ...(options.statusEvents !== undefined ? { statusEvents: options.statusEvents } : {}),
+    };
+    if (ownsOAuthRuntime)
+        owner.addCleanup(() => shutdownOAuth(oauthRuntime));
+    manager.setMetadataListChangedListener?.((serverName, reason) => {
+        if (!owner.isActive())
+            return;
+        updateServerMetadata(state, serverName);
+        updateMetadataCache(state, serverName, { preserveEmptyResources: false });
+        notifyToolMetadataUpdated(state, serverName, reason);
+        updateStatusBar(state);
+    });
+    owner.addCleanup(() => lifecycle.gracefulShutdown());
+    owner.addCleanup(() => {
+        if (state.uiServer) {
+            state.uiServer.close("runtime_owner_stopped");
+            state.uiServer = null;
+        }
+    });
+    const allServerEntries = Object.entries(config.mcpServers);
+    const serverEntries = allServerEntries.filter(([, definition]) => !isServerDisabled(definition));
+    if (serverEntries.length === 0) {
+        if (allServerEntries.length > 0 && hasUI) {
+            ui?.notify(`MCP: All ${allServerEntries.length} server(s) are disabled`, "info");
+        }
+        publishMcpStatusSnapshot(state);
+        return state;
+    }
+    const idleSetting = typeof config.settings?.idleTimeout === "number" ? config.settings.idleTimeout : 10;
+    lifecycle.setGlobalIdleTimeout(idleSetting);
+    const cachePath = getMetadataCachePath();
+    const cacheFileExists = existsSync(cachePath);
+    let cache = loadMetadataCache();
+    let bootstrapAll = false;
+    if (!cacheFileExists) {
+        bootstrapAll = true;
+        saveMetadataCache({ version: 1, servers: {} });
+    }
+    else if (!cache) {
+        cache = { version: 1, servers: {} };
+        saveMetadataCache(cache);
+    }
+    const prefix = config.settings?.toolPrefix ?? "server";
+    let cachedSelectorCandidateIndex;
+    for (const [name, definition] of serverEntries) {
+        const lifecycleMode = definition.lifecycle ?? "lazy";
+        const persistsAfterFirstSpawn = lifecycleMode === "eager" || lifecycleMode === "lazy-keep-alive";
+        const idleOverride = definition.idleTimeout ?? (persistsAfterFirstSpawn ? 0 : undefined);
+        lifecycle.registerServer(name, definition, idleOverride !== undefined ? { idleTimeout: idleOverride } : undefined);
+        if (lifecycleMode === "keep-alive") {
+            lifecycle.markKeepAlive(name, definition);
+        }
+        const cachedEntry = cache?.servers?.[name];
+        if (cachedEntry && isServerCacheValid(cachedEntry, definition)) {
+            const hasToolFilters = (Array.isArray(definition.includeTools) && definition.includeTools.length > 0) ||
+                (Array.isArray(definition.excludeTools) && definition.excludeTools.length > 0);
+            if (hasToolFilters && !cachedSelectorCandidateIndex && cache) {
+                cachedSelectorCandidateIndex = createCachedToolSelectorCandidateIndex(config.mcpServers, cache, prefix);
+            }
+            const metadata = reconstructToolMetadata(name, cachedEntry, prefix, definition, config.mcpServers, cache ?? undefined, cachedSelectorCandidateIndex);
+            toolMetadata.set(name, metadata);
+            if (Array.isArray(cachedEntry.resources)) {
+                resourceCounts.set(name, cachedEntry.resources.length);
+            }
+            if (cachedEntry.prompts?.length) {
+                promptMetadata.set(name, reconstructPromptMetadata(name, cachedEntry.prompts ?? [], prefix, definition));
+            }
+            if (cachedEntry.instructions) {
+                serverInstructions.set(name, cachedEntry.instructions);
+            }
+        }
+    }
+    const startupServers = bootstrapAll
+        ? serverEntries
+        : serverEntries.filter(([, definition]) => {
+            const mode = definition.lifecycle ?? "lazy";
+            return mode === "keep-alive" || mode === "eager";
+        });
+    if (ui && startupServers.length > 0) {
+        const status = formatMcpStatus(state.config, `connecting to ${startupServers.length} servers...`);
+        ui.setStatus("mcp", status);
+    }
+    const results = await parallelLimit(startupServers, 10, async ([name, definition]) => {
+        try {
+            const connection = await manager.connect(name, definition, runtimeSignal);
+            if (connection.status === "needs-auth") {
+                return {
+                    name,
+                    definition,
+                    connection: null,
+                    error: `OAuth authentication required. Run /mcp-auth ${name}.`,
+                };
+            }
+            return { name, definition, connection, error: null };
+        }
+        catch (error) {
+            if (isAbortError(error, runtimeSignal)) {
+                if (owner.signal.aborted)
+                    throw error;
+                return { name, definition, connection: null, error: null };
+            }
+            const message = error instanceof Error ? error.message : String(error);
+            return { name, definition, connection: null, error: message };
+        }
+    });
+    if (initialSignal?.aborted)
+        return state;
+    owner.throwIfInactive();
+    const startupKnownMetadata = new Map();
+    for (const { name, definition, connection } of results) {
+        if (!connection)
+            continue;
+        const effectivePrefix = resolveToolPrefix(definition, prefix);
+        const metadata = [
+            ...connection.tools
+                .filter((tool) => tool?.name)
+                .map((tool) => ({
+                name: formatToolName(tool.name, name, effectivePrefix),
+                originalName: tool.name,
+                description: tool.description ?? "",
+            })),
+            ...(definition.exposeResources !== false
+                ? connection.resources
+                    .filter((resource) => resource?.name && resource?.uri)
+                    .map((resource) => {
+                    const originalName = `read_${resourceNameToToolName(resource.name)}`;
+                    return {
+                        name: formatToolName(originalName, name, effectivePrefix),
+                        originalName,
+                        description: resource.description ?? `Read resource: ${resource.uri}`,
+                        resourceUri: resource.uri,
+                    };
+                })
+                : []),
+        ];
+        startupKnownMetadata.set(name, metadata);
+    }
+    for (const { name, definition, connection, error } of results) {
+        owner.throwIfInactive();
+        if (error || !connection) {
+            if (initialSignal?.aborted)
+                continue;
+            if (error)
+                recordFailure(state, name, error);
+            const displayError = sanitizeTerminalText(error ?? "Unknown connection failure");
+            if (ui) {
+                ui.notify(`MCP: Failed to connect to ${name}: ${displayError}`, "error");
+            }
+            console.error(`MCP: Failed to connect to ${name}: ${displayError}`);
+            continue;
+        }
+        const { metadata, failedTools } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix, config.mcpServers, startupKnownMetadata, true);
+        toolMetadata.set(name, metadata);
+        resourceCounts.set(name, connection.resources.length);
+        if (!connection.promptDiscoveryFailed) {
+            promptMetadata.set(name, reconstructPromptMetadata(name, connection.prompts ?? [], prefix, definition));
+            promptMetadataLive.add(name);
+        }
+        if (connection.instructions) {
+            serverInstructions.set(name, connection.instructions);
+        }
+        else {
+            serverInstructions.delete(name);
+        }
+        updateMetadataCache(state, name);
+        notifyToolMetadataUpdated(state, name, "startup");
+        markKeepAliveAfterConnect(state, name);
+        if (failedTools.length > 0 && ui) {
+            ui.notify(`MCP: ${name} - ${failedTools.length} tools skipped`, "warning");
+        }
+    }
+    const connectedCount = results.filter((r) => r.connection).length;
+    const failedCount = results.filter((r) => r.error).length;
+    if (ui && connectedCount > 0 && config.settings?.notifyOnStartupConnect !== false) {
+        const totalTools = totalToolCount(state);
+        const msg = failedCount > 0
+            ? `MCP: ${connectedCount}/${startupServers.length} servers connected (${totalTools} tools)`
+            : `MCP: ${connectedCount} servers connected (${totalTools} tools)`;
+        ui.notify(msg, "info");
+    }
+    const envDirect = process.env.MCP_DIRECT_TOOLS;
+    if (envDirect !== "__none__") {
+        const currentCache = loadMetadataCache();
+        const envDirectToolOverride = envDirect
+            ?.split(",")
+            .map((selector) => selector.trim())
+            .filter(Boolean);
+        const missingCacheServers = getMissingConfiguredDirectToolServers(config, currentCache, envDirectToolOverride);
+        if (missingCacheServers.length > 0) {
+            const bootstrapResults = await parallelLimit(missingCacheServers.filter((name) => !results.some((r) => r.name === name && r.connection)), 10, async (name) => {
+                try {
+                    const definition = config.mcpServers[name];
+                    if (!definition)
+                        throw new Error(`MCP server "${name}" is not configured`);
+                    const connection = await manager.connect(name, definition, runtimeSignal);
+                    if (connection.status === "needs-auth") {
+                        return { name, ok: false };
+                    }
+                    updateServerMetadata(state, name);
+                    updateMetadataCache(state, name);
+                    notifyToolMetadataUpdated(state, name, "direct-tools-bootstrap");
+                    markKeepAliveAfterConnect(state, name);
+                    clearFailure(state, name);
+                    return { name, ok: true };
+                }
+                catch (error) {
+                    if (isAbortError(error, runtimeSignal)) {
+                        if (owner.signal.aborted)
+                            throw error;
+                        return { name, ok: false };
+                    }
+                    const message = error instanceof Error ? error.message : String(error);
+                    recordFailure(state, name, message);
+                    logger.debug(`MCP: direct-tools bootstrap failed for ${name}: ${sanitizeTerminalText(message)}`);
+                    return { name, ok: false };
+                }
+            });
+            const bootstrapped = bootstrapResults.filter((r) => r.ok).map((r) => r.name);
+            owner.throwIfInactive();
+            if (bootstrapped.length > 0 && ui) {
+                ui.notify(`MCP: direct tools for ${bootstrapped.join(", ")} will be available after restart`, "info");
+            }
+        }
+    }
+    lifecycle.setReconnectCallback((serverName) => {
+        if (!owner.isActive())
+            return;
+        updateServerMetadata(state, serverName);
+        updateMetadataCache(state, serverName);
+        notifyToolMetadataUpdated(state, serverName, "lifecycle-reconnect");
+        clearFailure(state, serverName);
+        updateStatusBar(state);
+    });
+    lifecycle.setReconnectFailureCallback((serverName, error) => {
+        if (!owner.isActive())
+            return;
+        const message = error instanceof Error ? error.message : String(error);
+        recordFailure(state, serverName, message);
+        updateStatusBar(state);
+    });
+    lifecycle.setHealthRestoredCallback((serverName) => {
+        if (!owner.isActive())
+            return;
+        clearFailure(state, serverName);
+        updateStatusBar(state);
+    });
+    lifecycle.setAuthRequiredCallback((serverName) => {
+        if (!owner.isActive())
+            return;
+        clearFailure(state, serverName);
+        updateStatusBar(state);
+    });
+    lifecycle.setIdleShutdownCallback((serverName) => {
+        if (!owner.isActive())
+            return;
+        const idleMinutes = getEffectiveIdleTimeoutMinutes(state, serverName);
+        logger.debug(`${serverName} shut down (idle ${idleMinutes}m)`);
+        updateStatusBar(state);
+    });
+    owner.throwIfInactive();
+    lifecycle.startHealthChecks(runtimeSignal);
+    if (config.settings?.mcpFooterStatus === "off") {
+        ui?.setStatus("mcp", undefined);
+    }
+    publishMcpStatusSnapshot(state);
+    return state;
+}
+export function markKeepAliveAfterConnect(state, serverName) {
+    const definition = state.config.mcpServers[serverName];
+    if (!definition || isServerDisabled(definition))
+        return;
+    if ((definition.lifecycle ?? "lazy") === "lazy-keep-alive") {
+        state.lifecycle.markKeepAlive(serverName, definition);
+    }
+}
+export function updateServerMetadata(state, serverName) {
+    const connection = state.manager.getConnection(serverName);
+    if (!connection || connection.status !== "connected")
+        return;
+    const definition = state.config.mcpServers[serverName];
+    if (!definition)
+        return;
+    if (isServerDisabled(definition)) {
+        state.toolMetadata.delete(serverName);
+        state.resourceCounts?.delete(serverName);
+        state.promptMetadata?.delete(serverName);
+        state.promptMetadataLive?.delete(serverName);
+        state.serverInstructions.delete(serverName);
+        return;
+    }
+    const prefix = state.config.settings?.toolPrefix ?? "server";
+    const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, serverName, prefix, state.config.mcpServers, state.toolMetadata);
+    state.toolMetadata.set(serverName, metadata);
+    state.resourceCounts?.set(serverName, connection.resources.length);
+    if (!connection.promptDiscoveryFailed) {
+        state.promptMetadata?.set(serverName, reconstructPromptMetadata(serverName, connection.prompts ?? [], prefix, definition));
+        state.promptMetadataLive?.add(serverName);
+    }
+    if (connection.instructions) {
+        state.serverInstructions?.set(serverName, connection.instructions);
+    }
+    else {
+        state.serverInstructions?.delete(serverName);
+    }
+}
+export function updateMetadataCache(state, serverName, options = {}) {
+    const connection = state.manager.getConnection(serverName);
+    if (!connection || connection.status !== "connected")
+        return;
+    const definition = state.config.mcpServers[serverName];
+    if (!definition || isServerDisabled(definition))
+        return;
+    const configHash = computeServerHash(definition);
+    const existing = loadMetadataCache();
+    const existingEntry = existing?.servers?.[serverName];
+    const tools = serializeTools(connection.tools);
+    let resources = definition.exposeResources === false ? [] : serializeResources(connection.resources);
+    const prompts = connection.promptDiscoveryFailed
+        ? existingEntry?.configHash === configHash
+            ? existingEntry.prompts
+            : undefined
+        : serializePrompts(connection.prompts ?? []);
+    if (definition.exposeResources !== false &&
+        resources.length === 0 &&
+        existingEntry?.resources?.length &&
+        existingEntry.configHash === configHash &&
+        options.preserveEmptyResources !== false) {
+        resources = existingEntry.resources;
+    }
+    const entry = {
+        configHash,
+        tools,
+        resources,
+        ...(prompts !== undefined ? { prompts } : {}),
+        ...(connection.instructions !== undefined ? { instructions: connection.instructions } : {}),
+        cachedAt: Date.now(),
+    };
+    saveMetadataCache({ version: 1, servers: { [serverName]: entry } });
+}
+export function notifyToolMetadataUpdated(state, serverName, reason) {
+    try {
+        const result = state.onToolMetadataUpdated?.(serverName, reason);
+        if (result && typeof result.catch === "function") {
+            result.catch((error) => {
+                const message = error instanceof Error ? error.message : String(error);
+                logger.debug(`MCP: metadata update hook failed for ${serverName}: ${message}`);
+            });
+        }
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        logger.debug(`MCP: metadata update hook failed for ${serverName}: ${message}`);
+    }
+}
+export function flushMetadataCache(state) {
+    for (const [name, connection] of state.manager.getAllConnections()) {
+        if (connection.status === "connected") {
+            updateMetadataCache(state, name);
+        }
+    }
+}
+export function updateStatusBar(state) {
+    publishMcpStatusSnapshot(state);
+    const ui = state.ui;
+    if (!ui)
+        return;
+    const entries = Object.entries(state.config.mcpServers);
+    const disabledCount = entries.filter(([, definition]) => isServerDisabled(definition)).length;
+    const enabledCount = entries.length - disabledCount;
+    if (entries.length === 0) {
+        ui.setStatus("mcp", undefined);
+        return;
+    }
+    const connectedCount = [...state.manager.getAllConnections()].filter(([name, connection]) => {
+        const definition = state.config.mcpServers[name];
+        return connection.status === "connected" && definition !== undefined && !isServerDisabled(definition);
+    }).length;
+    const footerStatus = state.config.settings?.mcpFooterStatus ?? "full";
+    if (footerStatus === "off") {
+        ui.setStatus("mcp", undefined);
+        return;
+    }
+    let status = footerStatus === "compact"
+        ? `MCP ${connectedCount}/${enabledCount}`
+        : `${enabledCount} ${enabledCount === 1 ? "server" : "servers"} enabled`;
+    if (footerStatus === "full") {
+        if (connectedCount > 0)
+            status += ` (${connectedCount} connected)`;
+        if (disabledCount > 0)
+            status += ` (${disabledCount} disabled)`;
+    }
+    const formattedStatus = footerStatus === "compact" ? status : formatMcpStatus(state.config, status);
+    if (formattedStatus === undefined) {
+        ui.setStatus("mcp", undefined);
+        return;
+    }
+    ui.setStatus("mcp", ui.theme ? ui.theme.fg("accent", formattedStatus) : formattedStatus);
+}
+export function getFailureAgeSeconds(state, serverName) {
+    const failedAt = state.failureTracker.get(serverName);
+    if (!failedAt)
+        return null;
+    const ageMs = Date.now() - failedAt;
+    if (ageMs > FAILURE_BACKOFF_MS)
+        return null;
+    return Math.round(ageMs / 1000);
+}
+export function getFailureMessage(state, serverName) {
+    if (getFailureAgeSeconds(state, serverName) === null)
+        return null;
+    return state.failureMessages?.get(serverName) ?? null;
+}
+export async function lazyConnect(state, serverName, signal) {
+    const ownedSignal = combineAbortSignals(state.owner?.signal, signal);
+    throwIfAborted(ownedSignal);
+    const connection = state.manager.getConnection(serverName);
+    if (connection?.status === "needs-auth") {
+        return false;
+    }
+    if (connection?.status === "connected") {
+        updateServerMetadata(state, serverName);
+        markKeepAliveAfterConnect(state, serverName);
+        return true;
+    }
+    const failedAgo = getFailureAgeSeconds(state, serverName);
+    if (failedAgo !== null)
+        return false;
+    const definition = state.config.mcpServers[serverName];
+    if (!definition || isServerDisabled(definition))
+        return false;
+    try {
+        if (state.ui) {
+            const status = formatMcpStatus(state.config, `connecting to ${serverName}...`);
+            state.ui.setStatus("mcp", status);
+        }
+        const newConnection = await state.manager.connect(serverName, definition, ownedSignal);
+        if (newConnection.status === "needs-auth") {
+            return false;
+        }
+        clearFailure(state, serverName);
+        updateServerMetadata(state, serverName);
+        updateMetadataCache(state, serverName);
+        notifyToolMetadataUpdated(state, serverName, "lazy-connect");
+        markKeepAliveAfterConnect(state, serverName);
+        updateStatusBar(state);
+        return true;
+    }
+    catch (error) {
+        if (isAbortError(error, ownedSignal)) {
+            throwIfAborted(ownedSignal);
+        }
+        const message = error instanceof Error ? error.message : String(error);
+        recordFailure(state, serverName, message);
+        logger.debug(`MCP: lazy connect failed for ${serverName}: ${sanitizeTerminalText(message)}`);
+        updateStatusBar(state);
+        return false;
+    }
+}
+function getEffectiveIdleTimeoutMinutes(state, serverName) {
+    const definition = state.config.mcpServers[serverName];
+    if (!definition) {
+        return typeof state.config.settings?.idleTimeout === "number" ? state.config.settings.idleTimeout : 10;
+    }
+    if (typeof definition.idleTimeout === "number")
+        return definition.idleTimeout;
+    const mode = definition.lifecycle ?? "lazy";
+    if (mode === "eager" || mode === "lazy-keep-alive")
+        return 0;
+    return typeof state.config.settings?.idleTimeout === "number" ? state.config.settings.idleTimeout : 10;
+}
+//# sourceMappingURL=init.js.map