import { z } from "astro/zod";
import { externalPropertyResponse, filePropertyResponse } from "./file.js";

export const pageObjectSchema = z.object({
  icon: z
    .discriminatedUnion("type", [
      externalPropertyResponse,
      filePropertyResponse,
      z.object({
        type: z.literal("emoji"),
        emoji: z.string(),
      }),
    ])
    .nullable(),
  cover: z
    .discriminatedUnion("type", [
      externalPropertyResponse,
      filePropertyResponse,
    ])
    .nullable(),
  archived: z.boolean(),
  in_trash: z.boolean(),
  url: z.string().url(),
  public_url: z.string().url().nullable(),
  properties: z.object({}).catchall(
    z
      .object({
        type: z.string(),
        id: z.string(),
      })
      .passthrough(),
  ),
});

/**
 * Defines a schema for a Notion page with a specific set of properties.
 * @example
 * const schema = notionPageSchema({
 *   properties: {
 *     Name: z.object({}),
 *     Hidden: transformedPropertySchema.checkbox.optional(),
 *   }
 * });
 */
export function notionPageSchema<Schema extends z.ZodTypeAny>({
  properties,
}: {
  properties: Schema;
}) {
  return pageObjectSchema.extend({ properties });
}