aegiscode 3.1.10 → 3.2.0

package/src/tools/registry.ts

@@ -1,304 +0,0 @@
-/**
- * 
- * 
- * 
- */
-
-import { EventEmitter } from 'events';
-import type { Tool, FunctionDeclaration } from './types.js';
-
-/**
- * 
- */
-export interface ToolRegisteredEvent {
-  type: 'builtin' | 'mcp';
-  tool: Tool;
-}
-
-/**
- * 
- */
-export class ToolRegistry extends EventEmitter {
-  /** 内置工具 */
-  private tools = new Map<string, Tool>();
-  
-  /** MCP 工具 */
-  private mcpTools = new Map<string, Tool>();
-  
-  /** 分类索引 */
-  private categories = new Map<string, Set<string>>();
-  
-  /** 标签索引 */
-  private tagIndex = new Map<string, Set<string>>();
-
-  /**
-   * 
-   */
-  register(tool: Tool): void {
-    if (this.tools.has(tool.name) || this.mcpTools.has(tool.name)) {
-      throw new Error(`工具 '${tool.name}' 已注册`);
-    }
-    
-    this.tools.set(tool.name, tool);
-    this.updateIndexes(tool);
-    this.emit('toolRegistered', { type: 'builtin', tool } as ToolRegisteredEvent);
-  }
-
-  /**
-   * 
-   */
-  registerMcpTool(tool: Tool): void {
-    if (this.tools.has(tool.name) || this.mcpTools.has(tool.name)) {
-      throw new Error(`工具 '${tool.name}' 已注册`);
-    }
-    
-    this.mcpTools.set(tool.name, tool);
-    this.updateIndexes(tool);
-    this.emit('toolRegistered', { type: 'mcp', tool } as ToolRegisteredEvent);
-  }
-
-  /**
-   * 
-   */
-  registerMcpTools(tools: Tool[]): void {
-    for (const tool of tools) {
-      this.registerMcpTool(tool);
-    }
-  }
-
-  /**
-   * 
-   */
-  unregisterMcpTool(name: string): boolean {
-    const tool = this.mcpTools.get(name);
-    if (!tool) return false;
-
-    this.mcpTools.delete(name);
-    this.removeFromIndexes(tool);
-    this.emit('toolUnregistered', { type: 'mcp', tool });
-    return true;
-  }
-
-  /**
-   * 
-   */
-  clearMcpTools(): void {
-    for (const tool of this.mcpTools.values()) {
-      this.removeFromIndexes(tool);
-    }
-    this.mcpTools.clear();
-  }
-
-  /**
-   * 
-   */
-  registerAll(tools: Tool[]): void {
-    for (const tool of tools) {
-      this.register(tool);
-    }
-  }
-
-  /**
-   * 
-   */
-  get(name: string): Tool | undefined {
-    return this.tools.get(name) || this.mcpTools.get(name);
-  }
-
-  /**
-   * 
-   */
-  has(name: string): boolean {
-    return this.tools.has(name) || this.mcpTools.has(name);
-  }
-
-  /**
-   * 
-   */
-  getAll(): Tool[] {
-    return [...this.tools.values(), ...this.mcpTools.values()];
-  }
-
-  /**
-   * 
-   */
-  getBuiltinTools(): Tool[] {
-    return [...this.tools.values()];
-  }
-
-  /**
-   * 
-   */
-  getMcpTools(): Tool[] {
-    return [...this.mcpTools.values()];
-  }
-
-  /**
-   * 
-   */
-  getNames(): string[] {
-    return [...this.tools.keys(), ...this.mcpTools.keys()];
-  }
-
-  /**
-   * 
-   */
-  getReadOnlyTools(): Tool[] {
-    return this.getAll().filter(tool => tool.isReadOnly);
-  }
-
-  /**
-   * 
-   */
-  getWriteTools(): Tool[] {
-    return this.getAll().filter(tool => !tool.isReadOnly);
-  }
-
-  /**
-   * 
-   */
-  getByCategory(category: string): Tool[] {
-    const names = this.categories.get(category);
-    if (!names) return [];
-    return [...names].map(name => this.tools.get(name)!).filter(Boolean);
-  }
-
-  /**
-   * 
-   */
-  getByTag(tag: string): Tool[] {
-    const names = this.tagIndex.get(tag);
-    if (!names) return [];
-    return [...names].map(name => this.tools.get(name)!).filter(Boolean);
-  }
-
-  /**
-   * 
-   */
-  getCategories(): string[] {
-    return [...this.categories.keys()];
-  }
-
-  /**
-   * 
-   */
-  getTags(): string[] {
-    return [...this.tagIndex.keys()];
-  }
-
-  /**
-   * 
-   */
-  getFunctionDeclarations(): FunctionDeclaration[] {
-    return this.getAll().map(tool => tool.getFunctionDeclaration());
-  }
-
-  /**
-   * 
-   */
-  getFunctionDeclarationsByMode(mode?: string): FunctionDeclaration[] {
-    if (mode === 'plan') {
-      // Plan 模式只返回只读工
-      return this.getReadOnlyTools().map(t => t.getFunctionDeclaration());
-    }
-    return this.getFunctionDeclarations();
-  }
-
-  /**
-   * 
-   */
-  search(query: string): Tool[] {
-    const lowerQuery = query.toLowerCase();
-    return this.getAll().filter(tool => 
-      tool.name.toLowerCase().includes(lowerQuery) ||
-      tool.displayName.toLowerCase().includes(lowerQuery) ||
-      tool.description.short.toLowerCase().includes(lowerQuery)
-    );
-  }
-
-  /**
-   * 
-   */
-  get size(): number {
-    return this.tools.size + this.mcpTools.size;
-  }
-
-  /**
-   * 
-   */
-  get builtinSize(): number {
-    return this.tools.size;
-  }
-
-  /**
-   * 
-   */
-  get mcpSize(): number {
-    return this.mcpTools.size;
-  }
-
-  /**
-   * 
-   */
-  clear(): void {
-    this.tools.clear();
-    this.mcpTools.clear();
-    this.categories.clear();
-    this.tagIndex.clear();
-  }
-
-  /**
-   * 
-   */
-  private updateIndexes(tool: Tool): void {
-    // 更新分类索
-    if (tool.category) {
-      if (!this.categories.has(tool.category)) {
-        this.categories.set(tool.category, new Set());
-      }
-      this.categories.get(tool.category)!.add(tool.name);
-    }
-
-    // 更新标签索
-    for (const tag of tool.tags) {
-      if (!this.tagIndex.has(tag)) {
-        this.tagIndex.set(tag, new Set());
-      }
-      this.tagIndex.get(tag)!.add(tool.name);
-    }
-  }
-
-  /**
-   * 
-   */
-  private removeFromIndexes(tool: Tool): void {
-    // 从分类索引中移
-    if (tool.category) {
-      const categorySet = this.categories.get(tool.category);
-      if (categorySet) {
-        categorySet.delete(tool.name);
-        if (categorySet.size === 0) {
-          this.categories.delete(tool.category);
-        }
-      }
-    }
-
-    // 从标签索引中移
-    for (const tag of tool.tags) {
-      const tagSet = this.tagIndex.get(tag);
-      if (tagSet) {
-        tagSet.delete(tool.name);
-        if (tagSet.size === 0) {
-          this.tagIndex.delete(tag);
-        }
-      }
-    }
-  }
-}
-
-/**
- * 
- */
-export function createToolRegistry(): ToolRegistry {
-  return new ToolRegistry();
-}

package/src/tools/schemas.ts

@@ -1,109 +0,0 @@
-/**
- * 
- * 
- * 
- */
-
-import { z } from 'zod';
-
-/**
- * 
- */
-export const ToolSchemas = {
-  /**
-   * 
-   */
-  filePath: (options?: { description?: string }) =>
-    z.string()
-      .min(1, '文件路径不能为空')
-      .describe(options?.description || '文件路径'),
-
-  /**
-   * 
-   */
-  absolutePath: (options?: { description?: string }) =>
-    z.string()
-      .min(1, '路径不能为空')
-      .refine(
-        (path) => path.startsWith('/') || path.startsWith('~'),
-        '必须是绝对路径（以 / 或 ~ 开头）'
-      )
-      .describe(options?.description || '绝对路径'),
-
-  /**
-   * 
-   */
-  lineNumber: (options?: { description?: string }) =>
-    z.number()
-      .int('必须是整数')
-      .min(0, '行号不能为负')
-      .describe(options?.description || '行号（从 0 开始）'),
-
-  /**
-   * 
-   */
-  lineLimit: (options?: { description?: string; max?: number }) =>
-    z.number()
-      .int('必须是整数')
-      .min(1, '至少读取 1 行')
-      .max(options?.max || 10000, `最多读取 ${options?.max || 10000} 行`)
-      .describe(options?.description || '读取行数'),
-
-  /**
-   * 
-   */
-  encoding: () =>
-    z.enum(['utf8', 'utf-8', 'ascii', 'base64', 'binary', 'hex'])
-      .default('utf8')
-      .describe('文件编码格式'),
-
-  /**
-   * Glob 模式
-   */
-  globPattern: (options?: { description?: string }) =>
-    z.string()
-      .min(1, '模式不能为空')
-      .describe(options?.description || 'Glob 匹配模式'),
-
-  /**
-   * 
-   */
-  regexPattern: (options?: { description?: string }) =>
-    z.string()
-      .min(1, '模式不能为空')
-      .describe(options?.description || '正则表达式模式'),
-
-  /**
-   * 
-   */
-  command: (options?: { description?: string }) =>
-    z.string()
-      .min(1, '命令不能为空')
-      .describe(options?.description || 'Shell 命令'),
-
-  /**
-   * 
-   */
-  timeout: (options?: { max?: number; default?: number }) =>
-    z.number()
-      .int()
-      .min(0)
-      .max(options?.max || 600000)
-      .default(options?.default || 30000)
-      .describe('超时时间（毫秒）'),
-
-  /**
-   * 
-   */
-  booleanWithDefault: (defaultValue: boolean, description?: string) =>
-    z.boolean()
-      .default(defaultValue)
-      .describe(description || '布尔选项'),
-};
-
-/**
- * 
- */
-export function optional<T extends z.ZodType>(schema: T): z.ZodOptional<T> {
-  return schema.optional();
-}

package/src/tools/test.ts

@@ -1,220 +0,0 @@
-/**
- * 
- * 
- * 
- */
-
-import { ToolRegistry, getBuiltinTools, ToolKind, createTool } from './index.js';
-import { z } from 'zod';
-import path from 'path';
-import fs from 'fs/promises';
-import os from 'os';
-
-async function testToolSystem() {
-  console.log('=== 工具系统测试 ===\n');
-
-  // 1. 创建注册表并注册内置工
-  console.log('1. 创建工具注册表并注册内置工具');
-  const registry = new ToolRegistry();
-  const builtinTools = getBuiltinTools();
-  registry.registerAll(builtinTools);
-  console.log(`   已注册 ${registry.size} 个工具: ${registry.getNames().join(', ')}`);
-  console.log(`   - 内置工具: ${registry.builtinSize}`);
-  console.log(`   - MCP 工具: ${registry.mcpSize}\n`);
-
-  // 2. 测试工具获
-  console.log('2. 测试工具获取');
-  const readTool = registry.get('Read');
-  console.log(`   Read 工具: ${readTool ? '✅ 存在' : '❌ 不存在'}`);
-  console.log(`   - 类型: ${readTool?.kind}`);
-  console.log(`   - 只读: ${readTool?.isReadOnly}`);
-  console.log(`   - 并发安全: ${readTool?.isConcurrencySafe}`);
-  console.log(`   - 分类: ${readTool?.category}`);
-  console.log(`   - 标签: ${readTool?.tags.join(', ')}\n`);
-
-  // 3. 测试只
-  console.log('3. 测试只读/写入工具过滤');
-  const readOnlyTools = registry.getReadOnlyTools();
-  const writeTools = registry.getWriteTools();
-  console.log(`   只读工具: ${readOnlyTools.map(t => t.name).join(', ')}`);
-  console.log(`   写入工具: ${writeTools.map(t => t.name).join(', ')}\n`);
-
-  // 4. 测试函数声明生
-  console.log('4. 测试函数声明生成');
-  const declarations = registry.getFunctionDeclarations();
-  for (const decl of declarations) {
-    console.log(`   ${decl.name}:`);
-    console.log(`     描述: ${decl.description.slice(0, 60)}...`);
-    console.log(`     参数: ${Object.keys(decl.parameters.properties || {}).join(', ')}`);
-  }
-  console.log();
-
-  // 5. 测试 Plan 模式函数声
-  console.log('5. 测试 Plan 模式函数声明');
-  const planDeclarations = registry.getFunctionDeclarationsByMode('plan');
-  console.log(`   Plan 模式工具数: ${planDeclarations.length}`);
-  console.log(`   Plan 模式工具: ${planDeclarations.map(d => d.name).join(', ')}\n`);
-
-  // 6. 测试 Read 工具执
-  console.log('6. 测试 Read 工具执行');
-  if (readTool) {
-    const packageJsonPath = path.join(process.cwd(), 'package.json');
-    const result = await readTool.execute({ file_path: packageJsonPath });
-    console.log(`   结果: ${result.success ? '✅ 成功' : '❌ 失败'}`);
-    console.log(`   显示内容: ${result.displayContent}`);
-    console.log(`   LLM 内容长度: ${result.llmContent.length} 字符\n`);
-  }
-
-  // 7. 测试 Edit 工
-  console.log('7. 测试 Edit 工具');
-  const editTool = registry.get('Edit');
-  if (editTool) {
-    // 创建临时文
-    const tmpDir = os.tmpdir();
-    const testFile = path.join(tmpDir, `aegis-test-${Date.now()}.txt`);
-    await fs.writeFile(testFile, 'Hello World\nFoo Bar\n', 'utf8');
-    
-    // 测试编
-    const result = await editTool.execute({
-      file_path: testFile,
-      old_string: 'World',
-      new_string: 'AEGIS',
-    });
-    console.log(`   结果: ${result.success ? '✅ 成功' : '❌ 失败'}`);
-    console.log(`   显示内容: ${result.displayContent}`);
-    
-    // 验证内
-    const content = await fs.readFile(testFile, 'utf8');
-    console.log(`   验证: ${content.includes('AEGIS') ? '✅ 替换正确' : '❌ 替换失败'}`);
-    
-    // 测试多重匹
-    await fs.writeFile(testFile, 'AAA BBB AAA', 'utf8');
-    const multiResult = await editTool.execute({
-      file_path: testFile,
-      old_string: 'AAA',
-      new_string: 'CCC',
-    });
-    console.log(`   多重匹配: ${multiResult.success ? '❌ 应该失败' : '✅ 正确拒绝'}`);
-    
-    // 清
-    await fs.unlink(testFile);
-    console.log();
-  }
-
-  // 8. 测试 Write 工
-  console.log('8. 测试 Write 工具');
-  const writeTool = registry.get('Write');
-  if (writeTool) {
-    const tmpDir = os.tmpdir();
-    const testFile = path.join(tmpDir, `aegis-write-${Date.now()}.txt`);
-    
-    const result = await writeTool.execute({
-      file_path: testFile,
-      contents: 'Test content\nLine 2\n',
-    });
-    console.log(`   结果: ${result.success ? '✅ 成功' : '❌ 失败'}`);
-    console.log(`   显示内容: ${result.displayContent}`);
-    
-    // 验
-    const exists = await fs.access(testFile).then(() => true).catch(() => false);
-    console.log(`   文件存在: ${exists ? '✅ 是' : '❌ 否'}`);
-    
-    // 清
-    if (exists) await fs.unlink(testFile);
-    console.log();
-  }
-
-  // 9. 测试 Bash 工
-  console.log('9. 测试 Bash 工具');
-  const bashTool = registry.get('Bash');
-  if (bashTool) {
-    const result = await bashTool.execute({
-      command: 'echo "Hello from Bash"',
-      description: '测试 echo 命令',
-    });
-    console.log(`   结果: ${result.success ? '✅ 成功' : '❌ 失败'}`);
-    console.log(`   显示内容: ${result.displayContent}`);
-    console.log(`   输出: ${result.llmContent.trim()}`);
-    
-    // 测试失败命
-    const failResult = await bashTool.execute({
-      command: 'exit 1',
-    });
-    console.log(`   失败命令: ${failResult.success ? '❌ 应该失败' : '✅ 正确失败'}\n`);
-  }
-
-  // 10. 测试 Glob 工具执
-  console.log('10. 测试 Glob 工具执行');
-  const globTool = registry.get('Glob');
-  if (globTool) {
-    const result = await globTool.execute({ pattern: '*.ts', path: path.join(process.cwd(), 'src') });
-    console.log(`   结果: ${result.success ? '✅ 成功' : '❌ 失败'}`);
-    console.log(`   显示内容: ${result.displayContent}`);
-    if (result.metadata?.count) {
-      console.log(`   找到文件数: ${result.metadata.count}\n`);
-    }
-  }
-
-  // 11. 测试 Grep 工具执
-  console.log('11. 测试 Grep 工具执行');
-  const grepTool = registry.get('Grep');
-  if (grepTool) {
-    const result = await grepTool.execute({ 
-      pattern: 'export', 
-      include: '*.ts',
-      path: path.join(process.cwd(), 'src/tools'),
-    });
-    console.log(`   结果: ${result.success ? '✅ 成功' : '❌ 失败'}`);
-    console.log(`   显示内容: ${result.displayContent}`);
-    if (result.metadata?.matches) {
-      console.log(`   匹配数: ${result.metadata.matches}\n`);
-    }
-  }
-
-  // 12. 测试 MCP 工具注
-  console.log('12. 测试 MCP 工具注册');
-  const mockMcpTool = createTool({
-    name: 'MockMcpTool',
-    displayName: 'Mock MCP Tool',
-    kind: ToolKind.ReadOnly,
-    schema: z.object({ input: z.string() }),
-    description: { short: 'A mock MCP tool for testing' },
-    async execute(params) {
-      return {
-        success: true,
-        llmContent: `Mock result for: ${params.input}`,
-        displayContent: '✅ Mock MCP 工具执行成功',
-      };
-    },
-  });
-  
-  registry.registerMcpTool(mockMcpTool);
-  console.log(`   MCP 工具注册: ✅`);
-  console.log(`   总工具数: ${registry.size}`);
-  console.log(`   MCP 工具数: ${registry.mcpSize}`);
-  console.log(`   获取 MCP 工具: ${registry.get('MockMcpTool') ? '✅ 存在' : '❌ 不存在'}`);
-  
-  // 测试注
-  registry.unregisterMcpTool('MockMcpTool');
-  console.log(`   注销后: ${registry.get('MockMcpTool') ? '❌ 仍存在' : '✅ 已移除'}\n`);
-
-  // 13. 测试 build 方
-  console.log('13. 测试 build 方法');
-  if (readTool) {
-    const invocation = readTool.build({ file_path: '/test/path.ts' });
-    console.log(`   工具名: ${invocation.toolName}`);
-    console.log(`   参数: ${JSON.stringify(invocation.params)}\n`);
-  }
-
-  // 14. 测试参数验
-  console.log('14. 测试参数验证');
-  if (readTool) {
-    const invalidResult = await readTool.execute({ file_path: '' });
-    console.log(`   空路径测试: ${invalidResult.success ? '❌ 应该失败' : '✅ 正确失败'}`);
-    console.log(`   错误信息: ${invalidResult.displayContent}\n`);
-  }
-
-  console.log('=== 测试完成 ===');
-}
-
-testToolSystem().catch(console.error);