fix(notion-api-zod-schema): formula schema

alanlu-dev · May 3, 2024 · 2d9da9c · 2d9da9c
1 parent e1683dc
commit 2d9da9c
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/.changeset/fast-pants-play.md b/.changeset/fast-pants-play.md
@@ -0,0 +1,5 @@
+---
+"@alanlu-dev/notion-api-zod-schema": patch
+---
+
+fix(notion-api-zod-schema): formula schema
diff --git a/packages/notion-api-zod-schema/src/NotionDatabaseSchema.ts b/packages/notion-api-zod-schema/src/NotionDatabaseSchema.ts
@@ -166,10 +166,21 @@ export type NotionButtonType = z.infer<typeof NotionButtonSchema>
 // formula
 export const NotionFormulaSchema = z.object({
   type: z.literal('formula'),
-  formula: z.object({
-    type: z.literal('string'),
-    string: z.string().nullable(),
-  }),
+  formula: z.discriminatedUnion('type', [
+    z.object({
+      type: z.literal('string'),
+      string: z.string().nullable(),
+    }),
+    z.object({
+      type: z.literal('number'),
+      number: z.number().nullable(),
+    }),
+    z.object({
+      type: z.literal('boolean'),
+      boolean: z.boolean().nullable(),
+    }),
+    NotionDateSchema,
+  ]),
 })
 export type NotionFormulaType = z.infer<typeof NotionFormulaSchema>