import { prisma } from '@app/lib/prisma';
import { SleepRecord } from '@prisma/client';

import { CreateSleepRecordInput } from '@app/schemas';
import { rollingPeriod } from "@app/lib/queue/processors/insights/utils";
import { InsightJobName } from "@app/lib/queue/types";
import { getQueue } from "@app/lib/queue";

export const createSleepRecord = async (
  userId: number, input: CreateSleepRecordInput
): Promise<SleepRecord> => {
  const sleepRecord = await prisma.sleepRecord.create({
    data: {
      ...input,
      userId
    }
  });

  void enqueueEnergySleepInsight(userId);

  return sleepRecord;
};

export const getSleepRecordsByUserId = async (
  userId: number,
  params?: {
    limit?: number;
    page?: number;
    from?: string;
    to?: string;
  }
): Promise<{ entries: SleepRecord[]; total: number; page: number; nextPage: number | null }> => {
  const page  = params?.page  ?? 1;
  const limit = params?.limit ?? 20;
  const skip  = (page - 1) * limit;

  const where = {
    userId,
    ...(params?.from || params?.to ? {
      date: {
        ...(params.from && { gte: new Date(params.from) }),
        ...(params.to   && { lte: new Date(params.to)   }),
      }
    } : {}),
  };

  const [entries, total] = await Promise.all([
    prisma.sleepRecord.findMany({
      where,
      orderBy: { date: 'desc' },
      take: limit,
      skip,
    }),
    prisma.sleepRecord.count({ where }),
  ]);

  const totalPages = Math.ceil(total / limit);

  return {
    entries,
    total,
    page,
    nextPage: page < totalPages ? page + 1 : null,
  };
};

export const destroySleepRecordById = async (
  userId: number, id: number
): Promise<SleepRecord> => {
  return await prisma.sleepRecord.delete({
    where: {
      id,
      userId
    }
  })
}

async function enqueueEnergySleepInsight(userId: number): Promise<void> {
  await getQueue('insights').add(
    InsightJobName.EnergySleep,
    { userId, ...rollingPeriod(7) },
    {
      jobId: `energy-sleep-${userId}-${new Date().toDateString()}`,
      removeOnComplete: true,
    }
  );
}