aegiscode 3.1.10 → 3.4.2

package/src/main.tsx

@@ -1,596 +0,0 @@
-/**
- * AEGIS CLI - 主入口
- *
- * 
- * 1. 早期解析 --debug 参数（确保日志可用）
- * 2. 启动版本检查（不等待，与后续流程并行）
- * 3. 创建 yargs CLI 实例
- * 4. 注册全局选项和命令
- * 5. 执行中间件链（validatePermissions → loadConfiguration → validateOutput）
- * 6. 执行默认命令 → 启动 React UI（传递 versionCheckPromise）
- *
- * 
- * 1. 默认配置
- * 2. 用户配置 (~/.aegiscode/config.json)
- * 3. 项目配置 (./.aegiscode/config.json)
- * 4. 环境变量 (OPENAI_API_KEY, OPENAI_BASE_URL, OPENAI_MODEL)
- * 5. CLI 参数 (--api-key, --base-url, --model)
- */
-
-// RAF polyfill — MUST run before any module that uses requestAnimationFrame.
-// Pass Date.now() as the timestamp so RAF callbacks that throttle by timestamp
-// (e.g. MessageList's RAF_INTERVAL_MS check) work correctly. Without this,
-// `now` is undefined and `undefined - ref < 30` is always false, causing the
-// streaming render loop to fire at the raw setTimeout rate (~16ms) instead of 30ms.
-if (typeof globalThis.requestAnimationFrame === 'undefined') {
-  (globalThis as any).requestAnimationFrame = (cb: FrameRequestCallback) => setTimeout(() => cb(Date.now()), 16);
-  (globalThis as any).cancelAnimationFrame  = (id: number) => clearTimeout(id);
-}
-
-import { config as dotenvConfig } from 'dotenv';
-import { resolve } from 'path';
-import { homedir } from 'os';
-// --print's whole point is clean, pipeable stdout (text or JSON) — the
-// dotenvx promo banner would otherwise land on stdout ahead of the result.
-const isPrintMode = process.argv.includes('--print') || process.argv.includes('-p');
-// SetupWizard saves keys to ~/.aegiscode/.env (the only place a globally-installed
-// `aegis` binary can reliably find them — process.cwd() is wherever the user happens
-// to invoke it from). Load that first, then ./.env so a project-local file can override it.
-dotenvConfig({ path: resolve(homedir(), '.aegiscode', '.env'), quiet: isPrintMode });
-dotenvConfig({ path: resolve(process.cwd(), '.env'), quiet: isPrintMode, override: true });
-import React from 'react';
-import yargs from 'yargs';
-import { hideBin } from 'yargs/helpers';
-import { App } from './ui/App.js';
-import { configManager } from './config/index.js';
-import { cliConfig, globalOptions, middlewareChain } from './cli/index.js';
-import { checkVersionOnStartup } from './services/index.js';
-import { runCouncil } from './slash-commands/council.js';
-import { getLatestSessionFile } from './context/index.js';
-import { themeManager } from './ui/themes/index.js';
-import { setGlobalDebug } from './utils/debug.js';
-import type { CliArguments } from './cli/types.js';
-import type { VersionCheckResult } from './services/VersionChecker.js';
-import * as path from 'node:path';
-
-// Rendering debugger (auto-starts with --debug-rendering flag)
-const ENABLE_RENDER_DEBUG = process.argv.includes('--debug-rendering');
-
-// ========== 全局状
-let isDebugMode = false;
-let versionCheckPromise: Promise<VersionCheckResult | null> | undefined;
-
-/**
- * 
- *
- * 
- * - Logger 在各模块中被创建
- * - 如果等 yargs 解析完再设置 debug，部分初始化日志会丢失
- * - 早期解析确保所有日志都能正确输出
- */
-function parseDebugEarly(): void {
-  const rawArgs = hideBin(process.argv)
-  const debugIndex = rawArgs.indexOf('--debug')
-  const shortDebugIndex = rawArgs.indexOf('-d')
-
-  if (debugIndex !== -1 || shortDebugIndex !== -1) {
-    isDebugMode = true
-    setGlobalDebug(true)  // 设置全局 debug 状态，供其他模块使
-    console.log('[DEBUG] Debug mode enabled via early parsing')
-  }
-}
-
-/**
- * 
- */
-async function main(): Promise<void> {
-  // 1. 早期解
-  parseDebugEarly();
-
-  // One-shot memory introspection flags for external callers (e.g. aegiscode-gui's
-  // Electron main process) — single source of truth instead of duplicating the
-  // SQLite schema/read logic in another codebase. MUST run before anything below
-  // imports SharedMemory (its singleton constructor fires init() — TTL eviction +
-  // cloud sync — synchronously on first import) or makes network calls (version
-  // check, token verification) that a fast read-only call has no business paying for.
-  const earlyArgs = process.argv.slice(2);
-  // One-shot account re-verification for external callers (aegiscode-gui) — same
-  // single-source-of-truth reasoning as the memory flags below.
-  if (earlyArgs[0] === '--verify-account-json') {
-    const { verifyAccount } = await import('./auth/login.js');
-    console.log(JSON.stringify(await verifyAccount()));
-    process.exit(0);
-  }
-
-  if (earlyArgs[0] === '--memory-stats-json' || earlyArgs[0] === '--memory-search-json' || earlyArgs[0] === '--memory-clear-json' || earlyArgs[0] === '--memory-upload-json' || earlyArgs[0] === '--memory-download-json') {
-    process.env.AEGIS_MEMORY_READONLY = '1';
-    const { sharedMemory } = await import('./memory/SharedMemory.js');
-    await sharedMemory.whenReady();
-
-    if (earlyArgs[0] === '--memory-stats-json') {
-      console.log(JSON.stringify(sharedMemory.getStats()));
-    } else if (earlyArgs[0] === '--memory-search-json') {
-      const query = earlyArgs[1] || '';
-      const limit = parseInt(earlyArgs[2] || '50', 10);
-      const results = query ? await sharedMemory.search(query, limit) : sharedMemory.recent(limit);
-      console.log(JSON.stringify(results));
-    } else if (earlyArgs[0] === '--memory-upload-json') {
-      console.log(JSON.stringify(await sharedMemory.pushAll()));
-    } else if (earlyArgs[0] === '--memory-download-json') {
-      console.log(JSON.stringify(await sharedMemory.pullAll()));
-    } else {
-      sharedMemory.clear();
-      console.log(JSON.stringify({ ok: true }));
-    }
-    process.exit(0);
-  }
-
-  // 2. 启动版本检查（不等待，与后续流程并行执
-  versionCheckPromise = checkVersionOnStartup();
-
-  // Memory is available for all logged-in users — no separate token verification needed
-
-  // Handle --model flag: aegis --model deepseek "question"
-  // or aegis --model council "question" for council vote
-  const modelArg = process.argv.find(a => a.startsWith('--model='))?.split('=')[1]
-    || (() => { const i = process.argv.indexOf('--model'); return i > -1 ? process.argv[i+1] : null; })();
-
-  if (modelArg) {
-    const { readFileSync } = await import('fs');
-    const { homedir } = await import('os');
-    try {
-      const cfg = JSON.parse(readFileSync(`${homedir()}/.aegiscode/config.json`, 'utf8'));
-      const found = (cfg.models || []).find((m: any) => m.id === modelArg);
-      if (found) {
-        process.env.OPENAI_API_KEY = found.apiKey;
-        process.env.OPENAI_BASE_URL = found.baseUrl || found.baseURL;
-        // Sätt currentModelId så getDefaultModel() väljer rätt
-        cfg.currentModelId = found.id;
-        const { writeFileSync } = await import('fs');
-        writeFileSync(`${homedir()}/.aegiscode/config.json`, JSON.stringify(cfg, null, 2));
-        console.log(`\x1b[38;2;0;229;192m[AEGIS] Model: ${found.name || found.id}\x1b[0m`);
-      }
-    } catch {}
-  }
-
-  // Handle /council command
-  const args = process.argv.slice(2);
-
-  if (args[0] === '/council' || args[0] === 'council') {
-    const question = args.slice(1).join(' ');
-    if (!question) {
-      console.error('Usage: aegis /council <question>');
-      process.exit(1);
-    }
-    runCouncil(question).then(() => process.exit(0));
-    return;
-  }
-  if (isDebugMode) {
-    console.log('[DEBUG] Version check started (running in parallel)');
-  }
-
-  // 3. 创建 yargs CLI 实
-  const cli = yargs(hideBin(process.argv))
-    .scriptName(cliConfig.scriptName)
-    .usage(cliConfig.usage)
-    .version(cliConfig.version)
-    // 注册全局选
-    .options(globalOptions)
-
-    // 注册中间
-    .middleware(middlewareChain)
-
-    // ── auth commands ──────────────────────────────────────────────────────
-    .command(
-      'login',
-      'Log in to aegiscloud (Google or username/password), or Claude Code Pro/Max',
-      (y) => y
-        .option('password', {
-          alias: 'p',
-          type: 'boolean',
-          describe: 'Log in with username and password instead of browser',
-          default: false,
-        })
-        .option('claude-pro', {
-          type: 'boolean',
-          describe: 'Authenticate with a Claude Code Pro/Max subscription token instead of an API key',
-          default: false,
-        }),
-      async (argv) => {
-        const { runLogin, runLoginPassword, runLoginClaudePro } = await import('./auth/login.js');
-        try {
-          if ((argv as any).claudePro) {
-            await runLoginClaudePro();
-            process.exit(0);
-          } else if ((argv as any).password) {
-            await runLoginPassword();
-          } else {
-            await runLogin();
-          }
-          console.log('\n\x1b[32m✓ Logged in successfully.\x1b[0m');
-          console.log('\nRun \x1b[1maegis\x1b[0m to start coding.\n');
-        } catch (err) {
-          console.error('\n\x1b[31m✗ Login failed:\x1b[0m', (err as Error).message);
-          process.exit(1);
-        }
-        process.exit(0);
-      },
-    )
-
-    .command(
-      'logout',
-      'Log out and remove stored aegiscloud credentials',
-      () => {},
-      async () => {
-        const { runLogout } = await import('./auth/login.js');
-        runLogout();
-        process.exit(0);
-      },
-    )
-
-    // 示
-    .example('$0', 'Start interactive mode')
-    .example('$0 login', 'Log in via browser')
-    .example('$0 continue', 'Continue the most recent conversation')
-    .example('$0 resume <session-id>', 'Resume a specific conversation by ID')
-    .example('$0 "帮我分析这个项目"', 'Start with an initial message')
-    .example('$0 --model gpt-4', 'Use a specific model')
-    .example('$0 --router', 'Start with the auto-router on')
-    .example('$0 --debug', 'Enable debug mode')
-    .example('$0 --init', 'Create default config file')
-
-    // 帮助
-    .help()
-    .alias('h', 'help')
-
-    // 7. 错误处
-    .fail((msg, err, yargsInstance) => {
-      if (err) {
-        console.error('💥 An error occurred:')
-        console.error(err.message)
-        if (isDebugMode && err.stack) {
-          console.error('\nStack trace:')
-          console.error(err.stack)
-        }
-        process.exit(1)
-      }
-
-      if (msg) {
-        console.error('❌ Invalid arguments:')
-        console.error(msg)
-        console.error('')
-        yargsInstance.showHelp()
-        process.exit(1)
-      }
-    })
-
-    // 8. 严格模式（禁止未知选
-    .strict()
-
-    // 9. 默认命
-    .command(
-      '$0 [message..]',
-      'Start interactive mode',
-      (yargs) => {
-        return yargs.positional('message', {
-          type: 'string',
-          describe: 'Initial message to send (can be multiple words)',
-          array: true,
-        })
-      },
-      async (argv) => {
-        const args = argv as CliArguments
-
-        // 处理 --init 命
-        if (args.init) {
-          const configPath = await configManager.createDefaultConfig()
-          console.log(`✅ Created default config at: ${configPath}`)
-          console.log('')
-          console.log('Please edit the file and add your API key:')
-          console.log(`  vim ${configPath}`)
-          process.exit(0)
-        }
-
-        // ── Mandatory login check (Aegiscode Pro) ────────────────────────
-        // Aegiscode Pro requires login. No login = no API access.
-        const { readFileSync } = await import('node:fs');
-        const { homedir } = await import('node:os');
-        let hasCredentials = false;
-        try {
-          const cfg = JSON.parse(readFileSync(`${homedir()}/.aegiscode/config.json`, 'utf8'));
-          hasCredentials = !!(cfg?.aegiscloud?.api_key);
-        } catch {}
-
-        const needsLogin = hasCredentials
-          ? !(await (await import('./auth/login.js')).ensureAccountValid())
-          : true;
-
-        if (needsLogin) {
-          const { runLogin } = await import('./auth/login.js');
-          try {
-            await runLogin();
-            console.log('\n\x1b[32m✓ Logged in.\x1b[0m Starting ÆGIS...\n');
-            await configManager.initialize(process.cwd());
-          } catch (err) {
-            console.error('\n\x1b[31m✗ Login failed:\x1b[0m', (err as Error).message);
-            console.error('Run \x1b[1maegis login\x1b[0m to try again.');
-            process.exit(1);
-          }
-        }
-
-        const modelConfig = configManager.getDefaultModel()
-
-        // No API key yet — don't exit here. App.tsx already detects this on mount
-        // (initializeStoreState) and renders the interactive SetupWizard, which
-        // creates ~/.aegiscode/.env for the user. Exiting here with print-only
-        // instructions skipped that wizard entirely and left fresh installs
-        // stuck telling users to hand-create a hidden folder themselves.
-        // --print mode has no UI to fall back on, so it still needs to fail fast.
-        if (!modelConfig.apiKey && isPrintMode) {
-          process.stderr.write(
-            'No API key configured. Run `aegis` once (without --print) to set one up, ' +
-            'or set an environment variable: export OPENAI_API_KEY=sk-...\n',
-          );
-          process.exit(1);
-        }
-
-        // 获取初始消息（支持多个单
-        const messageArray = argv.message as string[] | undefined
-        let initialMessage =
-          messageArray && messageArray.length > 0
-            ? messageArray.join(' ')
-            : undefined
-
-        if (isDebugMode && initialMessage) {
-          console.log('[DEBUG] Initial message:', initialMessage)
-        }
-
-        // Handle `aegis continue` and `aegis resume <id>` without `--` prefix
-        // so they behave like `aegis --continue` and `aegis --resume <id>`.
-        if (!args.continue && !args.resume && initialMessage && !initialMessage.startsWith('/')) {
-          const trimmed = initialMessage.trim();
-          if (trimmed === 'continue') {
-            args.continue = true;
-            initialMessage = undefined;
-          } else if (trimmed.startsWith('resume ') || trimmed.startsWith('resume\t')) {
-            const parts = trimmed.split(/\s+/);
-            if (parts.length >= 2) {
-              args.resume = parts[1];
-              initialMessage = undefined;
-            }
-          }
-        }
-
-        // 处理 --continue 和 --resume 参
-        let resumeSessionId: string | undefined;
-        
-        if (args.continue) {
-          // 获取最近的会话文
-          const latestSession = await getLatestSessionFile(process.cwd());
-          if (latestSession) {
-            // 从文件路径提取 sessionId（去掉 .jsonl 扩展
-            resumeSessionId = path.basename(latestSession, '.jsonl');
-            if (isDebugMode) {
-              console.log('[DEBUG] Continuing session:', resumeSessionId);
-            }
-          } else {
-            console.log('No previous session found. Starting a new conversation.');
-          }
-        } else if (args.resume && typeof args.resume === 'string') {
-          resumeSessionId = args.resume;
-          if (isDebugMode) {
-            console.log('[DEBUG] Resuming session:', resumeSessionId);
-          }
-        }
-
-        // --print: headless mode, no Ink. Runs one turn, writes the result to
-        // stdout (text or JSON per --output-format), and exits — no TUI chrome,
-        // no spinners, no escape codes, so the output is safe to pipe/script.
-        if (args.print) {
-          if (!initialMessage) {
-            process.stderr.write('Error: --print requires a message, e.g. aegis --print "your question"\n');
-            process.exit(1);
-          }
-
-          const { startCostLedger } = await import('./services/CostLedger.js');
-          startCostLedger();
-          const { startHeartbeat } = await import('./services/Heartbeat.js');
-          startHeartbeat();
-
-          const { ContextManager } = await import('./context/index.js');
-          const { Agent } = await import('./agent/Agent.js');
-
-          const ctxManager = new ContextManager({ compressionThreshold: 100000 });
-          let sessionId: string;
-          let priorMessages: { role: 'system' | 'user' | 'assistant' | 'tool'; content: string }[] = [];
-
-          if (resumeSessionId) {
-            const loaded = await ctxManager.loadSession(resumeSessionId);
-            sessionId = loaded ? resumeSessionId : await ctxManager.createSession();
-            if (loaded) {
-              priorMessages = ctxManager.getMessages()
-                .filter(m => m.role === 'user' || m.role === 'assistant')
-                .map(m => ({ role: m.role as 'user' | 'assistant', content: m.content }));
-            }
-          } else {
-            sessionId = await ctxManager.createSession();
-          }
-
-          try {
-            const agent = await Agent.create({
-              apiKey: modelConfig.apiKey ?? '',
-              baseURL: modelConfig.baseURL,
-              model: modelConfig.model,
-              requireConfirmation: false,
-            });
-
-            const result = await agent.chatWithMetadata(initialMessage, {
-              sessionId,
-              messages: priorMessages,
-            });
-
-            if (!result.success) {
-              throw new Error(result.error?.message || 'Agent execution failed');
-            }
-            const finalMessage = result.finalMessage || '';
-
-            await ctxManager.addMessage('user', initialMessage);
-            await ctxManager.addMessage('assistant', finalMessage);
-            await ctxManager.flush();
-
-            if (args.outputFormat === 'json') {
-              process.stdout.write(JSON.stringify({
-                result: finalMessage,
-                session_id: sessionId,
-                num_turns: result.metadata?.turnsCount,
-                num_tool_calls: result.metadata?.toolCallsCount,
-                total_tokens: result.metadata?.totalTokens,
-              }) + '\n');
-            } else {
-              process.stdout.write(finalMessage + '\n');
-            }
-            process.exit(0);
-          } catch (error) {
-            process.stderr.write(`Error: ${error instanceof Error ? error.message : String(error)}\n`);
-            process.exit(1);
-          }
-        }
-
-        // 初始化主题（从用户配置加载，或自动检测终端颜色模
-        themeManager.initializeFromConfig();
-        
-        // CLI 参数覆盖（如果指定
-        if (args.theme && themeManager.hasTheme(args.theme)) {
-          themeManager.setTheme(args.theme);
-          if (isDebugMode) {
-            console.log('[DEBUG] Theme overridden by CLI to:', args.theme);
-          }
-        } else if (isDebugMode) {
-          console.log('[DEBUG] Theme:', themeManager.getCurrentThemeName());
-        }
-
-        // Check for --plain flag (plain text mode, no Ink)
-        const isPlain = args.plain === true;
-        if (isDebugMode) console.log('[DEBUG] Plain mode:', isPlain);
-
-        if (isPlain) {
-          // Plain text mode — just relay input/output
-          if (initialMessage) {
-            console.log(initialMessage);
-          }
-          const { createInterface } = await import('node:readline');
-          const rl = createInterface({ input: process.stdin, output: process.stdout });
-          rl.on('line', (line) => {
-            if (line.trim()) console.log(line);
-          });
-          rl.on('close', () => process.exit(0));
-          return;
-        }
-
-        // Ink rendering — with fallback for non-TTY environments
-        const isTTY = process.stdin.isTTY === true && process.stdout.isTTY === true;
-        if (isDebugMode) console.log('[DEBUG] TTY:', isTTY, '(stdin:', process.stdin.isTTY, 'stdout:', process.stdout.isTTY + ')');
-
-        // Auto-start rendering debugger if --debug-rendering is set
-        if (ENABLE_RENDER_DEBUG) {
-          const { startRenderDebugger } = await import('./ui/render-debugger.js');
-          startRenderDebugger({ reportInterval: 5000, verbose: false });
-        }
-
-        // Create proper stdin/stdout for Ink (mock if needed to prevent raw mode errors)
-        let renderStdin = process.stdin;
-        let renderStdout = process.stdout;
-
-        // Force isTTY on stdout so Ink renders properly in all terminal environments
-        if (!process.stdout.isTTY) {
-          (process.stdout as any).isTTY = true;
-          if (isDebugMode) console.log('[DEBUG] Forcing isTTY=true on stdout');
-        }
-
-        if (!isTTY) {
-          const { PassThrough } = await import('node:stream');
-          const mockStdin = new PassThrough();
-          (mockStdin as any).isTTY = true;
-          (mockStdin as any).setRawMode = () => {};
-          (mockStdin as any).ref = () => {};
-          (mockStdin as any).unref = () => {};
-          (mockStdin as any).setEncoding = () => {};
-          renderStdin = mockStdin as unknown as typeof process.stdin;
-          if (isDebugMode) console.log('[DEBUG] Using mock stdin for Ink render');
-        }
-
-        // Start background services (fire-and-forget, silent on failure)
-        const { startCostLedger } = await import('./services/CostLedger.js');
-        startCostLedger();
-        const { startHeartbeat } = await import('./services/Heartbeat.js');
-        startHeartbeat();
-
-        try {
-          const { render } = await import('ink');
-          render(
-            <App
-              apiKey={modelConfig.apiKey ?? ''}
-              baseURL={modelConfig.baseURL}
-              model={modelConfig.model}
-              initialMessage={initialMessage}
-              debug={args.debug}
-              permissionMode={args.permissionMode}
-              versionCheckPromise={versionCheckPromise}
-              resumeSessionId={resumeSessionId}
-              routerEnabled={args.router}
-            />,
-            {
-              exitOnCtrlC: false,
-              patchConsole: true,
-              stdin: renderStdin,
-              stdout: process.stdout,
-              // Disabled: the alt screen buffer blocks normal mouse text selection/copy
-              // in most terminal emulators. Real Claude Code renders inline instead.
-              alternateScreen: false,
-              maxFps: 30,
-              // Without this, Ink's default log-update mode erases and rewrites the
-              // ENTIRE terminal output on every re-render, even when only a single
-              // line changed (e.g. the input cursor blink, every ~530ms, forever).
-              // That full erase+rewrite invalidates any in-progress mouse text
-              // selection in terminals like Kitty — selecting a message becomes
-              // impossible because the screen keeps getting wiped out from under it.
-              // incrementalRendering does real line-level diffing and only rewrites
-              // lines that actually changed, leaving untouched lines (and any
-              // selection on them) alone.
-              incrementalRendering: true,
-            },
-          );
-
-          // Handle EOF (Ctrl+D) on real stdin to allow normal terminal closing
-          if (isTTY && process.stdin.isTTY) {
-            process.stdin.on('end', () => {
-              if (isDebugMode) console.log('[DEBUG] EOF received on stdin');
-              process.exit(0);
-            });
-          }
-        } catch (renderError) {
-          // Ink rendering failed, fall back to simple text mode
-          if (isDebugMode) console.log('[DEBUG] Ink rendering failed:', (renderError as Error).message, '- falling back to text mode');
-          if (initialMessage) {
-            console.log(initialMessage);
-          }
-          const { createInterface } = await import('node:readline');
-          const rl = createInterface({ input: process.stdin, output: process.stdout });
-          rl.on('line', (line) => {
-            if (line.trim()) console.log(line);
-          });
-          rl.on('close', () => process.exit(0));
-        }
-    })
-  await cli.parse()
-}
-
-// 运行主函
-main().catch((error) => {
-  console.error('Fatal error:', error.message)
-  if (isDebugMode && error.stack) {
-    console.error('\nStack trace:')
-    console.error(error.stack)
-  }
-  process.exit(1)
-})