refactor: enforce type validation using z.union for tmdbId and mbId

2025-03-15 13:09:19 +01:00
parent 38532bd28a
commit c212858221
2 changed files with 24 additions and 14 deletions
--- a/server/interfaces/api/watchlistCreate.ts
+++ b/server/interfaces/api/watchlistCreate.ts
@@ -1,10 +1,15 @@
 import { MediaType } from '@server/constants/media';
 import { z } from 'zod';

-export const watchlistCreate = z.object({
-  ratingKey: z.coerce.string().optional(),
-  tmdbId: z.coerce.number().optional(),
-  mbId: z.coerce.string().optional(),
-  mediaType: z.nativeEnum(MediaType),
-  title: z.coerce.string().optional(),
-});
+export const watchlistCreate = z
+  .object({
+    ratingKey: z.coerce.string().optional(),
+    mediaType: z.nativeEnum(MediaType),
+    title: z.coerce.string().optional(),
+  })
+  .and(
+    z.union([
+      z.object({ tmdbId: z.coerce.number() }),
+      z.object({ mbId: z.coerce.string() }),
+    ])
+  );