@zenstackhq/sdk 3.0.0-beta.9 → 3.1.0

package/dist/index.cjs

@@ -31,12 +31,9 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
-  DefaultOperationNodeVisitor: () => DefaultOperationNodeVisitor,
-  ExpressionVisitor: () => ExpressionVisitor,
   ModelUtils: () => model_utils_exports,
   PrismaSchemaGenerator: () => PrismaSchemaGenerator,
-  TsSchemaGenerator: () => TsSchemaGenerator,
-  ZModelCodeGenerator: () => ZModelCodeGenerator
+  TsSchemaGenerator: () => TsSchemaGenerator
 });
 module.exports = __toCommonJS(src_exports);
 
@@ -125,7 +122,7 @@ __name(resolved, "resolved");
 function getAuthDecl(model) {
   let found = model.declarations.find((d) => ((0, import_ast.isDataModel)(d) || (0, import_ast.isTypeDef)(d)) && d.attributes.some((attr) => attr.decl.$refText === "@@auth"));
   if (!found) {
-    found = model.declarations.find((d) => (0, import_ast.isDataModel)(d) && d.name === "User");
+    found = model.declarations.find((d) => ((0, import_ast.isDataModel)(d) || (0, import_ast.isTypeDef)(d)) && d.name === "User");
   }
   return found;
 }
@@ -136,365 +133,13 @@ function getIdFields(dm) {
 __name(getIdFields, "getIdFields");
 var DELEGATE_AUX_RELATION_PREFIX = "delegate_aux";
 
-// src/expression-utils.ts
-var import_ts_pattern = require("ts-pattern");
-var ExpressionVisitor = class {
-  static {
-    __name(this, "ExpressionVisitor");
-  }
-  visit(expr) {
-    (0, import_ts_pattern.match)(expr).with({
-      kind: "literal"
-    }, (e) => this.visitLiteral(e)).with({
-      kind: "array"
-    }, (e) => this.visitArray(e)).with({
-      kind: "field"
-    }, (e) => this.visitField(e)).with({
-      kind: "member"
-    }, (e) => this.visitMember(e)).with({
-      kind: "binary"
-    }, (e) => this.visitBinary(e)).with({
-      kind: "unary"
-    }, (e) => this.visitUnary(e)).with({
-      kind: "call"
-    }, (e) => this.visitCall(e)).with({
-      kind: "this"
-    }, (e) => this.visitThis(e)).with({
-      kind: "null"
-    }, (e) => this.visitNull(e)).exhaustive();
-  }
-  visitLiteral(_e) {
-  }
-  visitArray(e) {
-    e.items.forEach((item) => this.visit(item));
-  }
-  visitField(_e) {
-  }
-  visitMember(e) {
-    this.visit(e.receiver);
-  }
-  visitBinary(e) {
-    this.visit(e.left);
-    this.visit(e.right);
-  }
-  visitUnary(e) {
-    this.visit(e.operand);
-  }
-  visitCall(e) {
-    e.args?.forEach((arg) => this.visit(arg));
-  }
-  visitThis(_e) {
-  }
-  visitNull(_e) {
-  }
-};
-
-// src/default-operation-node-visitor.ts
-var import_kysely = require("kysely");
-var DefaultOperationNodeVisitor = class extends import_kysely.OperationNodeVisitor {
-  static {
-    __name(this, "DefaultOperationNodeVisitor");
-  }
-  defaultVisit(node) {
-    Object.values(node).forEach((value) => {
-      if (!value) {
-        return;
-      }
-      if (Array.isArray(value)) {
-        value.forEach((el) => this.defaultVisit(el));
-      }
-      if (typeof value === "object" && "kind" in value && typeof value.kind === "string") {
-        this.visitNode(value);
-      }
-    });
-  }
-  visitSelectQuery(node) {
-    this.defaultVisit(node);
-  }
-  visitSelection(node) {
-    this.defaultVisit(node);
-  }
-  visitColumn(node) {
-    this.defaultVisit(node);
-  }
-  visitAlias(node) {
-    this.defaultVisit(node);
-  }
-  visitTable(node) {
-    this.defaultVisit(node);
-  }
-  visitFrom(node) {
-    this.defaultVisit(node);
-  }
-  visitReference(node) {
-    this.defaultVisit(node);
-  }
-  visitAnd(node) {
-    this.defaultVisit(node);
-  }
-  visitOr(node) {
-    this.defaultVisit(node);
-  }
-  visitValueList(node) {
-    this.defaultVisit(node);
-  }
-  visitParens(node) {
-    this.defaultVisit(node);
-  }
-  visitJoin(node) {
-    this.defaultVisit(node);
-  }
-  visitRaw(node) {
-    this.defaultVisit(node);
-  }
-  visitWhere(node) {
-    this.defaultVisit(node);
-  }
-  visitInsertQuery(node) {
-    this.defaultVisit(node);
-  }
-  visitDeleteQuery(node) {
-    this.defaultVisit(node);
-  }
-  visitReturning(node) {
-    this.defaultVisit(node);
-  }
-  visitCreateTable(node) {
-    this.defaultVisit(node);
-  }
-  visitAddColumn(node) {
-    this.defaultVisit(node);
-  }
-  visitColumnDefinition(node) {
-    this.defaultVisit(node);
-  }
-  visitDropTable(node) {
-    this.defaultVisit(node);
-  }
-  visitOrderBy(node) {
-    this.defaultVisit(node);
-  }
-  visitOrderByItem(node) {
-    this.defaultVisit(node);
-  }
-  visitGroupBy(node) {
-    this.defaultVisit(node);
-  }
-  visitGroupByItem(node) {
-    this.defaultVisit(node);
-  }
-  visitUpdateQuery(node) {
-    this.defaultVisit(node);
-  }
-  visitColumnUpdate(node) {
-    this.defaultVisit(node);
-  }
-  visitLimit(node) {
-    this.defaultVisit(node);
-  }
-  visitOffset(node) {
-    this.defaultVisit(node);
-  }
-  visitOnConflict(node) {
-    this.defaultVisit(node);
-  }
-  visitOnDuplicateKey(node) {
-    this.defaultVisit(node);
-  }
-  visitCheckConstraint(node) {
-    this.defaultVisit(node);
-  }
-  visitDataType(node) {
-    this.defaultVisit(node);
-  }
-  visitSelectAll(node) {
-    this.defaultVisit(node);
-  }
-  visitIdentifier(node) {
-    this.defaultVisit(node);
-  }
-  visitSchemableIdentifier(node) {
-    this.defaultVisit(node);
-  }
-  visitValue(node) {
-    this.defaultVisit(node);
-  }
-  visitPrimitiveValueList(node) {
-    this.defaultVisit(node);
-  }
-  visitOperator(node) {
-    this.defaultVisit(node);
-  }
-  visitCreateIndex(node) {
-    this.defaultVisit(node);
-  }
-  visitDropIndex(node) {
-    this.defaultVisit(node);
-  }
-  visitList(node) {
-    this.defaultVisit(node);
-  }
-  visitPrimaryKeyConstraint(node) {
-    this.defaultVisit(node);
-  }
-  visitUniqueConstraint(node) {
-    this.defaultVisit(node);
-  }
-  visitReferences(node) {
-    this.defaultVisit(node);
-  }
-  visitWith(node) {
-    this.defaultVisit(node);
-  }
-  visitCommonTableExpression(node) {
-    this.defaultVisit(node);
-  }
-  visitCommonTableExpressionName(node) {
-    this.defaultVisit(node);
-  }
-  visitHaving(node) {
-    this.defaultVisit(node);
-  }
-  visitCreateSchema(node) {
-    this.defaultVisit(node);
-  }
-  visitDropSchema(node) {
-    this.defaultVisit(node);
-  }
-  visitAlterTable(node) {
-    this.defaultVisit(node);
-  }
-  visitDropColumn(node) {
-    this.defaultVisit(node);
-  }
-  visitRenameColumn(node) {
-    this.defaultVisit(node);
-  }
-  visitAlterColumn(node) {
-    this.defaultVisit(node);
-  }
-  visitModifyColumn(node) {
-    this.defaultVisit(node);
-  }
-  visitAddConstraint(node) {
-    this.defaultVisit(node);
-  }
-  visitDropConstraint(node) {
-    this.defaultVisit(node);
-  }
-  visitForeignKeyConstraint(node) {
-    this.defaultVisit(node);
-  }
-  visitCreateView(node) {
-    this.defaultVisit(node);
-  }
-  visitDropView(node) {
-    this.defaultVisit(node);
-  }
-  visitGenerated(node) {
-    this.defaultVisit(node);
-  }
-  visitDefaultValue(node) {
-    this.defaultVisit(node);
-  }
-  visitOn(node) {
-    this.defaultVisit(node);
-  }
-  visitValues(node) {
-    this.defaultVisit(node);
-  }
-  visitSelectModifier(node) {
-    this.defaultVisit(node);
-  }
-  visitCreateType(node) {
-    this.defaultVisit(node);
-  }
-  visitDropType(node) {
-    this.defaultVisit(node);
-  }
-  visitExplain(node) {
-    this.defaultVisit(node);
-  }
-  visitDefaultInsertValue(node) {
-    this.defaultVisit(node);
-  }
-  visitAggregateFunction(node) {
-    this.defaultVisit(node);
-  }
-  visitOver(node) {
-    this.defaultVisit(node);
-  }
-  visitPartitionBy(node) {
-    this.defaultVisit(node);
-  }
-  visitPartitionByItem(node) {
-    this.defaultVisit(node);
-  }
-  visitSetOperation(node) {
-    this.defaultVisit(node);
-  }
-  visitBinaryOperation(node) {
-    this.defaultVisit(node);
-  }
-  visitUnaryOperation(node) {
-    this.defaultVisit(node);
-  }
-  visitUsing(node) {
-    this.defaultVisit(node);
-  }
-  visitFunction(node) {
-    this.defaultVisit(node);
-  }
-  visitCase(node) {
-    this.defaultVisit(node);
-  }
-  visitWhen(node) {
-    this.defaultVisit(node);
-  }
-  visitJSONReference(node) {
-    this.defaultVisit(node);
-  }
-  visitJSONPath(node) {
-    this.defaultVisit(node);
-  }
-  visitJSONPathLeg(node) {
-    this.defaultVisit(node);
-  }
-  visitJSONOperatorChain(node) {
-    this.defaultVisit(node);
-  }
-  visitTuple(node) {
-    this.defaultVisit(node);
-  }
-  visitMergeQuery(node) {
-    this.defaultVisit(node);
-  }
-  visitMatched(node) {
-    this.defaultVisit(node);
-  }
-  visitAddIndex(node) {
-    this.defaultVisit(node);
-  }
-  visitCast(node) {
-    this.defaultVisit(node);
-  }
-  visitFetch(node) {
-    this.defaultVisit(node);
-  }
-  visitTop(node) {
-    this.defaultVisit(node);
-  }
-  visitOutput(node) {
-    this.defaultVisit(node);
-  }
-};
-
 // src/prisma/prisma-schema-generator.ts
 var import_common_helpers = require("@zenstackhq/common-helpers");
+var import_language = require("@zenstackhq/language");
 var import_ast2 = require("@zenstackhq/language/ast");
 var import_utils2 = require("@zenstackhq/language/utils");
 var import_langium = require("langium");
-var import_ts_pattern2 = require("ts-pattern");
+var import_ts_pattern = require("ts-pattern");
 
 // src/prisma/indent-string.ts
 function indentString(string, count = 4) {
@@ -888,6 +533,9 @@ var EnumField = class extends DeclarationBase {
 
 // src/prisma/prisma-schema-generator.ts
 var IDENTIFIER_NAME_MAX_LENGTH = 50 - DELEGATE_AUX_RELATION_PREFIX.length;
+var NON_PRISMA_DATASOURCE_FIELDS = [
+  "defaultSchema"
+];
 var PrismaSchemaGenerator = class {
   static {
     __name(this, "PrismaSchemaGenerator");
@@ -922,10 +570,13 @@ var PrismaSchemaGenerator = class {
           break;
       }
     }
+    if (!this.zmodel.declarations.some(import_ast2.isGeneratorDecl)) {
+      this.generateDefaultGenerator(prisma);
+    }
     return this.PRELUDE + prisma.toString();
   }
   generateDataSource(prisma, dataSource) {
-    const fields = dataSource.fields.map((f) => ({
+    const fields = dataSource.fields.filter((f) => !NON_PRISMA_DATASOURCE_FIELDS.includes(f.name)).map((f) => ({
       name: f.name,
       text: this.configExprToText(f.value)
     }));
@@ -962,6 +613,21 @@ var PrismaSchemaGenerator = class {
       text: this.configExprToText(f.value)
     })));
   }
+  generateDefaultGenerator(prisma) {
+    const gen = prisma.addGenerator("client", [
+      {
+        name: "provider",
+        text: '"prisma-client-js"'
+      }
+    ]);
+    const dataSource = this.zmodel.declarations.find(import_ast2.isDataSource);
+    if (dataSource?.fields.some((f) => f.name === "extensions")) {
+      gen.fields.push({
+        name: "previewFeatures",
+        text: '["postgresqlExtensions"]'
+      });
+    }
+  }
   generateModel(prisma, decl) {
     const model = decl.isView ? prisma.addView(decl.name) : prisma.addModel(decl.name);
     const allFields = (0, import_utils2.getAllFields)(decl, true);
@@ -973,20 +639,29 @@ var PrismaSchemaGenerator = class {
         this.generateModelField(model, field, decl);
       }
     }
-    const allAttributes = (0, import_utils2.getAllAttributes)(decl);
-    for (const attr of allAttributes.filter((attr2) => this.isPrismaAttribute(attr2) && !this.isInheritedMapAttribute(attr2, decl))) {
+    const allAttributes = (0, import_utils2.getAllAttributes)(decl).filter((attr) => this.isPrismaAttribute(attr));
+    for (const attr of allAttributes) {
       this.generateContainerAttribute(model, attr);
     }
+    if (this.datasourceHasSchemasSetting(decl.$container) && !allAttributes.some((attr) => attr.decl.ref?.name === "@@schema")) {
+      model.addAttribute("@@schema", [
+        new AttributeArg(void 0, new AttributeArgValue("String", this.getDefaultPostgresSchemaName(decl.$container)))
+      ]);
+    }
     decl.comments.forEach((c) => model.addComment(c));
     this.generateDelegateRelationForBase(model, decl);
     this.generateDelegateRelationForConcrete(model, decl);
   }
-  isInheritedMapAttribute(attr, contextModel) {
-    if (attr.$container === contextModel) {
-      return false;
-    }
-    const attrName = attr.decl.ref?.name ?? attr.decl.$refText;
-    return attrName === "@@map";
+  getDatasourceField(zmodel, fieldName) {
+    const dataSource = zmodel.declarations.find(import_ast2.isDataSource);
+    return dataSource?.fields.find((f) => f.name === fieldName);
+  }
+  datasourceHasSchemasSetting(zmodel) {
+    return !!this.getDatasourceField(zmodel, "schemas");
+  }
+  getDefaultPostgresSchemaName(zmodel) {
+    const defaultSchemaField = this.getDatasourceField(zmodel, "defaultSchema");
+    return (0, import_utils2.getStringLiteral)(defaultSchemaField?.value) ?? "public";
   }
   isPrismaAttribute(attr) {
     if (!attr.decl.ref) {
@@ -996,7 +671,7 @@ var PrismaSchemaGenerator = class {
   }
   getUnsupportedFieldType(fieldType) {
     if (fieldType.unsupported) {
-      const value = this.getStringLiteral(fieldType.unsupported.value);
+      const value = (0, import_utils2.getStringLiteral)(fieldType.unsupported.value);
       if (value) {
         return `Unsupported("${value}")`;
       } else {
@@ -1006,9 +681,6 @@ var PrismaSchemaGenerator = class {
       return void 0;
     }
   }
-  getStringLiteral(node) {
-    return (0, import_ast2.isStringLiteral)(node) ? node.value : void 0;
-  }
   generateModelField(model, field, contextModel, addToFront = false) {
     let fieldType;
     if (field.type.type) {
@@ -1065,7 +737,7 @@ var PrismaSchemaGenerator = class {
   }
   makeAttributeArgValue(node) {
     if ((0, import_ast2.isLiteralExpr)(node)) {
-      const argType = (0, import_ts_pattern2.match)(node.$type).with(import_ast2.StringLiteral, () => "String").with(import_ast2.NumberLiteral, () => "Number").with(import_ast2.BooleanLiteral, () => "Boolean").exhaustive();
+      const argType = (0, import_ts_pattern.match)(node.$type).with(import_ast2.StringLiteral, () => "String").with(import_ast2.NumberLiteral, () => "Number").with(import_ast2.BooleanLiteral, () => "Boolean").exhaustive();
       return new AttributeArgValue(argType, node.value);
     } else if ((0, import_ast2.isArrayExpr)(node)) {
       return new AttributeArgValue("Array", new Array(...node.items.map((item) => this.makeAttributeArgValue(item))));
@@ -1078,13 +750,13 @@ var PrismaSchemaGenerator = class {
     }
   }
   exprToText(expr) {
-    return new ZModelCodeGenerator({
+    return new import_language.ZModelCodeGenerator({
       quote: "double"
     }).generate(expr);
   }
   makeFunctionCall(node) {
     return new FunctionCall(node.function.ref.name, node.args.map((arg) => {
-      const val = (0, import_ts_pattern2.match)(arg.value).when(import_ast2.isStringLiteral, (v) => `"${v.value}"`).when(import_ast2.isLiteralExpr, (v) => v.value.toString()).when(import_ast2.isNullExpr, () => "null").otherwise(() => {
+      const val = (0, import_ts_pattern.match)(arg.value).when(import_ast2.isStringLiteral, (v) => `"${v.value}"`).when(import_ast2.isLiteralExpr, (v) => v.value.toString()).when(import_ast2.isNullExpr, () => "null").otherwise(() => {
         throw new Error("Function call argument must be literal or null");
       });
       return new FunctionCallArg(val);
@@ -1099,9 +771,15 @@ var PrismaSchemaGenerator = class {
     for (const field of decl.fields) {
       this.generateEnumField(_enum, field);
     }
-    for (const attr of decl.attributes.filter((attr2) => this.isPrismaAttribute(attr2))) {
+    const allAttributes = decl.attributes.filter((attr) => this.isPrismaAttribute(attr));
+    for (const attr of allAttributes) {
       this.generateContainerAttribute(_enum, attr);
     }
+    if (this.datasourceHasSchemasSetting(decl.$container) && !allAttributes.some((attr) => attr.decl.ref?.name === "@@schema")) {
+      _enum.addAttribute("@@schema", [
+        new AttributeArg(void 0, new AttributeArgValue("String", this.getDefaultPostgresSchemaName(decl.$container)))
+      ]);
+    }
     decl.comments.forEach((c) => _enum.addComment(c));
   }
   generateEnumField(_enum, field) {
@@ -1173,36 +851,51 @@ var import_ast3 = require("@zenstackhq/language/ast");
 var import_utils3 = require("@zenstackhq/language/utils");
 var import_node_fs = __toESM(require("fs"), 1);
 var import_node_path = __toESM(require("path"), 1);
-var import_ts_pattern3 = require("ts-pattern");
+var import_ts_pattern2 = require("ts-pattern");
 var ts = __toESM(require("typescript"), 1);
 var TsSchemaGenerator = class {
   static {
     __name(this, "TsSchemaGenerator");
   }
   usedExpressionUtils = false;
-  async generate(model, outputDir) {
-    import_node_fs.default.mkdirSync(outputDir, {
+  async generate(model, options) {
+    import_node_fs.default.mkdirSync(options.outDir, {
       recursive: true
     });
     this.usedExpressionUtils = false;
-    this.generateSchema(model, outputDir);
-    this.generateModelsAndTypeDefs(model, outputDir);
-    this.generateInputTypes(model, outputDir);
-  }
-  generateSchema(model, outputDir) {
-    const statements = [];
-    this.generateSchemaStatements(model, statements);
-    this.generateBannerComments(statements);
-    const schemaOutputFile = import_node_path.default.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);
-    import_node_fs.default.writeFileSync(schemaOutputFile, result);
+    this.generateSchema(model, options);
+    this.generateModelsAndTypeDefs(model, options);
+    this.generateInputTypes(model, options);
+  }
+  generateSchema(model, options) {
+    const targets = [];
+    if (!options.liteOnly) {
+      targets.push({
+        lite: false,
+        file: "schema.ts"
+      });
+    }
+    if (options.lite || options.liteOnly) {
+      targets.push({
+        lite: true,
+        file: "schema-lite.ts"
+      });
+    }
+    for (const { lite, file } of targets) {
+      const statements = [];
+      this.generateSchemaStatements(model, statements, lite);
+      this.generateBannerComments(statements);
+      const schemaOutputFile = import_node_path.default.join(options.outDir, file);
+      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);
+      import_node_fs.default.writeFileSync(schemaOutputFile, result);
+    }
   }
-  generateSchemaStatements(model, statements) {
+  generateSchemaStatements(model, statements, lite) {
     const hasComputedFields = model.declarations.some((d) => (0, import_ast3.isDataModel)(d) && d.fields.some((f) => hasAttribute(f, "@computed")));
-    const schemaObject = this.createSchemaObject(model);
-    const runtimeImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports([
+    const schemaClass = this.createSchemaClass(model, lite);
+    const runtimeImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(void 0, void 0, ts.factory.createNamedImports([
       ts.factory.createImportSpecifier(true, void 0, ts.factory.createIdentifier("SchemaDef")),
       ...hasComputedFields ? [
         ts.factory.createImportSpecifier(true, void 0, ts.factory.createIdentifier("OperandExpression"))
@@ -1210,64 +903,81 @@ var TsSchemaGenerator = class {
       ...this.usedExpressionUtils ? [
         ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("ExpressionUtils"))
       ] : []
-    ])), ts.factory.createStringLiteral("@zenstackhq/runtime/schema"));
+    ])), ts.factory.createStringLiteral("@zenstackhq/orm/schema"));
     statements.push(runtimeImportDecl);
-    const declaration = ts.factory.createVariableStatement([
+    statements.push(schemaClass);
+    const schemaDecl = ts.factory.createVariableStatement([
       ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
     ], ts.factory.createVariableDeclarationList([
-      ts.factory.createVariableDeclaration("schema", void 0, void 0, ts.factory.createSatisfiesExpression(ts.factory.createAsExpression(schemaObject, ts.factory.createTypeReferenceNode("const")), ts.factory.createTypeReferenceNode("SchemaDef")))
+      ts.factory.createVariableDeclaration("schema", void 0, void 0, ts.factory.createNewExpression(ts.factory.createIdentifier("SchemaType"), void 0, []))
     ], ts.NodeFlags.Const));
-    statements.push(declaration);
-    const typeDeclaration = ts.factory.createTypeAliasDeclaration([
-      ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
-    ], "SchemaType", void 0, ts.factory.createTypeReferenceNode("typeof schema"));
-    statements.push(typeDeclaration);
+    statements.push(schemaDecl);
   }
   createExpressionUtilsCall(method, args) {
     this.usedExpressionUtils = true;
     return ts.factory.createCallExpression(ts.factory.createPropertyAccessExpression(ts.factory.createIdentifier("ExpressionUtils"), method), void 0, args || []);
   }
-  createSchemaObject(model) {
-    const properties = [
+  createSchemaClass(model, lite) {
+    const members = [
       // provider
-      ts.factory.createPropertyAssignment("provider", this.createProviderObject(model)),
+      ts.factory.createPropertyDeclaration(void 0, "provider", void 0, void 0, this.createAsConst(this.createProviderObject(model))),
       // models
-      ts.factory.createPropertyAssignment("models", this.createModelsObject(model)),
+      ts.factory.createPropertyDeclaration(void 0, "models", void 0, void 0, this.createAsConst(this.createModelsObject(model, lite))),
       // typeDefs
       ...model.declarations.some(import_ast3.isTypeDef) ? [
-        ts.factory.createPropertyAssignment("typeDefs", this.createTypeDefsObject(model))
+        ts.factory.createPropertyDeclaration(void 0, "typeDefs", void 0, void 0, this.createAsConst(this.createTypeDefsObject(model, lite)))
       ] : []
     ];
     const enums = model.declarations.filter(import_ast3.isEnum);
     if (enums.length > 0) {
-      properties.push(ts.factory.createPropertyAssignment("enums", ts.factory.createObjectLiteralExpression(enums.map((e) => ts.factory.createPropertyAssignment(e.name, this.createEnumObject(e))), true)));
+      members.push(ts.factory.createPropertyDeclaration(void 0, "enums", void 0, void 0, this.createAsConst(ts.factory.createObjectLiteralExpression(enums.map((e) => ts.factory.createPropertyAssignment(e.name, this.createEnumObject(e))), true))));
     }
     const authType = getAuthDecl(model);
     if (authType) {
-      properties.push(ts.factory.createPropertyAssignment("authType", this.createLiteralNode(authType.name)));
+      members.push(ts.factory.createPropertyDeclaration(void 0, "authType", void 0, void 0, this.createAsConst(this.createLiteralNode(authType.name))));
     }
     const procedures = model.declarations.filter(import_ast3.isProcedure);
     if (procedures.length > 0) {
-      properties.push(ts.factory.createPropertyAssignment("procedures", this.createProceduresObject(procedures)));
+      members.push(ts.factory.createPropertyDeclaration(void 0, "procedures", void 0, void 0, this.createAsConst(this.createProceduresObject(procedures))));
     }
-    properties.push(ts.factory.createPropertyAssignment("plugins", ts.factory.createObjectLiteralExpression([], true)));
-    return ts.factory.createObjectLiteralExpression(properties, true);
+    members.push(ts.factory.createPropertyDeclaration(void 0, "plugins", void 0, void 0, ts.factory.createObjectLiteralExpression([], true)));
+    const schemaClass = ts.factory.createClassDeclaration([
+      ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
+    ], "SchemaType", void 0, [
+      ts.factory.createHeritageClause(ts.SyntaxKind.ImplementsKeyword, [
+        ts.factory.createExpressionWithTypeArguments(ts.factory.createIdentifier("SchemaDef"), void 0)
+      ])
+    ], members);
+    return schemaClass;
+  }
+  createAsConst(expr) {
+    return ts.factory.createAsExpression(expr, ts.factory.createTypeReferenceNode("const"));
   }
   createProviderObject(model) {
     const dsProvider = this.getDataSourceProvider(model);
+    const defaultSchema = this.getDataSourceDefaultSchema(model);
     return ts.factory.createObjectLiteralExpression([
-      ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(dsProvider.type))
+      ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(dsProvider)),
+      ...defaultSchema ? [
+        ts.factory.createPropertyAssignment("defaultSchema", ts.factory.createStringLiteral(defaultSchema))
+      ] : []
     ], true);
   }
-  createModelsObject(model) {
-    return ts.factory.createObjectLiteralExpression(model.declarations.filter((d) => (0, import_ast3.isDataModel)(d) && !hasAttribute(d, "@@ignore")).map((dm) => ts.factory.createPropertyAssignment(dm.name, this.createDataModelObject(dm))), true);
+  createModelsObject(model, lite) {
+    return ts.factory.createObjectLiteralExpression(this.getAllDataModels(model).map((dm) => ts.factory.createPropertyAssignment(dm.name, this.createDataModelObject(dm, lite))), true);
   }
-  createTypeDefsObject(model) {
-    return ts.factory.createObjectLiteralExpression(model.declarations.filter((d) => (0, import_ast3.isTypeDef)(d)).map((td) => ts.factory.createPropertyAssignment(td.name, this.createTypeDefObject(td))), true);
+  getAllDataModels(model) {
+    return model.declarations.filter((d) => (0, import_ast3.isDataModel)(d) && !hasAttribute(d, "@@ignore"));
   }
-  createDataModelObject(dm) {
+  getAllTypeDefs(model) {
+    return model.declarations.filter((d) => (0, import_ast3.isTypeDef)(d) && !hasAttribute(d, "@@ignore"));
+  }
+  createTypeDefsObject(model, lite) {
+    return ts.factory.createObjectLiteralExpression(this.getAllTypeDefs(model).map((td) => ts.factory.createPropertyAssignment(td.name, this.createTypeDefObject(td, lite))), true);
+  }
+  createDataModelObject(dm, lite) {
     const allFields = (0, import_utils3.getAllFields)(dm);
-    const allAttributes = (0, import_utils3.getAllAttributes)(dm).filter((attr) => {
+    const allAttributes = lite ? [] : (0, import_utils3.getAllAttributes)(dm).filter((attr) => {
       if (attr.decl.$refText === "@@delegate" && attr.$container !== dm) {
         return false;
       }
@@ -1282,7 +992,7 @@ var TsSchemaGenerator = class {
         ts.factory.createPropertyAssignment("baseModel", ts.factory.createStringLiteral(dm.baseModel.$refText))
       ] : [],
       // fields
-      ts.factory.createPropertyAssignment("fields", ts.factory.createObjectLiteralExpression(allFields.map((field) => ts.factory.createPropertyAssignment(field.name, this.createDataFieldObject(field, dm))), true)),
+      ts.factory.createPropertyAssignment("fields", ts.factory.createObjectLiteralExpression(allFields.map((field) => ts.factory.createPropertyAssignment(field.name, this.createDataFieldObject(field, dm, lite))), true)),
       // attributes
       ...allAttributes.length > 0 ? [
         ts.factory.createPropertyAssignment("attributes", ts.factory.createArrayLiteralExpression(allAttributes.map((attr) => this.createAttributeObject(attr)), true))
@@ -1312,14 +1022,14 @@ var TsSchemaGenerator = class {
   getSubModels(dm) {
     return dm.$container.declarations.filter(import_ast3.isDataModel).filter((d) => d.baseModel?.ref === dm).map((d) => d.name);
   }
-  createTypeDefObject(td) {
+  createTypeDefObject(td, lite) {
     const allFields = (0, import_utils3.getAllFields)(td);
     const allAttributes = (0, import_utils3.getAllAttributes)(td);
     const fields = [
       // name
       ts.factory.createPropertyAssignment("name", ts.factory.createStringLiteral(td.name)),
       // fields
-      ts.factory.createPropertyAssignment("fields", ts.factory.createObjectLiteralExpression(allFields.map((field) => ts.factory.createPropertyAssignment(field.name, this.createDataFieldObject(field, void 0))), true)),
+      ts.factory.createPropertyAssignment("fields", ts.factory.createObjectLiteralExpression(allFields.map((field) => ts.factory.createPropertyAssignment(field.name, this.createDataFieldObject(field, void 0, lite))), true)),
       // attributes
       ...allAttributes.length > 0 ? [
         ts.factory.createPropertyAssignment("attributes", ts.factory.createArrayLiteralExpression(allAttributes.map((attr) => this.createAttributeObject(attr)), true))
@@ -1342,7 +1052,7 @@ var TsSchemaGenerator = class {
     ], true))), true);
   }
   mapFieldTypeToTSType(type) {
-    let result = (0, import_ts_pattern3.match)(type.type).with("String", () => "string").with("Boolean", () => "boolean").with("Int", () => "number").with("Float", () => "number").with("BigInt", () => "bigint").with("Decimal", () => "number").otherwise(() => "unknown");
+    let result = (0, import_ts_pattern2.match)(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}[]`;
     }
@@ -1351,7 +1061,7 @@ var TsSchemaGenerator = class {
     }
     return result;
   }
-  createDataFieldObject(field, contextModel) {
+  createDataFieldObject(field, contextModel, lite) {
     const objectFields = [
       // name
       ts.factory.createPropertyAssignment("name", ts.factory.createStringLiteral(field.name)),
@@ -1373,6 +1083,9 @@ var TsSchemaGenerator = class {
     if (hasAttribute(field, "@updatedAt")) {
       objectFields.push(ts.factory.createPropertyAssignment("updatedAt", ts.factory.createTrue()));
     }
+    if (hasAttribute(field, "@omit")) {
+      objectFields.push(ts.factory.createPropertyAssignment("omit", ts.factory.createTrue()));
+    }
     if (contextModel && // id fields are duplicated in inherited models
     !isIdField(field, contextModel) && field.$container !== contextModel && isDelegateModel(field.$container)) {
       objectFields.push(ts.factory.createPropertyAssignment("originModel", ts.factory.createStringLiteral(field.$container.name)));
@@ -1380,7 +1093,7 @@ var TsSchemaGenerator = class {
     if (this.isDiscriminatorField(field)) {
       objectFields.push(ts.factory.createPropertyAssignment("isDiscriminator", ts.factory.createTrue()));
     }
-    if (field.attributes.length > 0) {
+    if (!lite && field.attributes.length > 0) {
       objectFields.push(ts.factory.createPropertyAssignment("attributes", ts.factory.createArrayLiteralExpression(field.attributes.map((attr) => this.createAttributeObject(attr)))));
     }
     const defaultValue = this.getFieldMappedDefault(field);
@@ -1433,11 +1146,18 @@ var TsSchemaGenerator = class {
     const dataSource = model.declarations.find(import_ast3.isDataSource);
     (0, import_common_helpers2.invariant)(dataSource, "No data source found in the model");
     const providerExpr = dataSource.fields.find((f) => f.name === "provider")?.value;
-    (0, import_common_helpers2.invariant)((0, import_ast3.isLiteralExpr)(providerExpr), "Provider must be a literal");
-    const type = providerExpr.value;
-    return {
-      type
-    };
+    (0, import_common_helpers2.invariant)((0, import_ast3.isLiteralExpr)(providerExpr) && typeof providerExpr.value === "string", "Provider must be a string literal");
+    return providerExpr.value;
+  }
+  getDataSourceDefaultSchema(model) {
+    const dataSource = model.declarations.find(import_ast3.isDataSource);
+    (0, import_common_helpers2.invariant)(dataSource, "No data source found in the model");
+    const defaultSchemaExpr = dataSource.fields.find((f) => f.name === "defaultSchema")?.value;
+    if (!defaultSchemaExpr) {
+      return void 0;
+    }
+    (0, import_common_helpers2.invariant)((0, import_ast3.isLiteralExpr)(defaultSchemaExpr) && typeof defaultSchemaExpr.value === "string", "Default schema must be a string literal");
+    return defaultSchemaExpr.value;
   }
   getFieldMappedDefault(field) {
     const defaultAttr = getAttribute(field, "@default");
@@ -1507,12 +1227,16 @@ var TsSchemaGenerator = class {
       relationFields.push(ts.factory.createPropertyAssignment("name", ts.factory.createStringLiteral(relationName)));
     }
     const relation = getAttribute(field, "@relation");
+    const fkFields = [];
     if (relation) {
       for (const arg of relation.args) {
         const param = arg.$resolvedParam.name;
         if (param === "fields" || param === "references") {
           const fieldNames = this.getReferenceNames(arg.value);
           if (fieldNames) {
+            if (param === "fields") {
+              fkFields.push(...fieldNames);
+            }
             relationFields.push(ts.factory.createPropertyAssignment(param, ts.factory.createArrayLiteralExpression(fieldNames.map((el) => ts.factory.createStringLiteral(el)))));
           }
         }
@@ -1522,6 +1246,15 @@ var TsSchemaGenerator = class {
         }
       }
     }
+    if (fkFields.length > 0) {
+      const allHaveDefault = fkFields.every((fieldName) => {
+        const fieldDef = field.$container.fields.find((f) => f.name === fieldName);
+        return fieldDef && hasAttribute(fieldDef, "@default");
+      });
+      if (allHaveDefault) {
+        relationFields.push(ts.factory.createPropertyAssignment("hasDefault", ts.factory.createTrue()));
+      }
+    }
     return ts.factory.createObjectLiteralExpression(relationFields);
   }
   getReferenceNames(expr) {
@@ -1590,7 +1323,11 @@ var TsSchemaGenerator = class {
     const seenKeys = /* @__PURE__ */ new Set();
     for (const attr of allAttributes) {
       if (attr.decl.$refText === "@@id" || attr.decl.$refText === "@@unique") {
-        const fieldNames = this.getReferenceNames(attr.args[0].value);
+        const fieldsArg = (0, import_utils3.getAttributeArg)(attr, "fields");
+        if (!fieldsArg) {
+          continue;
+        }
+        const fieldNames = this.getReferenceNames(fieldsArg);
         if (!fieldNames) {
           continue;
         }
@@ -1629,7 +1366,21 @@ var TsSchemaGenerator = class {
     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);
+    return ts.factory.createObjectLiteralExpression([
+      ts.factory.createPropertyAssignment("values", ts.factory.createObjectLiteralExpression(e.fields.map((f) => ts.factory.createPropertyAssignment(f.name, ts.factory.createStringLiteral(f.name))), true)),
+      // only generate `fields` if there are attributes on the fields
+      ...e.fields.some((f) => f.attributes.length > 0) ? [
+        ts.factory.createPropertyAssignment("fields", ts.factory.createObjectLiteralExpression(e.fields.map((field) => ts.factory.createPropertyAssignment(field.name, ts.factory.createObjectLiteralExpression([
+          ts.factory.createPropertyAssignment("name", ts.factory.createStringLiteral(field.name)),
+          ...field.attributes.length > 0 ? [
+            ts.factory.createPropertyAssignment("attributes", ts.factory.createArrayLiteralExpression(field.attributes?.map((attr) => this.createAttributeObject(attr)) ?? [], true))
+          ] : []
+        ], true))), true))
+      ] : [],
+      ...e.attributes.length > 0 ? [
+        ts.factory.createPropertyAssignment("attributes", ts.factory.createArrayLiteralExpression(e.attributes.map((attr) => this.createAttributeObject(attr)), true))
+      ] : []
+    ], true);
   }
   getLiteral(expr) {
     if (!(0, import_ast3.isLiteralExpr)(expr)) {
@@ -1646,7 +1397,10 @@ 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;
+    return arg === null ? ts.factory.createNull() : typeof arg === "string" ? ts.factory.createStringLiteral(arg) : typeof arg === "number" ? this.createNumberLiteral(arg) : arg === true ? ts.factory.createTrue() : arg === false ? ts.factory.createFalse() : void 0;
+  }
+  createNumberLiteral(arg) {
+    return arg < 0 ? ts.factory.createPrefixUnaryExpression(ts.SyntaxKind.MinusToken, ts.factory.createNumericLiteral(-arg)) : ts.factory.createNumericLiteral(arg);
   }
   createProceduresObject(procedures) {
     return ts.factory.createObjectLiteralExpression(procedures.map((proc) => ts.factory.createPropertyAssignment(proc.name, this.createProcedureObject(proc))), true);
@@ -1706,7 +1460,7 @@ var TsSchemaGenerator = class {
     ]);
   }
   createExpression(value) {
-    return (0, import_ts_pattern3.match)(value).when(import_ast3.isLiteralExpr, (expr) => this.createLiteralExpression(expr.$type, expr.value)).when(import_ast3.isInvocationExpr, (expr) => this.createCallExpression(expr)).when(import_ast3.isReferenceExpr, (expr) => this.createRefExpression(expr)).when(import_ast3.isArrayExpr, (expr) => this.createArrayExpression(expr)).when(import_ast3.isUnaryExpr, (expr) => this.createUnaryExpression(expr)).when(import_ast3.isBinaryExpr, (expr) => this.createBinaryExpression(expr)).when(import_ast3.isMemberAccessExpr, (expr) => this.createMemberExpression(expr)).when(import_ast3.isNullExpr, () => this.createNullExpression()).when(import_ast3.isThisExpr, () => this.createThisExpression()).otherwise(() => {
+    return (0, import_ts_pattern2.match)(value).when(import_ast3.isLiteralExpr, (expr) => this.createLiteralExpression(expr.$type, expr.value)).when(import_ast3.isInvocationExpr, (expr) => this.createCallExpression(expr)).when(import_ast3.isReferenceExpr, (expr) => this.createRefExpression(expr)).when(import_ast3.isArrayExpr, (expr) => this.createArrayExpression(expr)).when(import_ast3.isUnaryExpr, (expr) => this.createUnaryExpression(expr)).when(import_ast3.isBinaryExpr, (expr) => this.createBinaryExpression(expr)).when(import_ast3.isMemberAccessExpr, (expr) => this.createMemberExpression(expr)).when(import_ast3.isNullExpr, () => this.createNullExpression()).when(import_ast3.isThisExpr, () => this.createThisExpression()).otherwise(() => {
       throw new Error(`Unsupported attribute arg value: ${value.$type}`);
     });
   }
@@ -1768,7 +1522,7 @@ var TsSchemaGenerator = class {
     ]);
   }
   createLiteralExpression(type, value) {
-    return (0, import_ts_pattern3.match)(type).with("BooleanLiteral", () => this.createExpressionUtilsCall("literal", [
+    return (0, import_ts_pattern2.match)(type).with("BooleanLiteral", () => this.createExpressionUtilsCall("literal", [
       this.createLiteralNode(value)
     ])).with("NumberLiteral", () => this.createExpressionUtilsCall("literal", [
       ts.factory.createIdentifier(value)
@@ -1778,16 +1532,16 @@ var TsSchemaGenerator = class {
       throw new Error(`Unsupported literal type: ${type}`);
     });
   }
-  generateModelsAndTypeDefs(model, outputDir) {
+  generateModelsAndTypeDefs(model, options) {
     const statements = [];
-    statements.push(this.generateSchemaImport(model, true, true));
+    statements.push(this.generateSchemaImport(model, true, true, !!(options.lite || options.liteOnly), options.importWithFileExtension));
     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`)),
       ...model.declarations.some(import_ast3.isTypeDef) ? [
         ts.factory.createImportSpecifier(true, void 0, ts.factory.createIdentifier(`TypeDefResult as $TypeDefResult`))
       ] : []
-    ])), ts.factory.createStringLiteral("@zenstackhq/runtime")));
-    const dataModels = model.declarations.filter(import_ast3.isDataModel);
+    ])), ts.factory.createStringLiteral("@zenstackhq/orm")));
+    const dataModels = this.getAllDataModels(model);
     for (const dm of dataModels) {
       let modelType = ts.factory.createTypeAliasDeclaration([
         ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
@@ -1800,7 +1554,7 @@ var TsSchemaGenerator = class {
       }
       statements.push(modelType);
     }
-    const typeDefs = model.declarations.filter(import_ast3.isTypeDef);
+    const typeDefs = this.getAllTypeDefs(model);
     for (const td of typeDefs) {
       let typeDef = ts.factory.createTypeAliasDeclaration([
         ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)
@@ -1818,7 +1572,7 @@ var TsSchemaGenerator = class {
       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.factory.createVariableDeclaration(e.name, void 0, void 0, ts.factory.createPropertyAccessExpression(ts.factory.createPropertyAccessExpression(ts.factory.createPropertyAccessExpression(ts.factory.createIdentifier("$schema"), ts.factory.createIdentifier("enums")), ts.factory.createIdentifier(e.name)), ts.factory.createIdentifier("values")))
       ], ts.NodeFlags.Const));
       if (e.comments.length > 0) {
         enumDecl = this.generateDocs(enumDecl, e);
@@ -1833,13 +1587,13 @@ var TsSchemaGenerator = class {
       statements.push(typeAlias);
     }
     this.generateBannerComments(statements);
-    const outputFile = import_node_path.default.join(outputDir, "models.ts");
+    const outputFile = import_node_path.default.join(options.outDir, "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);
     import_node_fs.default.writeFileSync(outputFile, result);
   }
-  generateSchemaImport(model, schemaObject, schemaType) {
+  generateSchemaImport(model, schemaObject, schemaType, useLite, importWithFileExtension) {
     const importSpecifiers = [];
     if (schemaObject) {
       if (model.declarations.some(import_ast3.isEnum)) {
@@ -1849,17 +1603,21 @@ var TsSchemaGenerator = class {
     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"));
+    let importFrom = useLite ? "./schema-lite" : "./schema";
+    if (importWithFileExtension) {
+      importFrom += importWithFileExtension.startsWith(".") ? importWithFileExtension : `.${importWithFileExtension}`;
+    }
+    return ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports(importSpecifiers)), ts.factory.createStringLiteral(importFrom));
   }
   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(import_ast3.isDataModel);
+  generateInputTypes(model, options) {
+    const dataModels = this.getAllDataModels(model);
     const statements = [];
-    statements.push(this.generateSchemaImport(model, false, true));
+    statements.push(this.generateSchemaImport(model, false, true, !!(options.lite || options.liteOnly), options.importWithFileExtension));
     const inputTypes = [
       "FindManyArgs",
       "FindUniqueArgs",
@@ -1886,11 +1644,14 @@ var TsSchemaGenerator = class {
       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")),
+      ...inputTypes.map((inputType) => ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier(`${inputType} as $${inputType}`))),
+      ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("QueryOptions as $QueryOptions"))
+    ])), ts.factory.createStringLiteral("@zenstackhq/orm")));
+    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("SimplifiedPlainResult as $Result")),
       ts.factory.createImportSpecifier(false, void 0, ts.factory.createIdentifier("SelectIncludeOmit as $SelectIncludeOmit"))
-    ])), ts.factory.createStringLiteral("@zenstackhq/runtime")));
+    ])), ts.factory.createStringLiteral("@zenstackhq/orm")));
     for (const dm of dataModels) {
       for (const inputType of inputTypes) {
         const exportName = inputTypeNameFixes[inputType] ? `${dm.name}${inputTypeNameFixes[inputType]}` : `${dm.name}${inputType}`;
@@ -1908,528 +1669,31 @@ var TsSchemaGenerator = class {
           ts.factory.createTypeReferenceNode("$Schema"),
           ts.factory.createLiteralTypeNode(ts.factory.createStringLiteral(dm.name)),
           ts.factory.createLiteralTypeNode(ts.factory.createTrue())
+        ])),
+        ts.factory.createTypeParameterDeclaration(void 0, "Options", ts.factory.createTypeReferenceNode("$QueryOptions", [
+          ts.factory.createTypeReferenceNode("$Schema")
+        ]), ts.factory.createTypeReferenceNode("$QueryOptions", [
+          ts.factory.createTypeReferenceNode("$Schema")
         ]))
-      ], ts.factory.createTypeReferenceNode("$SimplifiedModelResult", [
+      ], ts.factory.createTypeReferenceNode("$Result", [
         ts.factory.createTypeReferenceNode("$Schema"),
         ts.factory.createLiteralTypeNode(ts.factory.createStringLiteral(dm.name)),
-        ts.factory.createTypeReferenceNode("Args")
+        ts.factory.createTypeReferenceNode("Args"),
+        ts.factory.createTypeReferenceNode("Options")
       ])));
     }
     this.generateBannerComments(statements);
-    const outputFile = import_node_path.default.join(outputDir, "input.ts");
+    const outputFile = import_node_path.default.join(options.outDir, "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);
     import_node_fs.default.writeFileSync(outputFile, result);
   }
 };
-
-// src/zmodel-code-generator.ts
-var import_ast4 = require("@zenstackhq/language/ast");
-function _ts_decorate(decorators, target, key, desc) {
-  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
-  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
-  else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
-  return c > 3 && r && Object.defineProperty(target, key, r), r;
-}
-__name(_ts_decorate, "_ts_decorate");
-function _ts_metadata(k, v) {
-  if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
-}
-__name(_ts_metadata, "_ts_metadata");
-var generationHandlers = /* @__PURE__ */ new Map();
-function gen(name) {
-  return function(_target, _propertyKey, descriptor) {
-    if (!generationHandlers.get(name)) {
-      generationHandlers.set(name, descriptor);
-    }
-    return descriptor;
-  };
-}
-__name(gen, "gen");
-var ZModelCodeGenerator = class {
-  static {
-    __name(this, "ZModelCodeGenerator");
-  }
-  options;
-  constructor(options) {
-    this.options = {
-      binaryExprNumberOfSpaces: options?.binaryExprNumberOfSpaces ?? 1,
-      unaryExprNumberOfSpaces: options?.unaryExprNumberOfSpaces ?? 0,
-      indent: options?.indent ?? 4,
-      quote: options?.quote ?? "single"
-    };
-  }
-  /**
-   * Generates ZModel source code from AST.
-   */
-  generate(ast) {
-    const handler = generationHandlers.get(ast.$type);
-    if (!handler) {
-      throw new Error(`No generation handler found for ${ast.$type}`);
-    }
-    return handler.value.call(this, ast);
-  }
-  _generateModel(ast) {
-    return ast.declarations.map((d) => this.generate(d)).join("\n\n");
-  }
-  _generateDataSource(ast) {
-    return `datasource ${ast.name} {
-${ast.fields.map((x) => this.indent + this.generate(x)).join("\n")}
-}`;
-  }
-  _generateEnum(ast) {
-    return `enum ${ast.name} {
-${ast.fields.map((x) => this.indent + this.generate(x)).join("\n")}
-}`;
-  }
-  _generateEnumField(ast) {
-    return `${ast.name}${ast.attributes.length > 0 ? " " + ast.attributes.map((x) => this.generate(x)).join(" ") : ""}`;
-  }
-  _generateGenerator(ast) {
-    return `generator ${ast.name} {
-${ast.fields.map((x) => this.indent + this.generate(x)).join("\n")}
-}`;
-  }
-  _generateConfigField(ast) {
-    return `${ast.name} = ${this.generate(ast.value)}`;
-  }
-  _generateConfigArrayExpr(ast) {
-    return `[${ast.items.map((x) => this.generate(x)).join(", ")}]`;
-  }
-  _generateConfigInvocationExpr(ast) {
-    if (ast.args.length === 0) {
-      return ast.name;
-    } else {
-      return `${ast.name}(${ast.args.map((x) => (x.name ? x.name + ": " : "") + this.generate(x.value)).join(", ")})`;
-    }
-  }
-  _generatePlugin(ast) {
-    return `plugin ${ast.name} {
-${ast.fields.map((x) => this.indent + this.generate(x)).join("\n")}
-}`;
-  }
-  _generatePluginField(ast) {
-    return `${ast.name} = ${this.generate(ast.value)}`;
-  }
-  _generateDataModel(ast) {
-    return `${ast.isView ? "view" : "model"} ${ast.name}${ast.mixins.length > 0 ? " mixes " + ast.mixins.map((x) => x.ref?.name).join(", ") : ""} {
-${ast.fields.map((x) => this.indent + this.generate(x)).join("\n")}${ast.attributes.length > 0 ? "\n\n" + ast.attributes.map((x) => this.indent + this.generate(x)).join("\n") : ""}
-}`;
-  }
-  _generateDataField(ast) {
-    return `${ast.name} ${this.fieldType(ast.type)}${ast.attributes.length > 0 ? " " + ast.attributes.map((x) => this.generate(x)).join(" ") : ""}`;
-  }
-  fieldType(type) {
-    const baseType = type.type ? type.type : type.$type == "DataFieldType" && type.unsupported ? "Unsupported(" + this.generate(type.unsupported.value) + ")" : type.reference?.$refText;
-    return `${baseType}${type.array ? "[]" : ""}${type.optional ? "?" : ""}`;
-  }
-  _generateDataModelAttribute(ast) {
-    return this.attribute(ast);
-  }
-  _generateDataFieldAttribute(ast) {
-    return this.attribute(ast);
-  }
-  attribute(ast) {
-    const args = ast.args.length ? `(${ast.args.map((x) => this.generate(x)).join(", ")})` : "";
-    return `${resolved(ast.decl).name}${args}`;
-  }
-  _generateAttributeArg(ast) {
-    if (ast.name) {
-      return `${ast.name}: ${this.generate(ast.value)}`;
-    } else {
-      return this.generate(ast.value);
-    }
-  }
-  _generateObjectExpr(ast) {
-    return `{ ${ast.fields.map((field) => this.objectField(field)).join(", ")} }`;
-  }
-  objectField(field) {
-    return `${field.name}: ${this.generate(field.value)}`;
-  }
-  _generateArrayExpr(ast) {
-    return `[${ast.items.map((item) => this.generate(item)).join(", ")}]`;
-  }
-  _generateLiteralExpr(ast) {
-    return this.options.quote === "single" ? `'${ast.value}'` : `"${ast.value}"`;
-  }
-  _generateNumberLiteral(ast) {
-    return ast.value.toString();
-  }
-  _generateBooleanLiteral(ast) {
-    return ast.value.toString();
-  }
-  _generateUnaryExpr(ast) {
-    return `${ast.operator}${this.unaryExprSpace}${this.generate(ast.operand)}`;
-  }
-  _generateBinaryExpr(ast) {
-    const operator = ast.operator;
-    const isCollectionPredicate = this.isCollectionPredicateOperator(operator);
-    const rightExpr = this.generate(ast.right);
-    const { left: isLeftParenthesis, right: isRightParenthesis } = this.isParenthesesNeededForBinaryExpr(ast);
-    return `${isLeftParenthesis ? "(" : ""}${this.generate(ast.left)}${isLeftParenthesis ? ")" : ""}${isCollectionPredicate ? "" : this.binaryExprSpace}${operator}${isCollectionPredicate ? "" : this.binaryExprSpace}${isRightParenthesis ? "(" : ""}${isCollectionPredicate ? `[${rightExpr}]` : rightExpr}${isRightParenthesis ? ")" : ""}`;
-  }
-  _generateReferenceExpr(ast) {
-    const args = ast.args.length ? `(${ast.args.map((x) => this.generate(x)).join(", ")})` : "";
-    return `${ast.target.ref?.name}${args}`;
-  }
-  _generateReferenceArg(ast) {
-    return `${ast.name}:${this.generate(ast.value)}`;
-  }
-  _generateMemberExpr(ast) {
-    return `${this.generate(ast.operand)}.${ast.member.ref?.name}`;
-  }
-  _generateInvocationExpr(ast) {
-    return `${ast.function.ref?.name}(${ast.args.map((x) => this.argument(x)).join(", ")})`;
-  }
-  _generateNullExpr() {
-    return "null";
-  }
-  _generateThisExpr() {
-    return "this";
-  }
-  _generateAttribute(ast) {
-    return `attribute ${ast.name}(${ast.params.map((x) => this.generate(x)).join(", ")})`;
-  }
-  _generateAttributeParam(ast) {
-    return `${ast.default ? "_ " : ""}${ast.name}: ${this.generate(ast.type)}`;
-  }
-  _generateAttributeParamType(ast) {
-    return `${ast.type ?? ast.reference?.$refText}${ast.array ? "[]" : ""}${ast.optional ? "?" : ""}`;
-  }
-  _generateFunctionDecl(ast) {
-    return `function ${ast.name}(${ast.params.map((x) => this.generate(x)).join(", ")}) ${ast.returnType ? ": " + this.generate(ast.returnType) : ""} {}`;
-  }
-  _generateFunctionParam(ast) {
-    return `${ast.name}: ${this.generate(ast.type)}`;
-  }
-  _generateFunctionParamType(ast) {
-    return `${ast.type ?? ast.reference?.$refText}${ast.array ? "[]" : ""}`;
-  }
-  _generateTypeDef(ast) {
-    return `type ${ast.name} {
-${ast.fields.map((x) => this.indent + this.generate(x)).join("\n")}${ast.attributes.length > 0 ? "\n\n" + ast.attributes.map((x) => this.indent + this.generate(x)).join("\n") : ""}
-}`;
-  }
-  argument(ast) {
-    return this.generate(ast.value);
-  }
-  get binaryExprSpace() {
-    return " ".repeat(this.options.binaryExprNumberOfSpaces);
-  }
-  get unaryExprSpace() {
-    return " ".repeat(this.options.unaryExprNumberOfSpaces);
-  }
-  get indent() {
-    return " ".repeat(this.options.indent);
-  }
-  isParenthesesNeededForBinaryExpr(ast) {
-    const result = {
-      left: false,
-      right: false
-    };
-    const operator = ast.operator;
-    const isCollectionPredicate = this.isCollectionPredicateOperator(operator);
-    const currentPriority = import_ast4.BinaryExprOperatorPriority[operator];
-    if (ast.left.$type === import_ast4.BinaryExpr && import_ast4.BinaryExprOperatorPriority[ast.left["operator"]] < currentPriority) {
-      result.left = true;
-    }
-    if (!isCollectionPredicate && ast.right.$type === import_ast4.BinaryExpr && import_ast4.BinaryExprOperatorPriority[ast.right["operator"]] <= currentPriority) {
-      result.right = true;
-    }
-    return result;
-  }
-  isCollectionPredicateOperator(op) {
-    return [
-      "?",
-      "!",
-      "^"
-    ].includes(op);
-  }
-};
-_ts_decorate([
-  gen(import_ast4.Model),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.Model === "undefined" ? Object : import_ast4.Model
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateModel", null);
-_ts_decorate([
-  gen(import_ast4.DataSource),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.DataSource === "undefined" ? Object : import_ast4.DataSource
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateDataSource", null);
-_ts_decorate([
-  gen(import_ast4.Enum),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.Enum === "undefined" ? Object : import_ast4.Enum
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateEnum", null);
-_ts_decorate([
-  gen(import_ast4.EnumField),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.EnumField === "undefined" ? Object : import_ast4.EnumField
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateEnumField", null);
-_ts_decorate([
-  gen(import_ast4.GeneratorDecl),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.GeneratorDecl === "undefined" ? Object : import_ast4.GeneratorDecl
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateGenerator", null);
-_ts_decorate([
-  gen(import_ast4.ConfigField),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ConfigField === "undefined" ? Object : import_ast4.ConfigField
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateConfigField", null);
-_ts_decorate([
-  gen(import_ast4.ConfigArrayExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ConfigArrayExpr === "undefined" ? Object : import_ast4.ConfigArrayExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateConfigArrayExpr", null);
-_ts_decorate([
-  gen(import_ast4.ConfigInvocationExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ConfigInvocationExpr === "undefined" ? Object : import_ast4.ConfigInvocationExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateConfigInvocationExpr", null);
-_ts_decorate([
-  gen(import_ast4.Plugin),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.Plugin === "undefined" ? Object : import_ast4.Plugin
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generatePlugin", null);
-_ts_decorate([
-  gen(import_ast4.PluginField),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.PluginField === "undefined" ? Object : import_ast4.PluginField
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generatePluginField", null);
-_ts_decorate([
-  gen(import_ast4.DataModel),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.DataModel === "undefined" ? Object : import_ast4.DataModel
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateDataModel", null);
-_ts_decorate([
-  gen(import_ast4.DataField),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.DataField === "undefined" ? Object : import_ast4.DataField
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateDataField", null);
-_ts_decorate([
-  gen(import_ast4.DataModelAttribute),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.DataModelAttribute === "undefined" ? Object : import_ast4.DataModelAttribute
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateDataModelAttribute", null);
-_ts_decorate([
-  gen(import_ast4.DataFieldAttribute),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.DataFieldAttribute === "undefined" ? Object : import_ast4.DataFieldAttribute
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateDataFieldAttribute", null);
-_ts_decorate([
-  gen(import_ast4.AttributeArg),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.AttributeArg === "undefined" ? Object : import_ast4.AttributeArg
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateAttributeArg", null);
-_ts_decorate([
-  gen(import_ast4.ObjectExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ObjectExpr === "undefined" ? Object : import_ast4.ObjectExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateObjectExpr", null);
-_ts_decorate([
-  gen(import_ast4.ArrayExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ArrayExpr === "undefined" ? Object : import_ast4.ArrayExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateArrayExpr", null);
-_ts_decorate([
-  gen(import_ast4.StringLiteral),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.LiteralExpr === "undefined" ? Object : import_ast4.LiteralExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateLiteralExpr", null);
-_ts_decorate([
-  gen(import_ast4.NumberLiteral),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.NumberLiteral === "undefined" ? Object : import_ast4.NumberLiteral
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateNumberLiteral", null);
-_ts_decorate([
-  gen(import_ast4.BooleanLiteral),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.BooleanLiteral === "undefined" ? Object : import_ast4.BooleanLiteral
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateBooleanLiteral", null);
-_ts_decorate([
-  gen(import_ast4.UnaryExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.UnaryExpr === "undefined" ? Object : import_ast4.UnaryExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateUnaryExpr", null);
-_ts_decorate([
-  gen(import_ast4.BinaryExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.BinaryExpr === "undefined" ? Object : import_ast4.BinaryExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateBinaryExpr", null);
-_ts_decorate([
-  gen(import_ast4.ReferenceExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ReferenceExpr === "undefined" ? Object : import_ast4.ReferenceExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateReferenceExpr", null);
-_ts_decorate([
-  gen(import_ast4.ReferenceArg),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.ReferenceArg === "undefined" ? Object : import_ast4.ReferenceArg
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateReferenceArg", null);
-_ts_decorate([
-  gen(import_ast4.MemberAccessExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.MemberAccessExpr === "undefined" ? Object : import_ast4.MemberAccessExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateMemberExpr", null);
-_ts_decorate([
-  gen(import_ast4.InvocationExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.InvocationExpr === "undefined" ? Object : import_ast4.InvocationExpr
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateInvocationExpr", null);
-_ts_decorate([
-  gen(import_ast4.NullExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", []),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateNullExpr", null);
-_ts_decorate([
-  gen(import_ast4.ThisExpr),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", []),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateThisExpr", null);
-_ts_decorate([
-  gen(import_ast4.Attribute),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.Attribute === "undefined" ? Object : import_ast4.Attribute
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateAttribute", null);
-_ts_decorate([
-  gen(import_ast4.AttributeParam),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.AttributeParam === "undefined" ? Object : import_ast4.AttributeParam
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateAttributeParam", null);
-_ts_decorate([
-  gen(import_ast4.AttributeParamType),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.AttributeParamType === "undefined" ? Object : import_ast4.AttributeParamType
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateAttributeParamType", null);
-_ts_decorate([
-  gen(import_ast4.FunctionDecl),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.FunctionDecl === "undefined" ? Object : import_ast4.FunctionDecl
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateFunctionDecl", null);
-_ts_decorate([
-  gen(import_ast4.FunctionParam),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.FunctionParam === "undefined" ? Object : import_ast4.FunctionParam
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateFunctionParam", null);
-_ts_decorate([
-  gen(import_ast4.FunctionParamType),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.FunctionParamType === "undefined" ? Object : import_ast4.FunctionParamType
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateFunctionParamType", null);
-_ts_decorate([
-  gen(import_ast4.TypeDef),
-  _ts_metadata("design:type", Function),
-  _ts_metadata("design:paramtypes", [
-    typeof import_ast4.TypeDef === "undefined" ? Object : import_ast4.TypeDef
-  ]),
-  _ts_metadata("design:returntype", void 0)
-], ZModelCodeGenerator.prototype, "_generateTypeDef", null);
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  DefaultOperationNodeVisitor,
-  ExpressionVisitor,
   ModelUtils,
   PrismaSchemaGenerator,
-  TsSchemaGenerator,
-  ZModelCodeGenerator
+  TsSchemaGenerator
 });
 //# sourceMappingURL=index.cjs.map