import { PrismaClient } from "@prisma/client";
import { ProcessingStatusManager } from "./lib/processingStatusManager";

const prisma = new PrismaClient();
const statusManager = new ProcessingStatusManager(prisma);

async function checkRefactoredPipelineStatus() {
  try {
    console.log("=== REFACTORED PIPELINE STATUS ===\n");

    // Get pipeline status using the new system
    const pipelineStatus = await statusManager.getPipelineStatus();

    console.log(`Total Sessions: ${pipelineStatus.totalSessions}\n`);

    // Display status for each stage
    const stages = [
      "CSV_IMPORT",
      "TRANSCRIPT_FETCH",
      "SESSION_CREATION",
      "AI_ANALYSIS",
      "QUESTION_EXTRACTION",
    ];

    for (const stage of stages) {
      console.log(`${stage}:`);
      const stageData = pipelineStatus.pipeline[stage] || {};

      const pending = stageData.PENDING || 0;
      const inProgress = stageData.IN_PROGRESS || 0;
      const completed = stageData.COMPLETED || 0;
      const failed = stageData.FAILED || 0;
      const skipped = stageData.SKIPPED || 0;

      console.log(`  PENDING: ${pending}`);
      console.log(`  IN_PROGRESS: ${inProgress}`);
      console.log(`  COMPLETED: ${completed}`);
      console.log(`  FAILED: ${failed}`);
      console.log(`  SKIPPED: ${skipped}`);
      console.log("");
    }

    // Show what needs processing
    console.log("=== WHAT NEEDS PROCESSING ===");

    for (const stage of stages) {
      const stageData = pipelineStatus.pipeline[stage] || {};
      const pending = stageData.PENDING || 0;
      const failed = stageData.FAILED || 0;

      if (pending > 0 || failed > 0) {
        console.log(`• ${stage}: ${pending} pending, ${failed} failed`);
      }
    }

    // Show failed sessions if any
    const failedSessions = await statusManager.getFailedSessions();
    if (failedSessions.length > 0) {
      console.log("\n=== FAILED SESSIONS ===");
      failedSessions.slice(0, 5).forEach((failure) => {
        console.log(
          `  ${failure.session.import?.externalSessionId || failure.sessionId}: ${failure.stage} - ${failure.errorMessage}`
        );
      });

      if (failedSessions.length > 5) {
        console.log(
          `  ... and ${failedSessions.length - 5} more failed sessions`
        );
      }
    }

    // Show sessions ready for AI processing
    const readyForAI =
      await statusManager.getSessionsNeedingProcessing(
        "AI_ANALYSIS",
        5
      );
    if (readyForAI.length > 0) {
      console.log("\n=== SESSIONS READY FOR AI PROCESSING ===");
      readyForAI.forEach((status) => {
        console.log(
          `  ${status.session.import?.externalSessionId || status.sessionId} (created: ${status.session.createdAt})`
        );
      });
    }
  } catch (error) {
    console.error("Error checking pipeline status:", error);
  } finally {
    await prisma.$disconnect();
  }
}

checkRefactoredPipelineStatus();