@ashx-j/lunr 0.2.17 → 0.2.18

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

@@ -1 +1 @@
-{"version":3,"file":"server-manager.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/server-manager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAKnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8CAA8C,CAAC;AACnF,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,2BAA2B,EACjC,MAAM,oCAAoC,CAAC;AAC5C,OAAO,KAAK,EACV,OAAO,EACP,WAAW,EACX,gBAAgB,EAChB,mCAAmC,EACnC,SAAS,EACV,MAAM,YAAY,CAAC;AAMpB,OAAO,EAA2B,KAAK,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC3F,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,0BAA0B,CAAC;AAIlC,UAAU,gBAAgB;IACxB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,SAAS,CAAC;IACrB,UAAU,EAAE,gBAAgB,CAAC;IAC7B,KAAK,EAAE,OAAO,EAAE,CAAC;IACjB,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,YAAY,CAAC;CAC/C;AAED,KAAK,gBAAgB,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE,mCAAmC,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC;AAElH,qBAAa,gBAAgB;IAUf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;IATxC,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,eAAe,CAAgD;IACvE,OAAO,CAAC,iBAAiB,CAAuC;IAChE,OAAO,CAAC,cAAc,CAAmC;IACzD,OAAO,CAAC,iBAAiB,CAAsC;IAC/D,OAAO,CAAC,uBAAuB,CAAkC;IACjE,OAAO,CAAC,uBAAuB,CAAqB;IAEpD,sEAAsE;IACtE,YAA6B,UAAU,CAAC,oBAAQ,EAAI;IAEpD,iBAAiB,CAAC,MAAM,EAAE,oBAAoB,GAAG,SAAS,GAAG,IAAI,CAEhE;IAED,oBAAoB,CAAC,MAAM,EAAE,uBAAuB,GAAG,SAAS,GAAG,IAAI,CAEtE;IAED,0BAA0B,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAE9D;IAED,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,cAAc,GAAG,SAAS,CAGhF;IAED,OAAO,CAAC,2BAA2B;IAOnC,OAAO,CAAC,mBAAmB;IAgBrB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,CAwBzG;YAEa,gBAAgB;IAqF9B,OAAO,CAAC,uBAAuB;IAc/B,OAAO,CAAC,YAAY;IA6Bd,4BAA4B,CAChC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,2BAA2B,GACjC,OAAO,CAAC,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC,CAW1C;IAED,OAAO,CAAC,sBAAsB;YAShB,mBAAmB;YA0EnB,aAAa;YAab,iBAAiB;IAqB/B,OAAO,CAAC,iCAAiC;IAQzC,wBAAwB,CAAC,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,GAAG,IAAI,CAE9E;IAED,sBAAsB,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAEhD;IAEK,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAc/F;IAEK,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAYvC;IAEK,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAG9B;IAED,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,gBAAgB,GAAG,SAAS,CAExD;IAED,iBAAiB,IAAI,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAEjD;IAED,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAKxB;IAED,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAKpC;IAED,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAKpC;IAED,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAK/C;CACF","sourcesContent":["// @ts-nocheck\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { UnauthorizedError } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type { RequestOptions } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\nimport {\n  ElicitationCompleteNotificationSchema,\n  type ReadResourceResult,\n  type UrlElicitationRequiredError,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type {\n  McpTool,\n  McpResource,\n  ServerDefinition,\n  ServerStreamResultPatchNotification,\n  Transport,\n} from \"./types.ts\";\nimport { serverStreamResultPatchNotificationSchema } from \"./types.ts\";\nimport { resolveNpxBinary } from \"./npx-resolver.ts\";\nimport { logger } from \"./logger.ts\";\nimport { McpOAuthProvider } from \"./mcp-oauth-provider.ts\";\nimport { extractOAuthConfig, supportsOAuth } from \"./mcp-auth-flow.ts\";\nimport { registerSamplingHandler, type ServerSamplingConfig } from \"./sampling-handler.ts\";\nimport {\n  handleUrlElicitation,\n  registerElicitationHandler,\n  type ServerElicitationConfig,\n} from \"./elicitation-handler.ts\";\nimport { interpolateEnvRecord, resolveBearerToken, resolveConfigPath } from \"./utils.ts\";\nimport { abortable, throwIfAborted } from \"./abort.ts\";\n\ninterface ServerConnection {\n  client: Client;\n  transport: Transport;\n  definition: ServerDefinition;\n  tools: McpTool[];\n  resources: McpResource[];\n  lastUsedAt: number;\n  inFlight: number;\n  status: \"connected\" | \"closed\" | \"needs-auth\";\n}\n\ntype UiStreamListener = (serverName: string, notification: ServerStreamResultPatchNotification[\"params\"]) => void;\n\nexport class McpServerManager {\n  private connections = new Map<string, ServerConnection>();\n  private connectPromises = new Map<string, Promise<ServerConnection>>();\n  private uiStreamListeners = new Map<string, UiStreamListener>();\n  private samplingConfig: ServerSamplingConfig | undefined;\n  private elicitationConfig: ServerElicitationConfig | undefined;\n  private acceptedUrlElicitations = new Map<string, Set<string>>();\n  private defaultRequestTimeoutMs: number | undefined;\n\n  /** Default cwd for stdio servers without an explicit config `cwd`. */\n  constructor(private readonly defaultCwd?: string) {}\n\n  setSamplingConfig(config: ServerSamplingConfig | undefined): void {\n    this.samplingConfig = config;\n  }\n\n  setElicitationConfig(config: ServerElicitationConfig | undefined): void {\n    this.elicitationConfig = config;\n  }\n\n  setDefaultRequestTimeoutMs(timeoutMs: number | undefined): void {\n    this.defaultRequestTimeoutMs = normalizeRequestTimeoutMs(timeoutMs);\n  }\n\n  getRequestOptions(name: string, signal?: AbortSignal): RequestOptions | undefined {\n    const connection = this.connections.get(name);\n    return this.buildRequestOptions(connection?.definition, signal);\n  }\n\n  private getResolvedRequestTimeoutMs(definition?: ServerDefinition): number | undefined {\n    if (definition?.requestTimeoutMs !== undefined) {\n      return normalizeRequestTimeoutMs(definition.requestTimeoutMs);\n    }\n    return this.defaultRequestTimeoutMs;\n  }\n\n  private buildRequestOptions(\n    definition?: ServerDefinition,\n    signal?: AbortSignal,\n  ): RequestOptions | undefined {\n    const timeout = this.getResolvedRequestTimeoutMs(definition);\n\n    if (!signal && timeout === undefined) {\n      return undefined;\n    }\n\n    return {\n      ...(signal ? { signal } : {}),\n      ...(timeout !== undefined ? { timeout } : {}),\n    };\n  }\n\n  async connect(name: string, definition: ServerDefinition, signal?: AbortSignal): Promise<ServerConnection> {\n    throwIfAborted(signal);\n    // Dedupe concurrent connection attempts\n    if (this.connectPromises.has(name)) {\n      return abortable(this.connectPromises.get(name)!, signal);\n    }\n\n    // Reuse existing connection if healthy\n    const existing = this.connections.get(name);\n    if (existing?.status === \"connected\") {\n      existing.lastUsedAt = Date.now();\n      return existing;\n    }\n\n    const promise = this.createConnection(name, definition, signal);\n    this.connectPromises.set(name, promise);\n\n    try {\n      const connection = await promise;\n      this.connections.set(name, connection);\n      return connection;\n    } finally {\n      this.connectPromises.delete(name);\n    }\n  }\n\n  private async createConnection(\n    name: string,\n    definition: ServerDefinition,\n    signal?: AbortSignal,\n  ): Promise<ServerConnection> {\n    throwIfAborted(signal);\n    const client = this.createClient(name);\n\n    let transport: Transport;\n\n    if (definition.command) {\n      let command = definition.command;\n      let args = definition.args ?? [];\n\n      if (command === \"npx\" || command === \"npm\") {\n        const resolved = await resolveNpxBinary(command, args);\n        if (resolved) {\n          command = resolved.isJs ? \"node\" : resolved.binPath;\n          args = resolved.isJs ? [resolved.binPath, ...resolved.extraArgs] : resolved.extraArgs;\n          logger.debug(`${name} resolved to ${resolved.binPath} (skipping npm parent)`);\n        }\n      }\n\n      transport = new StdioClientTransport({\n        command,\n        args,\n        env: resolveEnv(definition.env),\n        cwd: resolveConfigPath(definition.cwd) ?? this.defaultCwd,\n        stderr: definition.debug ? \"inherit\" : \"ignore\",\n      });\n    } else if (definition.url) {\n      // HTTP transport with fallback\n      transport = await this.createHttpTransport(definition, name, signal);\n    } else {\n      throw new Error(`Server ${name} has no command or url`);\n    }\n\n    const requestOptions = this.buildRequestOptions(definition, signal);\n\n    try {\n      await client.connect(transport, requestOptions);\n      this.attachAdapterNotificationHandlers(name, client);\n\n      // Discover tools and resources\n      const [tools, resources] = await Promise.all([\n        this.fetchAllTools(client, requestOptions),\n        this.fetchAllResources(client, requestOptions),\n      ]);\n\n      return {\n        client,\n        transport,\n        definition,\n        tools,\n        resources,\n        lastUsedAt: Date.now(),\n        inFlight: 0,\n        status: \"connected\",\n      };\n    } catch (error) {\n      // Check for UnauthorizedError - server requires OAuth\n      if (error instanceof UnauthorizedError && supportsOAuth(definition)) {\n        // Clean up both client and transport before reporting needs-auth.\n        await client.close().catch(() => {});\n        await transport.close().catch(() => {});\n\n        return {\n          client,\n          transport,\n          definition,\n          tools: [],\n          resources: [],\n          lastUsedAt: Date.now(),\n          inFlight: 0,\n          status: \"needs-auth\",\n        };\n      }\n\n      // Clean up both client and transport on any error\n      await client.close().catch(() => {});\n      await transport.close().catch(() => {});\n      throw error;\n    }\n  }\n\n  private buildClientCapabilities() {\n    return {\n      ...(this.samplingConfig ? { sampling: {} } : {}),\n      ...(this.elicitationConfig\n        ? {\n            elicitation: {\n              form: {},\n              ...(this.elicitationConfig.allowUrl ? { url: {} } : {}),\n            },\n          }\n        : {}),\n    };\n  }\n\n  private createClient(serverName: string): Client {\n    const capabilities = this.buildClientCapabilities();\n    const client = new Client(\n      { name: `pi-mcp-${serverName}`, version: \"1.0.0\" },\n      Object.keys(capabilities).length > 0 ? { capabilities } : undefined,\n    );\n    if (this.samplingConfig) {\n      registerSamplingHandler(client, { ...this.samplingConfig, serverName });\n    }\n    if (this.elicitationConfig) {\n      registerElicitationHandler(client, {\n        ...this.elicitationConfig,\n        serverName,\n        onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),\n      });\n      if (this.elicitationConfig.allowUrl) {\n        client.setNotificationHandler(ElicitationCompleteNotificationSchema, notification => {\n          const accepted = this.acceptedUrlElicitations.get(serverName);\n          if (!accepted?.delete(notification.params.elicitationId)) return;\n          this.elicitationConfig?.ui.notify(\n            `MCP browser interaction for ${serverName} completed. You can retry the tool now.`,\n            \"info\",\n          );\n        });\n      }\n    }\n    return client;\n  }\n\n  async handleUrlElicitationRequired(\n    serverName: string,\n    error: UrlElicitationRequiredError,\n  ): Promise<\"accept\" | \"decline\" | \"cancel\"> {\n    if (!this.elicitationConfig?.allowUrl) return \"cancel\";\n    for (const params of error.elicitations) {\n      const result = await handleUrlElicitation({\n        ...this.elicitationConfig,\n        serverName,\n        onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),\n      }, params);\n      if (result.action !== \"accept\") return result.action;\n    }\n    return \"accept\";\n  }\n\n  private rememberUrlElicitation(serverName: string, elicitationId: string): void {\n    let accepted = this.acceptedUrlElicitations.get(serverName);\n    if (!accepted) {\n      accepted = new Set();\n      this.acceptedUrlElicitations.set(serverName, accepted);\n    }\n    accepted.add(elicitationId);\n  }\n\n  private async createHttpTransport(\n    definition: ServerDefinition,\n    serverName: string,\n    signal?: AbortSignal,\n  ): Promise<Transport> {\n    throwIfAborted(signal);\n    const url = new URL(definition.url!);\n\n    // Build headers first (including any bearer token)\n    const headers = resolveHeaders(definition.headers) ?? {};\n\n    // For bearer auth, add the token to headers BEFORE creating requestInit\n    if (definition.auth === \"bearer\") {\n      const token = resolveBearerToken(definition);\n      if (token) {\n        headers[\"Authorization\"] = `Bearer ${token}`;\n      }\n    }\n\n    // Create request init with headers (Authorization now included for bearer auth)\n    const requestInit = Object.keys(headers).length > 0 ? { headers } : undefined;\n\n    // For OAuth servers, create an auth provider\n    let authProvider: McpOAuthProvider | undefined;\n    if (supportsOAuth(definition)) {\n      const oauthConfig = extractOAuthConfig(definition);\n      authProvider = new McpOAuthProvider(\n        serverName,\n        definition.url!,\n        oauthConfig,\n        {\n          onRedirect: async (_authUrl) => {\n            // URL is captured by startAuth, no need to log\n          },\n        }\n      );\n    }\n\n    // Try StreamableHTTP first (modern MCP servers)\n    const streamableTransport = new StreamableHTTPClientTransport(url, {\n      requestInit,\n      authProvider,\n    });\n\n    try {\n      // Create a test client to verify the transport works\n      const testClient = new Client({ name: \"pi-mcp-probe\", version: \"2.1.2\" });\n      await testClient.connect(streamableTransport, this.buildRequestOptions(definition, signal));\n      await testClient.close().catch(() => {});\n      // Close probe transport before creating fresh one\n      await streamableTransport.close().catch(() => {});\n\n      // StreamableHTTP works - create fresh transport for actual use\n      return new StreamableHTTPClientTransport(url, { requestInit, authProvider });\n    } catch (error) {\n      // StreamableHTTP failed, close and try SSE fallback\n      await streamableTransport.close().catch(() => {});\n\n      // Host cancellation is not transport capability evidence; do not fall\n      // through to SSE when the caller is trying to cancel the connect.\n      if (signal?.aborted) {\n        throwIfAborted(signal);\n      }\n\n      // If this was an UnauthorizedError, don't try SSE - the server needs auth\n      if (error instanceof UnauthorizedError) {\n        throw error;\n      }\n\n      // SSE is the legacy transport\n      return new SSEClientTransport(url, { requestInit, authProvider });\n    }\n  }\n\n  private async fetchAllTools(client: Client, requestOptions?: RequestOptions): Promise<McpTool[]> {\n    const allTools: McpTool[] = [];\n    let cursor: string | undefined;\n\n    do {\n      const result = await client.listTools(cursor ? { cursor } : undefined, requestOptions);\n      allTools.push(...(result.tools ?? []));\n      cursor = result.nextCursor;\n    } while (cursor);\n\n    return allTools;\n  }\n\n  private async fetchAllResources(client: Client, requestOptions?: RequestOptions): Promise<McpResource[]> {\n    try {\n      const allResources: McpResource[] = [];\n      let cursor: string | undefined;\n\n      do {\n        const result = await client.listResources(cursor ? { cursor } : undefined, requestOptions);\n        allResources.push(...(result.resources ?? []));\n        cursor = result.nextCursor;\n      } while (cursor);\n\n      return allResources;\n    } catch {\n      if (requestOptions?.signal?.aborted) {\n        throwIfAborted(requestOptions.signal);\n      }\n      // Server may not support resources\n      return [];\n    }\n  }\n\n  private attachAdapterNotificationHandlers(serverName: string, client: Client): void {\n    client.setNotificationHandler(serverStreamResultPatchNotificationSchema, (notification) => {\n      const listener = this.uiStreamListeners.get(notification.params.streamToken);\n      if (!listener) return;\n      listener(serverName, notification.params);\n    });\n  }\n\n  registerUiStreamListener(streamToken: string, listener: UiStreamListener): void {\n    this.uiStreamListeners.set(streamToken, listener);\n  }\n\n  removeUiStreamListener(streamToken: string): void {\n    this.uiStreamListeners.delete(streamToken);\n  }\n\n  async readResource(name: string, uri: string, signal?: AbortSignal): Promise<ReadResourceResult> {\n    const connection = this.connections.get(name);\n    if (!connection || connection.status !== \"connected\") {\n      throw new Error(`Server \"${name}\" is not connected`);\n    }\n\n    try {\n      this.touch(name);\n      this.incrementInFlight(name);\n      return await connection.client.readResource({ uri }, this.getRequestOptions(name, signal));\n    } finally {\n      this.decrementInFlight(name);\n      this.touch(name);\n    }\n  }\n\n  async close(name: string): Promise<void> {\n    const connection = this.connections.get(name);\n    if (!connection) return;\n\n    // Delete from map BEFORE async cleanup to prevent a race where a\n    // concurrent connect() creates a new connection that our deferred\n    // delete() would then remove, orphaning the new server process.\n    connection.status = \"closed\";\n    this.connections.delete(name);\n    this.acceptedUrlElicitations.delete(name);\n    await connection.client.close().catch(() => {});\n    await connection.transport.close().catch(() => {});\n  }\n\n  async closeAll(): Promise<void> {\n    const names = [...this.connections.keys()];\n    await Promise.all(names.map(name => this.close(name)));\n  }\n\n  getConnection(name: string): ServerConnection | undefined {\n    return this.connections.get(name);\n  }\n\n  getAllConnections(): Map<string, ServerConnection> {\n    return new Map(this.connections);\n  }\n\n  touch(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection) {\n      connection.lastUsedAt = Date.now();\n    }\n  }\n\n  incrementInFlight(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection) {\n      connection.inFlight = (connection.inFlight ?? 0) + 1;\n    }\n  }\n\n  decrementInFlight(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection && connection.inFlight) {\n      connection.inFlight--;\n    }\n  }\n\n  isIdle(name: string, timeoutMs: number): boolean {\n    const connection = this.connections.get(name);\n    if (!connection || connection.status !== \"connected\") return false;\n    if (connection.inFlight > 0) return false;\n    return (Date.now() - connection.lastUsedAt) > timeoutMs;\n  }\n}\n\n/**\n * Resolve environment variables with interpolation.\n */\nfunction resolveEnv(env?: Record<string, string>): Record<string, string> {\n  // Copy process.env, filtering out undefined values\n  const resolved: Record<string, string> = {};\n  for (const [key, value] of Object.entries(process.env)) {\n    if (value !== undefined) {\n      resolved[key] = value;\n    }\n  }\n\n  if (!env) return resolved;\n\n  const overrides = interpolateEnvRecord(env);\n  return overrides ? { ...resolved, ...overrides } : resolved;\n}\n\n/**\n * Resolve headers with environment variable interpolation.\n */\nfunction resolveHeaders(headers?: Record<string, string>): Record<string, string> | undefined {\n  return interpolateEnvRecord(headers);\n}\n\nfunction normalizeRequestTimeoutMs(timeoutMs: number | undefined): number | undefined {\n  return typeof timeoutMs === \"number\" && Number.isFinite(timeoutMs) && timeoutMs > 0\n    ? timeoutMs\n    : undefined;\n}\n"]}
+{"version":3,"file":"server-manager.d.ts","sourceRoot":"","sources":["../../../src/builtin-extensions/pi-mcp-adapter/server-manager.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AAKnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,8CAA8C,CAAC;AACnF,OAAO,EAEL,KAAK,kBAAkB,EACvB,KAAK,2BAA2B,EACjC,MAAM,oCAAoC,CAAC;AAC5C,OAAO,KAAK,EACV,OAAO,EACP,WAAW,EACX,gBAAgB,EAChB,mCAAmC,EACnC,SAAS,EACV,MAAM,YAAY,CAAC;AAMpB,OAAO,EAA2B,KAAK,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC3F,OAAO,EAGL,KAAK,uBAAuB,EAC7B,MAAM,0BAA0B,CAAC;AAIlC,UAAU,iBAAiB;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,SAAS,CAAC;CACtB;AAED,UAAU,gBAAiB,SAAQ,iBAAiB;IAClD,UAAU,EAAE,gBAAgB,CAAC;IAC7B,KAAK,EAAE,OAAO,EAAE,CAAC;IACjB,SAAS,EAAE,WAAW,EAAE,CAAC;IACzB,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,YAAY,CAAC;CAC/C;AAED,KAAK,gBAAgB,GAAG,CAAC,UAAU,EAAE,MAAM,EAAE,YAAY,EAAE,mCAAmC,CAAC,QAAQ,CAAC,KAAK,IAAI,CAAC;AAElH,qBAAa,gBAAgB;IAYf,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC;IAXxC,OAAO,CAAC,WAAW,CAAuC;IAC1D,OAAO,CAAC,eAAe,CAAgD;IACvE,OAAO,CAAC,kBAAkB,CAAwC;IAClE,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,iBAAiB,CAAuC;IAChE,OAAO,CAAC,cAAc,CAAmC;IACzD,OAAO,CAAC,iBAAiB,CAAsC;IAC/D,OAAO,CAAC,uBAAuB,CAAkC;IACjE,OAAO,CAAC,uBAAuB,CAAqB;IAEpD,sEAAsE;IACtE,YAA6B,UAAU,CAAC,oBAAQ,EAAI;IAEpD,iBAAiB,CAAC,MAAM,EAAE,oBAAoB,GAAG,SAAS,GAAG,IAAI,CAEhE;IAED,oBAAoB,CAAC,MAAM,EAAE,uBAAuB,GAAG,SAAS,GAAG,IAAI,CAEtE;IAED,0BAA0B,CAAC,SAAS,EAAE,MAAM,GAAG,SAAS,GAAG,IAAI,CAE9D;IAED,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,cAAc,GAAG,SAAS,CAGhF;IAED,OAAO,CAAC,2BAA2B;IAOnC,OAAO,CAAC,mBAAmB;IAgBrB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,CA+BzG;YAEa,gBAAgB;IAyF9B,OAAO,CAAC,uBAAuB;IAc/B,OAAO,CAAC,YAAY;IA6Bd,4BAA4B,CAChC,UAAU,EAAE,MAAM,EAClB,KAAK,EAAE,2BAA2B,GACjC,OAAO,CAAC,QAAQ,GAAG,SAAS,GAAG,QAAQ,CAAC,CAW1C;IAED,OAAO,CAAC,sBAAsB;YAShB,mBAAmB;YA0EnB,aAAa;YAab,iBAAiB;IAqB/B,OAAO,CAAC,iCAAiC;IAQzC,wBAAwB,CAAC,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,gBAAgB,GAAG,IAAI,CAE9E;IAED,sBAAsB,CAAC,WAAW,EAAE,MAAM,GAAG,IAAI,CAEhD;IAEK,YAAY,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAc/F;YAEa,iBAAiB;IAKzB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAWvC;IAEK,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC,CAQ9B;IAED,aAAa,CAAC,IAAI,EAAE,MAAM,GAAG,gBAAgB,GAAG,SAAS,CAExD;IAED,iBAAiB,IAAI,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAEjD;IAED,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAKxB;IAED,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAKpC;IAED,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAKpC;IAED,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,GAAG,OAAO,CAK/C;CACF","sourcesContent":["// @ts-nocheck\nimport { Client } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { StdioClientTransport } from \"@modelcontextprotocol/sdk/client/stdio.js\";\nimport { StreamableHTTPClientTransport } from \"@modelcontextprotocol/sdk/client/streamableHttp.js\";\nimport { SSEClientTransport } from \"@modelcontextprotocol/sdk/client/sse.js\";\nimport { UnauthorizedError } from \"@modelcontextprotocol/sdk/client/auth.js\";\nimport type { RequestOptions } from \"@modelcontextprotocol/sdk/shared/protocol.js\";\nimport {\n  ElicitationCompleteNotificationSchema,\n  type ReadResourceResult,\n  type UrlElicitationRequiredError,\n} from \"@modelcontextprotocol/sdk/types.js\";\nimport type {\n  McpTool,\n  McpResource,\n  ServerDefinition,\n  ServerStreamResultPatchNotification,\n  Transport,\n} from \"./types.ts\";\nimport { serverStreamResultPatchNotificationSchema } from \"./types.ts\";\nimport { resolveNpxBinary } from \"./npx-resolver.ts\";\nimport { logger } from \"./logger.ts\";\nimport { McpOAuthProvider } from \"./mcp-oauth-provider.ts\";\nimport { extractOAuthConfig, supportsOAuth } from \"./mcp-auth-flow.ts\";\nimport { registerSamplingHandler, type ServerSamplingConfig } from \"./sampling-handler.ts\";\nimport {\n  handleUrlElicitation,\n  registerElicitationHandler,\n  type ServerElicitationConfig,\n} from \"./elicitation-handler.ts\";\nimport { interpolateEnvRecord, resolveBearerToken, resolveConfigPath } from \"./utils.ts\";\nimport { abortable, throwIfAborted } from \"./abort.ts\";\n\ninterface PendingConnection {\n  client: Client;\n  transport: Transport;\n}\n\ninterface ServerConnection extends PendingConnection {\n  definition: ServerDefinition;\n  tools: McpTool[];\n  resources: McpResource[];\n  lastUsedAt: number;\n  inFlight: number;\n  status: \"connected\" | \"closed\" | \"needs-auth\";\n}\n\ntype UiStreamListener = (serverName: string, notification: ServerStreamResultPatchNotification[\"params\"]) => void;\n\nexport class McpServerManager {\n  private connections = new Map<string, ServerConnection>();\n  private connectPromises = new Map<string, Promise<ServerConnection>>();\n  private pendingConnections = new Map<string, PendingConnection>();\n  private closing = false;\n  private uiStreamListeners = new Map<string, UiStreamListener>();\n  private samplingConfig: ServerSamplingConfig | undefined;\n  private elicitationConfig: ServerElicitationConfig | undefined;\n  private acceptedUrlElicitations = new Map<string, Set<string>>();\n  private defaultRequestTimeoutMs: number | undefined;\n\n  /** Default cwd for stdio servers without an explicit config `cwd`. */\n  constructor(private readonly defaultCwd?: string) {}\n\n  setSamplingConfig(config: ServerSamplingConfig | undefined): void {\n    this.samplingConfig = config;\n  }\n\n  setElicitationConfig(config: ServerElicitationConfig | undefined): void {\n    this.elicitationConfig = config;\n  }\n\n  setDefaultRequestTimeoutMs(timeoutMs: number | undefined): void {\n    this.defaultRequestTimeoutMs = normalizeRequestTimeoutMs(timeoutMs);\n  }\n\n  getRequestOptions(name: string, signal?: AbortSignal): RequestOptions | undefined {\n    const connection = this.connections.get(name);\n    return this.buildRequestOptions(connection?.definition, signal);\n  }\n\n  private getResolvedRequestTimeoutMs(definition?: ServerDefinition): number | undefined {\n    if (definition?.requestTimeoutMs !== undefined) {\n      return normalizeRequestTimeoutMs(definition.requestTimeoutMs);\n    }\n    return this.defaultRequestTimeoutMs;\n  }\n\n  private buildRequestOptions(\n    definition?: ServerDefinition,\n    signal?: AbortSignal,\n  ): RequestOptions | undefined {\n    const timeout = this.getResolvedRequestTimeoutMs(definition);\n\n    if (!signal && timeout === undefined) {\n      return undefined;\n    }\n\n    return {\n      ...(signal ? { signal } : {}),\n      ...(timeout !== undefined ? { timeout } : {}),\n    };\n  }\n\n  async connect(name: string, definition: ServerDefinition, signal?: AbortSignal): Promise<ServerConnection> {\n    throwIfAborted(signal);\n    if (this.closing) throw new Error(\"MCP server manager is closed\");\n    // Dedupe concurrent connection attempts\n    if (this.connectPromises.has(name)) {\n      return abortable(this.connectPromises.get(name)!, signal);\n    }\n\n    // Reuse existing connection if healthy\n    const existing = this.connections.get(name);\n    if (existing?.status === \"connected\") {\n      existing.lastUsedAt = Date.now();\n      return existing;\n    }\n\n    const promise = this.createConnection(name, definition, signal).then(async connection => {\n      if (this.closing || signal?.aborted) {\n        await this.disposeConnection(connection);\n        throwIfAborted(signal);\n        throw new Error(\"MCP server manager closed during connection\");\n      }\n      this.connections.set(name, connection);\n      return connection;\n    });\n    this.connectPromises.set(name, promise);\n\n    try {\n      return await promise;\n    } finally {\n      if (this.connectPromises.get(name) === promise) this.connectPromises.delete(name);\n    }\n  }\n\n  private async createConnection(\n    name: string,\n    definition: ServerDefinition,\n    signal?: AbortSignal,\n  ): Promise<ServerConnection> {\n    throwIfAborted(signal);\n    const client = this.createClient(name);\n\n    let transport: Transport;\n\n    if (definition.command) {\n      let command = definition.command;\n      let args = definition.args ?? [];\n\n      if (command === \"npx\" || command === \"npm\") {\n        const resolved = await resolveNpxBinary(command, args);\n        if (resolved) {\n          command = resolved.isJs ? \"node\" : resolved.binPath;\n          args = resolved.isJs ? [resolved.binPath, ...resolved.extraArgs] : resolved.extraArgs;\n          logger.debug(`${name} resolved to ${resolved.binPath} (skipping npm parent)`);\n        }\n      }\n\n      transport = new StdioClientTransport({\n        command,\n        args,\n        env: resolveEnv(definition.env),\n        cwd: resolveConfigPath(definition.cwd) ?? this.defaultCwd,\n        stderr: definition.debug ? \"inherit\" : \"ignore\",\n      });\n    } else if (definition.url) {\n      // HTTP transport with fallback\n      transport = await this.createHttpTransport(definition, name, signal);\n    } else {\n      throw new Error(`Server ${name} has no command or url`);\n    }\n\n    const requestOptions = this.buildRequestOptions(definition, signal);\n    const pending = { client, transport };\n    this.pendingConnections.set(name, pending);\n\n    try {\n      await client.connect(transport, requestOptions);\n      this.attachAdapterNotificationHandlers(name, client);\n\n      // Discover tools and resources\n      const [tools, resources] = await Promise.all([\n        this.fetchAllTools(client, requestOptions),\n        this.fetchAllResources(client, requestOptions),\n      ]);\n\n      return {\n        client,\n        transport,\n        definition,\n        tools,\n        resources,\n        lastUsedAt: Date.now(),\n        inFlight: 0,\n        status: \"connected\",\n      };\n    } catch (error) {\n      // Check for UnauthorizedError - server requires OAuth\n      if (error instanceof UnauthorizedError && supportsOAuth(definition)) {\n        // Clean up both client and transport before reporting needs-auth.\n        await client.close().catch(() => {});\n        await transport.close().catch(() => {});\n\n        return {\n          client,\n          transport,\n          definition,\n          tools: [],\n          resources: [],\n          lastUsedAt: Date.now(),\n          inFlight: 0,\n          status: \"needs-auth\",\n        };\n      }\n\n      // Clean up both client and transport on any error\n      await client.close().catch(() => {});\n      await transport.close().catch(() => {});\n      throw error;\n    } finally {\n      if (this.pendingConnections.get(name) === pending) this.pendingConnections.delete(name);\n    }\n  }\n\n  private buildClientCapabilities() {\n    return {\n      ...(this.samplingConfig ? { sampling: {} } : {}),\n      ...(this.elicitationConfig\n        ? {\n            elicitation: {\n              form: {},\n              ...(this.elicitationConfig.allowUrl ? { url: {} } : {}),\n            },\n          }\n        : {}),\n    };\n  }\n\n  private createClient(serverName: string): Client {\n    const capabilities = this.buildClientCapabilities();\n    const client = new Client(\n      { name: `pi-mcp-${serverName}`, version: \"1.0.0\" },\n      Object.keys(capabilities).length > 0 ? { capabilities } : undefined,\n    );\n    if (this.samplingConfig) {\n      registerSamplingHandler(client, { ...this.samplingConfig, serverName });\n    }\n    if (this.elicitationConfig) {\n      registerElicitationHandler(client, {\n        ...this.elicitationConfig,\n        serverName,\n        onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),\n      });\n      if (this.elicitationConfig.allowUrl) {\n        client.setNotificationHandler(ElicitationCompleteNotificationSchema, notification => {\n          const accepted = this.acceptedUrlElicitations.get(serverName);\n          if (!accepted?.delete(notification.params.elicitationId)) return;\n          this.elicitationConfig?.ui.notify(\n            `MCP browser interaction for ${serverName} completed. You can retry the tool now.`,\n            \"info\",\n          );\n        });\n      }\n    }\n    return client;\n  }\n\n  async handleUrlElicitationRequired(\n    serverName: string,\n    error: UrlElicitationRequiredError,\n  ): Promise<\"accept\" | \"decline\" | \"cancel\"> {\n    if (!this.elicitationConfig?.allowUrl) return \"cancel\";\n    for (const params of error.elicitations) {\n      const result = await handleUrlElicitation({\n        ...this.elicitationConfig,\n        serverName,\n        onUrlAccepted: elicitationId => this.rememberUrlElicitation(serverName, elicitationId),\n      }, params);\n      if (result.action !== \"accept\") return result.action;\n    }\n    return \"accept\";\n  }\n\n  private rememberUrlElicitation(serverName: string, elicitationId: string): void {\n    let accepted = this.acceptedUrlElicitations.get(serverName);\n    if (!accepted) {\n      accepted = new Set();\n      this.acceptedUrlElicitations.set(serverName, accepted);\n    }\n    accepted.add(elicitationId);\n  }\n\n  private async createHttpTransport(\n    definition: ServerDefinition,\n    serverName: string,\n    signal?: AbortSignal,\n  ): Promise<Transport> {\n    throwIfAborted(signal);\n    const url = new URL(definition.url!);\n\n    // Build headers first (including any bearer token)\n    const headers = resolveHeaders(definition.headers) ?? {};\n\n    // For bearer auth, add the token to headers BEFORE creating requestInit\n    if (definition.auth === \"bearer\") {\n      const token = resolveBearerToken(definition);\n      if (token) {\n        headers[\"Authorization\"] = `Bearer ${token}`;\n      }\n    }\n\n    // Create request init with headers (Authorization now included for bearer auth)\n    const requestInit = Object.keys(headers).length > 0 ? { headers } : undefined;\n\n    // For OAuth servers, create an auth provider\n    let authProvider: McpOAuthProvider | undefined;\n    if (supportsOAuth(definition)) {\n      const oauthConfig = extractOAuthConfig(definition);\n      authProvider = new McpOAuthProvider(\n        serverName,\n        definition.url!,\n        oauthConfig,\n        {\n          onRedirect: async (_authUrl) => {\n            // URL is captured by startAuth, no need to log\n          },\n        }\n      );\n    }\n\n    // Try StreamableHTTP first (modern MCP servers)\n    const streamableTransport = new StreamableHTTPClientTransport(url, {\n      requestInit,\n      authProvider,\n    });\n\n    try {\n      // Create a test client to verify the transport works\n      const testClient = new Client({ name: \"pi-mcp-probe\", version: \"2.1.2\" });\n      await testClient.connect(streamableTransport, this.buildRequestOptions(definition, signal));\n      await testClient.close().catch(() => {});\n      // Close probe transport before creating fresh one\n      await streamableTransport.close().catch(() => {});\n\n      // StreamableHTTP works - create fresh transport for actual use\n      return new StreamableHTTPClientTransport(url, { requestInit, authProvider });\n    } catch (error) {\n      // StreamableHTTP failed, close and try SSE fallback\n      await streamableTransport.close().catch(() => {});\n\n      // Host cancellation is not transport capability evidence; do not fall\n      // through to SSE when the caller is trying to cancel the connect.\n      if (signal?.aborted) {\n        throwIfAborted(signal);\n      }\n\n      // If this was an UnauthorizedError, don't try SSE - the server needs auth\n      if (error instanceof UnauthorizedError) {\n        throw error;\n      }\n\n      // SSE is the legacy transport\n      return new SSEClientTransport(url, { requestInit, authProvider });\n    }\n  }\n\n  private async fetchAllTools(client: Client, requestOptions?: RequestOptions): Promise<McpTool[]> {\n    const allTools: McpTool[] = [];\n    let cursor: string | undefined;\n\n    do {\n      const result = await client.listTools(cursor ? { cursor } : undefined, requestOptions);\n      allTools.push(...(result.tools ?? []));\n      cursor = result.nextCursor;\n    } while (cursor);\n\n    return allTools;\n  }\n\n  private async fetchAllResources(client: Client, requestOptions?: RequestOptions): Promise<McpResource[]> {\n    try {\n      const allResources: McpResource[] = [];\n      let cursor: string | undefined;\n\n      do {\n        const result = await client.listResources(cursor ? { cursor } : undefined, requestOptions);\n        allResources.push(...(result.resources ?? []));\n        cursor = result.nextCursor;\n      } while (cursor);\n\n      return allResources;\n    } catch {\n      if (requestOptions?.signal?.aborted) {\n        throwIfAborted(requestOptions.signal);\n      }\n      // Server may not support resources\n      return [];\n    }\n  }\n\n  private attachAdapterNotificationHandlers(serverName: string, client: Client): void {\n    client.setNotificationHandler(serverStreamResultPatchNotificationSchema, (notification) => {\n      const listener = this.uiStreamListeners.get(notification.params.streamToken);\n      if (!listener) return;\n      listener(serverName, notification.params);\n    });\n  }\n\n  registerUiStreamListener(streamToken: string, listener: UiStreamListener): void {\n    this.uiStreamListeners.set(streamToken, listener);\n  }\n\n  removeUiStreamListener(streamToken: string): void {\n    this.uiStreamListeners.delete(streamToken);\n  }\n\n  async readResource(name: string, uri: string, signal?: AbortSignal): Promise<ReadResourceResult> {\n    const connection = this.connections.get(name);\n    if (!connection || connection.status !== \"connected\") {\n      throw new Error(`Server \"${name}\" is not connected`);\n    }\n\n    try {\n      this.touch(name);\n      this.incrementInFlight(name);\n      return await connection.client.readResource({ uri }, this.getRequestOptions(name, signal));\n    } finally {\n      this.decrementInFlight(name);\n      this.touch(name);\n    }\n  }\n\n  private async disposeConnection(connection: PendingConnection): Promise<void> {\n    await connection.client.close().catch(() => {});\n    await connection.transport.close().catch(() => {});\n  }\n\n  async close(name: string): Promise<void> {\n    const connection = this.connections.get(name);\n    if (!connection) return;\n\n    // Delete from map BEFORE async cleanup to prevent a race where a\n    // concurrent connect() creates a new connection that our deferred\n    // delete() would then remove, orphaning the new server process.\n    connection.status = \"closed\";\n    this.connections.delete(name);\n    this.acceptedUrlElicitations.delete(name);\n    await this.disposeConnection(connection);\n  }\n\n  async closeAll(): Promise<void> {\n    this.closing = true;\n    const names = [...this.connections.keys()];\n    const pending = [...this.pendingConnections.values()];\n    await Promise.all([\n      ...names.map(name => this.close(name)),\n      ...pending.map(connection => this.disposeConnection(connection)),\n    ]);\n  }\n\n  getConnection(name: string): ServerConnection | undefined {\n    return this.connections.get(name);\n  }\n\n  getAllConnections(): Map<string, ServerConnection> {\n    return new Map(this.connections);\n  }\n\n  touch(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection) {\n      connection.lastUsedAt = Date.now();\n    }\n  }\n\n  incrementInFlight(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection) {\n      connection.inFlight = (connection.inFlight ?? 0) + 1;\n    }\n  }\n\n  decrementInFlight(name: string): void {\n    const connection = this.connections.get(name);\n    if (connection && connection.inFlight) {\n      connection.inFlight--;\n    }\n  }\n\n  isIdle(name: string, timeoutMs: number): boolean {\n    const connection = this.connections.get(name);\n    if (!connection || connection.status !== \"connected\") return false;\n    if (connection.inFlight > 0) return false;\n    return (Date.now() - connection.lastUsedAt) > timeoutMs;\n  }\n}\n\n/**\n * Resolve environment variables with interpolation.\n */\nfunction resolveEnv(env?: Record<string, string>): Record<string, string> {\n  // Copy process.env, filtering out undefined values\n  const resolved: Record<string, string> = {};\n  for (const [key, value] of Object.entries(process.env)) {\n    if (value !== undefined) {\n      resolved[key] = value;\n    }\n  }\n\n  if (!env) return resolved;\n\n  const overrides = interpolateEnvRecord(env);\n  return overrides ? { ...resolved, ...overrides } : resolved;\n}\n\n/**\n * Resolve headers with environment variable interpolation.\n */\nfunction resolveHeaders(headers?: Record<string, string>): Record<string, string> | undefined {\n  return interpolateEnvRecord(headers);\n}\n\nfunction normalizeRequestTimeoutMs(timeoutMs: number | undefined): number | undefined {\n  return typeof timeoutMs === \"number\" && Number.isFinite(timeoutMs) && timeoutMs > 0\n    ? timeoutMs\n    : undefined;\n}\n"]}

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

@@ -18,6 +18,8 @@ export class McpServerManager {
     defaultCwd;
     connections = new Map();
     connectPromises = new Map();
+    pendingConnections = new Map();
+    closing = false;
     uiStreamListeners = new Map();
     samplingConfig;
     elicitationConfig;
@@ -58,6 +60,8 @@ export class McpServerManager {
     }
     async connect(name, definition, signal) {
         throwIfAborted(signal);
+        if (this.closing)
+            throw new Error("MCP server manager is closed");
         // Dedupe concurrent connection attempts
         if (this.connectPromises.has(name)) {
             return abortable(this.connectPromises.get(name), signal);
@@ -68,15 +72,22 @@ export class McpServerManager {
             existing.lastUsedAt = Date.now();
             return existing;
         }
-        const promise = this.createConnection(name, definition, signal);
-        this.connectPromises.set(name, promise);
-        try {
-            const connection = await promise;
+        const promise = this.createConnection(name, definition, signal).then(async connection => {
+            if (this.closing || signal?.aborted) {
+                await this.disposeConnection(connection);
+                throwIfAborted(signal);
+                throw new Error("MCP server manager closed during connection");
+            }
             this.connections.set(name, connection);
             return connection;
+        });
+        this.connectPromises.set(name, promise);
+        try {
+            return await promise;
         }
         finally {
-            this.connectPromises.delete(name);
+            if (this.connectPromises.get(name) === promise)
+                this.connectPromises.delete(name);
         }
     }
     async createConnection(name, definition, signal) {
@@ -110,6 +121,8 @@ export class McpServerManager {
             throw new Error(`Server ${name} has no command or url`);
         }
         const requestOptions = this.buildRequestOptions(definition, signal);
+        const pending = { client, transport };
+        this.pendingConnections.set(name, pending);
         try {
             await client.connect(transport, requestOptions);
             this.attachAdapterNotificationHandlers(name, client);
@@ -151,6 +164,10 @@ export class McpServerManager {
             await transport.close().catch(() => { });
             throw error;
         }
+        finally {
+            if (this.pendingConnections.get(name) === pending)
+                this.pendingConnections.delete(name);
+        }
     }
     buildClientCapabilities() {
         return {
@@ -323,6 +340,10 @@ export class McpServerManager {
             this.touch(name);
         }
     }
+    async disposeConnection(connection) {
+        await connection.client.close().catch(() => { });
+        await connection.transport.close().catch(() => { });
+    }
     async close(name) {
         const connection = this.connections.get(name);
         if (!connection)
@@ -333,12 +354,16 @@ export class McpServerManager {
         connection.status = "closed";
         this.connections.delete(name);
         this.acceptedUrlElicitations.delete(name);
-        await connection.client.close().catch(() => { });
-        await connection.transport.close().catch(() => { });
+        await this.disposeConnection(connection);
     }
     async closeAll() {
+        this.closing = true;
         const names = [...this.connections.keys()];
-        await Promise.all(names.map(name => this.close(name)));
+        const pending = [...this.pendingConnections.values()];
+        await Promise.all([
+            ...names.map(name => this.close(name)),
+            ...pending.map(connection => this.disposeConnection(connection)),
+        ]);
     }
     getConnection(name) {
         return this.connections.get(name);

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

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