@zenstackhq/sdk 3.0.0-beta.15 → 3.0.0-beta.17

package/dist/index.cjs

@@ -825,28 +825,43 @@ var TsSchemaGenerator = class {
     __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 schemaObject = this.createSchemaObject(model, lite);
     const runtimeImportDecl = ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports([
       ts.factory.createImportSpecifier(true, void 0, ts.factory.createIdentifier("SchemaDef")),
       ...hasComputedFields ? [
@@ -872,15 +887,15 @@ var TsSchemaGenerator = class {
     this.usedExpressionUtils = true;
     return ts.factory.createCallExpression(ts.factory.createPropertyAccessExpression(ts.factory.createIdentifier("ExpressionUtils"), method), void 0, args || []);
   }
-  createSchemaObject(model) {
+  createSchemaObject(model, lite) {
     const properties = [
       // provider
       ts.factory.createPropertyAssignment("provider", this.createProviderObject(model)),
       // models
-      ts.factory.createPropertyAssignment("models", this.createModelsObject(model)),
+      ts.factory.createPropertyAssignment("models", this.createModelsObject(model, lite)),
       // typeDefs
       ...model.declarations.some(import_ast3.isTypeDef) ? [
-        ts.factory.createPropertyAssignment("typeDefs", this.createTypeDefsObject(model))
+        ts.factory.createPropertyAssignment("typeDefs", this.createTypeDefsObject(model, lite))
       ] : []
     ];
     const enums = model.declarations.filter(import_ast3.isEnum);
@@ -904,15 +919,15 @@ var TsSchemaGenerator = class {
       ts.factory.createPropertyAssignment("type", ts.factory.createStringLiteral(dsProvider.type))
     ], 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(model.declarations.filter((d) => (0, import_ast3.isDataModel)(d) && !hasAttribute(d, "@@ignore")).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);
+  createTypeDefsObject(model, lite) {
+    return ts.factory.createObjectLiteralExpression(model.declarations.filter((d) => (0, import_ast3.isTypeDef)(d)).map((td) => ts.factory.createPropertyAssignment(td.name, this.createTypeDefObject(td, lite))), true);
   }
-  createDataModelObject(dm) {
+  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;
       }
@@ -927,7 +942,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))
@@ -957,14 +972,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))
@@ -996,7 +1011,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)),
@@ -1025,7 +1040,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);
@@ -1423,9 +1438,9 @@ 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)));
     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) ? [
@@ -1478,13 +1493,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) {
     const importSpecifiers = [];
     if (schemaObject) {
       if (model.declarations.some(import_ast3.isEnum)) {
@@ -1494,17 +1509,17 @@ 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"));
+    return ts.factory.createImportDeclaration(void 0, ts.factory.createImportClause(false, void 0, ts.factory.createNamedImports(importSpecifiers)), ts.factory.createStringLiteral(useLite ? "./schema-lite" : "./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) {
+  generateInputTypes(model, options) {
     const dataModels = model.declarations.filter(import_ast3.isDataModel);
     const statements = [];
-    statements.push(this.generateSchemaImport(model, false, true));
+    statements.push(this.generateSchemaImport(model, false, true, !!(options.lite || options.liteOnly)));
     const inputTypes = [
       "FindManyArgs",
       "FindUniqueArgs",
@@ -1561,7 +1576,7 @@ var TsSchemaGenerator = class {
       ])));
     }
     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);