export interface SubagentTask {
  task: string
  tools: string[]           // 授予子 Agent 的权限子集
  budget: { maxTokens: number }
  systemPrompt?: string      // 定制化人设
  workingDirectory?: string  // 隔离的工作目录
}

export interface SubagentResult {
  success: boolean
  summary: string           // 纯文本结论，不含中间过程
  toolCallCount: number
}

// 主 Agent 中派生 SubAgent
export async function spawnSubagent(taskConfig: SubagentTask): Promise<SubagentResult> {
  // 1. 初始化隔离的 ContextEngine 实例
  const subEngine = new DefaultContextEngine({
    systemPrompt: taskConfig.systemPrompt || '你是一个专注完成特定子任务的专业助手。'
  })

  // 2. 过滤工具集（最小权限原则）
  const scopedTools = filterAvailableTools(taskConfig.tools)

  // 3. 运行独立的 Agent Loop
  let finalSummary = ''
  let callCount = 0

  for await (const event of agentLoop({
    tools: scopedTools,
    contextEngine: subEngine,
    initialPrompt: taskConfig.task,
    budget: taskConfig.budget
  })) {
    if (event.type === 'tool_execution_start') callCount++
    if (event.type === 'message_end' && event.content.role === 'assistant') {
      finalSummary = event.content.content
    }
  }

  return {
    success: true,
    summary: finalSummary,
    toolCallCount: callCount
  }
}

import { exec } from 'child_process'
import { promisify } from 'util'
import { v4 as uuidv4 } from 'uuid'

const execAsync = promisify(exec)

export async function runInDedicatedWorktree<T>(
  repoPath: string,
  fn: (worktreePath: string) => Promise<T>
): Promise<T> {
  const branchName = `agent-work-${uuidv4().slice(0, 8)}`
  const worktreeDir = `/tmp/${branchName}`

  // 1. 创建独立分支与工作树目录
  await execAsync(`git worktree add -b ${branchName} ${worktreeDir} HEAD`, { cwd: repoPath })

  try {
    // 2. 子 Agent 在全新、独立的工作树中执行代码读写
    const result = await fn(worktreeDir)
    return result
  } finally {
    // 3. 执行完毕清理临时工作树
    await execAsync(`git worktree remove --force ${worktreeDir}`, { cwd: repoPath }).catch(() => {})
    await execAsync(`git branch -D ${branchName}`, { cwd: repoPath }).catch(() => {})
  }
}

export class SessionRouter {
  private sessions = new Map<string, AgentSession>()

  async resolveSession(msg: InboundMessage): Promise<AgentSession> {
    // 优先级 1: 显式声明的目标 Agent ID (Exact Peer ID)
    if (msg.targetAgentId) {
      return this.getOrCreateSession(`agent:${msg.targetAgentId}:${msg.userId}`)
    }

    // 优先级 2: 继承已有会话线程 (Thread Inherit)
    if (msg.threadId && this.sessions.has(msg.threadId)) {
      return this.sessions.get(msg.threadId)!
    }

    // 优先级 3: 角色指令路由 (如消息开头带有 @security-bot)
    const roleMatch = this.matchRoleMention(msg.text)
    if (roleMatch) {
      return this.getOrCreateSession(`role:${roleMatch}:${msg.userId}`)
    }

    // 优先级 4: 默认用户隔离会话 (Account Default)
    const userKey = `${msg.channel}:${msg.userId}`
    return this.getOrCreateSession(userKey)
  }
}