refactor: try using swc to compile generated code

zenstackhq · Jul 20, 2024 · a0b4c0d · a0b4c0d
1 parent 96f6516
commit a0b4c0d
Show file tree

Hide file tree

Showing 4 changed files with 184 additions and 46 deletions.
diff --git a/packages/sdk/package.json b/packages/sdk/package.json
@@ -20,6 +20,7 @@
     "dependencies": {
         "@prisma/generator-helper": "5.17.x",
         "@prisma/internals": "5.17.x",
+        "@swc/core": "^1.7.0",
         "@zenstackhq/language": "workspace:*",
         "@zenstackhq/runtime": "workspace:*",
         "langium": "1.3.1",

diff --git a/packages/sdk/src/code-gen.ts b/packages/sdk/src/code-gen.ts
@@ -1,5 +1,7 @@
-import { CompilerOptions, DiagnosticCategory, ModuleKind, Project, ScriptTarget } from 'ts-morph';
-import { PluginError } from './types';
+import { transformFile } from '@swc/core';
+import fs from 'fs';
+import path from 'path';
+import { CompilerOptions, ModuleKind, Project, ScriptTarget } from 'ts-morph';
 
 /**
  * Creates a TS code generation project
@@ -31,21 +33,27 @@ export async function saveProject(project: Project) {
  * Emit a TS project to JS files.
  */
 export async function emitProject(project: Project) {
-    const errors = project.getPreEmitDiagnostics().filter((d) => d.getCategory() === DiagnosticCategory.Error);
-    if (errors.length > 0) {
-        console.error('Error compiling generated code:');
-        console.error(project.formatDiagnosticsWithColorAndContext(errors.slice(0, 10)));
-        await project.save();
-        throw new PluginError('', `Error compiling generated code`);
-    }
-
-    const result = await project.emit();
+    await project.save();
 
-    const emitErrors = result.getDiagnostics().filter((d) => d.getCategory() === DiagnosticCategory.Error);
-    if (emitErrors.length > 0) {
-        console.error('Some generated code is not emitted:');
-        console.error(project.formatDiagnosticsWithColorAndContext(emitErrors.slice(0, 10)));
-        await project.save();
-        throw new PluginError('', `Error emitting generated code`);
-    }
+    await Promise.all(
+        project
+            .getSourceFiles()
+            .map((sf) => sf.getFilePath())
+            .filter((file) => !file.endsWith('.d.ts'))
+            .map(async (file) => {
+                const output = await transformFile(file, {
+                    jsc: {
+                        parser: {
+                            syntax: 'typescript',
+                            tsx: false,
+                        },
+                        target: 'es2020',
+                    },
+                    module: {
+                        type: 'commonjs',
+                    },
+                });
+                fs.writeFileSync(path.join(path.dirname(file), path.basename(file, '.ts') + '.js'), output.code);
+            })
+    );
 }
diff --git a/packages/testtools/src/schema.ts b/packages/testtools/src/schema.ts
@@ -294,7 +294,7 @@ export async function loadSchema(schema: string, options?: SchemaLoadOptions) {
         tsconfig.include = ['**/*.ts'];
         tsconfig.exclude = ['node_modules'];
         fs.writeFileSync(path.join(projectDir, './tsconfig.json'), JSON.stringify(tsconfig, null, 2));
-        run('npx tsc --project tsconfig.json');
+        run('npx tsc --project tsconfig.json --noEmit');
     }
 
     if (options?.getPrismaOnly) {