prisma-guard 1.6.1 → 1.7.0

package/dist/runtime/index.cjs

@@ -2224,21 +2224,67 @@ function createQueryBuilder(typeMap, enumMap, uniqueMap, scalarBase) {
       }
     }
     if (shape.orderBy) {
-      if (method === "groupBy" && shape.orderBy === true && shape.by) {
+      if (method === "groupBy" && shape.by) {
         const sortEnum = import_zod7.z.enum(["asc", "desc"]);
-        const groupByOrderFields = {};
-        for (const field of shape.by) {
-          groupByOrderFields[field] = sortEnum.optional();
-        }
-        groupByOrderFields["_count"] = sortEnum.optional();
-        const fieldKeys = Object.keys(groupByOrderFields);
-        const singleSchema = import_zod7.z.object(groupByOrderFields).strict().refine(
-          (v) => fieldKeys.some(
-            (k) => v[k] !== void 0
-          ),
-          { message: "orderBy must specify at least one field" }
-        );
-        schemaFields["orderBy"] = import_zod7.z.union([singleSchema, import_zod7.z.array(singleSchema).min(1)]).optional();
+        const bySet = new Set(shape.by);
+        if (shape.orderBy === true) {
+          const groupByOrderFields = {};
+          for (const field of shape.by) {
+            groupByOrderFields[field] = sortEnum.optional();
+          }
+          groupByOrderFields["_count"] = sortEnum.optional();
+          const fieldKeys = Object.keys(groupByOrderFields);
+          const singleSchema = import_zod7.z.object(groupByOrderFields).strict().refine(
+            (v) => fieldKeys.some(
+              (k) => v[k] !== void 0
+            ),
+            { message: "orderBy must specify at least one field" }
+          );
+          schemaFields["orderBy"] = import_zod7.z.union([singleSchema, import_zod7.z.array(singleSchema).min(1)]).optional();
+        } else {
+          const groupByOrderFields = {};
+          for (const [fieldName, config] of Object.entries(shape.orderBy)) {
+            if (fieldName === "_count") {
+              if (config === true) {
+                groupByOrderFields["_count"] = sortEnum.optional();
+              } else if (typeof config === "object" && config !== null) {
+                const countFields = {};
+                for (const countField of Object.keys(config)) {
+                  if (!bySet.has(countField)) {
+                    throw new ShapeError(
+                      `orderBy _count field "${countField}" must be included in "by" for groupBy`
+                    );
+                  }
+                  countFields[countField] = sortEnum.optional();
+                }
+                const countKeys = Object.keys(countFields);
+                groupByOrderFields["_count"] = import_zod7.z.object(countFields).strict().refine(
+                  (v) => countKeys.some(
+                    (k) => v[k] !== void 0
+                  ),
+                  {
+                    message: "orderBy._count must specify at least one field"
+                  }
+                ).optional();
+              }
+              continue;
+            }
+            if (!bySet.has(fieldName)) {
+              throw new ShapeError(
+                `orderBy field "${fieldName}" must be included in "by" for groupBy`
+              );
+            }
+            groupByOrderFields[fieldName] = sortEnum.optional();
+          }
+          const fieldKeys = Object.keys(groupByOrderFields);
+          const singleSchema = import_zod7.z.object(groupByOrderFields).strict().refine(
+            (v) => fieldKeys.some(
+              (k) => v[k] !== void 0
+            ),
+            { message: "orderBy must specify at least one field" }
+          );
+          schemaFields["orderBy"] = import_zod7.z.union([singleSchema, import_zod7.z.array(singleSchema).min(1)]).optional();
+        }
       } else {
         schemaFields["orderBy"] = argsBuilder.buildOrderBySchema(
           model,