aegiscode 3.1.8 → 3.1.10

package/src/hooks/HookService.ts

@@ -0,0 +1,269 @@
+/**
+ * Hook Service - Hooks 系统的简洁 API 层
+ * 
+ * 
+ * 
+ */
+
+import { getHookManager } from './HookManager.js';
+import type { HookContext } from './types.js';
+import type { PermissionMode } from '../agent/types.js';
+
+/**
+ * 
+ */
+function createContext(
+  sessionId: string,
+  projectDir?: string,
+  permissionMode?: string
+): HookContext {
+  return {
+    sessionId,
+    projectDir: projectDir || process.cwd(),
+    permissionMode: (permissionMode || 'default') as PermissionMode,
+  };
+}
+
+/**
+ * 
+ */
+export function isHooksAvailable(): boolean {
+  const manager = getHookManager();
+  return manager.isInitialized() && manager.isEnabled();
+}
+
+// ==================== 生命周
+
+/**
+ * 
+ */
+export async function onSessionStart(sessionId: string, projectDir?: string): Promise<void> {
+  if (!isHooksAvailable()) return;
+
+  try {
+    await getHookManager().executeSessionStartHooks(
+      createContext(sessionId, projectDir)
+    );
+  } catch (error) {
+    console.warn('[HookService] SessionStart hook error:', error);
+  }
+}
+
+/**
+ * 
+ */
+export async function onSessionEnd(sessionId: string, projectDir?: string): Promise<void> {
+  if (!isHooksAvailable()) return;
+
+  try {
+    await getHookManager().executeSessionEndHooks(
+      createContext(sessionId, projectDir)
+    );
+  } catch (error) {
+    console.warn('[HookService] SessionEnd hook error:', error);
+  }
+}
+
+/**
+ * 
+ * @returns 注入的上下文（如果有）
+ */
+export async function onUserPromptSubmit(
+  promptContent: string,
+  sessionId: string,
+  projectDir?: string
+): Promise<string | undefined> {
+  if (!isHooksAvailable()) return undefined;
+
+  try {
+    const result = await getHookManager().executeUserPromptHooks(
+      promptContent,
+      createContext(sessionId, projectDir)
+    );
+    return result.injectedContext;
+  } catch (error) {
+    console.warn('[HookService] UserPromptSubmit hook error:', error);
+    return undefined;
+  }
+}
+
+// ==================== 控制
+
+/**
+ * Agent 停止
+ * @returns true 表示应该继续执行
+ */
+export async function onStop(
+  stopReason: string | undefined,
+  sessionId: string,
+  projectDir?: string
+): Promise<boolean> {
+  if (!isHooksAvailable()) return false;
+
+  try {
+    const result = await getHookManager().executeStopHooks(
+      stopReason,
+      createContext(sessionId, projectDir)
+    );
+    return result.shouldContinue;
+  } catch (error) {
+    console.warn('[HookService] Stop hook error:', error);
+    return false;
+  }
+}
+
+/**
+ * 
+ * @returns true 表示应该阻止压缩
+ */
+export async function onCompaction(
+  preTokens: number,
+  messageCount: number,
+  sessionId: string,
+  projectDir?: string
+): Promise<boolean> {
+  if (!isHooksAvailable()) return false;
+
+  try {
+    const result = await getHookManager().executeCompactionHooks(
+      preTokens,
+      messageCount,
+      createContext(sessionId, projectDir)
+    );
+    return result.shouldPrevent;
+  } catch (error) {
+    console.warn('[HookService] Compaction hook error:', error);
+    return false;
+  }
+}
+
+// ==================== 工具执
+
+/**
+ * PreToolUse Hook 结果
+ */
+export interface PreToolHookResult {
+  decision: 'allow' | 'deny' | 'ask';
+  reason?: string;
+  modifiedInput?: Record<string, unknown>;
+  warning?: string;
+}
+
+/**
+ * PostToolUse Hook 结果
+ */
+export interface PostToolHookResult {
+  additionalContext?: string;
+  modifiedOutput?: unknown;
+}
+
+/**
+ * 
+ */
+export async function onPreToolUse(
+  toolName: string,
+  toolUseId: string,
+  toolInput: Record<string, unknown>,
+  sessionId: string,
+  projectDir?: string,
+  permissionMode?: string
+): Promise<PreToolHookResult> {
+  if (!isHooksAvailable()) {
+    return { decision: 'allow' };
+  }
+
+  try {
+    return await getHookManager().executePreToolHooks(
+      toolName,
+      toolUseId,
+      toolInput,
+      createContext(sessionId, projectDir, permissionMode)
+    );
+  } catch (error) {
+    console.warn('[HookService] PreToolUse hook error:', error);
+    return { decision: 'allow' };
+  }
+}
+
+/**
+ * 
+ */
+export async function onPostToolUse(
+  toolName: string,
+  toolUseId: string,
+  toolInput: Record<string, unknown>,
+  toolResult: { success: boolean; llmContent?: string },
+  sessionId: string,
+  projectDir?: string,
+  permissionMode?: string
+): Promise<PostToolHookResult> {
+  if (!isHooksAvailable()) {
+    return {};
+  }
+
+  try {
+    return await getHookManager().executePostToolHooks(
+      toolName,
+      toolUseId,
+      toolInput,
+      toolResult,
+      createContext(sessionId, projectDir, permissionMode)
+    );
+  } catch (error) {
+    console.warn('[HookService] PostToolUse hook error:', error);
+    return {};
+  }
+}
+
+/**
+ * 
+ */
+export async function onPostToolUseFailure(
+  toolName: string,
+  toolUseId: string,
+  toolInput: Record<string, unknown>,
+  errorMessage: string,
+  sessionId: string,
+  projectDir?: string,
+  permissionMode?: string
+): Promise<void> {
+  if (!isHooksAvailable()) return;
+
+  try {
+    await getHookManager().executePostToolFailureHooks(
+      toolName,
+      toolUseId,
+      toolInput,
+      errorMessage,
+      createContext(sessionId, projectDir, permissionMode)
+    );
+  } catch (error) {
+    console.warn('[HookService] PostToolUseFailure hook error:', error);
+  }
+}
+
+/**
+ * 
+ * @returns 'approve' | 'deny' | 'ask'
+ */
+export async function onPermissionRequest(
+  toolName: string,
+  toolInput: Record<string, unknown>,
+  sessionId: string,
+  projectDir?: string,
+  permissionMode?: string
+): Promise<'approve' | 'deny' | 'ask'> {
+  if (!isHooksAvailable()) return 'ask';
+
+  try {
+    const result = await getHookManager().executePermissionHooks(
+      toolName,
+      toolInput,
+      createContext(sessionId, projectDir, permissionMode)
+    );
+    return result.decision;
+  } catch (error) {
+    console.warn('[HookService] PermissionRequest hook error:', error);
+    return 'ask';
+  }
+}

package/src/hooks/Matcher.ts

@@ -0,0 +1,157 @@
+/**
+ * Hook 匹配器
+ * 
+ * 
+ * 
+ */
+
+import { minimatch } from 'minimatch';
+import type { MatcherConfig, MatchContext, HookMatcher, Hook } from './types.js';
+
+/**
+ * Hook 匹配器
+ */
+export class Matcher {
+  /**
+   * 
+   */
+  matches(config: MatcherConfig | undefined, context: MatchContext): boolean {
+    // 无匹配器配置 = 匹配所
+    if (!config) {
+      return true;
+    }
+
+    // 检查工具名匹
+    if (config.tools && context.toolName) {
+      if (!this.matchesPattern(context.toolName, config.tools)) {
+        return false;
+      }
+    }
+
+    // 检查文件路径匹
+    if (config.paths && context.filePath) {
+      if (!this.matchesGlob(context.filePath, config.paths)) {
+        return false;
+      }
+    }
+
+    // 检查命令匹
+    if (config.commands && context.command) {
+      if (!this.matchesPattern(context.command, config.commands)) {
+        return false;
+      }
+    }
+
+    return true;
+  }
+
+  /**
+   * 
+   */
+  getMatchingHooks(matchers: HookMatcher[] | undefined, context: MatchContext): Hook[] {
+    if (!matchers || matchers.length === 0) {
+      return [];
+    }
+
+    const result: Hook[] = [];
+
+    for (const matcher of matchers) {
+      if (this.matches(matcher.matcher, context)) {
+        result.push(...matcher.hooks);
+      }
+    }
+
+    return result;
+  }
+
+  /**
+   * 
+   * 
+   * 
+   * - 简单字符串: "Read"
+   * - 管道分隔: "Read|Write|Edit"
+   * - 正则表达式: "Bash\\(.*\\)"
+   */
+  private matchesPattern(value: string, pattern: string): boolean {
+    // 尝试作为管道分隔的简单模
+    if (!pattern.includes('\\') && !pattern.includes('^') && !pattern.includes('$')) {
+      const parts = pattern.split('|');
+      for (const part of parts) {
+        if (value === part.trim()) {
+          return true;
+        }
+      }
+    }
+
+    // 尝试作为正则表达
+    try {
+      const regex = new RegExp(`^(${pattern})$`);
+      return regex.test(value);
+    } catch {
+      // 正则无效，使用简单字符串比
+      return value === pattern;
+    }
+  }
+
+  /**
+   * Glob 模式匹配
+   * 
+   * 
+   * - 通配符: "*.ts"
+   * - 双星: "**\/*.tsx"
+   * - 多模式: "**\/*.{ts,tsx,js,jsx}"
+   */
+  private matchesGlob(filePath: string, pattern: string): boolean {
+    // 处理管道分隔的多
+    const patterns = pattern.split('|').map(p => p.trim());
+    
+    for (const p of patterns) {
+      if (minimatch(filePath, p, { matchBase: true })) {
+        return true;
+      }
+    }
+
+    return false;
+  }
+}
+
+/**
+ * 
+ */
+export function extractFilePath(toolInput: Record<string, unknown>): string | undefined {
+  // 常见的文件路径字段
+  const pathFields = ['file_path', 'path', 'filePath', 'file', 'target'];
+  
+  for (const field of pathFields) {
+    const value = toolInput[field];
+    if (typeof value === 'string' && value.length > 0) {
+      return value;
+    }
+  }
+
+  return undefined;
+}
+
+/**
+ * 
+ */
+export function extractCommand(
+  toolName: string,
+  toolInput: Record<string, unknown>
+): string | undefined {
+  // 只有 Bash 工具有命
+  if (toolName !== 'Bash' && toolName !== 'Shell') {
+    return undefined;
+  }
+
+  const commandFields = ['command', 'cmd', 'script'];
+  
+  for (const field of commandFields) {
+    const value = toolInput[field];
+    if (typeof value === 'string' && value.length > 0) {
+      return value;
+    }
+  }
+
+  return undefined;
+}

package/src/hooks/index.ts

@@ -0,0 +1,63 @@
+/**
+ * Hooks 系统模块
+ * 
+ * 
+ */
+
+// 类型导
+export * from './types.js';
+
+// 核心类导
+export { HookManager, getHookManager } from './HookManager.js';
+export { HookExecutor } from './HookExecutor.js';
+export { Matcher, extractFilePath, extractCommand } from './Matcher.js';
+
+// Hook Service - 简洁 API 层（独立函
+export {
+  isHooksAvailable,
+  // 生命周
+  onSessionStart,
+  onSessionEnd,
+  onUserPromptSubmit,
+  // 控制
+  onStop,
+  onCompaction,
+  // 工具执
+  onPreToolUse,
+  onPostToolUse,
+  onPostToolUseFailure,
+  onPermissionRequest,
+  // 类
+  type PreToolHookResult,
+  type PostToolHookResult,
+} from './HookService.js';
+
+// 便捷函
+import { getHookManager } from './HookManager.js';
+import type { HookConfig } from './types.js';
+
+/**
+ * 
+ */
+export function initializeHooks(config: Partial<HookConfig>): void {
+  const manager = getHookManager();
+  manager.loadConfig(config);
+}
+
+/**
+ * 
+ */
+export function isHooksEnabled(): boolean {
+  return getHookManager().isEnabled();
+}
+
+/**
+ * 
+ */
+export function getHookStats(): { enabled: boolean; counts: Record<string, number> } {
+  const manager = getHookManager();
+  return {
+    enabled: manager.isEnabled(),
+    counts: manager.getHookCounts(),
+  };
+}