zodvex 0.3.2 → 0.4.0

package/dist/index.js

@@ -725,6 +725,20 @@ function safeOmit(schema, mask) {
 }
 
 // src/custom.ts
+function customCtxWithHooks(fn) {
+  return {
+    args: {},
+    input: async (ctx) => {
+      const result = await fn(ctx);
+      return {
+        ctx: result.ctx ?? {},
+        args: result.args ?? {},
+        hooks: result.hooks,
+        transforms: result.transforms
+      };
+    }
+  };
+}
 function customFnBuilder(builder, customization) {
   const customInput = customization.input ?? NoOp.input;
   const inputArgs = customization.args ?? NoOp.args;
@@ -777,17 +791,18 @@ function customFnBuilder(builder, customization) {
             } catch (e) {
               handleZodValidationError(e, "returns");
             }
-            if (added?.onSuccess) {
-              await added.onSuccess({
+            if (added?.hooks?.onSuccess) {
+              await added.hooks.onSuccess({
                 ctx,
                 args: parsed.data,
                 result: validated
               });
             }
-            return toConvexJS(returns, validated);
+            const transformed = added?.transforms?.output ? await added.transforms.output(validated, returns) : validated;
+            return toConvexJS(returns, transformed);
           }
-          if (added?.onSuccess) {
-            await added.onSuccess({ ctx, args: parsed.data, result: ret });
+          if (added?.hooks?.onSuccess) {
+            await added.hooks.onSuccess({ ctx, args: parsed.data, result: ret });
           }
           return ret;
         }
@@ -814,13 +829,14 @@ function customFnBuilder(builder, customization) {
           } catch (e) {
             handleZodValidationError(e, "returns");
           }
-          if (added?.onSuccess) {
-            await added.onSuccess({ ctx, args: allArgs, result: validated });
+          if (added?.hooks?.onSuccess) {
+            await added.hooks.onSuccess({ ctx, args: allArgs, result: validated });
           }
-          return toConvexJS(returns, validated);
+          const transformed = added?.transforms?.output ? await added.transforms.output(validated, returns) : validated;
+          return toConvexJS(returns, transformed);
         }
-        if (added?.onSuccess) {
-          await added.onSuccess({ ctx, args: allArgs, result: ret });
+        if (added?.hooks?.onSuccess) {
+          await added.hooks.onSuccess({ ctx, args: allArgs, result: ret });
         }
         return ret;
       }
@@ -1098,6 +1114,6 @@ function zodTable(name, schemaOrShape) {
   }
 }
 
-export { addSystemFields, composeOverrides, convexCodec, customFnBuilder, findBaseCodec, formatZodIssues, fromConvexJS, getObjectShape, getZidTableName, handleZodValidationError, isDateSchema, isZidSchema, makeUnion, mapDateFieldToNumber, pick, pickShape, registerBaseCodec, registryHelpers, returnsAs, safeOmit, safePick, toConvexJS, toJSONSchema, zActionBuilder, zCustomAction, zCustomActionBuilder, zCustomMutation, zCustomMutationBuilder, zCustomQuery, zCustomQueryBuilder, zMutationBuilder, zPaginated, zQueryBuilder, zid, zodDoc, zodDocOrNull, zodTable, zodToConvex, zodToConvexFields, zodvexJSONSchemaOverride };
+export { addSystemFields, composeOverrides, convexCodec, customCtxWithHooks, customFnBuilder, findBaseCodec, formatZodIssues, fromConvexJS, getObjectShape, getZidTableName, handleZodValidationError, isDateSchema, isZidSchema, makeUnion, mapDateFieldToNumber, pick, pickShape, registerBaseCodec, registryHelpers, returnsAs, safeOmit, safePick, toConvexJS, toJSONSchema, zActionBuilder, zCustomAction, zCustomActionBuilder, zCustomMutation, zCustomMutationBuilder, zCustomQuery, zCustomQueryBuilder, zMutationBuilder, zPaginated, zQueryBuilder, zid, zodDoc, zodDocOrNull, zodTable, zodToConvex, zodToConvexFields, zodvexJSONSchemaOverride };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map