cal.diy/packages/features/bookings/lib/service/RecurringBookingService.ts

import type { CreateBookingMeta, CreateRecurringBookingData } from "@calcom/features/bookings/lib/dto/types";
import type { BookingResponse } from "@calcom/features/bookings/types";
import type { FeaturesRepository } from "@calcom/features/flags/features.repository";
import { CreationSource, SchedulingType } from "@calcom/prisma/enums";
import type { AppsStatus } from "@calcom/types/Calendar";
import { v4 as uuidv4 } from "uuid";
import type { BookingStatus } from "@calcom/prisma/enums";
import type { IBookingService } from "../interfaces/IBookingService";
import type { RegularBookingService } from "./RegularBookingService";
import type { BookingEventHandlerService } from "../onBookingEvents/BookingEventHandlerService";
import { getBookingAuditActorForNewBooking } from "../handleNewBooking/getBookingAuditActorForNewBooking";
import { criticalLogger } from "@calcom/lib/logger.server";
import { getAuditActionSource } from "../handleNewBooking/getAuditActionSource";
import { safeStringify } from "@calcom/lib/safeStringify";
import {
  buildBookingCreatedAuditData,
  buildBookingRescheduledAuditData,
} from "../handleNewBooking/buildBookingEventAuditData";
export type BookingHandlerInput = {
  bookingData: CreateRecurringBookingData;
} & CreateBookingMeta;

export const handleNewRecurringBooking = async function (
  this: RecurringBookingService,
  {
    input,
    deps,
    creationSource,
  }: {
    input: BookingHandlerInput;
    deps: IRecurringBookingServiceDependencies;
    creationSource: CreationSource;
  }
): Promise<BookingResponse[]> {
  const data = input.bookingData;
  const { regularBookingService } = deps;
  const createdBookings: BookingResponse[] = [];
  const allRecurringDates: { start: string; end: string | undefined }[] = data.map((booking) => {
    return { start: booking.start, end: booking.end };
  });
  const appsStatus: AppsStatus[] | undefined = undefined;

  const numSlotsToCheckForAvailability = 1;

  let thirdPartyRecurringEventId = null;

  // for round robin, the first slot needs to be handled first to define the lucky user
  const firstBooking = data[0];
  const isRoundRobin = firstBooking.schedulingType === SchedulingType.ROUND_ROBIN;

  let luckyUsers = undefined;

  const handleBookingMeta = {
    userId: input.userId,
    platformClientId: input.platformClientId,
    platformRescheduleUrl: input.platformRescheduleUrl,
    platformCancelUrl: input.platformCancelUrl,
    platformBookingUrl: input.platformBookingUrl,
    platformBookingLocation: input.platformBookingLocation,
    areCalendarEventsEnabled: input.areCalendarEventsEnabled,
  };

  if (isRoundRobin) {
    const recurringEventData = {
      ...firstBooking,
      appsStatus,
      allRecurringDates,
      isFirstRecurringSlot: true,
      thirdPartyRecurringEventId,
      numSlotsToCheckForAvailability,
      currentRecurringIndex: 0,
      noEmail: input.noEmail !== undefined ? input.noEmail : false,
    };

    const firstBookingResult = await regularBookingService.createBooking({
      bookingData: recurringEventData,
      bookingMeta: {
        hostname: input.hostname || "",
        forcedSlug: input.forcedSlug as string | undefined,
        ...handleBookingMeta,
      },
    });
    luckyUsers = firstBookingResult.luckyUsers;
  }

  for (let key = isRoundRobin ? 1 : 0; key < data.length; key++) {
    const booking = data[key];
    // Disable AppStatus in Recurring Booking Email as it requires us to iterate backwards to be able to compute the AppsStatus for all the bookings except the very first slot and then send that slot's email with statuses
    // It is also doubtful that how useful is to have the AppsStatus of all the bookings in the email.
    // It is more important to iterate forward and check for conflicts for only first few bookings defined by 'numSlotsToCheckForAvailability'
    // if (key === 0) {
    //   const calcAppsStatus: { [key: string]: AppsStatus } = createdBookings
    //     .flatMap((book) => (book.appsStatus !== undefined ? book.appsStatus : []))
    //     .reduce((prev, curr) => {
    //       if (prev[curr.type]) {
    //         prev[curr.type].failures += curr.failures;
    //         prev[curr.type].success += curr.success;
    //       } else {
    //         prev[curr.type] = curr;
    //       }
    //       return prev;
    //     }, {} as { [key: string]: AppsStatus });
    //   appsStatus = Object.values(calcAppsStatus);
    // }

    const recurringEventData = {
      ...booking,
      appsStatus,
      allRecurringDates,
      isFirstRecurringSlot: key == 0,
      thirdPartyRecurringEventId,
      numSlotsToCheckForAvailability,
      currentRecurringIndex: key,
      noEmail: input.noEmail !== undefined ? input.noEmail : key !== 0,
      luckyUsers,
    };

    const promiseEachRecurringBooking = regularBookingService.createBooking({
      bookingData: recurringEventData,
      bookingMeta: {
        hostname: input.hostname || "",
        forcedSlug: input.forcedSlug as string | undefined,
        ...handleBookingMeta,
      },
    });

    const eachRecurringBooking = await promiseEachRecurringBooking;

    createdBookings.push(eachRecurringBooking);

    if (!thirdPartyRecurringEventId) {
      if (eachRecurringBooking.references && eachRecurringBooking.references.length > 0) {
        for (const reference of eachRecurringBooking.references) {
          if (reference.thirdPartyRecurringEventId) {
            thirdPartyRecurringEventId = reference.thirdPartyRecurringEventId;
            break;
          }
        }
      }
    }
  }

  if (createdBookings.length > 0) {
    await this.fireBookingEvents({
      createdBookings,
      eventTypeId: firstBooking.eventTypeId,
      rescheduleUid: firstBooking.rescheduleUid ?? null,
      actorUserUuid: input.userUuid ?? null,
      rescheduledBy: firstBooking.rescheduledBy ?? null,
      creationSource,
    });
  }

  return createdBookings;
};

export interface IRecurringBookingServiceDependencies {
  regularBookingService: RegularBookingService;
  bookingEventHandler: BookingEventHandlerService;
  featuresRepository: FeaturesRepository;
}

/**
 * Recurring Booking Service takes care of creating/rescheduling recurring bookings.
 */
export class RecurringBookingService implements IBookingService {
  constructor(private readonly deps: IRecurringBookingServiceDependencies) {}

  async fireBookingEvents({
    createdBookings,
    eventTypeId,
    rescheduleUid,
    actorUserUuid,
    rescheduledBy,
    creationSource,
  }: {
    createdBookings: BookingResponse[];
    eventTypeId: number;
    rescheduleUid: string | null;
    actorUserUuid: string | null;
    rescheduledBy: string | null;
    creationSource: CreationSource | undefined;
  }) {
    try {
      type ValidBooking = BookingResponse & {
        uid: string;
        startTime: Date;
        endTime: Date;
        status: BookingStatus;
        userUuid: string | null;
      };
      type ValidRescheduledBooking = ValidBooking & {
        previousBooking: ValidBooking & { status: BookingStatus };
      };

      const isReschedule = !!rescheduleUid;
      const firstCreatedBooking = createdBookings[0];
      const eventOrganizationId = firstCreatedBooking.organizationId;
      const bookerAttendee = firstCreatedBooking.attendees?.[0];
      const bookerAttendeeId = bookerAttendee?.id;
      const bookerName = bookerAttendee?.name || "";
      const bookerEmail = bookerAttendee?.email || "";

      const isBookingAuditEnabled = eventOrganizationId
        ? await this.deps.featuresRepository.checkIfTeamHasFeature(eventOrganizationId, "booking-audit")
        : false;

      const rescheduledByAttendeeId = firstCreatedBooking.attendees?.find(
        (attendee) => attendee.email === rescheduledBy
      )?.id;
      // TODO: Note that user.email is always null here as RegularBookingService intentionally sets it to null. To fix, we need to separate out external facing .createBooking and one that is used by RecurringBookingService, so that if we expose something there it doesn't get exposed elsewhere
      const rescheduledByUserUuid =
        firstCreatedBooking.user?.email === rescheduledBy ? firstCreatedBooking.userUuid : null;

      const auditActor = getBookingAuditActorForNewBooking({
        bookerAttendeeId: bookerAttendeeId ?? null,
        actorUserUuid,
        bookerEmail,
        bookerName,
        rescheduledBy: rescheduledBy
          ? {
              attendeeId: rescheduledByAttendeeId ?? null,
              userUuid: rescheduledByUserUuid ?? null,
              email: rescheduledBy,
            }
          : null,
        logger: criticalLogger,
      });

      const actionSource = getAuditActionSource({ creationSource, eventTypeId, rescheduleUid });

      const operationId = uuidv4();

      const isValidBooking = (booking: BookingResponse): booking is ValidBooking => {
        return !!(booking.uid && booking.startTime && booking.endTime && booking.status);
      };

      const isValidRescheduledBooking = (booking: BookingResponse): booking is ValidRescheduledBooking => {
        return !!(
          isValidBooking(booking) &&
          booking.previousBooking &&
          booking.previousBooking.uid &&
          booking.previousBooking.startTime &&
          booking.previousBooking.endTime
        );
      };

      if (isReschedule) {
        const bulkRescheduledBookings = createdBookings.filter(isValidRescheduledBooking).map((booking) => ({
          bookingUid: booking.previousBooking.uid,
          auditData: buildBookingRescheduledAuditData({
            oldBooking: booking.previousBooking,
            newBooking: booking,
          }),
        }));

        if (bulkRescheduledBookings.length > 0) {
          await this.deps.bookingEventHandler.onBulkBookingsRescheduled({
            bookings: bulkRescheduledBookings,
            actor: auditActor,
            organizationId: eventOrganizationId,
            operationId,
            source: actionSource,
            isBookingAuditEnabled,
          });
        }
      } else {
        // For new bookings
        const bulkCreatedBookings = createdBookings.filter(isValidBooking).map((booking) => ({
          bookingUid: booking.uid,
          auditData: buildBookingCreatedAuditData({ booking, attendeeSeatId: null }),
        }));

        if (bulkCreatedBookings.length > 0) {
          await this.deps.bookingEventHandler.onBulkBookingsCreated({
            bookings: bulkCreatedBookings,
            actor: auditActor,
            organizationId: eventOrganizationId,
            operationId,
            source: actionSource,
            isBookingAuditEnabled,
          });
        }
      }
    } catch (error) {
      criticalLogger.error("Error while firing booking events", safeStringify(error));
    }
  }

  async createBooking(input: {
    bookingData: CreateRecurringBookingData;
    bookingMeta?: CreateBookingMeta;
    creationSource: CreationSource;
  }): Promise<BookingResponse[]> {
    const handlerInput = { bookingData: input.bookingData, ...(input.bookingMeta || {}) };
    return handleNewRecurringBooking.bind(this)({
      input: handlerInput,
      deps: this.deps,
      creationSource: input.creationSource,
    });
  }

  async rescheduleBooking(input: {
    bookingData: CreateRecurringBookingData;
    bookingMeta?: CreateBookingMeta;
    creationSource: CreationSource;
  }): Promise<BookingResponse[]> {
    const handlerInput = { bookingData: input.bookingData, ...(input.bookingMeta || {}) };
    return handleNewRecurringBooking.bind(this)({
      input: handlerInput,
      deps: this.deps,
      creationSource: input.creationSource,
    });
  }
}