import { z } from "zod"

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

import { phoneValidator } from "@/utils/zod/phoneValidator"

// MUTATION
export const createBookingSchema = z
  .object({
    data: z.object({
      attributes: z.object({
        reservationStatus: z.string(),
        paymentUrl: z.string().nullable(),
        rooms: z
          .array(
            z.object({
              confirmationNumber: z.string(),
              cancellationNumber: z.string().nullable(),
              priceChangedMetadata: z
                .object({
                  roomPrice: z.number(),
                  totalPrice: z.number(),
                })
                .nullable()
                .optional(),
            })
          )
          .default([]),
        errors: z
          .array(
            z.object({
              confirmationNumber: z.string().nullable().optional(),
              errorCode: z.string(),
              description: z.string().nullable().optional(),
              meta: z
                .record(z.string(), z.union([z.string(), z.number()]))
                .nullable()
                .optional(),
            })
          )
          .default([]),
      }),
      type: z.string(),
      id: z.string(),
      links: z.object({
        self: z.object({
          href: z.string().url(),
          meta: z.object({
            method: z.string(),
          }),
        }),
      }),
    }),
  })
  .transform((d) => ({
    id: d.data.id,
    links: d.data.links,
    type: d.data.type,
    reservationStatus: d.data.attributes.reservationStatus,
    paymentUrl: d.data.attributes.paymentUrl,
    rooms: d.data.attributes.rooms,
    errors: d.data.attributes.errors,
  }))

// QUERY
const extraBedTypesSchema = z.object({
  bedType: z.nativeEnum(ChildBedTypeEnum),
  quantity: z.number().int(),
})

const guestSchema = z.object({
  email: z.string().email().nullable().default(""),
  firstName: z.string().nullable().default(""),
  lastName: z.string().nullable().default(""),
  membershipNumber: z.string().nullable().default(""),
  phoneNumber: phoneValidator().nullable().default(""),
})

const packageSchema = z.object({
  code: z.string().nullable().default(""),
  currency: z.string(),
  quantity: z.number().int(),
  totalPrice: z.number(),
  totalQuantity: z.number().int(),
  unitPrice: z.number(),
})

const rateDefinitionSchema = z.object({
  breakfastIncluded: z.boolean().default(false),
  cancellationRule: z.string().nullable().default(""),
  cancellationText: z.string().nullable().default(""),
  generalTerms: z.array(z.string()).default([]),
  isMemberRate: z.boolean().default(false),
  mustBeGuaranteed: z.boolean().default(false),
  rateCode: z.string().nullable().default(""),
  title: z.string().nullable().default(""),
})

export const bookingConfirmationSchema = z
  .object({
    data: z.object({
      attributes: z.object({
        adults: z.number().int(),
        checkInDate: z.date({ coerce: true }),
        checkOutDate: z.date({ coerce: true }),
        createDateTime: z.date({ coerce: true }),
        childrenAges: z.array(z.number().int()).default([]),
        extraBedTypes: z.array(extraBedTypesSchema).default([]),
        computedReservationStatus: z.string().nullable().default(""),
        confirmationNumber: z.string().nullable().default(""),
        currencyCode: z.string(),
        guest: guestSchema,
        hotelId: z.string(),
        packages: z.array(packageSchema).default([]),
        rateDefinition: rateDefinitionSchema,
        reservationStatus: z.string().nullable().default(""),
        roomPrice: z.number(),
        roomTypeCode: z.string().nullable().default(""),
        totalPrice: z.number(),
        totalPriceExVat: z.number(),
        vatAmount: z.number(),
        vatPercentage: z.number(),
      }),
      id: z.string(),
      type: z.literal("booking"),
    }),
  })
  .transform(({ data }) => data.attributes)