Merge pull request #2 from exsjabe/enum

Feat: Support for enums

__tests__/testCases.ts

@@ -108,6 +108,16 @@ const testCases: TestCaseGroup[] = [
       ],
     ],
   },
+  {
+    name: 'z.enum()',
+    testCases: [
+      [
+        'z.enum(["a", "b"])',
+        z.enum(['a', 'b']),
+        { bsonType: 'string', enum: ['a', 'b'] },
+      ],
+    ],
+  },
 ]
 
 export default testCases

src/defParsers/parseEnumDef.ts

@@ -0,0 +1,13 @@
+import { type ZodEnumDef } from 'zod'
+
+import { type DefParser } from '../DefParser'
+import { type MongoSchema } from '../MongoSchema'
+
+const parseEnumDef: DefParser<ZodEnumDef> = (def): MongoSchema => {
+  return {
+    bsonType: 'string',
+    enum: def.values,
+  }
+}
+
+export default parseEnumDef

src/parseDef.ts

@@ -4,6 +4,7 @@ import { DefParser, ZodDef } from './DefParser'
 import { MongoSchema, MongoSchemaBool, MongoSchemaNull } from './MongoSchema'
 import parseArrayDef from './defParsers/parseArrayDef'
 import parseDateDef from './defParsers/parseDateDef'
+import parseEnumDef from './defParsers/parseEnumDef'
 import parseNullableDef from './defParsers/parseNullableDef'
 import parseNumberDef from './defParsers/parseNumberDef'
 import parseObjectDef from './defParsers/parseObjectDef'
@@ -46,4 +47,5 @@ const parseFnByKind: ParserFnByKind = {
   }),
   [ZodFirstPartyTypeKind.ZodArray]: parseArrayDef,
   [ZodFirstPartyTypeKind.ZodNullable]: parseNullableDef,
+  [ZodFirstPartyTypeKind.ZodEnum]: parseEnumDef,
 }