From f0c470cb0883793f7a64652ba265bae8bf27f981 Mon Sep 17 00:00:00 2001
From: Adrien de Peretti <adrien.deperetti@gmail.com>
Date: Fri, 13 Sep 2024 09:11:45 +0200
Subject: [PATCH] chore(workflows-sdk): Update hook functions type (#9117)

RESOLVES FRMW-2703

**What**
Update types to reflect both invoke and compensate function for hooks
---
 packages/core/workflows-sdk/src/utils/composer/type.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/packages/core/workflows-sdk/src/utils/composer/type.ts b/packages/core/workflows-sdk/src/utils/composer/type.ts
index 74615fd6fb..e291fe712e 100644
--- a/packages/core/workflows-sdk/src/utils/composer/type.ts
+++ b/packages/core/workflows-sdk/src/utils/composer/type.ts
@@ -10,6 +10,7 @@ import {
 import { Context, LoadedModule, MedusaContainer } from "@medusajs/types"
 import { ExportedWorkflow } from "../../helper"
 import { Hook } from "./create-hook"
+import { CompensateFn, InvokeFn } from "./create-step"
 
 export type StepFunctionResult<TOutput extends unknown | unknown[] = unknown> =
   (this: CreateWorkflowComposerContext) => WorkflowData<TOutput>
@@ -32,8 +33,9 @@ export type HookHandler = (...args: any[]) => void | Promise<void>
 type ConvertHooksToFunctions<THooks extends any[]> = {
   [K in keyof THooks]: THooks[K] extends Hook<infer Name, infer Input>
     ? {
-        [Fn in Name]: (
-          callback: (input: Input, context: StepExecutionContext) => any
+        [Fn in Name]: <TOutput, TCompensateInput>(
+          invoke: InvokeFn<Input, TOutput, TCompensateInput>,
+          compensate?: CompensateFn<TCompensateInput>
         ) => void
       }
     : never