@ashx-j/lunr 0.1.0

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

@@ -0,0 +1,296 @@
+import { existsSync } from "node:fs";
+import { loadMcpConfig } from "./config.js";
+import { ConsentManager } from "./consent-manager.js";
+import { McpLifecycleManager } from "./lifecycle.js";
+import { computeServerHash, getMetadataCachePath, isServerCacheValid, loadMetadataCache, reconstructToolMetadata, saveMetadataCache, serializeResources, serializeTools, } from "./metadata-cache.js";
+import { McpServerManager } from "./server-manager.js";
+import { buildToolMetadata, totalToolCount } from "./tool-metadata.js";
+import { UiResourceHandler } from "./ui-resource-handler.js";
+import { openUrl, parallelLimit } from "./utils.js";
+import { logger } from "./logger.js";
+import { getMissingConfiguredDirectToolServers } from "./direct-tools.js";
+import { throwIfAborted } from "./abort.js";
+const FAILURE_BACKOFF_MS = 60 * 1000;
+export function isTuiMode(ctx) {
+    return ctx.hasUI && ctx.mode === "tui";
+}
+export async function initializeMcp(pi, ctx) {
+    const configPath = pi.getFlag("mcp-config");
+    const config = loadMcpConfig(configPath, ctx.cwd);
+    const manager = new McpServerManager(ctx.cwd);
+    manager.setDefaultRequestTimeoutMs(config.settings?.requestTimeoutMs);
+    const samplingAutoApprove = config.settings?.samplingAutoApprove === true;
+    if (config.settings?.sampling !== false && (ctx.hasUI || samplingAutoApprove)) {
+        manager.setSamplingConfig({
+            autoApprove: samplingAutoApprove,
+            ui: ctx.hasUI ? ctx.ui : undefined,
+            modelRegistry: ctx.modelRegistry,
+            getCurrentModel: () => ctx.model,
+            getSignal: () => ctx.signal,
+        });
+    }
+    const elicitationEnabled = config.settings?.elicitation !== false && ctx.hasUI;
+    if (elicitationEnabled) {
+        manager.setElicitationConfig({
+            ui: ctx.ui,
+            allowUrl: isTuiMode(ctx),
+        });
+    }
+    const lifecycle = new McpLifecycleManager(manager);
+    const toolMetadata = new Map();
+    const failureTracker = new Map();
+    const uiResourceHandler = new UiResourceHandler(manager);
+    const consentManager = new ConsentManager("once-per-server");
+    const ui = ctx.hasUI ? ctx.ui : undefined;
+    const state = {
+        manager,
+        lifecycle,
+        toolMetadata,
+        config,
+        failureTracker,
+        uiResourceHandler,
+        consentManager,
+        uiServer: null,
+        completedUiSessions: [],
+        openBrowser: (url) => openUrl(pi, url, process.env.BROWSER),
+        ui,
+        sendMessage: (message, options) => pi.sendMessage(message, options),
+    };
+    const serverEntries = Object.entries(config.mcpServers);
+    if (serverEntries.length === 0) {
+        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";
+    for (const [name, definition] of serverEntries) {
+        const lifecycleMode = definition.lifecycle ?? "lazy";
+        const idleOverride = definition.idleTimeout ?? (lifecycleMode === "eager" ? 0 : undefined);
+        lifecycle.registerServer(name, definition, idleOverride !== undefined ? { idleTimeout: idleOverride } : undefined);
+        if (lifecycleMode === "keep-alive") {
+            lifecycle.markKeepAlive(name, definition);
+        }
+        if (cache?.servers?.[name] && isServerCacheValid(cache.servers[name], definition)) {
+            const metadata = reconstructToolMetadata(name, cache.servers[name], prefix, definition);
+            toolMetadata.set(name, metadata);
+        }
+    }
+    const startupServers = bootstrapAll
+        ? serverEntries
+        : serverEntries.filter(([, definition]) => {
+            const mode = definition.lifecycle ?? "lazy";
+            return mode === "keep-alive" || mode === "eager";
+        });
+    if (ctx.hasUI && startupServers.length > 0) {
+        ctx.ui.setStatus("mcp", `MCP: connecting to ${startupServers.length} servers...`);
+    }
+    const results = await parallelLimit(startupServers, 10, async ([name, definition]) => {
+        try {
+            const connection = await manager.connect(name, definition, ctx.signal);
+            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) {
+            const message = error instanceof Error ? error.message : String(error);
+            return { name, definition, connection: null, error: message };
+        }
+    });
+    for (const { name, definition, connection, error } of results) {
+        if (error || !connection) {
+            if (ctx.hasUI) {
+                ctx.ui.notify(`MCP: Failed to connect to ${name}: ${error}`, "error");
+            }
+            console.error(`MCP: Failed to connect to ${name}: ${error}`);
+            continue;
+        }
+        const { metadata, failedTools } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix);
+        toolMetadata.set(name, metadata);
+        updateMetadataCache(state, name);
+        if (failedTools.length > 0 && ctx.hasUI) {
+            ctx.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 (ctx.hasUI && connectedCount > 0) {
+        const totalTools = totalToolCount(state);
+        const msg = failedCount > 0
+            ? `MCP: ${connectedCount}/${startupServers.length} servers connected (${totalTools} tools)`
+            : `MCP: ${connectedCount} servers connected (${totalTools} tools)`;
+        ctx.ui.notify(msg, "info");
+    }
+    const envDirect = process.env.MCP_DIRECT_TOOLS;
+    if (envDirect !== "__none__") {
+        const currentCache = loadMetadataCache();
+        const missingCacheServers = getMissingConfiguredDirectToolServers(config, currentCache);
+        if (missingCacheServers.length > 0) {
+            const bootstrapResults = await parallelLimit(missingCacheServers.filter(name => !results.some(r => r.name === name && r.connection)), 10, async (name) => {
+                const definition = config.mcpServers[name];
+                try {
+                    const connection = await manager.connect(name, definition, ctx.signal);
+                    if (connection.status === "needs-auth") {
+                        return { name, ok: false };
+                    }
+                    const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix);
+                    toolMetadata.set(name, metadata);
+                    updateMetadataCache(state, name);
+                    return { name, ok: true };
+                }
+                catch (error) {
+                    const message = error instanceof Error ? error.message : String(error);
+                    logger.debug(`MCP: direct-tools bootstrap failed for ${name}: ${message}`);
+                    return { name, ok: false };
+                }
+            });
+            const bootstrapped = bootstrapResults.filter(r => r.ok).map(r => r.name);
+            if (bootstrapped.length > 0 && ctx.hasUI) {
+                ctx.ui.notify(`MCP: direct tools for ${bootstrapped.join(", ")} will be available after restart`, "info");
+            }
+        }
+    }
+    lifecycle.setReconnectCallback((serverName) => {
+        updateServerMetadata(state, serverName);
+        updateMetadataCache(state, serverName);
+        state.failureTracker.delete(serverName);
+        updateStatusBar(state);
+    });
+    lifecycle.setIdleShutdownCallback((serverName) => {
+        const idleMinutes = getEffectiveIdleTimeoutMinutes(state, serverName);
+        logger.debug(`${serverName} shut down (idle ${idleMinutes}m)`);
+        updateStatusBar(state);
+    });
+    lifecycle.startHealthChecks();
+    return state;
+}
+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;
+    const prefix = state.config.settings?.toolPrefix ?? "server";
+    const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, serverName, prefix);
+    state.toolMetadata.set(serverName, metadata);
+}
+export function updateMetadataCache(state, serverName) {
+    const connection = state.manager.getConnection(serverName);
+    if (!connection || connection.status !== "connected")
+        return;
+    const definition = state.config.mcpServers[serverName];
+    if (!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);
+    if (definition.exposeResources !== false &&
+        resources.length === 0 &&
+        existingEntry?.resources?.length &&
+        existingEntry.configHash === configHash) {
+        resources = existingEntry.resources;
+    }
+    const entry = {
+        configHash,
+        tools,
+        resources,
+        cachedAt: Date.now(),
+    };
+    saveMetadataCache({ version: 1, servers: { [serverName]: entry } });
+}
+export function flushMetadataCache(state) {
+    for (const [name, connection] of state.manager.getAllConnections()) {
+        if (connection.status === "connected") {
+            updateMetadataCache(state, name);
+        }
+    }
+}
+export function updateStatusBar(state) {
+    const ui = state.ui;
+    if (!ui)
+        return;
+    const total = Object.keys(state.config.mcpServers).length;
+    if (total === 0) {
+        ui.setStatus("mcp", undefined);
+        return;
+    }
+    const connectedCount = state.manager.getAllConnections().size;
+    ui.setStatus("mcp", ui.theme.fg("accent", `MCP: ${connectedCount}/${total} servers`));
+}
+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 async function lazyConnect(state, serverName, signal) {
+    const connection = state.manager.getConnection(serverName);
+    if (connection?.status === "needs-auth") {
+        return false;
+    }
+    if (connection?.status === "connected") {
+        updateServerMetadata(state, serverName);
+        return true;
+    }
+    const failedAgo = getFailureAgeSeconds(state, serverName);
+    if (failedAgo !== null)
+        return false;
+    const definition = state.config.mcpServers[serverName];
+    if (!definition)
+        return false;
+    try {
+        if (state.ui) {
+            state.ui.setStatus("mcp", `MCP: connecting to ${serverName}...`);
+        }
+        const newConnection = await state.manager.connect(serverName, definition, signal);
+        if (newConnection.status === "needs-auth") {
+            return false;
+        }
+        state.failureTracker.delete(serverName);
+        updateServerMetadata(state, serverName);
+        updateMetadataCache(state, serverName);
+        updateStatusBar(state);
+        return true;
+    }
+    catch (error) {
+        if (signal?.aborted) {
+            throwIfAborted(signal);
+        }
+        state.failureTracker.set(serverName, Date.now());
+        const message = error instanceof Error ? error.message : String(error);
+        logger.debug(`MCP: lazy connect failed for ${serverName}: ${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")
+        return 0;
+    return typeof state.config.settings?.idleTimeout === "number" ? state.config.settings.idleTimeout : 10;
+}
+//# sourceMappingURL=init.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"init.js","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/init.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,EACL,iBAAiB,EACjB,oBAAoB,EACpB,kBAAkB,EAClB,iBAAiB,EACjB,uBAAuB,EACvB,iBAAiB,EACjB,kBAAkB,EAClB,cAAc,GAEf,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACvE,OAAO,EAAE,iBAAiB,EAAE,MAAM,0BAA0B,CAAC;AAC7D,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,qCAAqC,EAAE,MAAM,mBAAmB,CAAC;AAC1E,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAE5C,MAAM,kBAAkB,GAAG,EAAE,GAAG,IAAI,CAAC;AAErC,MAAM,UAAU,SAAS,CAAC,GAA6C,EAAW;IAChF,OAAO,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,CAAC;AAAA,CACxC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,EAAgB,EAChB,GAAqB,EACO;IAC5B,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,YAAY,CAAuB,CAAC;IAClE,MAAM,MAAM,GAAG,aAAa,CAAC,UAAU,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;IAElD,MAAM,OAAO,GAAG,IAAI,gBAAgB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC9C,OAAO,CAAC,0BAA0B,CAAC,MAAM,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC;IACtE,MAAM,mBAAmB,GAAG,MAAM,CAAC,QAAQ,EAAE,mBAAmB,KAAK,IAAI,CAAC;IAC1E,IAAI,MAAM,CAAC,QAAQ,EAAE,QAAQ,KAAK,KAAK,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,mBAAmB,CAAC,EAAE,CAAC;QAC9E,OAAO,CAAC,iBAAiB,CAAC;YACxB,WAAW,EAAE,mBAAmB;YAChC,EAAE,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS;YAClC,aAAa,EAAE,GAAG,CAAC,aAAa;YAChC,eAAe,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,KAAK;YAChC,SAAS,EAAE,GAAG,EAAE,CAAC,GAAG,CAAC,MAAM;SAC5B,CAAC,CAAC;IACL,CAAC;IACD,MAAM,kBAAkB,GAAG,MAAM,CAAC,QAAQ,EAAE,WAAW,KAAK,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC;IAC/E,IAAI,kBAAkB,EAAE,CAAC;QACvB,OAAO,CAAC,oBAAoB,CAAC;YAC3B,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,QAAQ,EAAE,SAAS,CAAC,GAAG,CAAC;SACzB,CAAC,CAAC;IACL,CAAC;IACD,MAAM,SAAS,GAAG,IAAI,mBAAmB,CAAC,OAAO,CAAC,CAAC;IACnD,MAAM,YAAY,GAAG,IAAI,GAAG,EAA0B,CAAC;IACvD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;IACjD,MAAM,iBAAiB,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACzD,MAAM,cAAc,GAAG,IAAI,cAAc,CAAC,iBAAiB,CAAC,CAAC;IAC7D,MAAM,EAAE,GAAG,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1C,MAAM,KAAK,GAAsB;QAC/B,OAAO;QACP,SAAS;QACT,YAAY;QACZ,MAAM;QACN,cAAc;QACd,iBAAiB;QACjB,cAAc;QACd,QAAQ,EAAE,IAAI;QACd,mBAAmB,EAAE,EAAE;QACvB,WAAW,EAAE,CAAC,GAAW,EAAE,EAAE,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC;QACnE,EAAE;QACF,WAAW,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC,EAAE,CAAC,WAAW,CAAC,OAA0D,EAAE,OAAO,CAAC;KACvH,CAAC;IAEF,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACxD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC/B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,MAAM,CAAC,QAAQ,EAAE,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;IACxG,SAAS,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;IAE5C,MAAM,SAAS,GAAG,oBAAoB,EAAE,CAAC;IACzC,MAAM,eAAe,GAAG,UAAU,CAAC,SAAS,CAAC,CAAC;IAC9C,IAAI,KAAK,GAAG,iBAAiB,EAAE,CAAC;IAChC,IAAI,YAAY,GAAG,KAAK,CAAC;IAEzB,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,YAAY,GAAG,IAAI,CAAC;QACpB,iBAAiB,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;IACjD,CAAC;SAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QAClB,KAAK,GAAG,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QACpC,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAC3B,CAAC;IAED,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,EAAE,UAAU,IAAI,QAAQ,CAAC;IAEvD,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,aAAa,EAAE,CAAC;QAC/C,MAAM,aAAa,GAAG,UAAU,CAAC,SAAS,IAAI,MAAM,CAAC;QACrD,MAAM,YAAY,GAAG,UAAU,CAAC,WAAW,IAAI,CAAC,aAAa,KAAK,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QAC3F,SAAS,CAAC,cAAc,CACtB,IAAI,EACJ,UAAU,EACV,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CACvE,CAAC;QACF,IAAI,aAAa,KAAK,YAAY,EAAE,CAAC;YACnC,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,KAAK,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,IAAI,kBAAkB,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,UAAU,CAAC,EAAE,CAAC;YAClF,MAAM,QAAQ,GAAG,uBAAuB,CAAC,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,CAAC;YACxF,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,YAAY;QACjC,CAAC,CAAC,aAAa;QACf,CAAC,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC;YACvC,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,IAAI,MAAM,CAAC;YAC5C,OAAO,IAAI,KAAK,YAAY,IAAI,IAAI,KAAK,OAAO,CAAC;QAAA,CAClD,CAAC,CAAC;IAEP,IAAI,GAAG,CAAC,KAAK,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3C,GAAG,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,sBAAsB,cAAc,CAAC,MAAM,aAAa,CAAC,CAAC;IACpF,CAAC;IAED,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,cAAc,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,IAAI,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC;QACpF,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;YACvE,IAAI,UAAU,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;gBACvC,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,gDAAgD,IAAI,GAAG,EAAE,CAAC;YAChH,CAAC;YACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;QACvD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvE,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;QAChE,CAAC;IAAA,CACF,CAAC,CAAC;IAEH,KAAK,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,IAAI,OAAO,EAAE,CAAC;QAC9D,IAAI,KAAK,IAAI,CAAC,UAAU,EAAE,CAAC;YACzB,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACd,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,6BAA6B,IAAI,KAAK,KAAK,EAAE,EAAE,OAAO,CAAC,CAAC;YACxE,CAAC;YACD,OAAO,CAAC,KAAK,CAAC,6BAA6B,IAAI,KAAK,KAAK,EAAE,CAAC,CAAC;YAC7D,SAAS;QACX,CAAC;QAED,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,iBAAiB,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;QACtH,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACjC,mBAAmB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAEjC,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;YACxC,GAAG,CAAC,EAAE,CAAC,MAAM,CACX,QAAQ,IAAI,MAAM,WAAW,CAAC,MAAM,gBAAgB,EACpD,SAAS,CACV,CAAC;QACJ,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC;IAChE,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC;IACxD,IAAI,GAAG,CAAC,KAAK,IAAI,cAAc,GAAG,CAAC,EAAE,CAAC;QACpC,MAAM,UAAU,GAAG,cAAc,CAAC,KAAK,CAAC,CAAC;QACzC,MAAM,GAAG,GAAG,WAAW,GAAG,CAAC;YACzB,CAAC,CAAC,QAAQ,cAAc,IAAI,cAAc,CAAC,MAAM,uBAAuB,UAAU,SAAS;YAC3F,CAAC,CAAC,QAAQ,cAAc,uBAAuB,UAAU,SAAS,CAAC;QACrE,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;IAC/C,IAAI,SAAS,KAAK,UAAU,EAAE,CAAC;QAC7B,MAAM,YAAY,GAAG,iBAAiB,EAAE,CAAC;QACzC,MAAM,mBAAmB,GAAG,qCAAqC,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAExF,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,gBAAgB,GAAG,MAAM,aAAa,CAC1C,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC,EACvF,EAAE,EACF,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC;gBACd,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBAC3C,IAAI,CAAC;oBACH,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;oBACvE,IAAI,UAAU,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;wBACvC,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;oBAC7B,CAAC;oBACD,MAAM,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC;oBACzG,YAAY,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;oBACjC,mBAAmB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;oBACjC,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC;gBAC5B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACvE,MAAM,CAAC,KAAK,CAAC,0CAA0C,IAAI,KAAK,OAAO,EAAE,CAAC,CAAC;oBAC3E,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;gBAC7B,CAAC;YAAA,CACF,CACF,CAAC;YACF,MAAM,YAAY,GAAG,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YACzE,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;gBACzC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,yBAAyB,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,kCAAkC,EAAE,MAAM,CAAC,CAAC;YAC5G,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,CAAC,oBAAoB,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC;QAC7C,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACxC,mBAAmB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACvC,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACxC,eAAe,CAAC,KAAK,CAAC,CAAC;IAAA,CACxB,CAAC,CAAC;IAEH,SAAS,CAAC,uBAAuB,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC;QAChD,MAAM,WAAW,GAAG,8BAA8B,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,GAAG,UAAU,oBAAoB,WAAW,IAAI,CAAC,CAAC;QAC/D,eAAe,CAAC,KAAK,CAAC,CAAC;IAAA,CACxB,CAAC,CAAC;IAEH,SAAS,CAAC,iBAAiB,EAAE,CAAC;IAE9B,OAAO,KAAK,CAAC;AAAA,CACd;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAwB,EAAE,UAAkB,EAAQ;IACvF,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3D,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW;QAAE,OAAO;IAE7D,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU;QAAE,OAAO;IAExB,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,IAAI,QAAQ,CAAC;IAE7D,MAAM,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,UAAU,CAAC,KAAK,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAC/G,KAAK,CAAC,YAAY,CAAC,GAAG,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;AAAA,CAC9C;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAwB,EAAE,UAAkB,EAAQ;IACtF,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3D,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW;QAAE,OAAO;IAE7D,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU;QAAE,OAAO;IAExB,MAAM,UAAU,GAAG,iBAAiB,CAAC,UAAU,CAAC,CAAC;IACjD,MAAM,QAAQ,GAAG,iBAAiB,EAAE,CAAC;IACrC,MAAM,aAAa,GAAG,QAAQ,EAAE,OAAO,EAAE,CAAC,UAAU,CAAC,CAAC;IAEtD,MAAM,KAAK,GAAG,cAAc,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IAC/C,IAAI,SAAS,GAAG,UAAU,CAAC,eAAe,KAAK,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,kBAAkB,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IAErG,IACE,UAAU,CAAC,eAAe,KAAK,KAAK;QACpC,SAAS,CAAC,MAAM,KAAK,CAAC;QACtB,aAAa,EAAE,SAAS,EAAE,MAAM;QAChC,aAAa,CAAC,UAAU,KAAK,UAAU,EACvC,CAAC;QACD,SAAS,GAAG,aAAa,CAAC,SAAS,CAAC;IACtC,CAAC;IAED,MAAM,KAAK,GAAqB;QAC9B,UAAU;QACV,KAAK;QACL,SAAS;QACT,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,CAAC;IAEF,iBAAiB,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;AAAA,CACrE;AAED,MAAM,UAAU,kBAAkB,CAAC,KAAwB,EAAQ;IACjE,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,EAAE,CAAC;QACnE,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YACtC,mBAAmB,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;AAAA,CACF;AAED,MAAM,UAAU,eAAe,CAAC,KAAwB,EAAQ;IAC9D,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;IACpB,IAAI,CAAC,EAAE;QAAE,OAAO;IAChB,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,MAAM,CAAC;IAC1D,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;QAChB,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC/B,OAAO;IACT,CAAC;IACD,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,CAAC,iBAAiB,EAAE,CAAC,IAAI,CAAC;IAC9D,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,QAAQ,cAAc,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC;AAAA,CACvF;AAED,MAAM,UAAU,oBAAoB,CAAC,KAAwB,EAAE,UAAkB,EAAiB;IAChG,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACtD,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC;IACpC,IAAI,KAAK,GAAG,kBAAkB;QAAE,OAAO,IAAI,CAAC;IAC5C,OAAO,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC;AAAA,CACjC;AAED,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAwB,EAAE,UAAkB,EAAE,MAAoB,EAAoB;IACtH,MAAM,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;IAC3D,IAAI,UAAU,EAAE,MAAM,KAAK,YAAY,EAAE,CAAC;QACxC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,IAAI,UAAU,EAAE,MAAM,KAAK,WAAW,EAAE,CAAC;QACvC,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACxC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,SAAS,GAAG,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IAC1D,IAAI,SAAS,KAAK,IAAI;QAAE,OAAO,KAAK,CAAC;IAErC,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU;QAAE,OAAO,KAAK,CAAC;IAE9B,IAAI,CAAC;QACH,IAAI,KAAK,CAAC,EAAE,EAAE,CAAC;YACb,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC,KAAK,EAAE,sBAAsB,UAAU,KAAK,CAAC,CAAC;QACnE,CAAC;QACD,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;QAClF,IAAI,aAAa,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;YAC1C,OAAO,KAAK,CAAC;QACf,CAAC;QACD,KAAK,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QACxC,oBAAoB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACxC,mBAAmB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;QACvC,eAAe,CAAC,KAAK,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;YACpB,cAAc,CAAC,MAAM,CAAC,CAAC;QACzB,CAAC;QACD,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QACjD,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACvE,MAAM,CAAC,KAAK,CAAC,gCAAgC,UAAU,KAAK,OAAO,EAAE,CAAC,CAAC;QACvE,eAAe,CAAC,KAAK,CAAC,CAAC;QACvB,OAAO,KAAK,CAAC;IACf,CAAC;AAAA,CACF;AAED,SAAS,8BAA8B,CAAC,KAAwB,EAAE,UAAkB,EAAU;IAC5F,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,OAAO,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;IACzG,CAAC;IACD,IAAI,OAAO,UAAU,CAAC,WAAW,KAAK,QAAQ;QAAE,OAAO,UAAU,CAAC,WAAW,CAAC;IAC9E,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS,IAAI,MAAM,CAAC;IAC5C,IAAI,IAAI,KAAK,OAAO;QAAE,OAAO,CAAC,CAAC;IAC/B,OAAO,OAAO,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;AAAA,CACxG","sourcesContent":["// @ts-nocheck\nimport type { ExtensionAPI, ExtensionContext } from \"@earendil-works/pi-coding-agent\";\nimport type { McpExtensionState } from \"./state.ts\";\nimport type { ToolMetadata } from \"./types.ts\";\nimport { existsSync } from \"node:fs\";\nimport { loadMcpConfig } from \"./config.ts\";\nimport { ConsentManager } from \"./consent-manager.ts\";\nimport { McpLifecycleManager } from \"./lifecycle.ts\";\nimport {\n  computeServerHash,\n  getMetadataCachePath,\n  isServerCacheValid,\n  loadMetadataCache,\n  reconstructToolMetadata,\n  saveMetadataCache,\n  serializeResources,\n  serializeTools,\n  type ServerCacheEntry,\n} from \"./metadata-cache.ts\";\nimport { McpServerManager } from \"./server-manager.ts\";\nimport { buildToolMetadata, totalToolCount } from \"./tool-metadata.ts\";\nimport { UiResourceHandler } from \"./ui-resource-handler.ts\";\nimport { openUrl, parallelLimit } from \"./utils.ts\";\nimport { logger } from \"./logger.ts\";\nimport { getMissingConfiguredDirectToolServers } from \"./direct-tools.ts\";\nimport { throwIfAborted } from \"./abort.ts\";\n\nconst FAILURE_BACKOFF_MS = 60 * 1000;\n\nexport function isTuiMode(ctx: Pick<ExtensionContext, \"hasUI\" | \"mode\">): boolean {\n  return ctx.hasUI && ctx.mode === \"tui\";\n}\n\nexport async function initializeMcp(\n  pi: ExtensionAPI,\n  ctx: ExtensionContext\n): Promise<McpExtensionState> {\n  const configPath = pi.getFlag(\"mcp-config\") as string | undefined;\n  const config = loadMcpConfig(configPath, ctx.cwd);\n\n  const manager = new McpServerManager(ctx.cwd);\n  manager.setDefaultRequestTimeoutMs(config.settings?.requestTimeoutMs);\n  const samplingAutoApprove = config.settings?.samplingAutoApprove === true;\n  if (config.settings?.sampling !== false && (ctx.hasUI || samplingAutoApprove)) {\n    manager.setSamplingConfig({\n      autoApprove: samplingAutoApprove,\n      ui: ctx.hasUI ? ctx.ui : undefined,\n      modelRegistry: ctx.modelRegistry,\n      getCurrentModel: () => ctx.model,\n      getSignal: () => ctx.signal,\n    });\n  }\n  const elicitationEnabled = config.settings?.elicitation !== false && ctx.hasUI;\n  if (elicitationEnabled) {\n    manager.setElicitationConfig({\n      ui: ctx.ui,\n      allowUrl: isTuiMode(ctx),\n    });\n  }\n  const lifecycle = new McpLifecycleManager(manager);\n  const toolMetadata = new Map<string, ToolMetadata[]>();\n  const failureTracker = new Map<string, number>();\n  const uiResourceHandler = new UiResourceHandler(manager);\n  const consentManager = new ConsentManager(\"once-per-server\");\n  const ui = ctx.hasUI ? ctx.ui : undefined;\n  const state: McpExtensionState = {\n    manager,\n    lifecycle,\n    toolMetadata,\n    config,\n    failureTracker,\n    uiResourceHandler,\n    consentManager,\n    uiServer: null,\n    completedUiSessions: [],\n    openBrowser: (url: string) => openUrl(pi, url, process.env.BROWSER),\n    ui,\n    sendMessage: (message, options) => pi.sendMessage(message as unknown as Parameters<typeof pi.sendMessage>[0], options),\n  };\n\n  const serverEntries = Object.entries(config.mcpServers);\n  if (serverEntries.length === 0) {\n    return state;\n  }\n\n  const idleSetting = typeof config.settings?.idleTimeout === \"number\" ? config.settings.idleTimeout : 10;\n  lifecycle.setGlobalIdleTimeout(idleSetting);\n\n  const cachePath = getMetadataCachePath();\n  const cacheFileExists = existsSync(cachePath);\n  let cache = loadMetadataCache();\n  let bootstrapAll = false;\n\n  if (!cacheFileExists) {\n    bootstrapAll = true;\n    saveMetadataCache({ version: 1, servers: {} });\n  } else if (!cache) {\n    cache = { version: 1, servers: {} };\n    saveMetadataCache(cache);\n  }\n\n  const prefix = config.settings?.toolPrefix ?? \"server\";\n\n  for (const [name, definition] of serverEntries) {\n    const lifecycleMode = definition.lifecycle ?? \"lazy\";\n    const idleOverride = definition.idleTimeout ?? (lifecycleMode === \"eager\" ? 0 : undefined);\n    lifecycle.registerServer(\n      name,\n      definition,\n      idleOverride !== undefined ? { idleTimeout: idleOverride } : undefined\n    );\n    if (lifecycleMode === \"keep-alive\") {\n      lifecycle.markKeepAlive(name, definition);\n    }\n\n    if (cache?.servers?.[name] && isServerCacheValid(cache.servers[name], definition)) {\n      const metadata = reconstructToolMetadata(name, cache.servers[name], prefix, definition);\n      toolMetadata.set(name, metadata);\n    }\n  }\n\n  const startupServers = bootstrapAll\n    ? serverEntries\n    : serverEntries.filter(([, definition]) => {\n        const mode = definition.lifecycle ?? \"lazy\";\n        return mode === \"keep-alive\" || mode === \"eager\";\n      });\n\n  if (ctx.hasUI && startupServers.length > 0) {\n    ctx.ui.setStatus(\"mcp\", `MCP: connecting to ${startupServers.length} servers...`);\n  }\n\n  const results = await parallelLimit(startupServers, 10, async ([name, definition]) => {\n    try {\n      const connection = await manager.connect(name, definition, ctx.signal);\n      if (connection.status === \"needs-auth\") {\n        return { name, definition, connection: null, error: `OAuth authentication required. Run /mcp-auth ${name}.` };\n      }\n      return { name, definition, connection, error: null };\n    } catch (error) {\n      const message = error instanceof Error ? error.message : String(error);\n      return { name, definition, connection: null, error: message };\n    }\n  });\n\n  for (const { name, definition, connection, error } of results) {\n    if (error || !connection) {\n      if (ctx.hasUI) {\n        ctx.ui.notify(`MCP: Failed to connect to ${name}: ${error}`, \"error\");\n      }\n      console.error(`MCP: Failed to connect to ${name}: ${error}`);\n      continue;\n    }\n\n    const { metadata, failedTools } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix);\n    toolMetadata.set(name, metadata);\n    updateMetadataCache(state, name);\n\n    if (failedTools.length > 0 && ctx.hasUI) {\n      ctx.ui.notify(\n        `MCP: ${name} - ${failedTools.length} tools skipped`,\n        \"warning\"\n      );\n    }\n  }\n\n  const connectedCount = results.filter(r => r.connection).length;\n  const failedCount = results.filter(r => r.error).length;\n  if (ctx.hasUI && connectedCount > 0) {\n    const totalTools = totalToolCount(state);\n    const msg = failedCount > 0\n      ? `MCP: ${connectedCount}/${startupServers.length} servers connected (${totalTools} tools)`\n      : `MCP: ${connectedCount} servers connected (${totalTools} tools)`;\n    ctx.ui.notify(msg, \"info\");\n  }\n\n  const envDirect = process.env.MCP_DIRECT_TOOLS;\n  if (envDirect !== \"__none__\") {\n    const currentCache = loadMetadataCache();\n    const missingCacheServers = getMissingConfiguredDirectToolServers(config, currentCache);\n\n    if (missingCacheServers.length > 0) {\n      const bootstrapResults = await parallelLimit(\n        missingCacheServers.filter(name => !results.some(r => r.name === name && r.connection)),\n        10,\n        async (name) => {\n          const definition = config.mcpServers[name];\n          try {\n            const connection = await manager.connect(name, definition, ctx.signal);\n            if (connection.status === \"needs-auth\") {\n              return { name, ok: false };\n            }\n            const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, name, prefix);\n            toolMetadata.set(name, metadata);\n            updateMetadataCache(state, name);\n            return { name, ok: true };\n          } catch (error) {\n            const message = error instanceof Error ? error.message : String(error);\n            logger.debug(`MCP: direct-tools bootstrap failed for ${name}: ${message}`);\n            return { name, ok: false };\n          }\n        },\n      );\n      const bootstrapped = bootstrapResults.filter(r => r.ok).map(r => r.name);\n      if (bootstrapped.length > 0 && ctx.hasUI) {\n        ctx.ui.notify(`MCP: direct tools for ${bootstrapped.join(\", \")} will be available after restart`, \"info\");\n      }\n    }\n  }\n\n  lifecycle.setReconnectCallback((serverName) => {\n    updateServerMetadata(state, serverName);\n    updateMetadataCache(state, serverName);\n    state.failureTracker.delete(serverName);\n    updateStatusBar(state);\n  });\n\n  lifecycle.setIdleShutdownCallback((serverName) => {\n    const idleMinutes = getEffectiveIdleTimeoutMinutes(state, serverName);\n    logger.debug(`${serverName} shut down (idle ${idleMinutes}m)`);\n    updateStatusBar(state);\n  });\n\n  lifecycle.startHealthChecks();\n\n  return state;\n}\n\nexport function updateServerMetadata(state: McpExtensionState, serverName: string): void {\n  const connection = state.manager.getConnection(serverName);\n  if (!connection || connection.status !== \"connected\") return;\n\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) return;\n\n  const prefix = state.config.settings?.toolPrefix ?? \"server\";\n\n  const { metadata } = buildToolMetadata(connection.tools, connection.resources, definition, serverName, prefix);\n  state.toolMetadata.set(serverName, metadata);\n}\n\nexport function updateMetadataCache(state: McpExtensionState, serverName: string): void {\n  const connection = state.manager.getConnection(serverName);\n  if (!connection || connection.status !== \"connected\") return;\n\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) return;\n\n  const configHash = computeServerHash(definition);\n  const existing = loadMetadataCache();\n  const existingEntry = existing?.servers?.[serverName];\n\n  const tools = serializeTools(connection.tools);\n  let resources = definition.exposeResources === false ? [] : serializeResources(connection.resources);\n\n  if (\n    definition.exposeResources !== false &&\n    resources.length === 0 &&\n    existingEntry?.resources?.length &&\n    existingEntry.configHash === configHash\n  ) {\n    resources = existingEntry.resources;\n  }\n\n  const entry: ServerCacheEntry = {\n    configHash,\n    tools,\n    resources,\n    cachedAt: Date.now(),\n  };\n\n  saveMetadataCache({ version: 1, servers: { [serverName]: entry } });\n}\n\nexport function flushMetadataCache(state: McpExtensionState): void {\n  for (const [name, connection] of state.manager.getAllConnections()) {\n    if (connection.status === \"connected\") {\n      updateMetadataCache(state, name);\n    }\n  }\n}\n\nexport function updateStatusBar(state: McpExtensionState): void {\n  const ui = state.ui;\n  if (!ui) return;\n  const total = Object.keys(state.config.mcpServers).length;\n  if (total === 0) {\n    ui.setStatus(\"mcp\", undefined);\n    return;\n  }\n  const connectedCount = state.manager.getAllConnections().size;\n  ui.setStatus(\"mcp\", ui.theme.fg(\"accent\", `MCP: ${connectedCount}/${total} servers`));\n}\n\nexport function getFailureAgeSeconds(state: McpExtensionState, serverName: string): number | null {\n  const failedAt = state.failureTracker.get(serverName);\n  if (!failedAt) return null;\n  const ageMs = Date.now() - failedAt;\n  if (ageMs > FAILURE_BACKOFF_MS) return null;\n  return Math.round(ageMs / 1000);\n}\n\nexport async function lazyConnect(state: McpExtensionState, serverName: string, signal?: AbortSignal): Promise<boolean> {\n  const connection = state.manager.getConnection(serverName);\n  if (connection?.status === \"needs-auth\") {\n    return false;\n  }\n  if (connection?.status === \"connected\") {\n    updateServerMetadata(state, serverName);\n    return true;\n  }\n\n  const failedAgo = getFailureAgeSeconds(state, serverName);\n  if (failedAgo !== null) return false;\n\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) return false;\n\n  try {\n    if (state.ui) {\n      state.ui.setStatus(\"mcp\", `MCP: connecting to ${serverName}...`);\n    }\n    const newConnection = await state.manager.connect(serverName, definition, signal);\n    if (newConnection.status === \"needs-auth\") {\n      return false;\n    }\n    state.failureTracker.delete(serverName);\n    updateServerMetadata(state, serverName);\n    updateMetadataCache(state, serverName);\n    updateStatusBar(state);\n    return true;\n  } catch (error) {\n    if (signal?.aborted) {\n      throwIfAborted(signal);\n    }\n    state.failureTracker.set(serverName, Date.now());\n    const message = error instanceof Error ? error.message : String(error);\n    logger.debug(`MCP: lazy connect failed for ${serverName}: ${message}`);\n    updateStatusBar(state);\n    return false;\n  }\n}\n\nfunction getEffectiveIdleTimeoutMinutes(state: McpExtensionState, serverName: string): number {\n  const definition = state.config.mcpServers[serverName];\n  if (!definition) {\n    return typeof state.config.settings?.idleTimeout === \"number\" ? state.config.settings.idleTimeout : 10;\n  }\n  if (typeof definition.idleTimeout === \"number\") return definition.idleTimeout;\n  const mode = definition.lifecycle ?? \"lazy\";\n  if (mode === \"eager\") return 0;\n  return typeof state.config.settings?.idleTimeout === \"number\" ? state.config.settings.idleTimeout : 10;\n}\n"]}

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

@@ -0,0 +1,30 @@
+import type { ServerDefinition } from "./types.ts";
+import type { McpServerManager } from "./server-manager.ts";
+export type ReconnectCallback = (serverName: string) => void;
+export declare class McpLifecycleManager {
+    private manager;
+    private keepAliveServers;
+    private allServers;
+    private serverSettings;
+    private globalIdleTimeout;
+    private healthCheckInterval?;
+    private onReconnect?;
+    private onIdleShutdown?;
+    constructor(manager: McpServerManager);
+    /**
+     * Set callback to be invoked after a successful auto-reconnect.
+     * Use this to update tool metadata when a server reconnects.
+     */
+    setReconnectCallback(callback: ReconnectCallback): void;
+    markKeepAlive(name: string, definition: ServerDefinition): void;
+    registerServer(name: string, definition: ServerDefinition, settings?: {
+        idleTimeout?: number;
+    }): void;
+    setGlobalIdleTimeout(minutes: number): void;
+    setIdleShutdownCallback(callback: (serverName: string) => void): void;
+    startHealthChecks(intervalMs?: number): void;
+    private checkConnections;
+    private getIdleTimeout;
+    gracefulShutdown(): Promise<void>;
+}
+//# sourceMappingURL=lifecycle.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"lifecycle.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/lifecycle.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AACnD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAG5D,MAAM,MAAM,iBAAiB,GAAG,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,CAAC;AAE7D,qBAAa,mBAAmB;IAC9B,OAAO,CAAC,OAAO,CAAmB;IAClC,OAAO,CAAC,gBAAgB,CAAuC;IAC/D,OAAO,CAAC,UAAU,CAAuC;IACzD,OAAO,CAAC,cAAc,CAA+C;IACrE,OAAO,CAAC,iBAAiB,CAA0B;IACnD,OAAO,CAAC,mBAAmB,CAAC,CAAiB;IAC7C,OAAO,CAAC,WAAW,CAAC,CAAoB;IACxC,OAAO,CAAC,cAAc,CAAC,CAA+B;IAEtD,YAAY,OAAO,EAAE,gBAAgB,EAEpC;IAED;;;OAGG;IACH,oBAAoB,CAAC,QAAQ,EAAE,iBAAiB,GAAG,IAAI,CAEtD;IAED,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,GAAG,IAAI,CAE9D;IAED,cAAc,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,QAAQ,CAAC,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAKpG;IAED,oBAAoB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI,CAE1C;IAED,uBAAuB,CAAC,QAAQ,EAAE,CAAC,UAAU,EAAE,MAAM,KAAK,IAAI,GAAG,IAAI,CAEpE;IAED,iBAAiB,CAAC,UAAU,SAAQ,GAAG,IAAI,CAK1C;YAEa,gBAAgB;IA0B9B,OAAO,CAAC,cAAc;IAMhB,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC,CAKtC;CACF","sourcesContent":["// @ts-nocheck\nimport type { ServerDefinition } from \"./types.ts\";\nimport type { McpServerManager } from \"./server-manager.ts\";\nimport { logger } from \"./logger.ts\";\n\nexport type ReconnectCallback = (serverName: string) => void;\n\nexport class McpLifecycleManager {\n  private manager: McpServerManager;\n  private keepAliveServers = new Map<string, ServerDefinition>();\n  private allServers = new Map<string, ServerDefinition>();\n  private serverSettings = new Map<string, { idleTimeout?: number }>();\n  private globalIdleTimeout: number = 10 * 60 * 1000;\n  private healthCheckInterval?: NodeJS.Timeout;\n  private onReconnect?: ReconnectCallback;\n  private onIdleShutdown?: (serverName: string) => void;\n  \n  constructor(manager: McpServerManager) {\n    this.manager = manager;\n  }\n  \n  /**\n   * Set callback to be invoked after a successful auto-reconnect.\n   * Use this to update tool metadata when a server reconnects.\n   */\n  setReconnectCallback(callback: ReconnectCallback): void {\n    this.onReconnect = callback;\n  }\n  \n  markKeepAlive(name: string, definition: ServerDefinition): void {\n    this.keepAliveServers.set(name, definition);\n  }\n\n  registerServer(name: string, definition: ServerDefinition, settings?: { idleTimeout?: number }): void {\n    this.allServers.set(name, definition);\n    if (settings?.idleTimeout !== undefined) {\n      this.serverSettings.set(name, settings);\n    }\n  }\n\n  setGlobalIdleTimeout(minutes: number): void {\n    this.globalIdleTimeout = minutes * 60 * 1000;\n  }\n\n  setIdleShutdownCallback(callback: (serverName: string) => void): void {\n    this.onIdleShutdown = callback;\n  }\n  \n  startHealthChecks(intervalMs = 30000): void {\n    this.healthCheckInterval = setInterval(() => {\n      this.checkConnections();\n    }, intervalMs);\n    this.healthCheckInterval.unref();\n  }\n  \n  private async checkConnections(): Promise<void> {\n    for (const [name, definition] of this.keepAliveServers) {\n      const connection = this.manager.getConnection(name);\n      \n      if (!connection || connection.status !== \"connected\") {\n        try {\n          await this.manager.connect(name, definition);\n          logger.debug(`Reconnected to ${name}`);\n          // Notify extension to update metadata\n          this.onReconnect?.(name);\n        } catch (error) {\n          console.error(`MCP: Failed to reconnect to ${name}:`, error);\n        }\n      }\n    }\n\n    for (const [name] of this.allServers) {\n      if (this.keepAliveServers.has(name)) continue;\n      const timeout = this.getIdleTimeout(name);\n      if (timeout > 0 && this.manager.isIdle(name, timeout)) {\n        await this.manager.close(name);\n        this.onIdleShutdown?.(name);\n      }\n    }\n  }\n\n  private getIdleTimeout(name: string): number {\n    const perServer = this.serverSettings.get(name)?.idleTimeout;\n    if (perServer !== undefined) return perServer * 60 * 1000;\n    return this.globalIdleTimeout;\n  }\n  \n  async gracefulShutdown(): Promise<void> {\n    if (this.healthCheckInterval) {\n      clearInterval(this.healthCheckInterval);\n    }\n    await this.manager.closeAll();\n  }\n}\n"]}

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

@@ -0,0 +1,80 @@
+import { logger } from "./logger.js";
+export class McpLifecycleManager {
+    manager;
+    keepAliveServers = new Map();
+    allServers = new Map();
+    serverSettings = new Map();
+    globalIdleTimeout = 10 * 60 * 1000;
+    healthCheckInterval;
+    onReconnect;
+    onIdleShutdown;
+    constructor(manager) {
+        this.manager = manager;
+    }
+    /**
+     * Set callback to be invoked after a successful auto-reconnect.
+     * Use this to update tool metadata when a server reconnects.
+     */
+    setReconnectCallback(callback) {
+        this.onReconnect = callback;
+    }
+    markKeepAlive(name, definition) {
+        this.keepAliveServers.set(name, definition);
+    }
+    registerServer(name, definition, settings) {
+        this.allServers.set(name, definition);
+        if (settings?.idleTimeout !== undefined) {
+            this.serverSettings.set(name, settings);
+        }
+    }
+    setGlobalIdleTimeout(minutes) {
+        this.globalIdleTimeout = minutes * 60 * 1000;
+    }
+    setIdleShutdownCallback(callback) {
+        this.onIdleShutdown = callback;
+    }
+    startHealthChecks(intervalMs = 30000) {
+        this.healthCheckInterval = setInterval(() => {
+            this.checkConnections();
+        }, intervalMs);
+        this.healthCheckInterval.unref();
+    }
+    async checkConnections() {
+        for (const [name, definition] of this.keepAliveServers) {
+            const connection = this.manager.getConnection(name);
+            if (!connection || connection.status !== "connected") {
+                try {
+                    await this.manager.connect(name, definition);
+                    logger.debug(`Reconnected to ${name}`);
+                    // Notify extension to update metadata
+                    this.onReconnect?.(name);
+                }
+                catch (error) {
+                    console.error(`MCP: Failed to reconnect to ${name}:`, error);
+                }
+            }
+        }
+        for (const [name] of this.allServers) {
+            if (this.keepAliveServers.has(name))
+                continue;
+            const timeout = this.getIdleTimeout(name);
+            if (timeout > 0 && this.manager.isIdle(name, timeout)) {
+                await this.manager.close(name);
+                this.onIdleShutdown?.(name);
+            }
+        }
+    }
+    getIdleTimeout(name) {
+        const perServer = this.serverSettings.get(name)?.idleTimeout;
+        if (perServer !== undefined)
+            return perServer * 60 * 1000;
+        return this.globalIdleTimeout;
+    }
+    async gracefulShutdown() {
+        if (this.healthCheckInterval) {
+            clearInterval(this.healthCheckInterval);
+        }
+        await this.manager.closeAll();
+    }
+}
+//# sourceMappingURL=lifecycle.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"lifecycle.js","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/lifecycle.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAIrC,MAAM,OAAO,mBAAmB;IACtB,OAAO,CAAmB;IAC1B,gBAAgB,GAAG,IAAI,GAAG,EAA4B,CAAC;IACvD,UAAU,GAAG,IAAI,GAAG,EAA4B,CAAC;IACjD,cAAc,GAAG,IAAI,GAAG,EAAoC,CAAC;IAC7D,iBAAiB,GAAW,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IAC3C,mBAAmB,CAAkB;IACrC,WAAW,CAAqB;IAChC,cAAc,CAAgC;IAEtD,YAAY,OAAyB,EAAE;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IAAA,CACxB;IAED;;;OAGG;IACH,oBAAoB,CAAC,QAA2B,EAAQ;QACtD,IAAI,CAAC,WAAW,GAAG,QAAQ,CAAC;IAAA,CAC7B;IAED,aAAa,CAAC,IAAY,EAAE,UAA4B,EAAQ;QAC9D,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;IAAA,CAC7C;IAED,cAAc,CAAC,IAAY,EAAE,UAA4B,EAAE,QAAmC,EAAQ;QACpG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACtC,IAAI,QAAQ,EAAE,WAAW,KAAK,SAAS,EAAE,CAAC;YACxC,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;IAAA,CACF;IAED,oBAAoB,CAAC,OAAe,EAAQ;QAC1C,IAAI,CAAC,iBAAiB,GAAG,OAAO,GAAG,EAAE,GAAG,IAAI,CAAC;IAAA,CAC9C;IAED,uBAAuB,CAAC,QAAsC,EAAQ;QACpE,IAAI,CAAC,cAAc,GAAG,QAAQ,CAAC;IAAA,CAChC;IAED,iBAAiB,CAAC,UAAU,GAAG,KAAK,EAAQ;QAC1C,IAAI,CAAC,mBAAmB,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YAC3C,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAAA,CACzB,EAAE,UAAU,CAAC,CAAC;QACf,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IAAA,CAClC;IAEO,KAAK,CAAC,gBAAgB,GAAkB;QAC9C,KAAK,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,IAAI,IAAI,CAAC,gBAAgB,EAAE,CAAC;YACvD,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YAEpD,IAAI,CAAC,UAAU,IAAI,UAAU,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;gBACrD,IAAI,CAAC;oBACH,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;oBAC7C,MAAM,CAAC,KAAK,CAAC,kBAAkB,IAAI,EAAE,CAAC,CAAC;oBACvC,sCAAsC;oBACtC,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC;gBAC3B,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,IAAI,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;QACH,CAAC;QAED,KAAK,MAAM,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC;gBAAE,SAAS;YAC9C,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;YAC1C,IAAI,OAAO,GAAG,CAAC,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,EAAE,CAAC;gBACtD,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC/B,IAAI,CAAC,cAAc,EAAE,CAAC,IAAI,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;IAAA,CACF;IAEO,cAAc,CAAC,IAAY,EAAU;QAC3C,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,CAC/B;IAED,KAAK,CAAC,gBAAgB,GAAkB;QACtC,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC7B,aAAa,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1C,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC;IAAA,CAC/B;CACF","sourcesContent":["// @ts-nocheck\nimport type { ServerDefinition } from \"./types.ts\";\nimport type { McpServerManager } from \"./server-manager.ts\";\nimport { logger } from \"./logger.ts\";\n\nexport type ReconnectCallback = (serverName: string) => void;\n\nexport class McpLifecycleManager {\n  private manager: McpServerManager;\n  private keepAliveServers = new Map<string, ServerDefinition>();\n  private allServers = new Map<string, ServerDefinition>();\n  private serverSettings = new Map<string, { idleTimeout?: number }>();\n  private globalIdleTimeout: number = 10 * 60 * 1000;\n  private healthCheckInterval?: NodeJS.Timeout;\n  private onReconnect?: ReconnectCallback;\n  private onIdleShutdown?: (serverName: string) => void;\n  \n  constructor(manager: McpServerManager) {\n    this.manager = manager;\n  }\n  \n  /**\n   * Set callback to be invoked after a successful auto-reconnect.\n   * Use this to update tool metadata when a server reconnects.\n   */\n  setReconnectCallback(callback: ReconnectCallback): void {\n    this.onReconnect = callback;\n  }\n  \n  markKeepAlive(name: string, definition: ServerDefinition): void {\n    this.keepAliveServers.set(name, definition);\n  }\n\n  registerServer(name: string, definition: ServerDefinition, settings?: { idleTimeout?: number }): void {\n    this.allServers.set(name, definition);\n    if (settings?.idleTimeout !== undefined) {\n      this.serverSettings.set(name, settings);\n    }\n  }\n\n  setGlobalIdleTimeout(minutes: number): void {\n    this.globalIdleTimeout = minutes * 60 * 1000;\n  }\n\n  setIdleShutdownCallback(callback: (serverName: string) => void): void {\n    this.onIdleShutdown = callback;\n  }\n  \n  startHealthChecks(intervalMs = 30000): void {\n    this.healthCheckInterval = setInterval(() => {\n      this.checkConnections();\n    }, intervalMs);\n    this.healthCheckInterval.unref();\n  }\n  \n  private async checkConnections(): Promise<void> {\n    for (const [name, definition] of this.keepAliveServers) {\n      const connection = this.manager.getConnection(name);\n      \n      if (!connection || connection.status !== \"connected\") {\n        try {\n          await this.manager.connect(name, definition);\n          logger.debug(`Reconnected to ${name}`);\n          // Notify extension to update metadata\n          this.onReconnect?.(name);\n        } catch (error) {\n          console.error(`MCP: Failed to reconnect to ${name}:`, error);\n        }\n      }\n    }\n\n    for (const [name] of this.allServers) {\n      if (this.keepAliveServers.has(name)) continue;\n      const timeout = this.getIdleTimeout(name);\n      if (timeout > 0 && this.manager.isIdle(name, timeout)) {\n        await this.manager.close(name);\n        this.onIdleShutdown?.(name);\n      }\n    }\n  }\n\n  private getIdleTimeout(name: string): number {\n    const perServer = this.serverSettings.get(name)?.idleTimeout;\n    if (perServer !== undefined) return perServer * 60 * 1000;\n    return this.globalIdleTimeout;\n  }\n  \n  async gracefulShutdown(): Promise<void> {\n    if (this.healthCheckInterval) {\n      clearInterval(this.healthCheckInterval);\n    }\n    await this.manager.closeAll();\n  }\n}\n"]}

package/dist/builtin-extensions/pi-mcp-adapter/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/builtin-extensions/pi-mcp-adapter/logger.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"logger.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/logger.ts"],"names":[],"mappings":"AACA;;;GAGG;AAEH,MAAM,MAAM,QAAQ,GAAG,OAAO,GAAG,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC;AAE3D,MAAM,WAAW,UAAU;IACzB,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;CACxB;AAED,MAAM,WAAW,QAAQ;IACvB,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;CACjB;AAED,KAAK,UAAU,GAAG,CAAC,KAAK,EAAE,QAAQ,KAAK,IAAI,CAAC;AAgB5C,cAAM,MAAM;IACV,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;IAoCZ,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;CACF;AAED,cAAM,WAAW;IAEb,OAAO,CAAC,MAAM;IACd,OAAO,CAAC,OAAO;IAFjB,YACU,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,UAAU,EACzB;IAEJ,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;CACF;AAcD,eAAO,MAAM,MAAM,QAAe,CAAC","sourcesContent":["// @ts-nocheck\n/**\n * Centralized logging for MCP UI operations.\n * Provides structured, contextual logs with levels.\n */\n\nexport type LogLevel = \"debug\" | \"info\" | \"warn\" | \"error\";\n\nexport interface LogContext {\n  server?: string;\n  session?: string;\n  tool?: string;\n  uri?: string;\n  [key: string]: unknown;\n}\n\nexport interface LogEntry {\n  level: LogLevel;\n  message: string;\n  context?: LogContext;\n  error?: Error;\n  timestamp: Date;\n}\n\ntype LogHandler = (entry: LogEntry) => void;\n\nconst LEVEL_PRIORITY: Record<LogLevel, number> = {\n  debug: 0,\n  info: 1,\n  warn: 2,\n  error: 3,\n};\n\nconst LEVEL_PREFIX: Record<LogLevel, string> = {\n  debug: \"[MCP-UI:DEBUG]\",\n  info: \"[MCP-UI]\",\n  warn: \"[MCP-UI:WARN]\",\n  error: \"[MCP-UI:ERROR]\",\n};\n\nclass Logger {\n  private minLevel: LogLevel = \"info\";\n  private handlers: LogHandler[] = [];\n  private defaultContext: LogContext = {};\n\n  setLevel(level: LogLevel): void {\n    this.minLevel = level;\n  }\n\n  setDefaultContext(context: LogContext): void {\n    this.defaultContext = context;\n  }\n\n  addHandler(handler: LogHandler): void {\n    this.handlers.push(handler);\n  }\n\n  clearHandlers(): void {\n    this.handlers = [];\n  }\n\n  private shouldLog(level: LogLevel): boolean {\n    return LEVEL_PRIORITY[level] >= LEVEL_PRIORITY[this.minLevel];\n  }\n\n  private emit(level: LogLevel, message: string, context?: LogContext, error?: Error): void {\n    if (!this.shouldLog(level)) return;\n\n    const entry: LogEntry = {\n      level,\n      message,\n      context: { ...this.defaultContext, ...context },\n      error,\n      timestamp: new Date(),\n    };\n\n    // Default console output\n    const prefix = LEVEL_PREFIX[level];\n    const contextStr = formatContext(entry.context);\n    const fullMessage = contextStr ? `${prefix} ${message} ${contextStr}` : `${prefix} ${message}`;\n\n    if (level === \"error\") {\n      console.error(fullMessage, error ?? \"\");\n    } else if (level === \"warn\") {\n      console.warn(fullMessage);\n    } else if (level === \"debug\") {\n      console.debug(fullMessage);\n    } else {\n      console.log(fullMessage);\n    }\n\n    // Custom handlers\n    for (const handler of this.handlers) {\n      try {\n        handler(entry);\n      } catch {\n        // Ignore handler errors\n      }\n    }\n  }\n\n  debug(message: string, context?: LogContext): void {\n    this.emit(\"debug\", message, context);\n  }\n\n  info(message: string, context?: LogContext): void {\n    this.emit(\"info\", message, context);\n  }\n\n  warn(message: string, context?: LogContext): void {\n    this.emit(\"warn\", message, context);\n  }\n\n  error(message: string, error?: Error, context?: LogContext): void {\n    this.emit(\"error\", message, context, error);\n  }\n\n  /**\n   * Create a child logger with additional default context.\n   */\n  child(context: LogContext): ChildLogger {\n    return new ChildLogger(this, context);\n  }\n}\n\nclass ChildLogger {\n  constructor(\n    private parent: Logger,\n    private context: LogContext\n  ) {}\n\n  debug(message: string, context?: LogContext): void {\n    this.parent.debug(message, { ...this.context, ...context });\n  }\n\n  info(message: string, context?: LogContext): void {\n    this.parent.info(message, { ...this.context, ...context });\n  }\n\n  warn(message: string, context?: LogContext): void {\n    this.parent.warn(message, { ...this.context, ...context });\n  }\n\n  error(message: string, error?: Error, context?: LogContext): void {\n    this.parent.error(message, error, { ...this.context, ...context });\n  }\n\n  child(context: LogContext): ChildLogger {\n    return new ChildLogger(this.parent, { ...this.context, ...context });\n  }\n}\n\nfunction formatContext(context?: LogContext): string {\n  if (!context || Object.keys(context).length === 0) return \"\";\n  const parts: string[] = [];\n  for (const [key, value] of Object.entries(context)) {\n    if (value !== undefined && value !== null) {\n      parts.push(`${key}=${typeof value === \"string\" ? value : JSON.stringify(value)}`);\n    }\n  }\n  return 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  logger.setLevel(\"debug\");\n}\n"]}