import { z } from "zod"

import { BlocksEnums } from "../../../../types/blocksEnum"
import { ContentEnum } from "../../../../types/content"
import {
  rawLinkRefsUnionSchema,
  rawLinkUnionSchema,
  transformPageLink,
} from "../pageLinks"
import {
  imageContainerRefsSchema,
  imageContainerSchema,
} from "./imageContainer"
import { sysAssetRefsSchema, sysAssetSchema } from "./sysAsset"

export const contentSchema = z.object({
  typename: z
    .literal(BlocksEnums.block.Content)
    .optional()
    .default(BlocksEnums.block.Content),
  content: z
    .object({
      content: z
        .object({
          json: z.any(), // JSON
          embedded_itemsConnection: z.object({
            edges: z.array(
              z.object({
                node: z
                  .discriminatedUnion("__typename", [
                    imageContainerSchema,
                    sysAssetSchema,
                    ...rawLinkUnionSchema.options,
                  ])
                  .transform((data) => {
                    const link = transformPageLink(data)
                    if (link) {
                      return link
                    }
                    return data
                  }),
              })
            ),
          }),
        })
        .nullish(),
    })
    .nullish()
    .transform((data) => {
      return data?.content
    }),
})

export const contentRefsSchema = z.object({
  content: z
    .object({
      content: z
        .object({
          embedded_itemsConnection: z.object({
            edges: z.array(
              z.object({
                node: z.discriminatedUnion("__typename", [
                  sysAssetRefsSchema,
                  imageContainerRefsSchema,
                  ...rawLinkRefsUnionSchema.options,
                ]),
              })
            ),
          }),
        })
        .nullish(),
    })
    .nullish()
    .transform((data) => {
      return data?.content?.embedded_itemsConnection.edges
        .map(({ node }) => {
          switch (node.__typename) {
            case ContentEnum.blocks.SysAsset:
              return node.system && (node.permanent_url || node.url)
                ? { system: node.system, url: node.permanent_url || node.url }
                : null
            default:
              return node.system
          }
        })
        .filter((node) => !!node)
    }),
})