import { z } from "zod"

import { BlocksEnums } from "../../../../types/blocksEnum"
import { transformedAlertSchema } from "../alert"

export const alertBlockSchema = z.object({
  typename: z.literal(BlocksEnums.block.Alert).default(BlocksEnums.block.Alert),
  alert: z
    .object({
      alertConnection: z.object({
        edges: z.array(
          z.object({
            node: transformedAlertSchema,
          })
        ),
      }),
    })
    .transform((data) => {
      const alert = data.alertConnection.edges[0]?.node
      if (!alert) {
        return null
      }
      return alert
    }),
})