import { createLogger } from '../utils/logger';
import { getContext, updateContext } from '../utils/contextStore';
import { getLlmClient } from '../llm/llmRouter';
import { qaEngineerPrompt } from '../utils/promptTemplates';

const logger = createLogger('qaEngineer');

export const runQaEngineer = async () => {
  logger.start();
  const context = getContext();
  const { spec, agents } = context;

  const llmClient = getLlmClient();
  const prompt = qaEngineerPrompt
    .replace('{{spec}}', JSON.stringify(spec, null, 2))
    .replace('{{backendCode}}', agents.backendDeveloper.output)
    .replace('{{frontendCode}}', agents.frontendDeveloper.output);

  const tests = await llmClient.generate(prompt);

  updateContext({ agents: { ...context.agents, qaEngineer: { tests, passed: true, completed: true } } });
  logger.stop();
  logger.success('Generated QA tests.');
  return tests;
};