aegiscode 3.1.10 → 3.4.2

package/src/mcp/HealthMonitor.ts

@@ -1,150 +0,0 @@
-/**
- * MCP 服务器健康监控
- * 
- */
-
-import { EventEmitter } from 'events';
-import type { HealthCheckConfig, McpClientInterface } from './types.js';
-import { McpConnectionStatus, DEFAULT_HEALTH_CHECK_CONFIG } from './types.js';
-import { createDebugLogger } from '../utils/debug.js';
-
-/**
- * 
- */
-export class HealthMonitor extends EventEmitter {
-  private checkTimer: NodeJS.Timeout | null = null;
-  private consecutiveFailures = 0;
-  private isRunning = false;
-  private lastCheckTime: Date | null = null;
-  private lastCheckResult: 'healthy' | 'unhealthy' | 'unknown' = 'unknown';
-  private debug: ReturnType<typeof createDebugLogger>;
-
-  constructor(
-    private client: McpClientInterface,
-    private config: HealthCheckConfig = DEFAULT_HEALTH_CHECK_CONFIG
-  ) {
-    super();
-    this.debug = createDebugLogger(`HealthMonitor:${client.serverName}`);
-  }
-
-  /**
-   * 
-   */
-  start(): void {
-    if (this.isRunning) {
-      return;
-    }
-
-    this.isRunning = true;
-    this.consecutiveFailures = 0;
-    this.scheduleNextCheck();
-
-    this.debug.log(
-      `健康监控已启动`,
-      `(间隔: ${this.config.intervalMs}ms, 超时: ${this.config.timeoutMs}ms)`
-    );
-  }
-
-  /**
-   * 
-   */
-  stop(): void {
-    this.isRunning = false;
-    if (this.checkTimer) {
-      clearTimeout(this.checkTimer);
-      this.checkTimer = null;
-    }
-    this.debug.log(`健康监控已停止`);
-  }
-
-  /**
-   * 
-   */
-  getStatus(): {
-    isHealthy: boolean;
-    consecutiveFailures: number;
-    lastCheckTime: Date | null;
-    lastCheckResult: 'healthy' | 'unhealthy' | 'unknown';
-  } {
-    return {
-      isHealthy: this.consecutiveFailures < this.config.maxFailures,
-      consecutiveFailures: this.consecutiveFailures,
-      lastCheckTime: this.lastCheckTime,
-      lastCheckResult: this.lastCheckResult,
-    };
-  }
-
-  /**
-   * 
-   */
-  private scheduleNextCheck(): void {
-    if (!this.isRunning) {
-      return;
-    }
-
-    this.checkTimer = setTimeout(async () => {
-      await this.performCheck();
-      this.scheduleNextCheck();
-    }, this.config.intervalMs);
-  }
-
-  /**
-   * 
-   */
-  private async performCheck(): Promise<void> {
-    if (!this.isRunning) {
-      return;
-    }
-
-    // 只在连接状态下进行检
-    if (this.client.connectionStatus !== McpConnectionStatus.CONNECTED) {
-      this.lastCheckResult = 'unknown';
-      return;
-    }
-
-    this.lastCheckTime = new Date();
-
-    try {
-      // 使用超时包装检
-      const checkPromise = this.doHealthCheck();
-      const timeoutPromise = new Promise<never>((_, reject) => {
-        setTimeout(() => reject(new Error('健康检查超时')), this.config.timeoutMs);
-      });
-
-      await Promise.race([checkPromise, timeoutPromise]);
-
-      // 检查成
-      this.consecutiveFailures = 0;
-      this.lastCheckResult = 'healthy';
-      this.emit('healthy');
-    } catch (error) {
-      // 检查失
-      this.consecutiveFailures++;
-      this.lastCheckResult = 'unhealthy';
-
-      this.debug.warn(
-        `健康检查失败`,
-        `(${this.consecutiveFailures}/${this.config.maxFailures}):`,
-        (error as Error).message
-      );
-
-      // 超过最大失败次数，触发不健康事
-      if (this.consecutiveFailures >= this.config.maxFailures) {
-        this.debug.error(
-          `服务器不健康，连续失败 ${this.consecutiveFailures} 次`
-        );
-        this.emit('unhealthy', this.consecutiveFailures, error);
-      }
-    }
-  }
-
-  /**
-   * 
-   * 
-   */
-  private async doHealthCheck(): Promise<void> {
-    // 简单的健康检查：尝试重新加载工具列
-    // 如果能成功获取，说明连接是健康
-    await this.client.reloadTools();
-  }
-}

package/src/mcp/McpClient.ts

@@ -1,491 +0,0 @@
-/**
- * MCP 客户端
- * 
- */
-
-import { EventEmitter } from 'events';
-import {
-  McpConnectionStatus,
-  McpToolDefinition,
-  McpToolCallResponse,
-  McpServerConfig,
-  HealthCheckConfig,
-  ErrorType,
-  ClassifiedError,
-  DEFAULT_CONNECTION_CONFIG,
-  type McpClientInterface,
-} from './types.js';
-import { HealthMonitor } from './HealthMonitor.js';
-import { createDebugLogger } from '../utils/debug.js';
-
-/**
- * 
- */
-function classifyError(error: unknown): ClassifiedError {
-  if (!(error instanceof Error)) {
-    return {
-      type: ErrorType.UNKNOWN,
-      isRetryable: false,
-      originalError: new Error(String(error)),
-    };
-  }
-
-  const msg = error.message.toLowerCase();
-
-  // 永久性配置错误（不重
-  const permanentErrors = [
-    'command not found',
-    'no such file',
-    'permission denied',
-    'invalid configuration',
-    'enoent',
-    'spawn',
-  ];
-  if (permanentErrors.some(p => msg.includes(p))) {
-    return { type: ErrorType.CONFIG_ERROR, isRetryable: false, originalError: error };
-  }
-
-  // 认证错误（需要用户介
-  if (msg.includes('unauthorized') || msg.includes('401') || msg.includes('authentication failed')) {
-    return { type: ErrorType.AUTH_ERROR, isRetryable: false, originalError: error };
-  }
-
-  // 临时网络错误（可重
-  const temporaryErrors = [
-    'timeout',
-    'connection refused',
-    'network error',
-    'rate limit',
-    '503',
-    '429',
-    'econnrefused',
-    'etimedout',
-  ];
-  if (temporaryErrors.some(t => msg.includes(t))) {
-    return { type: ErrorType.NETWORK_TEMPORARY, isRetryable: true, originalError: error };
-  }
-
-  // 默认允许重
-  return { type: ErrorType.UNKNOWN, isRetryable: true, originalError: error };
-}
-
-/**
- * MCP 客户端实现
- */
-export class McpClient extends EventEmitter implements McpClientInterface {
-  private status: McpConnectionStatus = McpConnectionStatus.DISCONNECTED;
-  private sdkClient: any = null;  // @modelcontextprotocol/sdk Client
-  private tools = new Map<string, McpToolDefinition>();
-  private serverInfo: { name: string; version: string } | null = null;
-
-  // 重连配
-  private reconnectAttempts = 0;
-  private reconnectTimer: NodeJS.Timeout | null = null;
-  private isManualDisconnect = false;
-
-  // 健康监
-  private healthMonitor: HealthMonitor | null = null;
-
-  // 服务器名称（用于日
-  public readonly serverName: string;
-  
-  // Debug logger
-  private debug: ReturnType<typeof createDebugLogger>;
-
-  constructor(
-    private config: McpServerConfig,
-    serverName?: string,
-    healthCheckConfig?: HealthCheckConfig
-  ) {
-    super();
-    this.serverName = serverName || 'default';
-    this.debug = createDebugLogger(`McpClient:${this.serverName}`);
-
-    // 初始化健康监
-    if (healthCheckConfig?.enabled) {
-      this.healthMonitor = new HealthMonitor(this, healthCheckConfig);
-      this.healthMonitor.on('unhealthy', (failures: number, error: Error) => {
-        this.emit('unhealthy', failures, error);
-        // 触发重
-        if (this.status === McpConnectionStatus.CONNECTED) {
-          this.handleUnexpectedClose();
-        }
-      });
-    }
-  }
-
-  /**
-   * 
-   */
-  get connectionStatus(): McpConnectionStatus {
-    return this.status;
-  }
-
-  /**
-   * 
-   */
-  get availableTools(): McpToolDefinition[] {
-    return Array.from(this.tools.values());
-  }
-
-  /**
-   * 
-   */
-  private setStatus(status: McpConnectionStatus): void {
-    const oldStatus = this.status;
-    this.status = status;
-    if (oldStatus !== status) {
-      this.emit('statusChanged', status, oldStatus);
-    }
-  }
-
-  /**
-   * 
-   */
-  async connectWithRetry(
-    maxRetries = DEFAULT_CONNECTION_CONFIG.maxRetries,
-    initialDelay = DEFAULT_CONNECTION_CONFIG.initialDelay
-  ): Promise<void> {
-    if (this.status !== McpConnectionStatus.DISCONNECTED) {
-      throw new Error('客户端已连接或正在连接中');
-    }
-
-    let lastError: Error | null = null;
-
-    for (let attempt = 1; attempt <= maxRetries; attempt++) {
-      try {
-        await this.doConnect();
-        this.reconnectAttempts = 0;
-        return;
-      } catch (error) {
-        lastError = error as Error;
-        const classified = classifyError(error);
-
-        this.debug.warn(
-          `连接失败（${attempt}/${maxRetries}）:`,
-          classified.type,
-          (error as Error).message
-        );
-
-        // 永久性错误不重
-        if (!classified.isRetryable) {
-          this.debug.error(`检测到永久性错误，放弃重试`);
-          throw error;
-        }
-
-        // 指数退避重
-        if (attempt < maxRetries) {
-          const delay = initialDelay * Math.pow(2, attempt - 1);
-          this.debug.log(`${delay}ms 后重试...`);
-          await new Promise(resolve => setTimeout(resolve, delay));
-        }
-      }
-    }
-
-    throw lastError || new Error('连接失败');
-  }
-
-  /**
-   * 
-   */
-  private async doConnect(): Promise<void> {
-    try {
-      this.setStatus(McpConnectionStatus.CONNECTING);
-
-      // 动态导
-      const { Client } = await import('@modelcontextprotocol/sdk/client/index.js');
-
-      // 创建 SDK 客户
-      this.sdkClient = new Client(
-        { name: 'aegis', version: '1.0.0' },
-        { capabilities: { roots: { listChanged: true }, sampling: {} } }
-      );
-
-      // 监听关闭事
-      this.sdkClient.onclose = () => this.handleUnexpectedClose();
-
-      // 创建传输
-      const transport = await this.createTransport();
-
-      // 连
-      await this.sdkClient.connect(transport);
-
-      // 获取服务器信
-      const serverVersion = this.sdkClient.getServerVersion?.();
-      this.serverInfo = {
-        name: serverVersion?.name || 'Unknown',
-        version: serverVersion?.version || '0.0.0',
-      };
-
-      // 加载工具列
-      await this.loadTools();
-
-      this.setStatus(McpConnectionStatus.CONNECTED);
-      this.emit('connected', this.serverInfo);
-
-      // 启动健康监
-      if (this.healthMonitor) {
-        this.healthMonitor.start();
-      }
-
-      this.debug.log(
-        `已连接到服务器:`,
-        this.serverInfo.name,
-        `v${this.serverInfo.version}`,
-        `(${this.tools.size} 个工具)`
-      );
-    } catch (error) {
-      this.setStatus(McpConnectionStatus.ERROR);
-      this.emit('error', error);
-      throw error;
-    }
-  }
-
-  /**
-   * 
-   */
-  private async createTransport(): Promise<any> {
-    const { type, command, args, env, cwd, url, headers } = this.config;
-
-    if (type === 'stdio') {
-      if (!command) {
-        throw new Error('stdio 传输需要 command 参数');
-      }
-
-      const { StdioClientTransport } = await import(
-        '@modelcontextprotocol/sdk/client/stdio.js'
-      );
-
-      // Whitelist — skicka bara säkra env-variabler till MCP child processes
-      const SAFE_ENV_KEYS = ['PATH', 'HOME', 'USER', 'SHELL', 'LANG', 'LC_ALL', 'TERM', 'TZ'];
-      const mergedEnv: Record<string, string> = {};
-      for (const key of SAFE_ENV_KEYS) {
-        const value = process.env[key];
-        if (value !== undefined) mergedEnv[key] = value;
-      }
-      // Tillåt explicit konfigurerade env-variabler från MCP-serverns config
-      if (env) {
-        Object.assign(mergedEnv, env);
-      }
-
-      return new StdioClientTransport({
-        command,
-        args: args || [],
-        env: mergedEnv,
-        cwd: cwd || process.cwd(),
-        stderr: 'ignore',
-      });
-    }
-
-    if (type === 'sse') {
-      if (!url) {
-        throw new Error('sse 传输需要 url 参数');
-      }
-
-      const { SSEClientTransport } = await import(
-        '@modelcontextprotocol/sdk/client/sse.js'
-      );
-
-      return new SSEClientTransport(new URL(url), {
-        requestInit: { headers: headers || {} },
-      });
-    }
-
-    if (type === 'http') {
-      if (!url) {
-        throw new Error('http 传输需要 url 参数');
-      }
-
-      // HTTP Streamable 传输（如果 SDK 支
-      try {
-        const { StreamableHTTPClientTransport } = await import(
-          '@modelcontextprotocol/sdk/client/streamableHttp.js'
-        );
-        return new StreamableHTTPClientTransport(new URL(url), {
-          requestInit: { headers: headers || {} },
-        });
-      } catch {
-        // 回退
-        const { SSEClientTransport } = await import(
-          '@modelcontextprotocol/sdk/client/sse.js'
-        );
-        return new SSEClientTransport(new URL(url), {
-          requestInit: { headers: headers || {} },
-        });
-      }
-    }
-
-    throw new Error(`不支持的传输类型: ${type}`);
-  }
-
-  /**
-   * 
-   */
-  private async loadTools(): Promise<void> {
-    if (!this.sdkClient) {
-      throw new Error('客户端未连接');
-    }
-
-    try {
-      const result = await this.sdkClient.listTools();
-      const oldCount = this.tools.size;
-
-      this.tools.clear();
-      if (result.tools && Array.isArray(result.tools)) {
-        for (const tool of result.tools) {
-          this.tools.set(tool.name, {
-            name: tool.name,
-            description: tool.description,
-            inputSchema: tool.inputSchema || { type: 'object' },
-          });
-        }
-      }
-
-      if (oldCount !== this.tools.size) {
-        this.emit('toolsUpdated', this.availableTools);
-      }
-    } catch (error) {
-      this.debug.error(`加载工具列表失败:`, error);
-      throw error;
-    }
-  }
-
-  /**
-   * 
-   */
-  async reloadTools(): Promise<void> {
-    await this.loadTools();
-  }
-
-  /**
-   * 
-   */
-  async callTool(
-    name: string,
-    arguments_: Record<string, any> = {}
-  ): Promise<McpToolCallResponse> {
-    if (!this.sdkClient) {
-      throw new Error('客户端未连接到服务器');
-    }
-
-    if (!this.tools.has(name)) {
-      throw new Error(`工具 "${name}" 不存在`);
-    }
-
-    try {
-      const result = await this.sdkClient.callTool({
-        name,
-        arguments: arguments_,
-      });
-
-      return result as McpToolCallResponse;
-    } catch (error) {
-      this.debug.error(`调用工具 "${name}" 失败:`, error);
-      throw error;
-    }
-  }
-
-  /**
-   * 
-   */
-  async disconnect(): Promise<void> {
-    this.isManualDisconnect = true;
-
-    // 停止健康监
-    if (this.healthMonitor) {
-      this.healthMonitor.stop();
-    }
-
-    // 清除重连计时
-    if (this.reconnectTimer) {
-      clearTimeout(this.reconnectTimer);
-      this.reconnectTimer = null;
-    }
-
-    // 关闭 SDK 客户
-    if (this.sdkClient) {
-      try {
-        await this.sdkClient.close();
-      } catch (error) {
-        this.debug.warn(`关闭连接时出错:`, error);
-      }
-      this.sdkClient = null;
-    }
-
-    this.tools.clear();
-    this.setStatus(McpConnectionStatus.DISCONNECTED);
-    this.emit('disconnected');
-
-    this.debug.log(`已断开连接`);
-  }
-
-  /**
-   * 
-   */
-  private handleUnexpectedClose(): void {
-    if (this.isManualDisconnect) {
-      return;
-    }
-
-    if (this.status === McpConnectionStatus.CONNECTED) {
-      this.debug.warn(`检测到意外断连，准备重连...`);
-      this.setStatus(McpConnectionStatus.ERROR);
-      this.emit('error', new Error('MCP服务器连接意外关闭'));
-      this.scheduleReconnect();
-    }
-  }
-
-  /**
-   * 
-   */
-  private scheduleReconnect(): void {
-    if (this.reconnectTimer) {
-      clearTimeout(this.reconnectTimer);
-      this.reconnectTimer = null;
-    }
-
-    if (this.reconnectAttempts >= DEFAULT_CONNECTION_CONFIG.maxReconnectAttempts) {
-      this.debug.error(`达到最大重连次数，放弃重连`);
-      this.emit('reconnectFailed');
-      return;
-    }
-
-    // 指数退避：1s, 2s, 4s, 8s, 16s（最
-    const delay = Math.min(
-      1000 * Math.pow(2, this.reconnectAttempts),
-      DEFAULT_CONNECTION_CONFIG.maxReconnectDelay
-    );
-    this.reconnectAttempts++;
-
-    this.debug.log(
-      `将在 ${delay}ms 后进行第 ${this.reconnectAttempts} 次重连...`
-    );
-
-    this.emit('reconnecting', this.reconnectAttempts);
-
-    this.reconnectTimer = setTimeout(async () => {
-      try {
-        // 清理旧连
-        if (this.sdkClient) {
-          await this.sdkClient.close().catch(() => {});
-          this.sdkClient = null;
-        }
-
-        this.setStatus(McpConnectionStatus.DISCONNECTED);
-        await this.doConnect();
-
-        this.debug.log(`重连成功`);
-        this.reconnectAttempts = 0;
-        this.emit('reconnected');
-      } catch (error) {
-        const classified = classifyError(error);
-        if (classified.isRetryable) {
-          this.scheduleReconnect();
-        } else {
-          this.debug.error(`检测到永久性错误，停止重连`);
-          this.emit('reconnectFailed');
-        }
-      }
-    }, delay);
-  }
-}