feat: pass args to task onSuccess and onFail callbacks

packages/payload/src/queues/config/types/taskTypes.ts

@@ -127,15 +127,18 @@ export type RunInlineTaskFunction = <TTaskInput extends object, TTaskOutput exte
   },
 ) => Promise<TTaskOutput>
 
-export type ShouldRestoreFn = (args: {
+export type TaskCallbackArgs = {
   /**
    * Input data passed to the task
    */
   input: object
   job: Job
   req: PayloadRequest
   taskStatus: SingleTaskStatus<string>
-}) => boolean | Promise<boolean>
+}
+
+export type ShouldRestoreFn = (args: TaskCallbackArgs) => boolean | Promise<boolean>
+export type TaskCallbackFn = (args: TaskCallbackArgs) => Promise<void> | void
 
 export type RetryConfig = {
   /**
@@ -220,11 +223,11 @@ export type TaskConfig<
   /**
    * Function to be executed if the task fails.
    */
-  onFail?: () => Promise<void> | void
+  onFail?: TaskCallbackFn
   /**
    * Function to be executed if the task succeeds.
    */
-  onSuccess?: () => Promise<void> | void
+  onSuccess?: TaskCallbackFn
   /**
    * Define the output field schema - payload will generate a type for this schema.
    */

packages/payload/src/queues/errors/handleTaskError.ts

@@ -47,8 +47,13 @@ export async function handleTaskError({
     workflowConfig,
   } = error.args
 
-  if (taskConfig?.onFail) {
-    await taskConfig.onFail()
+  if (taskConfig?.onFail && taskStatus) {
+    await taskConfig.onFail({
+      input: input!,
+      job,
+      req,
+      taskStatus,
+    })
   }
 
   const errorJSON = {

packages/payload/src/queues/operations/runJobs/runJob/getRunTaskFunction.ts

@@ -181,8 +181,13 @@ export const getRunTaskFunction = <TIsInline extends boolean>(
         output = taskHandlerResult.output
       }
 
-      if (taskConfig?.onSuccess) {
-        await taskConfig.onSuccess()
+      if (taskConfig?.onSuccess && taskStatus) {
+        await taskConfig.onSuccess({
+          input: input!,
+          job,
+          req,
+          taskStatus,
+        })
       }
 
       ;(job.log ??= []).push({