查询引擎

// QueryEngine.ts — 简化版
export class QueryEngine {
  // 核心状态
  private mutableMessages: Message[] = []
  private abortController: AbortController | null = null
  private permissionDenials = new Set<string>()
  private totalUsage: UsageTracker = createUsageTracker()
  private readFileState = new LRUCache<string, FileContent>({
    max: 100  // 最多缓存 100 个文件
  })
  private discoveredSkillNames = new Set<string>()
  private loadedNestedMemoryPaths = new Set<string>()

  // 配置
  private config: QueryEngineConfig

  constructor(config: QueryEngineConfig) {
    this.config = config
  }

  // 取消当前查询
  abort(): void {
    this.abortController?.abort()
  }

  // 获取 token 使用统计
  getUsage(): UsageSummary {
    return this.totalUsage.getSummary()
  }

  // 核心方法：提交消息（见下一个示例）
  async *submitMessage(userInput: string): AsyncGenerator<Message> {
    // ...
  }
}

// QueryEngine.ts — submitMessage 简化版
async *submitMessage(
  userInput: string
): AsyncGenerator<Message> {
  // 1. 清除上一轮的临时状态
  this.permissionDenials.clear()

  // 2. 创建新的 AbortController
  this.abortController = new AbortController()

  // 3. 包装工具权限检查
  const canUseTool = wrapCanUseTool(
    this.config.permissionMode,
    this.permissionDenials
  )

  // 4. 组装系统提示词（多源合并）
  const systemPromptParts = await fetchSystemPromptParts({
    tools: this.config.tools,
    memories: this.loadedNestedMemoryPaths,
    cwd: process.cwd(),
    skills: this.discoveredSkillNames,
  })

  // 5. 添加用户消息到对话历史
  this.mutableMessages.push({
    role: 'user',
    content: userInput,
  })

  // 6. 进入查询循环，yield 所有响应消息
  const queryGen = query({
    messages: this.mutableMessages,
    systemPrompt: systemPromptParts.join('\n'),
    model: this.config.model,
    tools: this.config.tools,
    canUseTool,
    abortSignal: this.abortController.signal,
    usage: this.totalUsage,
    readFileState: this.readFileState,
  })

  // 7. 逐个 yield SDK 消息供 UI 消费
  for await (const event of queryGen) {
    if (event.type === 'message') {
      this.mutableMessages.push(event.message)
      yield event.message
    }
  }

  // 8. 记录会话转录
  await recordTranscript(this.mutableMessages)
}

// fetchSystemPromptParts — 简化版
async function fetchSystemPromptParts(opts: {
  tools: Tool[]
  memories: Set<string>
  cwd: string
  skills: Set<string>
}): Promise<string[]> {
  const parts: string[] = []

  // 1. 基础系统提示词
  parts.push(BASE_SYSTEM_PROMPT)

  // 2. 工具使用说明
  for (const tool of opts.tools) {
    parts.push(formatToolDescription(tool))
  }

  // 3. CLAUDE.md 项目记忆
  const projectMemory = await loadClaudeMd(opts.cwd)
  if (projectMemory) {
    parts.push(`<project-memory>\n${projectMemory}\n</project-memory>`)
  }

  // 4. 用户级记忆（~/.claude/CLAUDE.md）
  const userMemory = await loadUserClaudeMd()
  if (userMemory) {
    parts.push(`<user-memory>\n${userMemory}\n</user-memory>`)
  }

  // 5. 环境上下文
  parts.push(`Current directory: ${opts.cwd}`)
  parts.push(`Platform: ${process.platform}`)

  // 6. Coordinator 模式额外上下文
  if (isCoordinatorMode()) {
    parts.push(await getCoordinatorContext())
  }

  return parts
}