@zenstackhq/orm 3.5.5 → 3.5.6

package/dist/index.js

@@ -787,17 +787,32 @@ var BaseCrudDialect = class {
       take = -take;
     }
     result = this.buildSkipTake(result, skip, take);
-    result = this.buildOrderBy(result, model, modelAlias, args.orderBy, negateOrderBy, take);
+    let distinctFields = [];
     if ("distinct" in args && args.distinct) {
-      const distinct = ensureArray(args.distinct);
+      distinctFields = ensureArray(args.distinct);
       if (this.supportsDistinctOn) {
-        result = result.distinctOn(distinct.map((f) => this.eb.ref(`${modelAlias}.${f}`)));
+        result = result.distinctOn(distinctFields.map((f) => this.eb.ref(`${modelAlias}.${f}`)));
       } else {
         throw createNotSupportedError(`"distinct" is not supported by "${this.schema.provider.type}" provider`);
       }
     }
+    let effectiveOrderBy = args.orderBy;
+    if (distinctFields.length > 0 && this.supportsDistinctOn) {
+      const existingOrderBy = enumerate(args.orderBy).filter((o) => Object.keys(o).length > 0);
+      const alreadySatisfied = distinctFields.every((f, i) => i < existingOrderBy.length && Object.keys(existingOrderBy[i])[0] === f);
+      if (existingOrderBy.length > 0 && !alreadySatisfied) {
+        const prependedOrderBy = distinctFields.map((f) => ({
+          [f]: "asc"
+        }));
+        effectiveOrderBy = [
+          ...prependedOrderBy,
+          ...existingOrderBy
+        ];
+      }
+    }
+    result = this.buildOrderBy(result, model, modelAlias, effectiveOrderBy, negateOrderBy, take);
     if (args.cursor) {
-      result = this.buildCursorFilter(model, result, args.cursor, args.orderBy, negateOrderBy, modelAlias);
+      result = this.buildCursorFilter(model, result, args.cursor, effectiveOrderBy, negateOrderBy, modelAlias);
     }
     return result;
   }
@@ -5755,6 +5770,9 @@ var ZodSchemaFactory = class {
       z.literal("asc"),
       z.literal("desc")
     ]);
+    const refineAtMostOneKey = /* @__PURE__ */ __name((s) => s.refine((v) => Object.keys(v).length <= 1, {
+      message: "Each orderBy element must have at most one key"
+    }), "refineAtMostOneKey");
     const nextOpts = this.nextOptions(options);
     for (const [field, fieldDef] of this.getModelFields(model)) {
       if (fieldDef.relation) {
@@ -5762,9 +5780,9 @@ var ZodSchemaFactory = class {
           fields[field] = z.lazy(() => {
             let relationOrderBy = this.makeOrderBySchema(fieldDef.type, withRelation, WithAggregation, nextOpts);
             if (fieldDef.array) {
-              relationOrderBy = relationOrderBy.extend({
+              relationOrderBy = refineAtMostOneKey(relationOrderBy.safeExtend({
                 _count: sort
-              });
+              }));
             }
             return relationOrderBy.optional();
           });
@@ -5798,7 +5816,7 @@ var ZodSchemaFactory = class {
         fields[agg] = z.lazy(() => this.makeOrderBySchema(model, true, false, options).optional());
       }
     }
-    return z.strictObject(fields);
+    return refineAtMostOneKey(z.strictObject(fields));
   }
   makeDistinctSchema(model) {
     const nonRelationFields = this.getModelFields(model).filter(([, def]) => !def.relation).map(([name]) => name);