feat: add ast factory

Author: svetch

packages/cli/src/actions/db.ts

@@ -3,20 +3,28 @@ import { ZModelCodeGenerator } from '@zenstackhq/sdk';
 import fs from 'node:fs';
 import path from 'node:path';
 import { execPackage } from '../utils/exec-utils';
-import { generateTempPrismaSchema, getSchemaFile, handleSubProcessError, loadSchemaDocumentWithServices } from './action-utils';
+import {
+    generateTempPrismaSchema,
+    getSchemaFile,
+    handleSubProcessError,
+    loadSchemaDocumentWithServices,
+} from './action-utils';
 import { syncEnums, syncRelation, syncTable, type Relation } from './pull';
 import { providers } from './pull/provider';
 import { getDatasource } from './pull/utils';
+import { config } from '@dotenvx/dotenvx';
 
 type PushOptions = {
     schema?: string;
     acceptDataLoss?: boolean;
     forceReset?: boolean;
 };
 
-type PullOptions = {
+export type PullOptions = {
     schema?: string;
     out?: string;
+    naming?: 'pascal' | 'camel' | 'snake' | 'kebab' | 'none';
+    alwaysMap?: boolean;
 };
 
 /**
@@ -63,62 +71,57 @@ async function runPush(options: PushOptions) {
 async function runPull(options: PullOptions) {
     const schemaFile = getSchemaFile(options.schema);
     const { model, services } = await loadSchemaDocumentWithServices(schemaFile);
-    await import("@dotenvx/dotenvx/config")
-    const SUPPORTED_PROVIDERS = ['sqlite', 'postgresql']
-    const datasource = getDatasource(model)
+    config();
+    const SUPPORTED_PROVIDERS = ['sqlite', 'postgresql'];
+    const datasource = getDatasource(model);
 
     if (!datasource) {
-        throw new Error('No datasource found in the schema.')
+        throw new Error('No datasource found in the schema.');
     }
 
     if (!SUPPORTED_PROVIDERS.includes(datasource.provider)) {
-        throw new Error(`Unsupported datasource provider: ${datasource.provider}`)
+        throw new Error(`Unsupported datasource provider: ${datasource.provider}`);
     }
 
     const provider = providers[datasource.provider];
 
     if (!provider) {
-        throw new Error(
-            `No introspection provider found for: ${datasource.provider}`
-        )
+        throw new Error(`No introspection provider found for: ${datasource.provider}`);
     }
 
-    const { enums, tables } = await provider.introspect(datasource.url)
+    const { enums, tables } = await provider.introspect(datasource.url);
 
     const newModel: Model = {
         $type: 'Model',
         $container: undefined,
         $containerProperty: undefined,
         $containerIndex: undefined,
-        declarations: [...model.declarations.filter(d => ["DataSource"].includes(d.$type))],
+        declarations: [...model.declarations.filter((d) => ['DataSource'].includes(d.$type))],
         imports: [],
     };
 
+    syncEnums({ dbEnums: enums, model: newModel, services, options });
 
-    syncEnums({ dbEnums: enums, model: newModel, services })
-
-
-
-    const resolvedRelations: Relation[] = []
+    const resolvedRelations: Relation[] = [];
     for (const table of tables) {
-        const relations = syncTable({ table, model: newModel, provider, services })
-        resolvedRelations.push(...relations)
+        const relations = syncTable({ table, model: newModel, provider, services, options });
+        resolvedRelations.push(...relations);
     }
 
     for (const relation of resolvedRelations) {
-        syncRelation({ model: newModel, relation, services });
+        syncRelation({ model: newModel, relation, services, options });
     }
 
     //TODO: diff models and apply changes only
 
-    const generator = await new ZModelCodeGenerator();
+    const generator = new ZModelCodeGenerator();
 
-    const zmodelSchema = await generator.generate(newModel)
+    const zmodelSchema = generator.generate(newModel);
 
     console.log(options.out ? `Writing to ${options.out}` : schemaFile);
 
     const outPath = options.out ? path.resolve(options.out) : schemaFile;
     console.log(outPath);
 
-    fs.writeFileSync(outPath, zmodelSchema)
+    fs.writeFileSync(outPath, zmodelSchema);
 }