zodvex 0.7.2 → 0.7.3

package/dist/mini/server/index.js

@@ -806,15 +806,10 @@ function parseObjectArgsOrThrow(argsSchema, rawArgs) {
   }
   return parsed.data;
 }
-async function runCustomizationInput(customInput, ctx, allArgs, inputArgs, extra) {
-  return await customInput(
-    ctx,
-    // Cast justification: customInput expects ObjectType<CustomArgsValidator>, but pick()
-    // returns Partial<T>. The cast is safe because inputArgs keys are derived from
-    // CustomArgsValidator at the type level.
-    pick(allArgs, Object.keys(inputArgs)),
-    extra
-  );
+async function runCustomizationInput(customInput, ctx, allArgs, inputArgs, extra, argsSchema) {
+  const picked = pick(allArgs, Object.keys(inputArgs));
+  const customArgs = argsSchema ? parseObjectArgsOrThrow(argsSchema, picked) : picked;
+  return await customInput(ctx, customArgs, extra);
 }
 function applyCustomizationResult(ctx, baseArgs, added) {
   const finalCtx = { ...ctx, ...added?.ctx ?? {} };
@@ -842,7 +837,11 @@ async function finalizeFunctionReturn(result, options) {
 // src/internal/custom.ts
 function customFnBuilder(builder, customization) {
   const customInput = customization.input ?? NoOp.input;
-  const inputArgs = customization.args ?? NoOp.args;
+  const rawInputArgs = customization.args ?? NoOp.args;
+  const customArgEntries = Object.entries(rawInputArgs);
+  const customArgsAreZod = customArgEntries.length > 0 && customArgEntries.every(([, v7]) => v7 instanceof $ZodType);
+  const inputArgs = customArgsAreZod ? zodToConvexFields(rawInputArgs) : rawInputArgs;
+  const customArgsSchema = customArgsAreZod ? z.object(rawInputArgs) : void 0;
   return function customBuilder(fn) {
     const { args, handler = fn, returns: maybeObject, ...extra } = fn;
     const skipConvexValidation = fn.skipConvexValidation ?? false;
@@ -865,7 +864,8 @@ function customFnBuilder(builder, customization) {
             ctx,
             allArgs,
             inputArgs,
-            extra
+            extra,
+            customArgsSchema
           );
           const argKeys = Object.keys(argsValidator);
           const rawArgs = pick(allArgs, argKeys);
@@ -888,7 +888,8 @@ function customFnBuilder(builder, customization) {
           ctx,
           allArgs,
           inputArgs,
-          extra
+          extra,
+          customArgsSchema
         );
         const { finalCtx, finalArgs } = applyCustomizationResult(ctx, baseArgs, added);
         const ret = await handler(finalCtx, finalArgs);