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

const logger = createLogger('reviewerAgent');

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

  // Safely access agent outputs with fallbacks
  const backendOutput = agents.backendDeveloper?.output || 'No backend code generated yet';
  const frontendOutput = agents.frontendDeveloper?.output || 'No frontend code generated yet';
  const docsOutput = agents.documentationAgent?.output || 'No documentation generated yet';

  const llmClient = getLlmClient();
  const prompt = reviewerAgentPrompt
    .replace('{{code}}', JSON.stringify({ 
      backend: backendOutput, 
      frontend: frontendOutput 
    }, null, 2))
    .replace('{{docs}}', docsOutput);

  const report = await llmClient.generate(prompt);

  updateContext({ agents: { ...agents, reviewerAgent: { output: report, completed: true } } });
  logger.stop();
  logger.success('Generated review.');
  return report;
};