Class McpManagerImpl

packages/mcp-client/src/manager.ts:71–235

// ============================================================================

class McpManagerImpl implements McpManager {
  private connections = new Map<string, MCPServerConnection>()
  private toolsCache = new Map<string, CoreTool[]>()
  private listeners = new Map<string, Set<EventHandler>>()
  private deps: McpClientDependencies
  private connectFn:
    | ((
        name: string,
        config: ScopedMcpServerConfig,
      ) => Promise<MCPServerConnection>)
    | null = null

  constructor(deps: McpClientDependencies) {
    this.deps = deps
  }

  /** Set the connect function — the host provides this with all transport logic */
  setConnectFn(
    fn: (
      name: string,
      config: ScopedMcpServerConfig,
    ) => Promise<MCPServerConnection>,
  ): void {
    this.connectFn = fn
  }

  async connect(
    name: string,
    config: McpServerConfig,
  ): Promise<MCPServerConnection> {
    if (!this.connectFn) {
      throw new Error(
        'McpManager: connectFn not set. Call setConnectFn() first.',
      )
    }

    const scopedConfig: ScopedMcpServerConfig = { ...config, scope: 'dynamic' }

    try {
      const connection = await this.connectFn(name, scopedConfig)
      this.connections.set(name, connection)

      if (connection.type === 'connected') {
        this.emit('connected', name)
        // Fetch tools for this server
        await this.refreshTools(name, connection)
      } else if (connection.type === 'needs-auth') {
        this.emit('authRequired', name)
      }

      return connection
    } catch (err) {
      const error = err instanceof Error ? err : new Error(String(err))
      this.emit('error', name, error)
      throw err
    }
  }