Class MCPServerManager

src/node/services/mcpServerManager.ts:599–2022

}

export class MCPServerManager {
  private readonly workspaceServers = new Map<string, WorkspaceServers>();
  private readonly workspaceLeases = new Map<string, number>();
  private readonly idleCheckInterval: ReturnType<typeof setInterval>;
  private inlineServers: Record<string, string> = {};
  private readonly policyService: PolicyService | null;
  private mcpOauthService: McpOauthService | null = null;
  private ignoreConfigFile = false;

  setMcpOauthService(service: McpOauthService): void {
    this.mcpOauthService = service;
  }
  constructor(
    private readonly configService: MCPConfigService,
    options?: MCPServerManagerOptions,
    policyService?: PolicyService
  ) {
    this.policyService = policyService ?? null;
    this.idleCheckInterval = setInterval(() => this.cleanupIdleServers(), IDLE_CHECK_INTERVAL_MS);
    this.idleCheckInterval.unref?.();
    if (options?.inlineServers) {
      this.inlineServers = options.inlineServers;
    }
    if (options?.ignoreConfigFile) {
      this.ignoreConfigFile = options.ignoreConfigFile;
    }
  }

  /**
   * Stop the idle cleanup interval. Call when shutting down.
   */
  dispose(): void {
    clearInterval(this.idleCheckInterval);
  }

  private getLeaseCount(workspaceId: string): number {
    return this.workspaceLeases.get(workspaceId) ?? 0;
  }

  /**
   * Mark a workspace's MCP servers as actively in-use.
   *
   * This prevents idle cleanup from shutting down MCP clients while a stream is
   * still running (which can otherwise surface as "Attempted to send a request
   * from a closed client").
   */
  acquireLease(workspaceId: string): void {
    const current = this.workspaceLeases.get(workspaceId) ?? 0;
    this.workspaceLeases.set(workspaceId, current + 1);
    this.markActivity(workspaceId);
  }

  /**
   * Release a previously-acquired lease.
   */
  releaseLease(workspaceId: string): void {
    const current = this.workspaceLeases.get(workspaceId) ?? 0;
    if (current <= 0) {