aegiscode 3.1.7 → 3.1.10

package/src/config/ConfigManager.ts

@@ -0,0 +1,587 @@
+/**
+ * ConfigManager - 配置管理器（双文件配置架构）
+ * 
+ * 
+ * - config.json: 基础配置（API、模型、UI、MCP）
+ * - settings.json: 行为配置（权限、Hooks、环境变量）
+ * - settings.local.json: 本地行为配置（不提交到 Git）
+ * 
+ * 
+ * 1. 默认配置
+ * 2. 用户 config.json (~/.aegiscode/config.json)
+ * 3. 用户 settings.json (~/.aegiscode/settings.json)
+ * 4. 项目 config.json (./.aegiscode/config.json)
+ * 5. 项目 settings.json (./.aegiscode/settings.json)
+ * 6. 本地 settings.local.json (./.aegiscode/settings.local.json)
+ * 7. 环境变量
+ * 8. CLI 参数
+ */
+
+import * as fs from 'node:fs';
+import * as path from 'node:path';
+import * as os from 'node:os';
+import {
+  Config,
+  ConfigSchema,
+  DEFAULT_CONFIG,
+  ModelConfig,
+  type PermissionConfig,
+  type McpServerConfig,
+} from './types.js';
+
+export class ConfigManager {
+  private static instance: ConfigManager;
+  private config: Config;
+  private configPaths: string[] = [];
+
+  private constructor() {
+    this.config = { ...DEFAULT_CONFIG };
+  }
+
+  /**
+   * 
+   */
+  static getInstance(): ConfigManager {
+    if (!ConfigManager.instance) {
+      ConfigManager.instance = new ConfigManager();
+    }
+    return ConfigManager.instance;
+  }
+
+  /**
+   * 
+   * @returns 加载后的配置
+   */
+  async initialize(projectPath?: string): Promise<Config> {
+    // 1. 从默认配置开
+    this.config = { ...DEFAULT_CONFIG };
+
+    const userConfigDir = this.getUserConfigDir();
+    const projectDir = projectPath || process.cwd();
+    const projectConfigDir = path.join(projectDir, '.aegiscode');
+
+    // 2. 加载用
+    await this.loadConfigFile(path.join(userConfigDir, 'config.json'));
+
+    // 3. 加载用
+    await this.loadConfigFile(path.join(userConfigDir, 'settings.json'));
+
+    // 4. 加载项
+    await this.loadConfigFile(path.join(projectConfigDir, 'config.json'));
+
+    // 5. 加载项
+    await this.loadConfigFile(path.join(projectConfigDir, 'settings.json'));
+
+    // 6. 加载本地 settings.local.json（不提交
+    await this.loadConfigFile(path.join(projectConfigDir, 'settings.local.json'));
+
+    // 7. 应用环境变
+    this.applyEnvironmentVariables();
+
+    return this.config;
+  }
+
+  /**
+   * 
+   */
+  getUserConfigDir(): string {
+    return path.join(os.homedir(), '.aegiscode');
+  }
+
+  /**
+   * 
+   */
+  getUserConfigPath(): string {
+    return path.join(this.getUserConfigDir(), 'config.json');
+  }
+
+  /**
+   * 
+   */
+  private async loadConfigFile(configPath: string): Promise<boolean> {
+    try {
+      if (!fs.existsSync(configPath)) {
+        return false;
+      }
+
+      const content = fs.readFileSync(configPath, 'utf-8');
+      const parsed = JSON.parse(content);
+
+      // Backward compat: if root has apiKey/baseURL/model, wrap in 'default'
+      if (parsed.apiKey || parsed.baseURL || parsed.baseUrl || parsed.model) {
+        if (!parsed.default) {
+          parsed.default = {};
+        }
+        if (parsed.apiKey && !parsed.default.apiKey) parsed.default.apiKey = parsed.apiKey;
+        if (parsed.baseURL && !parsed.default.baseURL) parsed.default.baseURL = parsed.baseURL;
+        if (parsed.baseUrl && !parsed.default.baseURL) parsed.default.baseURL = parsed.baseUrl;
+        if (parsed.model && !parsed.default.model) parsed.default.model = parsed.model;
+      }
+
+      const validated = ConfigSchema.partial().parse(parsed);
+
+      // 深度合并配
+      this.config = this.mergeConfig(this.config, validated);
+      this.configPaths.push(configPath);
+
+      return true;
+    } catch (error) {
+      console.warn(`Warning: Failed to load config from ${configPath}:`, (error as Error).message);
+      return false;
+    }
+  }
+
+
+  /**
+   * 
+   */
+  private applyEnvironmentVariables(): void {
+    // Load ~/.aegiscode/.env
+    try {
+      const lines = (fs.existsSync(path.join(os.homedir(), '.aegiscode', '.env')) ? fs.readFileSync(path.join(os.homedir(), '.aegiscode', '.env'), 'utf8') : '').split('\n');
+      for (const line of lines) {
+        const eq = line.indexOf('=');
+        if (eq < 0 || line.trim().startsWith('#')) continue;
+        const k = line.slice(0, eq).trim();
+        const v = line.slice(eq + 1).trim().replace(/^["']|["']$/g, '');
+        if (k && v && !v.startsWith('YOUR_') && !process.env[k]) process.env[k] = v;
+      }
+    } catch {}
+
+    this.config.default = this.config.default || {};
+    const defaultConfig = this.config.default;
+
+    // Normalize baseUrl -> baseURL
+    if (!defaultConfig.baseURL && (defaultConfig as any).baseUrl) {
+      defaultConfig.baseURL = (defaultConfig as any).baseUrl;
+    }
+
+    // Inject env keys into each model by provider
+    const allModels: any[] = (this.config as any).models || [];
+    for (const m of allModels) {
+      const bu: string = m.baseURL || m.baseUrl || '';
+      // Claude Code Pro/Max subscription OAuth token (from `claude setup-token`,
+      // or `aegis login --claude-pro`) works as a drop-in for ANTHROPIC_API_KEY
+      // against the same model — it does not change which model is selected.
+      // It takes priority over ANTHROPIC_API_KEY: logging in with --claude-pro
+      // is a deliberate choice and shouldn't be shadowed by a stale/out-of-credit
+      // API key still sitting in the environment.
+      if (bu.includes('anthropic') && process.env.CLAUDE_CODE_OAUTH_TOKEN) {
+        m.apiKey = process.env.CLAUDE_CODE_OAUTH_TOKEN;
+      } else if (bu.includes('anthropic') && process.env.ANTHROPIC_API_KEY) {
+        m.apiKey = process.env.ANTHROPIC_API_KEY;
+      } else if (bu.includes('deepseek') && process.env.DEEPSEEK_API_KEY) {
+        m.apiKey = process.env.DEEPSEEK_API_KEY;
+      } else if (bu.includes('groq') && process.env.GROQ_API_KEY) {
+        m.apiKey = process.env.GROQ_API_KEY;
+      } else if (bu.includes('sakana') && process.env.SAKANA_API_KEY) {
+        m.apiKey = process.env.SAKANA_API_KEY;
+      } else if (bu.includes('googleapis') && process.env.GEMINI_API_KEY) {
+        m.apiKey = process.env.GEMINI_API_KEY;
+      } else if (bu.includes('aegisintel') && process.env.NEXUS_API_KEY) {
+        // Nexus - AEGIS's pooled aegis-key, auto-routed across providers server-side.
+        // OpenAI-wire-compatible (streaming + tool_calls), billed via the token bank.
+        m.apiKey = process.env.NEXUS_API_KEY;
+      } else if (bu.includes('openai.com') && process.env.OPENAI_API_KEY) {
+        // Google's OpenAI-compatibility endpoint is
+        // generativelanguage.googleapis.com/v1beta/openai/ — it contains the
+        // bare substring "openai", so when GEMINI_API_KEY isn't set this
+        // branch's old `bu.includes('openai')` check fell through and matched
+        // it anyway, handing Gemini models the OpenAI key (wrong key format,
+        // Google's API rejects it with a bare 400). 'openai.com' only matches
+        // the real api.openai.com host.
+        m.apiKey = process.env.OPENAI_API_KEY;
+      }
+    }
+
+    // currentModelId defaults to a Claude model regardless of which provider's key
+    // the user actually set (env var, .env, or SetupWizard — none of them touch
+    // currentModelId). If the selected model still has no key but a key was found
+    // for a different provider above, switch to that model so getDefaultModel()
+    // doesn't keep returning a keyless Claude entry and reporting "No API key configured".
+    {
+      const currentId = (this.config as any).currentModelId;
+      const selected = allModels.find((m: any) => m.id === currentId);
+      if (!selected?.apiKey) {
+        const fallback = allModels.find((m: any) => m.apiKey);
+        if (fallback) (this.config as any).currentModelId = fallback.id;
+      }
+    }
+
+    // Use currentModelId to pick the right default — don't override with env key priority
+    if (!defaultConfig.apiKey) {
+      const currentId = (this.config as any).currentModelId;
+      const currentModel = allModels.find((m: any) => m.id === currentId);
+      if (currentModel && currentModel.apiKey) {
+        defaultConfig.apiKey  = currentModel.apiKey;
+        defaultConfig.baseURL = currentModel.baseURL || currentModel.baseUrl;
+        defaultConfig.model   = currentModel.model;
+      } else if (process.env.DEEPSEEK_API_KEY) {
+        defaultConfig.apiKey  = process.env.DEEPSEEK_API_KEY;
+        defaultConfig.baseURL = 'https://api.deepseek.com/v1';
+        defaultConfig.model   = 'deepseek-chat';
+      } else if (process.env.GROQ_API_KEY) {
+        defaultConfig.apiKey  = process.env.GROQ_API_KEY;
+        defaultConfig.baseURL = 'https://api.groq.com/openai/v1';
+        defaultConfig.model   = 'llama-3.3-70b-versatile';
+      } else if (process.env.OPENAI_API_KEY) {
+        defaultConfig.apiKey  = process.env.OPENAI_API_KEY;
+        if (process.env.OPENAI_BASE_URL) defaultConfig.baseURL = process.env.OPENAI_BASE_URL;
+        if (process.env.OPENAI_MODEL)    defaultConfig.model   = process.env.OPENAI_MODEL;
+        else if (!defaultConfig.model)   defaultConfig.model   = 'gpt-4o';
+      } else if (process.env.CLAUDE_CODE_OAUTH_TOKEN) {
+        defaultConfig.apiKey  = process.env.CLAUDE_CODE_OAUTH_TOKEN;
+        defaultConfig.baseURL = 'https://api.anthropic.com/v1';
+        defaultConfig.model   = defaultConfig.model || 'claude-sonnet-4-6';
+      } else if (process.env.ANTHROPIC_API_KEY) {
+        defaultConfig.apiKey  = process.env.ANTHROPIC_API_KEY;
+        defaultConfig.baseURL = 'https://api.anthropic.com/v1';
+        defaultConfig.model   = defaultConfig.model || 'claude-sonnet-4-6';
+      } else if (process.env.NEXUS_API_KEY) {
+        defaultConfig.apiKey  = process.env.NEXUS_API_KEY;
+        defaultConfig.baseURL = 'https://aegisintel.up.railway.app/api/v1';
+        defaultConfig.model   = defaultConfig.model || 'nexus-smart';
+      }
+    }
+  }
+
+  /**
+   * 
+   */
+  applyCliArgs(args: Partial<ModelConfig>): void {
+    this.config.default = this.config.default || {};
+    const defaultConfig = this.config.default;
+
+    if (args.apiKey) {
+      defaultConfig.apiKey = args.apiKey;
+    }
+    if (args.baseURL) {
+      defaultConfig.baseURL = args.baseURL;
+    }
+    if (args.model) {
+      const models: any[] = (this.config as any).models || [];
+      const found = models.find((m: any) => m.id === args.model || m.model === args.model);
+      if (found) {
+        // args.model is a model ID — switch fully to that model's config
+        (this.config as any).currentModelId = found.id;
+        this.config.default = { ...defaultConfig, ...found };
+      } else {
+        defaultConfig.model = args.model;
+      }
+    }
+  }
+
+  /**
+   * 
+   */
+  private mergeConfig(base: Config, override: Partial<Config>): Config {
+    const merged: Config = {
+      ...base,
+    };
+
+    if (override.default || base.default) {
+      merged.default = {
+        ...base.default,
+        ...override.default,
+      };
+    }
+
+    if (override.models) {
+      merged.models = override.models;
+    }
+
+    if (override.ui || base.ui) {
+      merged.ui = {
+        ...base.ui,
+        ...override.ui,
+      };
+    }
+
+    // 合并权限配置（数组合并而非覆
+    if (override.permissions || base.permissions) {
+      merged.permissions = {
+        allow: [
+          ...(base.permissions?.allow || []),
+          ...(override.permissions?.allow || []),
+        ],
+        deny: [
+          ...(base.permissions?.deny || []),
+          ...(override.permissions?.deny || []),
+        ],
+        ask: [
+          ...(base.permissions?.ask || []),
+          ...(override.permissions?.ask || []),
+        ],
+      };
+    }
+
+    // currentModelId — explicit merge
+    if ((override as any).currentModelId) {
+      (merged as any).currentModelId = (override as any).currentModelId;
+    }
+    // 其他字段使用后者覆
+    if (override.defaultPermissionMode) {
+      merged.defaultPermissionMode = override.defaultPermissionMode;
+    }
+    if (override.toolWhitelist) {
+      merged.toolWhitelist = override.toolWhitelist;
+    }
+    if (override.toolBlacklist) {
+      merged.toolBlacklist = override.toolBlacklist;
+    }
+
+    // 合并 MCP 服务器配置（深度合
+    if (override.mcpServers || base.mcpServers) {
+      merged.mcpServers = {
+        ...(base.mcpServers || {}),
+        ...(override.mcpServers || {}),
+      };
+    }
+
+    // MCP 启用状
+    if (override.mcpEnabled !== undefined) {
+      merged.mcpEnabled = override.mcpEnabled;
+    }
+
+    // 合并 Hooks 配置（深度合
+    if (override.hooks || base.hooks) {
+      merged.hooks = {
+        ...(base.hooks || {}),
+        ...(override.hooks || {}),
+      };
+    }
+
+    return merged;
+  }
+
+  /**
+   * 
+   */
+  getConfig(): Config {
+    return this.config;
+  }
+
+  /**
+   * 
+   */
+  getDefaultModel(): ModelConfig {
+    // Prefer currentModelId from config
+    // Läs currentModelId direkt från fil (Zod strippar det)
+    let currentId = (this.config as any).currentModelId;
+    if (!currentId) {
+      try {
+        const raw = JSON.parse(fs.readFileSync(
+          path.join(os.homedir(), '.aegiscode', 'config.json'), 'utf8'
+        ));
+        currentId = raw.currentModelId;
+      } catch {}
+    }
+    const models: any[] = (this.config as any).models || [];
+    if (currentId && models.length > 0) {
+      let found = models.find((m: any) => m.id === currentId);
+      // Fallback: match by model name
+      if (!found) found = models.find((m: any) => m.model === currentId);
+      if (found) {
+        const result: any = { ...found };
+        if (!result.baseURL && result.baseUrl) result.baseURL = result.baseUrl;
+        return result;
+      }
+    }
+    // Fallback to default
+    const base: any = { ...(this.config.default ?? {}) };
+    if (!base.baseURL && base.baseUrl) base.baseURL = base.baseUrl;
+    return base;
+  }
+
+  /**
+   * 
+   */
+  getLoadedConfigPaths(): string[] {
+    return [...this.configPaths];
+  }
+
+  /**
+   * 
+   */
+  getPermissionConfig(): PermissionConfig {
+    return this.config.permissions || { allow: [], deny: [], ask: [] };
+  }
+
+  /**
+   * Hämta modellspecifika verktygsbegränsningar
+   * 
+   */
+  getModelToolRestrictions(modelId?: string): { allowedTools?: string[]; disallowedTools?: string[] } {
+    if (!modelId) return {};
+    const models: any[] = (this.config as any).models || [];
+    let model = models.find((m: any) => m.id === modelId);
+    if (!model) model = models.find((m: any) => m.model === modelId);
+    if (!model && (this.config.default as any)?.model === modelId) {
+      model = this.config.default;
+    }
+    if (!model) return {};
+    return {
+      allowedTools: model.allowedTools,
+      disallowedTools: model.disallowedTools,
+    };
+  }
+
+  /**
+   * 
+   */
+  getDefaultPermissionMode(): string {
+    return this.config.defaultPermissionMode || 'default';
+  }
+
+  /**
+   * Update the default permission mode at runtime (used by /yolo toggle).
+   * This keeps ConfigManager in sync with config file changes without a full reload.
+   */
+  setDefaultPermissionMode(mode: 'default' | 'autoEdit' | 'yolo' | 'plan'): void {
+    this.config.defaultPermissionMode = mode;
+  }
+
+  /**
+   * 
+   */
+  getMcpServers(): Record<string, McpServerConfig> {
+    return this.config.mcpServers || {};
+  }
+
+  /**
+   * 
+   */
+  isMcpEnabled(): boolean {
+    return this.config.mcpEnabled !== false;
+  }
+
+  /**
+   * Extended-thinking budget tier — drives `thinking`/`output_config.effort`
+   * on the native Anthropic request path. Defaults to 'off'.
+   */
+  getThinkingBudget(): 'off' | 'low' | 'medium' | 'high' | 'max' {
+    return this.config.thinking?.budget || 'off';
+  }
+
+  /**
+   * 
+   */
+  async createDefaultConfig(): Promise<string> {
+    const configDir = this.getUserConfigDir();
+    const configPath = this.getUserConfigPath();
+
+    // 创建目
+    if (!fs.existsSync(configDir)) {
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+
+    // 写入默认配置（不包含敏感信
+    const defaultConfig: Config = {
+      default: {
+        model: 'claude-sonnet-4-6',
+        // apiKey: 'your-api-key',  // 需要用户自己填
+        // baseURL: 'https://api.openai.com/v1',
+      },
+      ui: {
+        theme: 'dark',
+      },
+      mcpEnabled: true,
+      mcpServers: {
+        // 示例 MCP 服务器配置（已注
+        // github: {
+        //   type: 'stdio',
+        //   command: 'npx',
+        //   args: ['-y', '@modelcontextprotocol/server-github'],
+        //   env: { GITHUB_TOKEN: '${GITHUB_TOKEN}' },
+        // },
+      },
+    };
+
+    const content = JSON.stringify(defaultConfig, null, 2);
+    fs.writeFileSync(configPath, content, 'utf-8');
+
+    return configPath;
+  }
+
+  /**
+   * 
+   */
+  isConfigValid(): boolean {
+    return !!this.config.default?.apiKey;
+  }
+
+  /**
+   * 
+   * 
+   * 
+   * 
+   */
+  getTheme(): string | null {
+    // 优先读取用户配置中的主题（用户主动设置
+    const userConfigPath = this.getUserConfigPath();
+    
+    if (fs.existsSync(userConfigPath)) {
+      try {
+        const content = fs.readFileSync(userConfigPath, 'utf-8');
+        const userConfig = JSON.parse(content);
+        
+        // 如果用户配置中有明确设置的主题，使用
+        if (userConfig.theme) {
+          return userConfig.theme;
+        }
+        if (userConfig.ui?.theme) {
+          return userConfig.ui.theme;
+        }
+      } catch {
+        // 解析失败，返
+      }
+    }
+    
+    // 没有用户设置的主题，返回 null 表示应该自动检
+    return null;
+  }
+
+  /**
+   * 
+   */
+  saveTheme(themeName: string): void {
+    const configPath = this.getUserConfigPath();
+    const configDir = this.getUserConfigDir();
+
+    // 确保目录存
+    if (!fs.existsSync(configDir)) {
+      fs.mkdirSync(configDir, { recursive: true });
+    }
+
+    // 读取现有配置或创建新配
+    let existingConfig: Partial<Config> = {};
+    if (fs.existsSync(configPath)) {
+      try {
+        const content = fs.readFileSync(configPath, 'utf-8');
+        existingConfig = JSON.parse(content);
+      } catch {
+        // 如果解析失败，使用空配
+      }
+    }
+
+    // 更新主
+    existingConfig.theme = themeName;
+    if (!existingConfig.ui) {
+      existingConfig.ui = {};
+    }
+    existingConfig.ui.theme = themeName;
+
+    // 同时更新内存中的配
+    this.config.theme = themeName;
+    if (this.config.ui) {
+      this.config.ui.theme = themeName;
+    }
+
+    // 写入文
+    const content = JSON.stringify(existingConfig, null, 2);
+    fs.writeFileSync(configPath, content, 'utf-8');
+  }
+}
+
+// 导出单
+export const configManager = ConfigManager.getInstance();

package/src/config/index.ts

@@ -0,0 +1,7 @@
+/**
+ * 
+ */
+
+export { ConfigManager, configManager } from './ConfigManager.js';
+export { ConfigSchema, ModelConfigSchema, DEFAULT_CONFIG } from './types.js';
+export type { Config, ModelConfig, UIConfig } from './types.js';