import { Credential, SelectedCalendar } from "@prisma/client";

import { getBusyCalendarTimes } from "@calcom/core/CalendarManager";
import { getBusyVideoTimes } from "@calcom/core/videoClient";
import notEmpty from "@calcom/lib/notEmpty";
import type { EventBusyDate } from "@calcom/types/Calendar";

import prisma from "@lib/prisma";

async function getBusyTimes(params: {
  credentials: Credential[];
  userId: number;
  eventTypeId?: number;
  startTime: string;
  endTime: string;
  selectedCalendars: SelectedCalendar[];
}) {
  const { credentials, userId, eventTypeId, startTime, endTime, selectedCalendars } = params;
  const busyTimes: EventBusyDate[] = await prisma.booking
    .findMany({
      where: {
        AND: [
          {
            userId,
            eventTypeId,
            startTime: { gte: new Date(startTime) },
            endTime: { lte: new Date(endTime) },
          },
          {
            OR: [
              {
                status: "ACCEPTED",
              },
              {
                status: "PENDING",
              },
            ],
          },
        ],
      },
      select: {
        startTime: true,
        endTime: true,
      },
    })
    .then((bookings) => bookings.map((booking) => ({ end: booking.endTime, start: booking.startTime })));

  if (credentials) {
    const calendarBusyTimes = await getBusyCalendarTimes(credentials, startTime, endTime, selectedCalendars);
    busyTimes.push(...calendarBusyTimes);
    const videoBusyTimes = (await getBusyVideoTimes(credentials)).filter(notEmpty);
    busyTimes.push(...videoBusyTimes);
  }

  return busyTimes;
}

export default getBusyTimes;
