fix: use current model for hooks when restoring (#218)

Co-authored-by: thommym <[email protected]>
smartive · Oct 21, 2024 · a66e2d8 · a66e2d8
1 parent 964be81
commit a66e2d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/resolvers/mutations.ts b/src/resolvers/mutations.ts
@@ -279,7 +279,7 @@ const restore = async (model: EntityModel, { where }: { where: any }, ctx: FullC
     const data = { prev: relatedEntity, input: {}, normalizedInput, next: { ...relatedEntity, ...normalizedInput } };
     if (ctx.mutationHook) {
       beforeHooks.push(async () => {
-        await ctx.mutationHook(model, 'restore', 'before', data, ctx);
+        await ctx.mutationHook(currentModel, 'restore', 'before', data, ctx);
       });
     }
     mutations.push(async () => {
@@ -288,7 +288,7 @@ const restore = async (model: EntityModel, { where }: { where: any }, ctx: FullC
     });
     if (ctx.mutationHook) {
       afterHooks.push(async () => {
-        await ctx.mutationHook(model, 'restore', 'after', data, ctx);
+        await ctx.mutationHook(currentModel, 'restore', 'after', data, ctx);
       });
     }