import { Agent } from './agents/index.js';
import { getOrchestratorTools } from './tools/registry.js';
import { PROMPTS } from './prompts/system.js';
import * as readline from 'node:readline/promises';
import { hooks } from './hooks/index.js';
import { registerHooks } from './hooks/registry.js';

await registerHooks();

const promptName = process.argv[2] || 'orchestrator';
const systemPrompt = PROMPTS[promptName as keyof typeof PROMPTS];

if (!systemPrompt) {
  console.error(`未知提示词: ${promptName}，可选: ${Object.keys(PROMPTS).join(', ')}`);
  process.exit(1);
}

// 主 orchestrator Agent，拥有全部工具（包括 spawn_agent / run_conversation）
const mainAgent = new Agent({
  name: 'Orchestrator',
  systemPrompt,
  tools: getOrchestratorTools(),
});

const rl = readline.createInterface({
  input: process.stdin,
  output: process.stdout,
});

console.log(`提示词模式: ${promptName}，输入 "exit" 退出。\n`);

while (true) {
  const userInput = await rl.question('我: ');
  if (userInput.toLowerCase() === 'exit') break;

  hooks.emit('agent:start', { userInput });

  const reply = await mainAgent.run(userInput);

  hooks.emit('agent:end', { userInput, reply });
  // agent:end hook 会打印回复，这里不需要重复打印
}

rl.close();