import {

  Mood,

  MoodComponent,

  SleepRecord,

  Trigger,

  TriggerMoodLink,

  User,

} from "@prisma/client";

import { prisma } from "@app/lib/prisma";

import { findUserById } from "@app/services/user.service";

export type ExportData = {

  user: User;

  moods: Mood[];

  moodComponents: MoodComponent[];

  sleepRecords: SleepRecord[];

  triggers: Trigger[];

  triggerMoodLinks: TriggerMoodLink[];

};

export async function gatherExportData(userId: number): Promise<ExportData> {

  const user = await findUserById(userId);

  if (!user) {

    throw new Error(`Usuário ${userId} não encontrado para exportação`);

  }

  const [moods, sleepRecords, triggers, triggerMoodLinks] = await Promise.all([

    prisma.mood.findMany({

      where: { userId },

      include: { moodComponents: true },

      orderBy: { moment: "asc" },

    }),

    prisma.sleepRecord.findMany({

      where: { userId },

      orderBy: { date: "asc" },

    }),

    prisma.trigger.findMany({

      where: { userId },

      orderBy: { moment: "asc" },

    }),

    prisma.triggerMoodLink.findMany({

      where: { trigger: { userId } },

      orderBy: { linkedAt: "asc" },

    }),

  ]);

  return {

    user,

    moods,

    moodComponents: moods.flatMap((mood) => mood.moodComponents),

    sleepRecords,

    triggers,

    triggerMoodLinks,

  };

}