sass-mohfarawati/lib/media-validation.ts

import { MediaKind } from "@/lib/db/enums";
import { z } from "zod";

const mediaModeSchema = z.enum(["library", "external", "upload"]);

const optionalTrimmedText = z.string().trim().optional().transform((value) => value ?? "");

export const mediaFieldInputSchema = z
  .object({
    mode: mediaModeSchema,
    assetId: optionalTrimmedText,
    url: optionalTrimmedText,
    label: optionalTrimmedText,
    kind: z.nativeEnum(MediaKind),
  })
  .superRefine((value, context) => {
    if (value.mode === "library" && !value.assetId) {
      context.addIssue({
        code: z.ZodIssueCode.custom,
        path: ["assetId"],
        message: "Library selection requires a media asset.",
      });
    }

    if (value.mode === "external" && !value.url) {
      context.addIssue({
        code: z.ZodIssueCode.custom,
        path: ["url"],
        message: "External media requires a URL.",
      });
    }

    const isRootRelative = value.url.startsWith("/") && !value.url.startsWith("//");

    if (value.url && !/^https?:\/\//i.test(value.url) && !isRootRelative) {
      context.addIssue({
        code: z.ZodIssueCode.custom,
        path: ["url"],
        message: "Media URL must be an absolute URL or start with /.",
      });
    }
  });

export type MediaFieldInput = z.infer<typeof mediaFieldInputSchema>;