fix(zod): avoid importing Prisma enum, recognize enum fields with default

fixes #2291

Author: ymc9

packages/schema/src/plugins/zod/generator.ts

@@ -396,10 +396,6 @@ export const ${typeDef.name}Schema = ${refineFuncName}(${noRefineSchema});
                 }
             }
         }
-        if (importEnums.size > 0) {
-            const prismaImport = computePrismaClientImport(path.join(output, 'models'), this.options);
-            writer.writeLine(`import { ${[...importEnums].join(', ')} } from '${prismaImport}';`);
-        }
 
         // import enum schemas
         const importedEnumSchemas = new Set<string>();
@@ -730,9 +726,7 @@ export const ${upperCaseFirst(model.name)}UpdateSchema = ${updateSchema};
     /**
     * Schema refinement function for applying \`@@validate\` rules.
     */
-    export function ${refineFuncName}<T>(schema: z.ZodType<T>) { return schema${refinements.join(
-                    '\n'
-                )};
+    export function ${refineFuncName}<T>(schema: z.ZodType<T>) { return schema${refinements.join('\n')};
     }
     `
             );

packages/schema/src/plugins/zod/utils/schema-gen.ts

@@ -1,5 +1,5 @@
 import { upperCaseFirst } from '@zenstackhq/runtime/local-helpers';
-import { getLiteral, hasAttribute, isFromStdlib } from '@zenstackhq/sdk';
+import { getLiteral, hasAttribute, isEnumFieldReference, isFromStdlib } from '@zenstackhq/sdk';
 import {
     DataModelField,
     DataModelFieldAttribute,
@@ -246,6 +246,8 @@ export function getFieldSchemaDefault(field: DataModelField | TypeDefField) {
             return arg.value.value;
         } else if (isBooleanLiteral(arg.value)) {
             return arg.value.value;
+        } else if (isEnumFieldReference(arg.value) && arg.value.target.ref) {
+            return JSON.stringify(arg.value.target.ref.name);
         } else if (
             isInvocationExpr(arg.value) &&
             isFromStdlib(arg.value.function.ref!) &&

tests/regression/tests/issue-2291.test.ts

@@ -0,0 +1,23 @@
+import { loadSchema } from '@zenstackhq/testtools';
+
+describe('Issue 2291', () => {
+    it('should work', async () => {
+        const { zodSchemas } = await loadSchema(
+            `
+enum SomeEnum {
+    Ex1
+    Ex2
+}
+
+/// Post model
+model Post {
+    id        String   @id @default(cuid())
+    e SomeEnum @default(Ex1)
+}
+`,
+            { fullZod: true }
+        );
+
+        expect(zodSchemas.models.PostSchema.parse({ id: '1' })).toEqual({ id: '1', e: 'Ex1' });
+    });
+});