import { z } from "zod";

import { extendedBookingCreateBody } from "@calcom/prisma/zod-utils";

import type { getBookingFieldsWithSystemFields } from "./getBookingFields";
import getBookingResponsesSchema from "./getBookingResponsesSchema";

const getBookingDataSchema = ({
  view = "booking",
  bookingFields,
}: {
  view: "booking" | "reschedule";
  bookingFields: Awaited<ReturnType<typeof getBookingFieldsWithSystemFields>>;
}) => {
  return extendedBookingCreateBody.merge(
    z.object({ responses: getBookingResponsesSchema({ bookingFields, view }) })
  );
};

export type TgetBookingDataSchema = z.infer<ReturnType<typeof getBookingDataSchema>>;

export default getBookingDataSchema;