fix --lite-only option and add CLI test

Author: ymc9

packages/cli/src/plugins/typescript.ts

@@ -19,7 +19,10 @@ const plugin: CliPlugin = {
         // lite mode
         const lite = pluginOptions['lite'] === true;
 
-        await new TsSchemaGenerator().generate(model, { outDir, lite });
+        // liteOnly mode
+        const liteOnly = pluginOptions['liteOnly'] === true;
+
+        await new TsSchemaGenerator().generate(model, { outDir, lite, liteOnly });
     },
 };
 

packages/cli/test/generate.test.ts

@@ -44,4 +44,18 @@ describe('CLI generate command test', () => {
         runCli('generate', workDir);
         expect(fs.existsSync(path.join(workDir, 'bar/schema.ts'))).toBe(true);
     });
+
+    it('should respect lite option', () => {
+        const workDir = createProject(model);
+        runCli('generate --lite', workDir);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/schema.ts'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/schema-lite.ts'))).toBe(true);
+    });
+
+    it('should respect liteOnly option', () => {
+        const workDir = createProject(model);
+        runCli('generate --lite-only', workDir);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/schema.ts'))).toBe(false);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/schema-lite.ts'))).toBe(true);
+    });
 });

packages/cli/test/ts-schema-gen.test.ts

@@ -420,4 +420,26 @@ model User {
             },
         });
     });
+
+    it('supports lite schema generation', async () => {
+        const { schemaLite } = await generateTsSchema(
+            `
+model User {
+    id String @id @default(uuid())
+    name String
+    email String @unique
+
+    @@map('users')
+}
+        `,
+            undefined,
+            undefined,
+            undefined,
+            true,
+        );
+
+        expect(schemaLite!.models.User.attributes).toBeUndefined();
+        expect(schemaLite!.models.User.fields.id.attributes).toBeUndefined();
+        expect(schemaLite!.models.User.fields.email.attributes).toBeUndefined();
+    });
 });

packages/testtools/src/schema.ts

@@ -1,6 +1,6 @@
 import { invariant } from '@zenstackhq/common-helpers';
-import { TsSchemaGenerator } from '@zenstackhq/sdk';
 import type { SchemaDef } from '@zenstackhq/schema';
+import { TsSchemaGenerator } from '@zenstackhq/sdk';
 import { execSync } from 'node:child_process';
 import crypto from 'node:crypto';
 import fs from 'node:fs';
@@ -37,6 +37,7 @@ export async function generateTsSchema(
     provider: 'sqlite' | 'postgresql' = 'sqlite',
     dbUrl?: string,
     extraSourceFiles?: Record<string, string>,
+    withLiteSchema?: boolean,
 ) {
     const workDir = createTestProject();
 
@@ -50,7 +51,7 @@ export async function generateTsSchema(
     }
 
     const generator = new TsSchemaGenerator();
-    await generator.generate(result.model, { outDir: workDir });
+    await generator.generate(result.model, { outDir: workDir, lite: withLiteSchema });
 
     if (extraSourceFiles) {
         for (const [fileName, content] of Object.entries(extraSourceFiles)) {
@@ -72,7 +73,14 @@ async function compileAndLoad(workDir: string) {
 
     // load the schema module
     const module = await import(path.join(workDir, 'schema.js'));
-    return { workDir, schema: module.schema as SchemaDef };
+
+    let moduleLite: any;
+    try {
+        moduleLite = await import(path.join(workDir, 'schema-lite.js'));
+    } catch {
+        // ignore
+    }
+    return { workDir, schema: module.schema as SchemaDef, schemaLite: moduleLite?.schema as SchemaDef | undefined };
 }
 
 export function generateTsSchemaFromFile(filePath: string) {