import { z } from "zod"

import { discriminatedUnionArray } from "@/lib/discriminatedUnion"

import {
  cardGridRefsSchema,
  cardsGridSchema,
} from "../schemas/blocks/cardsGrid"
import {
  contentRefsSchema as blockContentRefsSchema,
  contentSchema as blockContentSchema,
} from "../schemas/blocks/content"
import {
  dynamicContentRefsSchema,
  dynamicContentSchema as blockDynamicContentSchema,
} from "../schemas/blocks/dynamicContent"
import {
  shortcutsRefsSchema,
  shortcutsSchema,
} from "../schemas/blocks/shortcuts"
import { textColsRefsSchema, textColsSchema } from "../schemas/blocks/textCols"
import { uspGridRefsSchema, uspGridSchema } from "../schemas/blocks/uspGrid"
import { tempImageVaultAssetSchema } from "../schemas/imageVault"
import {
  contentRefsSchema as sidebarContentRefsSchema,
  contentSchema as sidebarContentSchema,
} from "../schemas/sidebar/content"
import { dynamicContentSchema as sidebarDynamicContentSchema } from "../schemas/sidebar/dynamicContent"
import {
  joinLoyaltyContactRefsSchema,
  joinLoyaltyContactSchema,
} from "../schemas/sidebar/joinLoyaltyContact"
import { systemSchema } from "../schemas/system"

import { ContentPageEnum } from "@/types/enums/contentPage"

// Block schemas
export const contentPageCards = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.CardsGrid),
  })
  .merge(cardsGridSchema)

export const contentPageContent = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.Content),
  })
  .merge(blockContentSchema)

export const contentPageDynamicContent = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.DynamicContent),
  })
  .merge(blockDynamicContentSchema)

export const contentPageShortcuts = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.Shortcuts),
  })
  .merge(shortcutsSchema)

export const contentPageTextCols = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.TextCols),
  })
  .merge(textColsSchema)

export const contentPageUspGrid = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.UspGrid),
  })
  .merge(uspGridSchema)

export const blocksSchema = z.discriminatedUnion("__typename", [
  contentPageCards,
  contentPageContent,
  contentPageDynamicContent,
  contentPageShortcuts,
  contentPageTextCols,
  contentPageUspGrid,
])

export const contentPageSidebarContent = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.sidebar.Content),
  })
  .merge(sidebarContentSchema)

export const contentPageSidebarDynamicContent = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.sidebar.DynamicContent),
  })
  .merge(sidebarDynamicContentSchema)

export const contentPageJoinLoyaltyContact = z
  .object({
    __typename: z.literal(
      ContentPageEnum.ContentStack.sidebar.JoinLoyaltyContact
    ),
  })
  .merge(joinLoyaltyContactSchema)

export const sidebarSchema = z.discriminatedUnion("__typename", [
  contentPageSidebarContent,
  contentPageSidebarDynamicContent,
  contentPageJoinLoyaltyContact,
])

// Content Page Schema and types
export const contentPageSchema = z.object({
  content_page: z.object({
    hero_image: tempImageVaultAssetSchema,
    blocks: discriminatedUnionArray(blocksSchema.options).nullable(),
    sidebar: discriminatedUnionArray(sidebarSchema.options).nullable(),
    title: z.string(),
    header: z.object({
      heading: z.string(),
      preamble: z.string(),
    }),
    system: systemSchema.merge(
      z.object({
        created_at: z.string(),
        updated_at: z.string(),
      })
    ),
  }),
})

/** REFS */
const contentPageCardsRefs = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.CardsGrid),
  })
  .merge(cardGridRefsSchema)

const contentPageBlockContentRefs = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.Content),
  })
  .merge(blockContentRefsSchema)

const contentPageDynamicContentRefs = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.DynamicContent),
  })
  .merge(dynamicContentRefsSchema)

const contentPageShortcutsRefs = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.Shortcuts),
  })
  .merge(shortcutsRefsSchema)

const contentPageTextColsRefs = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.TextCols),
  })
  .merge(textColsRefsSchema)

const contentPageUspGridRefs = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.blocks.UspGrid),
  })
  .merge(uspGridRefsSchema)

const contentPageBlockRefsItem = z.discriminatedUnion("__typename", [
  contentPageBlockContentRefs,
  contentPageShortcutsRefs,
  contentPageCardsRefs,
  contentPageDynamicContentRefs,
  contentPageTextColsRefs,
  contentPageUspGridRefs,
])

const contentPageSidebarContentRef = z
  .object({
    __typename: z.literal(ContentPageEnum.ContentStack.sidebar.Content),
  })
  .merge(sidebarContentRefsSchema)

const contentPageSidebarJoinLoyaltyContactRef = z
  .object({
    __typename: z.literal(
      ContentPageEnum.ContentStack.sidebar.JoinLoyaltyContact
    ),
  })
  .merge(joinLoyaltyContactRefsSchema)

const contentPageSidebarRefsItem = z.discriminatedUnion("__typename", [
  contentPageSidebarContentRef,
  contentPageSidebarJoinLoyaltyContactRef,
])

export const contentPageRefsSchema = z.object({
  content_page: z.object({
    blocks: discriminatedUnionArray(
      contentPageBlockRefsItem.options
    ).nullable(),
    sidebar: discriminatedUnionArray(
      contentPageSidebarRefsItem.options
    ).nullable(),
    system: systemSchema,
  }),
})