zenstackhq
diff --git a/‎packages/cli/package.json‎
Lines changed: 1 addition & 0 deletions b/‎packages/cli/package.json‎
Lines changed: 1 addition & 0 deletions
diff --git a/‎packages/cli/src/actions/action-utils.ts‎
Lines changed: 21 additions & 4 deletions b/‎packages/cli/src/actions/action-utils.ts‎
Lines changed: 21 additions & 4 deletions
diff --git a/‎packages/cli/src/actions/db.ts‎
Lines changed: 624 additions & 4 deletions b/‎packages/cli/src/actions/db.ts‎
Lines changed: 624 additions & 4 deletions
diff --git a/‎packages/cli/src/actions/pull/casing.ts‎
Lines changed: 43 additions & 0 deletions b/‎packages/cli/src/actions/pull/casing.ts‎
Lines changed: 43 additions & 0 deletions
@@ -37,6 +37,7 @@
     },
     "dependencies": {
         "@zenstackhq/common-helpers": "workspace:*",
+        "@zenstackhq/schema": "workspace:*",
         "@zenstackhq/language": "workspace:*",
         "@zenstackhq/orm": "workspace:*",
         "@zenstackhq/sdk": "workspace:*",
 
@@ -1,5 +1,5 @@
-import { loadDocument } from '@zenstackhq/language';
-import { isDataSource } from '@zenstackhq/language/ast';
+import { type ZModelServices, loadDocument } from '@zenstackhq/language';
+import { type Model, isDataSource } from '@zenstackhq/language/ast';
 import { PrismaSchemaGenerator } from '@zenstackhq/sdk';
 import colors from 'colors';
 import fs from 'node:fs';
@@ -41,8 +41,22 @@ export function getSchemaFile(file?: string) {
     }
 }
 
-export async function loadSchemaDocument(schemaFile: string) {
-    const loadResult = await loadDocument(schemaFile);
+export async function loadSchemaDocument(
+    schemaFile: string,
+    opts?: { mergeImports?: boolean; returnServices?: false },
+): Promise<Model>;
+export async function loadSchemaDocument(
+    schemaFile: string,
+    opts: { returnServices: true; mergeImports?: boolean },
+): Promise<{ model: Model; services: ZModelServices }>;
+export async function loadSchemaDocument(
+    schemaFile: string,
+    opts: { returnServices?: boolean; mergeImports?: boolean } = {},
+) {
+    const returnServices = opts.returnServices ?? false;
+    const mergeImports = opts.mergeImports ?? true;
+
+    const loadResult = await loadDocument(schemaFile, [], mergeImports);
     if (!loadResult.success) {
         loadResult.errors.forEach((err) => {
             console.error(colors.red(err));
@@ -52,6 +66,9 @@ export async function loadSchemaDocument(schemaFile: string) {
     loadResult.warnings.forEach((warn) => {
         console.warn(colors.yellow(warn));
     });
+
+    if (returnServices) return { model: loadResult.model, services: loadResult.services };
+
     return loadResult.model;
 }
 
 
@@ -0,0 +1,43 @@
+export function resolveNameCasing(casing: 'pascal' | 'camel' | 'snake' | 'none', originalName: string) {
+    let name = originalName;
+    const fieldPrefix = /[0-9]/g.test(name.charAt(0)) ? '_' : '';
+
+    switch (casing) {
+        case 'pascal':
+            name = toPascalCase(originalName);
+            break;
+        case 'camel':
+            name = toCamelCase(originalName);
+            break;
+        case 'snake':
+            name = toSnakeCase(originalName);
+            break;
+    }
+
+    return {
+        modified: name !== originalName || fieldPrefix !== '',
+        name: `${fieldPrefix}${name}`,
+    };
+}
+
+function isAllUpperCase(str: string): boolean {
+    return str === str.toUpperCase();
+}
+
+export function toPascalCase(str: string): string {
+    if (isAllUpperCase(str)) return str;
+    return str.replace(/[_\- ]+(\w)/g, (_, c) => c.toUpperCase()).replace(/^\w/, (c) => c.toUpperCase());
+}
+
+export function toCamelCase(str: string): string {
+    if (isAllUpperCase(str)) return str;
+    return str.replace(/[_\- ]+(\w)/g, (_, c) => c.toUpperCase()).replace(/^\w/, (c) => c.toLowerCase());
+}
+
+export function toSnakeCase(str: string): string {
+    if (isAllUpperCase(str)) return str;
+    return str
+        .replace(/[- ]+/g, '_')
+        .replace(/([a-z0-9])([A-Z])/g, '$1_$2')
+        .toLowerCase();
+}