web/apps/scandic-web/server/routers/booking/input.ts

import { z } from "zod"

import { ChildBedTypeEnum } from "@/constants/booking"
import { Lang, langToApiLang } from "@/constants/languages"

const signupSchema = z.discriminatedUnion("becomeMember", [
  z.object({
    dateOfBirth: z.string(),
    postalCode: z.string(),
    becomeMember: z.literal<boolean>(true),
  }),
  z.object({ becomeMember: z.literal<boolean>(false) }),
])

const roomsSchema = z.array(
  z.object({
    adults: z.number().int().nonnegative(),
    childrenAges: z
      .array(
        z.object({
          age: z.number().int().nonnegative(),
          bedType: z.nativeEnum(ChildBedTypeEnum),
        })
      )
      .default([]),
    rateCode: z.string(),
    roomTypeCode: z.coerce.string(),
    guest: z.intersection(
      z.object({
        firstName: z.string(),
        lastName: z.string(),
        email: z.string().email(),
        phoneNumber: z.string(),
        countryCode: z.string(),
        membershipNumber: z.string().optional(),
      }),
      signupSchema
    ),
    smsConfirmationRequested: z.boolean(),
    packages: z.object({
      breakfast: z.boolean(),
      allergyFriendly: z.boolean(),
      petFriendly: z.boolean(),
      accessibility: z.boolean(),
    }),
    roomPrice: z.object({
      memberPrice: z.number().nullish(),
      publicPrice: z.number().nullish(),
    }),
  })
)

const paymentSchema = z.object({
  paymentMethod: z.string(),
  card: z
    .object({
      alias: z.string(),
      expiryDate: z.string(),
      cardType: z.string(),
    })
    .optional(),
  cardHolder: z
    .object({
      email: z.string().email(),
      name: z.string(),
      phoneCountryCode: z.string(),
      phoneSubscriber: z.string(),
    })
    .optional(),
  success: z.string(),
  error: z.string(),
  cancel: z.string(),
})

// Mutation
export const createBookingInput = z.object({
  hotelId: z.string(),
  checkInDate: z.string(),
  checkOutDate: z.string(),
  rooms: roomsSchema,
  payment: paymentSchema.optional(),
  language: z.nativeEnum(Lang).transform((val) => langToApiLang[val]),
})

export const addPackageInput = z.object({
  confirmationNumber: z.string(),
  ancillaryComment: z.string(),
  ancillaryDeliveryTime: z.string().optional(),
  packages: z.array(
    z.object({
      code: z.string(),
      quantity: z.number(),
      comment: z.string().optional(),
    })
  ),
  language: z.nativeEnum(Lang).transform((val) => langToApiLang[val]),
})

export const removePackageInput = z.object({
  confirmationNumber: z.string(),
  codes: z.array(z.string()),
  language: z.nativeEnum(Lang).transform((val) => langToApiLang[val]),
})

export const priceChangeInput = z.object({
  confirmationNumber: z.string(),
})

export const cancelBookingInput = z.object({
  confirmationNumber: z.string(),
  language: z.nativeEnum(Lang).transform((val) => langToApiLang[val]),
})

export const guaranteeBookingInput = z.object({
  confirmationNumber: z.string(),
  card: z
    .object({
      alias: z.string(),
      expiryDate: z.string(),
      cardType: z.string(),
    })
    .optional(),
  language: z.nativeEnum(Lang).transform((val) => langToApiLang[val]),
  success: z.string().nullable(),
  error: z.string().nullable(),
  cancel: z.string().nullable(),
})

export const createRefIdInput = z.object({
  confirmationNumber: z
    .string()
    .trim()
    .regex(/^\s*[0-9]+(-[0-9])?\s*$/)
    .min(1),
  lastName: z.string().trim().max(250).min(1),
})

export const updateBookingInput = z.object({
  confirmationNumber: z.string(),
  checkInDate: z.string().optional(),
  checkOutDate: z.string().optional(),
  guest: z
    .object({
      email: z.string().optional(),
      phoneNumber: z.string().optional(),
      countryCode: z.string().optional(),
    })
    .optional(),
})

// Query
const confirmationNumberInput = z.object({
  confirmationNumber: z.string(),
  lang: z.nativeEnum(Lang).optional(),
})

export const bookingConfirmationInput = confirmationNumberInput

export const getBookingStatusInput = confirmationNumberInput