Allow handler to return a promise when using PropertyValidators (#31895)

GitOrigin-RevId: f210a4fb93772b0fa2c4dbca3561622e5ea2d157

src/server/registration.test.ts

@@ -430,6 +430,36 @@ describe("argument and return value validators can be objects or validators", ()
       },
     }),
 
+    queryAsync: query({
+      args: v.object({
+        arg: v.string(),
+      }),
+      returns: { arg: v.string() },
+      handler: async (_, { arg }) => {
+        return { arg: arg };
+      },
+    }),
+
+    mutationAsync: mutation({
+      args: v.object({
+        arg: v.string(),
+      }),
+      returns: { arg: v.string() },
+      handler: async (_, { arg }) => {
+        return { arg: arg };
+      },
+    }),
+
+    actionAsync: action({
+      args: v.object({
+        arg: v.string(),
+      }),
+      returns: { arg: v.string() },
+      handler: async (_, { arg }) => {
+        return { arg: arg };
+      },
+    }),
+
     // This is syntx that we no longer want to support when typechecking because they result in undefined behavior.
     mutationNoOptionalValidators: mutation({
       // @ts-expect-error Optional validators are not supported at the top level

src/server/registration.ts

@@ -583,7 +583,7 @@ export type ReturnValueForOptionalValidator<
 > = [ReturnsValidator] extends [Validator<any, any, any>]
   ? ValidatorTypeToReturnType<Infer<ReturnsValidator>>
   : [ReturnsValidator] extends [PropertyValidators]
-    ? ObjectType<ReturnsValidator>
+    ? ValidatorTypeToReturnType<ObjectType<ReturnsValidator>>
     : any;
 
 export type ArgsArrayForOptionalValidator<