import { z } from "zod"

import * as pageLinks from "@/server/routers/contentstack/schemas/pageLinks"

import { BlocksEnums } from "@/types/enums/blocks"
import { UspGridEnum } from "@/types/enums/uspGrid"

export const uspGridSchema = z.object({
  typename: z
    .literal(BlocksEnums.block.UspGrid)
    .optional()
    .default(BlocksEnums.block.UspGrid),
  usp_grid: z.object({
    usp_card: z.array(
      z.object({
        icon: UspGridEnum.uspIcons,
        text: z.object({
          json: z.any(), // JSON
          embedded_itemsConnection: z.object({
            edges: z.array(
              z.object({
                node: z
                  .discriminatedUnion("__typename", [
                    pageLinks.accountPageSchema,
                    pageLinks.contentPageSchema,
                    pageLinks.hotelPageSchema,
                    pageLinks.loyaltyPageSchema,
                  ])
                  .transform((data) => {
                    const link = pageLinks.transform(data)
                    if (link) {
                      return link
                    }
                    return data
                  }),
              })
            ),
          }),
        }),
      })
    ),
  }),
})

const actualRefs = z.discriminatedUnion("__typename", [
  pageLinks.accountPageRefSchema,
  pageLinks.contentPageRefSchema,
  pageLinks.hotelPageRefSchema,
  pageLinks.loyaltyPageRefSchema,
])

type Refs = {
  node: z.TypeOf<typeof actualRefs>
}

export const uspGridRefsSchema = z.object({
  usp_grid: z
    .object({
      usp_card: z.array(
        z.object({
          text: z.object({
            embedded_itemsConnection: z.object({
              edges: z.array(
                z.object({
                  node: z.discriminatedUnion("__typename", [
                    ...actualRefs.options,
                  ]),
                })
              ),
            }),
          }),
        })
      ),
    })
    .transform((data) => {
      return data.usp_card.flat()
    }),
})