79 lines
1.7 KiB
TypeScript
79 lines
1.7 KiB
TypeScript
import { z } from "zod"
|
|
|
|
import { discriminatedUnion } from "@/lib/discriminatedUnion"
|
|
import * as pageLinks from "@/server/routers/contentstack/schemas/pageLinks"
|
|
|
|
const linkUnionSchema = z.discriminatedUnion("__typename", [
|
|
pageLinks.contentPageSchema,
|
|
pageLinks.hotelPageSchema,
|
|
pageLinks.loyaltyPageSchema,
|
|
])
|
|
|
|
const titleSchema = z.object({
|
|
title: z.string().optional().default(""),
|
|
})
|
|
|
|
export const linkConnectionSchema = z
|
|
.object({
|
|
linkConnection: z.object({
|
|
edges: z.array(
|
|
z.object({
|
|
node: discriminatedUnion(linkUnionSchema.options),
|
|
})
|
|
),
|
|
}),
|
|
})
|
|
.transform((data) => {
|
|
if (data.linkConnection.edges.length) {
|
|
const linkNode = data.linkConnection.edges[0].node
|
|
if (linkNode) {
|
|
const link = pageLinks.transform(linkNode)
|
|
if (link) {
|
|
return {
|
|
link,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return {
|
|
link: null,
|
|
}
|
|
})
|
|
|
|
export const linkAndTitleSchema = z.intersection(
|
|
linkConnectionSchema,
|
|
titleSchema
|
|
)
|
|
|
|
const linkRefsUnionSchema = z.discriminatedUnion("__typename", [
|
|
pageLinks.contentPageRefSchema,
|
|
pageLinks.hotelPageRefSchema,
|
|
pageLinks.loyaltyPageRefSchema,
|
|
])
|
|
|
|
export const linkConnectionRefs = z
|
|
.object({
|
|
linkConnection: z.object({
|
|
edges: z.array(
|
|
z.object({
|
|
node: discriminatedUnion(linkRefsUnionSchema.options),
|
|
})
|
|
),
|
|
}),
|
|
})
|
|
.transform((data) => {
|
|
if (data.linkConnection.edges.length) {
|
|
const linkNode = data.linkConnection.edges[0].node
|
|
if (linkNode) {
|
|
const link = pageLinks.transformRef(linkNode)
|
|
if (link) {
|
|
return {
|
|
link,
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return { link: null }
|
|
})
|