@zenstackhq/sdk 3.4.3 → 3.4.5

package/dist/index.cjs

@@ -44,7 +44,9 @@ __export(model_utils_exports, {
   getAttribute: () => getAttribute,
   getAuthDecl: () => getAuthDecl,
   getContainingModel: () => getContainingModel,
+  getDelegateOriginModel: () => getDelegateOriginModel,
   getIdFields: () => getIdFields,
+  getOwnedFields: () => getOwnedFields,
   hasAttribute: () => hasAttribute,
   isDelegateModel: () => isDelegateModel,
   isFromStdlib: () => isFromStdlib,
@@ -89,6 +91,29 @@ function isDelegateModel(node) {
   return (0, import_ast.isDataModel)(node) && hasAttribute(node, "@@delegate");
 }
 __name(isDelegateModel, "isDelegateModel");
+function getOwnedFields(model) {
+  const fields = [
+    ...model.fields
+  ];
+  for (const mixin of model.mixins) {
+    if (mixin.ref) {
+      fields.push(...getOwnedFields(mixin.ref));
+    }
+  }
+  return fields;
+}
+__name(getOwnedFields, "getOwnedFields");
+function getDelegateOriginModel(field, contextModel) {
+  let base = contextModel.baseModel?.ref;
+  while (base) {
+    if (isDelegateModel(base) && getOwnedFields(base).some((f) => f.name === field.name)) {
+      return base.name;
+    }
+    base = base.baseModel?.ref;
+  }
+  return void 0;
+}
+__name(getDelegateOriginModel, "getDelegateOriginModel");
 function isUniqueField(field) {
   if (hasAttribute(field, "@unique")) {
     return true;
@@ -642,7 +667,7 @@ var PrismaSchemaGenerator = class {
       if (model_utils_exports.hasAttribute(field, "@computed")) {
         continue;
       }
-      if (model_utils_exports.isIdField(field, decl) || !this.isInheritedFromDelegate(field, decl)) {
+      if (model_utils_exports.isIdField(field, decl) || !getDelegateOriginModel(field, decl)) {
         this.generateModelField(model, field, decl);
       }
     }
@@ -714,7 +739,7 @@ var PrismaSchemaGenerator = class {
     const type = new ModelFieldType(fieldType, isArray, field.type.optional);
     const attributes = field.attributes.filter((attr) => this.isPrismaAttribute(attr)).filter((attr) => !this.isDefaultWithAuthInvocation(attr)).filter((attr) => (
       // when building physical schema, exclude `@default` for id fields inherited from delegate base
-      !(model_utils_exports.isIdField(field, contextModel) && this.isInheritedFromDelegate(field, contextModel) && attr.decl.$refText === "@default")
+      !(model_utils_exports.isIdField(field, contextModel) && getDelegateOriginModel(field, contextModel) && attr.decl.$refText === "@default")
     )).map((attr) => this.makeFieldAttribute(attr));
     const docs = [
       ...field.comments
@@ -732,9 +757,6 @@ var PrismaSchemaGenerator = class {
     }
     return import_langium.AstUtils.streamAst(expr).some(import_utils2.isAuthInvocation);
   }
-  isInheritedFromDelegate(field, contextModel) {
-    return field.$container !== contextModel && model_utils_exports.isDelegateModel(field.$container);
-  }
   makeFieldAttribute(attr) {
     const attrName = attr.decl.ref.name;
     return new FieldAttribute(attrName, attr.args.map((arg) => this.makeAttributeArg(arg)));
@@ -1099,8 +1121,11 @@ var TsSchemaGenerator = class {
       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)));
+    !isIdField(field, contextModel)) {
+      const delegateOrigin = getDelegateOriginModel(field, contextModel);
+      if (delegateOrigin) {
+        objectFields.push(ts.factory.createPropertyAssignment("originModel", ts.factory.createStringLiteral(delegateOrigin)));
+      }
     }
     if (this.isDiscriminatorField(field)) {
       objectFields.push(ts.factory.createPropertyAssignment("isDiscriminator", ts.factory.createTrue()));