import dayjs from "@calcom/dayjs"; import prisma from "@calcom/prisma"; import type { BookingLimit } from "@calcom/types/Calendar"; import { HttpError } from "../http-error"; import { parseBookingLimit } from "../isBookingLimits"; export async function checkBookingLimits( bookingLimits: any, eventStartDate: Date, eventId: number, returnBusyTimes?: boolean ) { const parsedBookingLimits = parseBookingLimit(bookingLimits); if (parsedBookingLimits) { const limitCalculations = Object.entries(parsedBookingLimits).map( async ([key, limitingNumber]) => await checkLimit({ key, limitingNumber, eventStartDate, eventId, returnBusyTimes }) ); await Promise.all(limitCalculations) .then((res) => { if (returnBusyTimes) { return res; } }) .catch((error) => { throw new HttpError({ message: error.message, statusCode: 401 }); }); return true; } return false; } export async function checkLimit({ eventStartDate, eventId, key, limitingNumber, returnBusyTimes = false, }: { eventStartDate: Date; eventId: number; key: string; limitingNumber: number; returnBusyTimes?: boolean; }) { { const limitKey = key as keyof BookingLimit; // Take PER_DAY and turn it into day and PER_WEEK into week etc. const filter = limitKey.split("_")[1].toLocaleLowerCase() as "day" | "week" | "month" | "year"; // Have to cast here const startDate = dayjs(eventStartDate).startOf(filter).toDate(); // this is parsed above with parseBookingLimit so we know it's safe. const endDate = dayjs(startDate).endOf(filter).toDate(); // This allows us to easily add it within dayjs const bookingsInPeriod = await prisma.booking.count({ where: { status: "ACCEPTED", eventType: { AND: [ { id: eventId, bookings: { every: { startTime: { gte: startDate, }, endTime: { lte: endDate, }, }, }, }, ], }, }, }); if (bookingsInPeriod >= limitingNumber) { // This is used when getting availbility if (returnBusyTimes) { return { start: startDate, end: endDate, }; } throw new HttpError({ message: `booking_limit_reached`, statusCode: 403, }); } } }