web/apps/scandic-web/components/Forms/BookingWidget/schema.ts

import { z } from "zod"

import { REDEMPTION } from "@/constants/booking"

import { ChildBedMapEnum } from "@/types/components/bookingWidget/enums"

export const guestRoomSchema = z
  .object({
    adults: z.number().default(1),
    childrenInRoom: z
      .array(
        z.object({
          age: z.number().min(0, "Age is required"),
          bed: z.number().min(0, "Bed choice is required"),
        })
      )
      .default([]),
  })
  .superRefine((value, ctx) => {
    const childrenInAdultsBed = value.childrenInRoom.filter(
      (c) => c.bed === ChildBedMapEnum.IN_ADULTS_BED
    )
    if (value.adults < childrenInAdultsBed.length) {
      const lastAdultBedIndex = value.childrenInRoom
        .map((c) => c.bed)
        .lastIndexOf(ChildBedMapEnum.IN_ADULTS_BED)

      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message:
          "You cannot have more children in adults bed than adults in the room",
        path: ["childrenInRoom", lastAdultBedIndex],
      })
    }
  })

export const guestRoomsSchema = z.array(guestRoomSchema)

export const bookingCodeSchema = z
  .object({
    value: z
      .string()
      .refine(
        (value) => {
          if (
            !value ||
            /(^D\d*$)|(^DSH[0-9a-z]*$)|(^L\d*$)|(^LH[0-9a-z]*$)|(^B[a-z]{3}\d{6})|(^VO[0-9a-z]*$)|^[0-9a-z]*$/i.test(
              value
            )
          ) {
            return true
          } else {
            return false
          }
        },
        { message: "Invalid booking code" }
      )
      .default(""),
    remember: z.boolean().default(false),
    flag: z.boolean().default(false),
  })
  .optional()

export const bookingWidgetSchema = z
  .object({
    bookingCode: bookingCodeSchema,
    date: z.object({
      // Update this as required once started working with Date picker in Nights component
      fromDate: z.string(),
      toDate: z.string(),
    }),
    redemption: z.boolean().default(false),
    rooms: guestRoomsSchema,
    search: z.string({ coerce: true }).min(1, "Required"),
    selectedSearch: z.string().optional(),
    hotel: z.number().optional(),
    city: z.string().optional(),
  })
  .superRefine((value, ctx) => {
    if (!value.hotel && !value.city) {
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Destination required",
        path: ["search"],
      })
    }
    if (value.rooms.length > 1 && value.bookingCode?.value.startsWith("VO")) {
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Multi-room booking is not available with this booking code.",
        path: ["bookingCode.value"],
      })
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Multi-room booking is not available with this booking code.",
        path: ["rooms"],
      })
    }
    if (value.rooms.length > 1 && value.redemption) {
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Multi-room booking is not available with reward night.",
        path: [REDEMPTION],
      })
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Multi-room booking is not available with reward night.",
        path: ["rooms"],
      })
    }
    if (value.bookingCode?.value && value.redemption) {
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Code and voucher is not available with reward night.",
        path: [REDEMPTION],
      })
      ctx.addIssue({
        code: z.ZodIssueCode.custom,
        message: "Code and voucher is not available with reward night.",
        path: ["bookingCode.value"],
      })
    }
  })