@zenstackhq/sdk 3.0.0-alpha.1 → 3.0.0-alpha.11

package/dist/index.d.cts

@@ -1,5 +1,6 @@
 import * as _zenstackhq_language_ast from '@zenstackhq/language/ast';
 import { DataModelField, DataModel, TypeDef, Enum, EnumField, FunctionDecl, Attribute, AttributeParam, TypeDefField, DataModelFieldAttribute, DataModelAttribute, AstNode, Model, Reference, InvocationExpr } from '@zenstackhq/language/ast';
+import { MaybePromise } from 'langium';
 
 declare function isIdField(field: DataModelField): boolean;
 declare function hasAttribute(decl: DataModel | TypeDef | DataModelField | Enum | EnumField | FunctionDecl | Attribute | AttributeParam, name: string): boolean;
@@ -36,6 +37,13 @@ declare namespace modelUtils {
   export { modelUtils_getAttribute as getAttribute, modelUtils_getAuthDecl as getAuthDecl, modelUtils_getContainingModel as getContainingModel, modelUtils_getModelIdFields as getModelIdFields, modelUtils_getModelUniqueFields as getModelUniqueFields, modelUtils_getRecursiveBases as getRecursiveBases, modelUtils_hasAttribute as hasAttribute, modelUtils_isDelegateModel as isDelegateModel, modelUtils_isFromStdlib as isFromStdlib, modelUtils_isIdField as isIdField, modelUtils_isUniqueField as isUniqueField, modelUtils_resolved as resolved };
 }
 
+type CliGeneratorContext = {
+    model: Model;
+    outputPath: string;
+    tsSchemaFile: string;
+};
+type CliGenerator = (context: CliGeneratorContext) => MaybePromise<void>;
+
 declare class FunctionCall {
     func: string;
     args: FunctionCallArg[];
@@ -82,20 +90,19 @@ declare class PrismaSchemaGenerator {
 }
 
 declare class TsSchemaGenerator {
-    generate(schemaFile: string, pluginModelFiles: string[], outputFile: string): Promise<{
-        model: Model;
-        warnings: string[];
-    }>;
+    generate(schemaFile: string, pluginModelFiles: string[], outputDir: string): Promise<void>;
+    private generateSchema;
     private generateSchemaStatements;
     private createSchemaObject;
     private createProviderObject;
     private createModelsObject;
     private createDataModelObject;
     private createComputedFieldsObject;
-    private mapTypeToTSSyntaxKeyword;
+    private mapFieldTypeToTSType;
     private createDataModelFieldObject;
     private getDataSourceProvider;
-    private getMappedDefault;
+    private getFieldMappedDefault;
+    private getMappedValue;
     private getMemberAccessChain;
     private isAuthMemberAccess;
     private isAuthInvocation;
@@ -106,10 +113,10 @@ declare class TsSchemaGenerator {
     private getRelationName;
     private getIdFields;
     private createUniqueFieldsObject;
+    private generateFieldTypeLiteral;
     private createEnumObject;
     private getLiteral;
     private createLiteralNode;
-    private createDialectConfigProvider;
     private createProceduresObject;
     private createProcedureObject;
     private generateBannerComments;
@@ -125,6 +132,10 @@ declare class TsSchemaGenerator {
     private createRefExpression;
     private createCallExpression;
     private createLiteralExpression;
+    private generateModels;
+    private generateSchemaTypeImport;
+    private generateDocs;
+    private generateInputTypes;
 }
 
 /**
@@ -193,4 +204,4 @@ declare class ZModelCodeGenerator {
     private isCollectionPredicateOperator;
 }
 
-export { modelUtils as ModelUtils, PrismaSchemaGenerator, TsSchemaGenerator, ZModelCodeGenerator, type ZModelCodeOptions };
+export { type CliGenerator, type CliGeneratorContext, modelUtils as ModelUtils, PrismaSchemaGenerator, TsSchemaGenerator, ZModelCodeGenerator, type ZModelCodeOptions };

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 import * as _zenstackhq_language_ast from '@zenstackhq/language/ast';
 import { DataModelField, DataModel, TypeDef, Enum, EnumField, FunctionDecl, Attribute, AttributeParam, TypeDefField, DataModelFieldAttribute, DataModelAttribute, AstNode, Model, Reference, InvocationExpr } from '@zenstackhq/language/ast';
+import { MaybePromise } from 'langium';
 
 declare function isIdField(field: DataModelField): boolean;
 declare function hasAttribute(decl: DataModel | TypeDef | DataModelField | Enum | EnumField | FunctionDecl | Attribute | AttributeParam, name: string): boolean;
@@ -36,6 +37,13 @@ declare namespace modelUtils {
   export { modelUtils_getAttribute as getAttribute, modelUtils_getAuthDecl as getAuthDecl, modelUtils_getContainingModel as getContainingModel, modelUtils_getModelIdFields as getModelIdFields, modelUtils_getModelUniqueFields as getModelUniqueFields, modelUtils_getRecursiveBases as getRecursiveBases, modelUtils_hasAttribute as hasAttribute, modelUtils_isDelegateModel as isDelegateModel, modelUtils_isFromStdlib as isFromStdlib, modelUtils_isIdField as isIdField, modelUtils_isUniqueField as isUniqueField, modelUtils_resolved as resolved };
 }
 
+type CliGeneratorContext = {
+    model: Model;
+    outputPath: string;
+    tsSchemaFile: string;
+};
+type CliGenerator = (context: CliGeneratorContext) => MaybePromise<void>;
+
 declare class FunctionCall {
     func: string;
     args: FunctionCallArg[];
@@ -82,20 +90,19 @@ declare class PrismaSchemaGenerator {
 }
 
 declare class TsSchemaGenerator {
-    generate(schemaFile: string, pluginModelFiles: string[], outputFile: string): Promise<{
-        model: Model;
-        warnings: string[];
-    }>;
+    generate(schemaFile: string, pluginModelFiles: string[], outputDir: string): Promise<void>;
+    private generateSchema;
     private generateSchemaStatements;
     private createSchemaObject;
     private createProviderObject;
     private createModelsObject;
     private createDataModelObject;
     private createComputedFieldsObject;
-    private mapTypeToTSSyntaxKeyword;
+    private mapFieldTypeToTSType;
     private createDataModelFieldObject;
     private getDataSourceProvider;
-    private getMappedDefault;
+    private getFieldMappedDefault;
+    private getMappedValue;
     private getMemberAccessChain;
     private isAuthMemberAccess;
     private isAuthInvocation;
@@ -106,10 +113,10 @@ declare class TsSchemaGenerator {
     private getRelationName;
     private getIdFields;
     private createUniqueFieldsObject;
+    private generateFieldTypeLiteral;
     private createEnumObject;
     private getLiteral;
     private createLiteralNode;
-    private createDialectConfigProvider;
     private createProceduresObject;
     private createProcedureObject;
     private generateBannerComments;
@@ -125,6 +132,10 @@ declare class TsSchemaGenerator {
     private createRefExpression;
     private createCallExpression;
     private createLiteralExpression;
+    private generateModels;
+    private generateSchemaTypeImport;
+    private generateDocs;
+    private generateInputTypes;
 }
 
 /**
@@ -193,4 +204,4 @@ declare class ZModelCodeGenerator {
     private isCollectionPredicateOperator;
 }
 
-export { modelUtils as ModelUtils, PrismaSchemaGenerator, TsSchemaGenerator, ZModelCodeGenerator, type ZModelCodeOptions };
+export { type CliGenerator, type CliGeneratorContext, modelUtils as ModelUtils, PrismaSchemaGenerator, TsSchemaGenerator, ZModelCodeGenerator, type ZModelCodeOptions };

package/dist/index.js

@@ -285,9 +285,9 @@ var Model = class extends ContainerDeclaration {
   }
   name;
   isView;
-  fields;
+  fields = [];
   constructor(name, isView, documentations = []) {
-    super(documentations), this.name = name, this.isView = isView, this.fields = [];
+    super(documentations), this.name = name, this.isView = isView;
   }
   addField(name, type, attributes = [], documentations = [], addToFront = false) {
     const field = new ModelField(name, type, attributes, documentations);
@@ -507,9 +507,9 @@ var Enum = class extends ContainerDeclaration {
     __name(this, "Enum");
   }
   name;
-  fields;
+  fields = [];
   constructor(name, documentations = []) {
-    super(documentations), this.name = name, this.fields = [];
+    super(documentations), this.name = name;
   }
   addField(name, attributes = [], documentations = []) {
     const field = new EnumField(name, attributes, documentations);
@@ -558,15 +558,14 @@ var PrismaSchemaGenerator = class {
     __name(this, "PrismaSchemaGenerator");
   }
   zmodel;
-  PRELUDE;
-  constructor(zmodel) {
-    this.zmodel = zmodel;
-    this.PRELUDE = `//////////////////////////////////////////////////////////////////////////////////////////////
+  PRELUDE = `//////////////////////////////////////////////////////////////////////////////////////////////
 // DO NOT MODIFY THIS FILE                                                                  //
 // This file is automatically generated by ZenStack CLI and should not be manually updated. //
 //////////////////////////////////////////////////////////////////////////////////////////////
 
 `;
+  constructor(zmodel) {
+    this.zmodel = zmodel;
   }
   async generate() {
     const prisma = new PrismaModel();
@@ -777,37 +776,39 @@ var PrismaSchemaGenerator = class {
 };
 
 // src/ts-schema-generator.ts
+import { invariant } from "@zenstackhq/common-helpers";
 import { loadDocument } from "@zenstackhq/language";
 import { isArrayExpr as isArrayExpr3, isBinaryExpr, isDataModel as isDataModel2, isDataModelField, isDataSource, isEnum, isEnumField, isInvocationExpr as isInvocationExpr2, isLiteralExpr as isLiteralExpr3, isMemberAccessExpr, isNullExpr as isNullExpr2, isProcedure, isReferenceExpr as isReferenceExpr3, isThisExpr, isUnaryExpr } from "@zenstackhq/language/ast";
-import fs from "node:fs";
-import path from "node:path";
-import invariant from "tiny-invariant";
+import fs from "fs";
+import path from "path";
 import { match as match2 } from "ts-pattern";
 import * as ts from "typescript";
 var TsSchemaGenerator = class {
   static {
     __name(this, "TsSchemaGenerator");
   }
-  async generate(schemaFile, pluginModelFiles, outputFile) {
+  async generate(schemaFile, pluginModelFiles, outputDir) {
     const loaded = await loadDocument(schemaFile, pluginModelFiles);
     if (!loaded.success) {
       throw new Error(`Error loading schema:${loaded.errors.join("\n")}`);
     }
-    const { model, warnings } = loaded;
+    const { model } = loaded;
+    fs.mkdirSync(outputDir, {
+      recursive: true
+    });
+    this.generateSchema(model, outputDir);
+    this.generateModels(model, outputDir);
+    this.generateInputTypes(model, outputDir);
+  }
+  generateSchema(model, outputDir) {
     const statements = [];
     this.generateSchemaStatements(model, statements);
     this.generateBannerComments(statements);
-    const sourceFile = ts.createSourceFile(outputFile, "", ts.ScriptTarget.ESNext, false, ts.ScriptKind.TS);
+    const schemaOutputFile = path.join(outputDir, "schema.ts");
+    const sourceFile = ts.createSourceFile(schemaOutputFile, "", ts.ScriptTarget.ESNext, false, ts.ScriptKind.TS);
     const printer = ts.createPrinter();
     const result = printer.printList(ts.ListFormat.MultiLine, ts.factory.createNodeArray(statements), sourceFile);
-    fs.mkdirSync(path.dirname(outputFile), {
-      recursive: true
-    });
-    fs.writeFileSync(outputFile, result);
-    return {
-      model,
-      warnings
-    };
+    fs.writeFileSync(schemaOutputFile, result);
   }
   generateSchemaStatements(model, statements) {
     const hasComputedFields = model.declarations.some((d) => isDataModel2(d) && d.fields.some((f) => hasAttribute(f, "@computed")));
@@ -819,27 +820,6 @@ var TsSchemaGenerator = class {
       ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("ExpressionUtils"))
     ])), ts.factory.createStringLiteral("@zenstackhq/runtime/schema"));
     statements.push(runtimeImportDecl);
-    const { type: providerType } = this.getDataSourceProvider(model);
-    switch (providerType) {
-      case "sqlite": {
-        const pathImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, ts.factory.createIdentifier("path"), void 0), ts.factory.createStringLiteral("node:path"));
-        statements.push(pathImportDecl);
-        const urlImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, ts.factory.createIdentifier("url"), void 0), ts.factory.createStringLiteral("node:url"));
-        statements.push(urlImportDecl);
-        const dialectConfigImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports([
-          ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("toDialectConfig"))
-        ])), ts.factory.createStringLiteral("@zenstackhq/runtime/utils/sqlite-utils"));
-        statements.push(dialectConfigImportDecl);
-        break;
-      }
-      case "postgresql": {
-        const dialectConfigImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports([
-          ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("toDialectConfig"))
-        ])), ts.factory.createStringLiteral("@zenstackhq/runtime/utils/pg-utils"));
-        statements.push(dialectConfigImportDecl);
-        break;
-      }
-    }
     const declaration = ts.factory.createVariableStatement([
       ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
     ], ts.factory.createVariableDeclarationList([
@@ -876,8 +856,7 @@ var TsSchemaGenerator = class {
   createProviderObject(model) {
     const dsProvider = this.getDataSourceProvider(model);
     return ts.factory.createObjectLiteralExpression([
-      ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(dsProvider.type)),
-      ts.factory.createPropertyAssignment("dialectConfigProvider", this.createDialectConfigProvider(dsProvider))
+      ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(dsProvider.type))
     ], true);
   }
   createModelsObject(model) {
@@ -904,19 +883,26 @@ var TsSchemaGenerator = class {
   }
   createComputedFieldsObject(fields) {
     return ts.factory.createObjectLiteralExpression(fields.map((field) => ts.factory.createMethodDeclaration(void 0, void 0, field.name, void 0, void 0, [], ts.factory.createTypeReferenceNode("OperandExpression", [
-      ts.factory.createKeywordTypeNode(this.mapTypeToTSSyntaxKeyword(field.type.type))
+      ts.factory.createTypeReferenceNode(this.mapFieldTypeToTSType(field.type))
     ]), ts.factory.createBlock([
       ts.factory.createThrowStatement(ts.factory.createNewExpression(ts.factory.createIdentifier("Error"), void 0, [
         ts.factory.createStringLiteral("This is a stub for computed field")
       ]))
     ], true))), true);
   }
-  mapTypeToTSSyntaxKeyword(type) {
-    return match2(type).with("String", () => ts.SyntaxKind.StringKeyword).with("Boolean", () => ts.SyntaxKind.BooleanKeyword).with("Int", () => ts.SyntaxKind.NumberKeyword).with("Float", () => ts.SyntaxKind.NumberKeyword).with("BigInt", () => ts.SyntaxKind.BigIntKeyword).with("Decimal", () => ts.SyntaxKind.NumberKeyword).otherwise(() => ts.SyntaxKind.UnknownKeyword);
+  mapFieldTypeToTSType(type) {
+    let result = match2(type.type).with("String", () => "string").with("Boolean", () => "boolean").with("Int", () => "number").with("Float", () => "number").with("BigInt", () => "bigint").with("Decimal", () => "number").otherwise(() => "unknown");
+    if (type.array) {
+      result = `${result}[]`;
+    }
+    if (type.optional) {
+      result = `${result} | null`;
+    }
+    return result;
   }
   createDataModelFieldObject(field) {
     const objectFields = [
-      ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(field.type.type ?? field.type.reference.$refText))
+      ts.factory.createPropertyAssignment("type", this.generateFieldTypeLiteral(field))
     ];
     if (isIdField(field)) {
       objectFields.push(ts.factory.createPropertyAssignment("id", ts.factory.createTrue()));
@@ -936,9 +922,9 @@ var TsSchemaGenerator = class {
     if (field.attributes.length > 0) {
       objectFields.push(ts.factory.createPropertyAssignment("attributes", ts.factory.createArrayLiteralExpression(field.attributes.map((attr) => this.createAttributeObject(attr)))));
     }
-    const defaultValue = this.getMappedDefault(field);
+    const defaultValue = this.getFieldMappedDefault(field);
     if (defaultValue !== void 0) {
-      if (typeof defaultValue === "object") {
+      if (typeof defaultValue === "object" && !Array.isArray(defaultValue)) {
         if ("call" in defaultValue) {
           objectFields.push(ts.factory.createPropertyAssignment("default", ts.factory.createCallExpression(ts.factory.createIdentifier("ExpressionUtils.call"), void 0, [
             ts.factory.createStringLiteral(defaultValue.call),
@@ -957,7 +943,11 @@ var TsSchemaGenerator = class {
           throw new Error(`Unsupported default value type for field ${field.name}`);
         }
       } else {
-        objectFields.push(ts.factory.createPropertyAssignment("default", typeof defaultValue === "string" ? ts.factory.createStringLiteral(defaultValue) : typeof defaultValue === "number" ? ts.factory.createNumericLiteral(defaultValue) : defaultValue === true ? ts.factory.createTrue() : ts.factory.createFalse()));
+        if (Array.isArray(defaultValue)) {
+          objectFields.push(ts.factory.createPropertyAssignment("default", ts.factory.createArrayLiteralExpression(defaultValue.map((item) => this.createLiteralNode(item)))));
+        } else {
+          objectFields.push(ts.factory.createPropertyAssignment("default", this.createLiteralNode(defaultValue)));
+        }
       }
     }
     if (hasAttribute(field, "@computed")) {
@@ -998,34 +988,39 @@ var TsSchemaGenerator = class {
       throw new Error("Unsupported URL type");
     }
   }
-  getMappedDefault(field) {
+  getFieldMappedDefault(field) {
     const defaultAttr = getAttribute(field, "@default");
     if (!defaultAttr) {
       return void 0;
     }
     const defaultValue = defaultAttr.args[0]?.value;
     invariant(defaultValue, "Expected a default value");
-    if (isLiteralExpr3(defaultValue)) {
-      const lit = defaultValue.value;
-      return field.type.type === "Boolean" ? lit : [
+    return this.getMappedValue(defaultValue, field.type);
+  }
+  getMappedValue(expr, fieldType) {
+    if (isLiteralExpr3(expr)) {
+      const lit = expr.value;
+      return fieldType.type === "Boolean" ? lit : [
         "Int",
         "Float",
         "Decimal",
         "BigInt"
-      ].includes(field.type.type) ? Number(lit) : lit;
-    } else if (isReferenceExpr3(defaultValue) && isEnumField(defaultValue.target.ref)) {
-      return defaultValue.target.ref.name;
-    } else if (isInvocationExpr2(defaultValue)) {
+      ].includes(fieldType.type) ? Number(lit) : lit;
+    } else if (isArrayExpr3(expr)) {
+      return expr.items.map((item) => this.getMappedValue(item, fieldType));
+    } else if (isReferenceExpr3(expr) && isEnumField(expr.target.ref)) {
+      return expr.target.ref.name;
+    } else if (isInvocationExpr2(expr)) {
       return {
-        call: defaultValue.function.$refText,
-        args: defaultValue.args.map((arg) => this.getLiteral(arg.value))
+        call: expr.function.$refText,
+        args: expr.args.map((arg) => this.getLiteral(arg.value))
       };
-    } else if (this.isAuthMemberAccess(defaultValue)) {
+    } else if (this.isAuthMemberAccess(expr)) {
       return {
-        authMember: this.getMemberAccessChain(defaultValue)
+        authMember: this.getMemberAccessChain(expr)
       };
     } else {
-      throw new Error(`Unsupported default value type for field ${field.name}`);
+      throw new Error(`Unsupported default value type for ${expr.$type}`);
     }
   }
   getMemberAccessChain(expr) {
@@ -1138,10 +1133,11 @@ var TsSchemaGenerator = class {
     for (const field of dm.fields) {
       if (hasAttribute(field, "@id") || hasAttribute(field, "@unique")) {
         properties.push(ts.factory.createPropertyAssignment(field.name, ts.factory.createObjectLiteralExpression([
-          ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(field.type.type))
+          ts.factory.createPropertyAssignment("type", this.generateFieldTypeLiteral(field))
         ])));
       }
     }
+    const seenKeys = /* @__PURE__ */ new Set();
     for (const attr of dm.attributes) {
       if (attr.decl.$refText === "@@id" || attr.decl.$refText === "@@unique") {
         const fieldNames = this.getReferenceNames(attr.args[0].value);
@@ -1151,13 +1147,18 @@ var TsSchemaGenerator = class {
         if (fieldNames.length === 1) {
           const fieldDef = dm.fields.find((f) => f.name === fieldNames[0]);
           properties.push(ts.factory.createPropertyAssignment(fieldNames[0], ts.factory.createObjectLiteralExpression([
-            ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(fieldDef.type.type))
+            ts.factory.createPropertyAssignment("type", this.generateFieldTypeLiteral(fieldDef))
           ])));
         } else {
+          const key = fieldNames.join("_");
+          if (seenKeys.has(key)) {
+            continue;
+          }
+          seenKeys.add(key);
           properties.push(ts.factory.createPropertyAssignment(fieldNames.join("_"), ts.factory.createObjectLiteralExpression(fieldNames.map((field) => {
             const fieldDef = dm.fields.find((f) => f.name === field);
             return ts.factory.createPropertyAssignment(field, ts.factory.createObjectLiteralExpression([
-              ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(fieldDef.type.type))
+              ts.factory.createPropertyAssignment("type", this.generateFieldTypeLiteral(fieldDef))
             ]));
           }))));
         }
@@ -1165,6 +1166,10 @@ var TsSchemaGenerator = class {
     }
     return ts.factory.createObjectLiteralExpression(properties, true);
   }
+  generateFieldTypeLiteral(field) {
+    invariant(field.type.type || field.type.reference || field.type.unsupported, "Field type must be a primitive, reference, or Unsupported");
+    return field.type.type ? ts.factory.createStringLiteral(field.type.type) : field.type.reference ? ts.factory.createStringLiteral(field.type.reference.$refText) : ts.factory.createStringLiteral("Unsupported");
+  }
   createEnumObject(e) {
     return ts.factory.createObjectLiteralExpression(e.fields.map((field) => ts.factory.createPropertyAssignment(field.name, ts.factory.createStringLiteral(field.name))), true);
   }
@@ -1185,44 +1190,6 @@ var TsSchemaGenerator = class {
   createLiteralNode(arg) {
     return arg === null ? ts.factory.createNull() : typeof arg === "string" ? ts.factory.createStringLiteral(arg) : typeof arg === "number" ? ts.factory.createNumericLiteral(arg) : arg === true ? ts.factory.createTrue() : arg === false ? ts.factory.createFalse() : void 0;
   }
-  createDialectConfigProvider(dsProvider) {
-    const type = dsProvider.type;
-    let urlExpr;
-    if (dsProvider.env !== void 0) {
-      urlExpr = ts.factory.createIdentifier(`process.env['${dsProvider.env}']`);
-    } else {
-      urlExpr = ts.factory.createStringLiteral(dsProvider.url);
-      if (type === "sqlite") {
-        let parsedUrl;
-        try {
-          parsedUrl = new URL(dsProvider.url);
-        } catch {
-        }
-        if (parsedUrl) {
-          if (parsedUrl.protocol !== "file:") {
-            throw new Error("Invalid SQLite URL: only file protocol is supported");
-          }
-          urlExpr = ts.factory.createStringLiteral(dsProvider.url.replace(/^file:/, ""));
-        }
-      }
-    }
-    return match2(type).with("sqlite", () => {
-      return ts.factory.createFunctionExpression(void 0, void 0, void 0, void 0, void 0, void 0, ts.factory.createBlock([
-        ts.factory.createReturnStatement(ts.factory.createCallExpression(ts.factory.createIdentifier("toDialectConfig"), void 0, [
-          urlExpr,
-          ts.factory.createIdentifier(`typeof __dirname !== 'undefined' ? __dirname : path.dirname(url.fileURLToPath(import.meta.url))`)
-        ]))
-      ], true));
-    }).with("postgresql", () => {
-      return ts.factory.createFunctionExpression(void 0, void 0, void 0, void 0, void 0, void 0, ts.factory.createBlock([
-        ts.factory.createReturnStatement(ts.factory.createCallExpression(ts.factory.createIdentifier("toDialectConfig"), void 0, [
-          urlExpr
-        ]))
-      ], true));
-    }).otherwise(() => {
-      throw new Error(`Unsupported provider: ${type}`);
-    });
-  }
   createProceduresObject(procedures) {
     return ts.factory.createObjectLiteralExpression(procedures.map((proc) => ts.factory.createPropertyAssignment(proc.name, this.createProcedureObject(proc))), true);
   }
@@ -1257,6 +1224,8 @@ var TsSchemaGenerator = class {
 // This file is automatically generated by ZenStack CLI and should not be manually updated. //
 //////////////////////////////////////////////////////////////////////////////////////////////
 
+/* eslint-disable */
+
 `;
     ts.addSyntheticLeadingComment(statements[0], ts.SyntaxKind.SingleLineCommentTrivia, banner);
   }
@@ -1351,6 +1320,132 @@ var TsSchemaGenerator = class {
       throw new Error(`Unsupported literal type: ${type}`);
     });
   }
+  generateModels(model, outputDir) {
+    const statements = [];
+    statements.push(this.generateSchemaTypeImport(true, true));
+    statements.push(ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports([
+      ts.factory.createImportSpecifier(true, void 0, ts.factory.createIdentifier(`ModelResult as $ModelResult`))
+    ])), ts.factory.createStringLiteral("@zenstackhq/runtime")));
+    const dataModels = model.declarations.filter(isDataModel2);
+    for (const dm of dataModels) {
+      let modelType = ts.factory.createTypeAliasDeclaration([
+        ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
+      ], dm.name, void 0, ts.factory.createTypeReferenceNode("$ModelResult", [
+        ts.factory.createTypeReferenceNode("$Schema"),
+        ts.factory.createLiteralTypeNode(ts.factory.createStringLiteral(dm.name))
+      ]));
+      if (dm.comments.length > 0) {
+        modelType = this.generateDocs(modelType, dm);
+      }
+      statements.push(modelType);
+    }
+    const enums = model.declarations.filter(isEnum);
+    for (const e of enums) {
+      let enumDecl = ts.factory.createVariableStatement([
+        ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
+      ], ts.factory.createVariableDeclarationList([
+        ts.factory.createVariableDeclaration(e.name, void 0, void 0, ts.factory.createPropertyAccessExpression(ts.factory.createPropertyAccessExpression(ts.factory.createIdentifier("$schema"), ts.factory.createIdentifier("enums")), ts.factory.createIdentifier(e.name)))
+      ], ts.NodeFlags.Const));
+      if (e.comments.length > 0) {
+        enumDecl = this.generateDocs(enumDecl, e);
+      }
+      statements.push(enumDecl);
+      let typeAlias = ts.factory.createTypeAliasDeclaration([
+        ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
+      ], e.name, void 0, ts.factory.createIndexedAccessTypeNode(ts.factory.createTypeQueryNode(ts.factory.createIdentifier(e.name)), ts.factory.createTypeOperatorNode(ts.SyntaxKind.KeyOfKeyword, ts.factory.createTypeQueryNode(ts.factory.createIdentifier(e.name)))));
+      if (e.comments.length > 0) {
+        typeAlias = this.generateDocs(typeAlias, e);
+      }
+      statements.push(typeAlias);
+    }
+    this.generateBannerComments(statements);
+    const outputFile = path.join(outputDir, "models.ts");
+    const sourceFile = ts.createSourceFile(outputFile, "", ts.ScriptTarget.ESNext, false, ts.ScriptKind.TS);
+    const printer = ts.createPrinter();
+    const result = printer.printList(ts.ListFormat.MultiLine, ts.factory.createNodeArray(statements), sourceFile);
+    fs.writeFileSync(outputFile, result);
+  }
+  generateSchemaTypeImport(schemaObject, schemaType) {
+    const importSpecifiers = [];
+    if (schemaObject) {
+      importSpecifiers.push(ts.factory.createImportSpecifier(false, ts.factory.createIdentifier("schema"), ts.factory.createIdentifier("$schema")));
+    }
+    if (schemaType) {
+      importSpecifiers.push(ts.factory.createImportSpecifier(true, ts.factory.createIdentifier("SchemaType"), ts.factory.createIdentifier("$Schema")));
+    }
+    return ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports(importSpecifiers)), ts.factory.createStringLiteral("./schema"));
+  }
+  generateDocs(tsDecl, decl) {
+    return ts.addSyntheticLeadingComment(tsDecl, ts.SyntaxKind.MultiLineCommentTrivia, `*
+ * ${decl.comments.map((c) => c.replace(/^\s*\/*\s*/, "")).join("\n * ")}
+ `, true);
+  }
+  generateInputTypes(model, outputDir) {
+    const dataModels = model.declarations.filter(isDataModel2);
+    const statements = [];
+    statements.push(this.generateSchemaTypeImport(false, true));
+    const inputTypes = [
+      "FindManyArgs",
+      "FindUniqueArgs",
+      "FindFirstArgs",
+      "CreateArgs",
+      "CreateManyArgs",
+      "CreateManyAndReturnArgs",
+      "UpdateArgs",
+      "UpdateManyArgs",
+      "UpdateManyAndReturnArgs",
+      "UpsertArgs",
+      "DeleteArgs",
+      "DeleteManyArgs",
+      "CountArgs",
+      "AggregateArgs",
+      "GroupByArgs",
+      "WhereInput",
+      "SelectInput",
+      "IncludeInput",
+      "OmitInput"
+    ];
+    const inputTypeNameFixes = {
+      SelectInput: "Select",
+      IncludeInput: "Include",
+      OmitInput: "Omit"
+    };
+    statements.push(ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(true, void 0, ts.factory.createNamedImports(inputTypes.map((inputType) => ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier(`${inputType} as $${inputType}`))))), ts.factory.createStringLiteral("@zenstackhq/runtime")));
+    statements.push(ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(true, void 0, ts.factory.createNamedImports([
+      ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("SimplifiedModelResult as $SimplifiedModelResult")),
+      ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("SelectIncludeOmit as $SelectIncludeOmit"))
+    ])), ts.factory.createStringLiteral("@zenstackhq/runtime")));
+    for (const dm of dataModels) {
+      for (const inputType of inputTypes) {
+        const exportName = inputTypeNameFixes[inputType] ? `${dm.name}${inputTypeNameFixes[inputType]}` : `${dm.name}${inputType}`;
+        statements.push(ts.factory.createTypeAliasDeclaration([
+          ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
+        ], exportName, void 0, ts.factory.createTypeReferenceNode(`$${inputType}`, [
+          ts.factory.createTypeReferenceNode("$Schema"),
+          ts.factory.createLiteralTypeNode(ts.factory.createStringLiteral(dm.name))
+        ])));
+      }
+      statements.push(ts.factory.createTypeAliasDeclaration([
+        ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
+      ], `${dm.name}GetPayload`, [
+        ts.factory.createTypeParameterDeclaration(void 0, "Args", ts.factory.createTypeReferenceNode("$SelectIncludeOmit", [
+          ts.factory.createTypeReferenceNode("$Schema"),
+          ts.factory.createLiteralTypeNode(ts.factory.createStringLiteral(dm.name)),
+          ts.factory.createLiteralTypeNode(ts.factory.createTrue())
+        ]))
+      ], ts.factory.createTypeReferenceNode("$SimplifiedModelResult", [
+        ts.factory.createTypeReferenceNode("$Schema"),
+        ts.factory.createLiteralTypeNode(ts.factory.createStringLiteral(dm.name)),
+        ts.factory.createTypeReferenceNode("Args")
+      ])));
+    }
+    this.generateBannerComments(statements);
+    const outputFile = path.join(outputDir, "input.ts");
+    const sourceFile = ts.createSourceFile(outputFile, "", ts.ScriptTarget.ESNext, false, ts.ScriptKind.TS);
+    const printer = ts.createPrinter();
+    const result = printer.printList(ts.ListFormat.MultiLine, ts.factory.createNodeArray(statements), sourceFile);
+    fs.writeFileSync(outputFile, result);
+  }
 };
 
 // src/zmodel-code-generator.ts