@zenstackhq/runtime 3.0.0-beta.1 → 3.0.0-beta.2

package/dist/index.js

@@ -4295,9 +4295,17 @@ var FindOperationHandler = class extends BaseOperationHandler {
   }
   async handle(operation, args, validateArgs = true) {
     const normalizedArgs = this.normalizeArgs(args);
-    const parsedArgs = validateArgs ? this.inputValidator.validateFindArgs(this.model, operation === "findUnique", normalizedArgs) : normalizedArgs;
+    const findOne = operation === "findFirst" || operation === "findUnique";
+    let parsedArgs = validateArgs ? this.inputValidator.validateFindArgs(this.model, normalizedArgs, {
+      unique: operation === "findUnique",
+      findOne
+    }) : normalizedArgs;
+    if (findOne) {
+      parsedArgs = parsedArgs ?? {};
+      parsedArgs.take = 1;
+    }
     const result = await this.read(this.client.$qb, this.model, parsedArgs);
-    const finalResult = operation === "findMany" ? result : result[0] ?? null;
+    const finalResult = findOne ? result[0] ?? null : result;
     return finalResult;
   }
 };
@@ -4503,10 +4511,8 @@ var InputValidator = class {
   constructor(schema) {
     this.schema = schema;
   }
-  validateFindArgs(model, unique, args) {
-    return this.validate(model, "find", {
-      unique
-    }, (model2, options) => this.makeFindSchema(model2, options), args);
+  validateFindArgs(model, args, options) {
+    return this.validate(model, "find", options, (model2, options2) => this.makeFindSchema(model2, options2), args);
   }
   validateCreateArgs(model, args) {
     return this.validate(model, "create", void 0, (model2) => this.makeCreateSchema(model2), args);
@@ -4575,7 +4581,11 @@ var InputValidator = class {
     fields["omit"] = this.makeOmitSchema(model).optional();
     if (!options.unique) {
       fields["skip"] = this.makeSkipSchema().optional();
-      fields["take"] = this.makeTakeSchema().optional();
+      if (options.findOne) {
+        fields["take"] = z.literal(1).optional();
+      } else {
+        fields["take"] = this.makeTakeSchema().optional();
+      }
       fields["orderBy"] = this.orArray(this.makeOrderBySchema(model, true, false), true).optional();
       fields["cursor"] = this.makeCursorSchema(model).optional();
       fields["distinct"] = this.makeDistinctSchema(model).optional();