import { Job } from "bullmq";
import {
  InsightJobName,
  InsightJobData,
  InsightPeriodPayload,
} from "@app/lib/queue/types";
import { getQueue } from "@app/lib/queue/QueueRegistry";
import { prisma } from "@app/lib/prisma";
import { processMoodTrend } from "@app/lib/queue/processors/insights/moodTrend";
import { processEnergySleep } from "@app/lib/queue/processors/insights/energySleep";
import { processTriggerPattern } from "@app/lib/queue/processors/insights/triggerPattern";
import { processDailyEnergy } from "@app/lib/queue/processors/insights/dailyEnergy";
import { processDailySleep } from "@app/lib/queue/processors/insights/dailySleep";

export async function insightProcessor(
  job: Job<InsightJobData["data"]>,
): Promise<void> {
  console.log(`Insight job ${job.name} started`);

  switch (job.name as InsightJobName) {
    case InsightJobName.FanOut:
      return fanOut();

    case InsightJobName.MoodTrend:
      return processMoodTrend(job.data as InsightPeriodPayload);

    case InsightJobName.EnergySleep:
      return processEnergySleep(job.data as InsightPeriodPayload);

    case InsightJobName.TriggerPattern:
      return processTriggerPattern(job.data as InsightPeriodPayload);

    case InsightJobName.DailyEnergy:
      return processDailyEnergy(job.data as InsightPeriodPayload);

    case InsightJobName.DailySleep:
      return processDailySleep(job.data as InsightPeriodPayload);

    default:
      throw new Error(`Unknown insight job: ${job.name}`);
  }
}

async function fanOut(): Promise<void> {
  const users = await prisma.user.findMany({
    where: { active: true },
    select: { id: true },
  });

  const queue = getQueue("insights");
  const periodEnd = new Date();
  const periodStart = new Date();
  periodStart.setDate(periodEnd.getDate() - 7);

  const payload: Omit<InsightPeriodPayload, "userId"> = {
    periodStart: periodStart.toISOString(),
    periodEnd: periodEnd.toISOString(),
  };

  await queue.addBulk(
    users.flatMap((u) => [
      { name: InsightJobName.MoodTrend, data: { userId: u.id, ...payload } },
      { name: InsightJobName.EnergySleep, data: { userId: u.id, ...payload } },
      {
        name: InsightJobName.TriggerPattern,
        data: { userId: u.id, ...payload },
      },
    ]),
  );
}