fix: generated zod schemas fail to compile when outputting to a custom dir (#1611)

Author: ymc9

packages/schema/src/plugins/enhancer/enhance/index.ts

@@ -95,7 +95,8 @@ export class EnhancerGenerator {
 
         const enhanceTs = this.project.createSourceFile(
             path.join(this.outDir, 'enhance.ts'),
-            `import { type EnhancementContext, type EnhancementOptions, type ZodSchemas, type AuthUser } from '@zenstackhq/runtime';
+            `/* eslint-disable */
+import { type EnhancementContext, type EnhancementOptions, type ZodSchemas, type AuthUser } from '@zenstackhq/runtime';
 import { createEnhancement } from '@zenstackhq/runtime/enhancements';
 import modelMeta from './model-meta';
 import policy from './policy';

packages/schema/src/plugins/enhancer/index.ts

@@ -24,9 +24,11 @@ const run: PluginFunction = async (model, options, _dmmf, globalOptions) => {
     let prismaClientPath: string | undefined;
     if (dmmf) {
         // a logical client is generated
-        if (typeof options.output === 'string') {
+        if (options.output || globalOptions?.output) {
+            // handle custom output path
+
             // get the absolute path of the prisma client types
-            const prismaClientPathAbs = path.resolve(options.output, 'models');
+            const prismaClientPathAbs = path.resolve(outDir, 'models');
 
             // resolve it relative to the schema path
             prismaClientPath = path.relative(path.dirname(options.schemaPath), prismaClientPathAbs);

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

@@ -0,0 +1,56 @@
+import { loadSchema } from '@zenstackhq/testtools';
+describe('issue 1610', () => {
+    it('regular prisma client', async () => {
+        await loadSchema(
+            `
+            model User {
+                id Int @id
+                posts Post[]
+            }
+            
+            model Post {
+                id Int @id
+                author User @relation(fields: [authorId], references: [id])
+                authorId Int
+            }
+            `,
+            { fullZod: true, output: './lib/zen' }
+        );
+    });
+
+    it('logical prisma client', async () => {
+        await loadSchema(
+            `
+            model User {
+                id Int @id
+                posts Post[]
+            }
+            
+            model Post {
+                id Int @id
+                author User @relation(fields: [authorId], references: [id])
+                authorId Int @default(auth().id)
+            }
+            `,
+            { fullZod: true, output: './lib/zen' }
+        );
+    });
+
+    it('no custom output', async () => {
+        await loadSchema(
+            `
+            model User {
+                id Int @id
+                posts Post[]
+            }
+            
+            model Post {
+                id Int @id
+                author User @relation(fields: [authorId], references: [id])
+                authorId Int @default(auth().id)
+            }
+            `,
+            { fullZod: true, preserveTsFiles: true }
+        );
+    });
+});