@webiny/tasks 0.0.0-unstable.e53eceafb5 → 0.0.0-unstable.e6f0dc8ca7

package/runner/TaskControl.js

@@ -1,16 +1,12 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.TaskControl = void 0;
-var _types = require("../types");
-var _TaskManager = require("./TaskManager");
-var _response = require("../response");
-var _TaskManagerStore = require("./TaskManagerStore");
-var _handlerGraphql = require("@webiny/handler-graphql");
-var _getErrorProperties = require("../utils/getErrorProperties");
-class TaskControl {
+import { TaskDataStatus } from "../types.js";
+import { TaskManager } from "./TaskManager.js";
+import { DatabaseResponse, TaskResponse } from "../response/index.js";
+import { TaskManagerStore } from "./TaskManagerStore.js";
+import { getErrorProperties } from "../utils/getErrorProperties.js";
+import { AuthenticatedIdentity } from "@webiny/api-core/features/security/IdentityContext/index.js";
+import { TaskExecutionContext } from "../features/TaskExecutionContext/index.js";
+import { TaskResultStatus } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+export class TaskControl {
   constructor(runner, response, context) {
     this.runner = runner;
     this.context = context;
@@ -27,14 +23,21 @@ class TaskControl {
     let task;
     try {
       task = await this.getTask(taskId);
-      this.context.security.setIdentity(task.createdBy);
+      this.context.security.setIdentity(new AuthenticatedIdentity({
+        id: task.createdBy.id,
+        type: task.createdBy.type,
+        displayName: task.createdBy.displayName ?? "",
+        context: {
+          canAccessTenant: true
+        }
+      }));
     } catch (error) {
       /**
        * TODO Refactor error handling.
        */
       // @ts-expect-error
       return this.response.error({
-        ...(0, _getErrorProperties.getErrorProperties)(error)
+        ...getErrorProperties(error)
       });
     }
     /**
@@ -52,14 +55,20 @@ class TaskControl {
         }
       });
     }
-    const disableDatabaseLogs = !!definition.disableDatabaseLogs;
+    /**
+     * Only enable logs if definition explicitly allows them.
+     */
+    const databaseLogs = definition.databaseLogs === true;
 
     /**
      * As this as a run of the task, we need to create a new log entry.
      */
     let taskLog;
     try {
-      taskLog = await this.getTaskLog(task, disableDatabaseLogs);
+      taskLog = await this.getTaskLog({
+        task,
+        databaseLogs
+      });
     } catch (error) {
       return this.response.error({
         error
@@ -69,14 +78,14 @@ class TaskControl {
      * Make sure that task does not run if it is aborted.
      * This will effectively end the Step Function execution with a "success" status.
      */
-    if (task.taskStatus === _types.TaskDataStatus.ABORTED) {
+    if (task.taskStatus === TaskDataStatus.ABORTED) {
       return this.response.aborted();
     }
     /**
      * Do not run if already a success (done).
      */
     //
-    else if (task.taskStatus === _types.TaskDataStatus.SUCCESS) {
+    else if (task.taskStatus === TaskDataStatus.SUCCESS) {
       return this.response.error({
         error: {
           message: "Task is already done, cannot run it again."
@@ -87,26 +96,35 @@ class TaskControl {
      * Do not run if already failed.
      */
     //
-    else if (task.taskStatus === _types.TaskDataStatus.FAILED) {
+    else if (task.taskStatus === TaskDataStatus.FAILED) {
       return this.response.error({
         error: {
           message: "Task has failed, cannot run it again."
         }
       });
     }
-    const taskResponse = new _response.TaskResponse(this.response);
-    const store = new _TaskManagerStore.TaskManagerStore({
+    const store = new TaskManagerStore({
       context: this.context,
       task,
       log: taskLog,
-      disableDatabaseLogs
+      databaseLogs
     });
-    const manager = new _TaskManager.TaskManager(this.runner, this.context, this.response, taskResponse, store);
-    const databaseResponse = new _response.DatabaseResponse(this.response, store);
+
+    // Populate TaskExecutionContext BEFORE executing task
+    const executionContext = this.context.container.resolve(TaskExecutionContext);
+    executionContext.setStore(store);
+    executionContext.setRunner(this.runner);
+    executionContext.setTimer(this.runner.timer);
+    executionContext.setResponse(new TaskResponse(this.response));
+    const manager = new TaskManager(this.context, this.response, store);
+    const databaseResponse = new DatabaseResponse(this.response, store);
     try {
       const result = await manager.run(definition);
-      await this.runEvents(result, definition, task);
-      return await databaseResponse.from(result);
+      const responseResult = await databaseResponse.from(result);
+
+      // Get the updated task from store (no database read needed - store maintains local cache)
+      await this.runEvents(result, definition, store.getTask());
+      return responseResult;
     } catch (ex) {
       /**
        * We always want to store the error in the task log.
@@ -122,28 +140,29 @@ class TaskControl {
           }
         }
       }));
+    } finally {
+      // Clear execution context after task completes
+      executionContext.clear();
     }
   }
   async runEvents(result, definition, task) {
-    if (result.status === _types.TaskResponseStatus.ERROR && definition.onError) {
+    if (result.status === TaskResultStatus.ERROR && definition.onError) {
       try {
         await definition.onError({
-          task,
-          context: this.context
+          task
         });
       } catch (ex) {
         console.error(`Error executing onError hook for task "${task.id}".`);
-        console.log((0, _getErrorProperties.getErrorProperties)(ex));
+        console.log(getErrorProperties(ex));
       }
-    } else if (result.status === _types.TaskResponseStatus.DONE && definition.onDone) {
+    } else if (result.status === TaskResultStatus.DONE && definition.onDone) {
       try {
         await definition.onDone({
-          task,
-          context: this.context
+          task
         });
       } catch (ex) {
         console.error(`Error executing onDone hook for task "${task.id}".`);
-        console.log((0, _getErrorProperties.getErrorProperties)(ex));
+        console.log(getErrorProperties(ex));
       }
     }
   }
@@ -170,11 +189,15 @@ class TaskControl {
       }
     });
   }
-  async getTaskLog(task, disableDatabaseLogs) {
+  async getTaskLog(params) {
+    const {
+      task,
+      databaseLogs
+    } = params;
     /**
      * If logs are disabled, let's return a mocked one.
      */
-    if (disableDatabaseLogs) {
+    if (!databaseLogs) {
       return {
         id: `${task.id}-log`,
         createdOn: task.createdOn,
@@ -195,7 +218,7 @@ class TaskControl {
       /**
        * If error is not the NotFoundError, we need to throw it.
        */
-      if (error instanceof _handlerGraphql.NotFoundError === false) {
+      if (error.code !== "NOT_FOUND") {
         throw this.response.error({
           error
         });
@@ -217,6 +240,5 @@ class TaskControl {
     }
   }
 }
-exports.TaskControl = TaskControl;
 
 //# sourceMappingURL=TaskControl.js.map

package/runner/TaskControl.js.map

@@ -1 +1 @@
-{"version":3,"names":["_types","require","_TaskManager","_response","_TaskManagerStore","_handlerGraphql","_getErrorProperties","TaskControl","constructor","runner","response","context","run","event","taskId","webinyTaskId","task","getTask","security","setIdentity","createdBy","error","getErrorProperties","definition","tasks","getDefinition","definitionId","message","id","code","data","disableDatabaseLogs","taskLog","getTaskLog","taskStatus","TaskDataStatus","ABORTED","aborted","SUCCESS","FAILED","taskResponse","TaskResponse","store","TaskManagerStore","log","manager","TaskManager","databaseResponse","DatabaseResponse","result","runEvents","from","ex","stack","input","status","TaskResponseStatus","ERROR","onError","console","DONE","onDone","createdOn","executionName","iteration","iterations","items","getLatestLog","NotFoundError","currentIteration","createLog","exports"],"sources":["TaskControl.ts"],"sourcesContent":["import type { ITaskEvent } from \"~/handler/types\";\nimport type { Context, ITask, ITaskDataInput, ITaskDefinition, ITaskLog } from \"~/types\";\nimport { TaskDataStatus, TaskResponseStatus } from \"~/types\";\nimport type { ITaskControl, ITaskRunner } from \"./abstractions\";\nimport { TaskManager } from \"./TaskManager\";\nimport type { IResponse, IResponseResult } from \"~/response/abstractions\";\nimport { DatabaseResponse, TaskResponse } from \"~/response\";\nimport { TaskManagerStore } from \"./TaskManagerStore\";\nimport { NotFoundError } from \"@webiny/handler-graphql\";\nimport { getErrorProperties } from \"~/utils/getErrorProperties\";\n\nexport class TaskControl implements ITaskControl {\n    public readonly runner: ITaskRunner;\n    public readonly response: IResponse;\n    public readonly context: Context;\n\n    public constructor(runner: ITaskRunner, response: IResponse, context: Context) {\n        this.runner = runner;\n        this.context = context;\n        this.response = response;\n    }\n\n    public async run(event: Pick<ITaskEvent, \"webinyTaskId\">): Promise<IResponseResult> {\n        const taskId = event.webinyTaskId;\n        /**\n         * This is the initial getTask idea.\n         * We will need to take care of child tasks:\n         * * child tasks can be in multiple levels (child task creates a child task, etc...).\n         * * child tasks could be executed in parallel.\n         */\n        let task: ITask<ITaskDataInput>;\n        try {\n            task = await this.getTask(taskId);\n            this.context.security.setIdentity(task.createdBy);\n        } catch (error) {\n            /**\n             * TODO Refactor error handling.\n             */\n            // @ts-expect-error\n            return this.response.error({\n                ...getErrorProperties(error)\n            });\n        }\n        /**\n         * Let's get the task definition.\n         */\n        const definition = this.context.tasks.getDefinition(task.definitionId);\n        if (!definition) {\n            return this.response.error({\n                error: {\n                    message: `Task \"${task.id}\" cannot be executed because there is no \"${task.definitionId}\" definition plugin.`,\n                    code: \"TASK_DEFINITION_ERROR\",\n                    data: {\n                        definitionId: task.definitionId\n                    }\n                }\n            });\n        }\n        const disableDatabaseLogs = !!definition.disableDatabaseLogs;\n\n        /**\n         * As this as a run of the task, we need to create a new log entry.\n         */\n        let taskLog: ITaskLog;\n        try {\n            taskLog = await this.getTaskLog(task, disableDatabaseLogs);\n        } catch (error) {\n            return this.response.error({\n                error\n            });\n        }\n        /**\n         * Make sure that task does not run if it is aborted.\n         * This will effectively end the Step Function execution with a \"success\" status.\n         */\n        if (task.taskStatus === TaskDataStatus.ABORTED) {\n            return this.response.aborted();\n        }\n        /**\n         * Do not run if already a success (done).\n         */\n        //\n        else if (task.taskStatus === TaskDataStatus.SUCCESS) {\n            return this.response.error({\n                error: {\n                    message: \"Task is already done, cannot run it again.\"\n                }\n            });\n        }\n        /**\n         * Do not run if already failed.\n         */\n        //\n        else if (task.taskStatus === TaskDataStatus.FAILED) {\n            return this.response.error({\n                error: {\n                    message: \"Task has failed, cannot run it again.\"\n                }\n            });\n        }\n\n        const taskResponse = new TaskResponse(this.response);\n        const store = new TaskManagerStore({\n            context: this.context,\n            task,\n            log: taskLog,\n            disableDatabaseLogs\n        });\n\n        const manager = new TaskManager(\n            this.runner,\n            this.context,\n            this.response,\n            taskResponse,\n            store\n        );\n\n        const databaseResponse = new DatabaseResponse(this.response, store);\n\n        try {\n            const result = await manager.run(definition);\n\n            await this.runEvents(result, definition, task);\n\n            return await databaseResponse.from(result);\n        } catch (ex) {\n            /**\n             * We always want to store the error in the task log.\n             */\n            return await databaseResponse.from(\n                this.response.error({\n                    error: {\n                        message: ex.message,\n                        code: ex.code || \"TASK_ERROR\",\n                        stack: ex.stack,\n                        data: {\n                            ...ex.data,\n                            input: task.input\n                        }\n                    }\n                })\n            );\n        }\n    }\n\n    private async runEvents(\n        result: IResponseResult,\n        definition: ITaskDefinition,\n        task: ITask\n    ): Promise<void> {\n        if (result.status === TaskResponseStatus.ERROR && definition.onError) {\n            try {\n                await definition.onError({\n                    task,\n                    context: this.context\n                });\n            } catch (ex) {\n                console.error(`Error executing onError hook for task \"${task.id}\".`);\n                console.log(getErrorProperties(ex));\n            }\n        } else if (result.status === TaskResponseStatus.DONE && definition.onDone) {\n            try {\n                await definition.onDone({\n                    task,\n                    context: this.context\n                });\n            } catch (ex) {\n                console.error(`Error executing onDone hook for task \"${task.id}\".`);\n                console.log(getErrorProperties(ex));\n            }\n        }\n    }\n\n    private async getTask<T = any>(id: string): Promise<ITask<T>> {\n        try {\n            const task = await this.runner.context.tasks.getTask<T>(id);\n            if (task) {\n                return task;\n            }\n        } catch (ex) {\n            throw this.response.error({\n                error: {\n                    message: ex.message,\n                    code: ex.code || \"TASK_ERROR\",\n                    stack: ex.stack,\n                    data: ex.data\n                }\n            });\n        }\n        throw this.response.error({\n            error: {\n                message: `Task \"${id}\" cannot be executed because it does not exist.`,\n                code: \"TASK_NOT_FOUND\"\n            }\n        });\n    }\n\n    private async getTaskLog(task: ITask, disableDatabaseLogs: boolean): Promise<ITaskLog> {\n        /**\n         * If logs are disabled, let's return a mocked one.\n         */\n        if (disableDatabaseLogs) {\n            return {\n                id: `${task.id}-log`,\n                createdOn: task.createdOn,\n                createdBy: task.createdBy,\n                executionName: task.executionName,\n                task: task.id,\n                iteration: task.iterations,\n                items: []\n            };\n        }\n        let taskLog: ITaskLog | null = null;\n        /**\n         * First we are trying to get existing latest log.\n         */\n        try {\n            taskLog = await this.context.tasks.getLatestLog(task.id);\n        } catch (error) {\n            /**\n             * If error is not the NotFoundError, we need to throw it.\n             */\n            if (error instanceof NotFoundError === false) {\n                throw this.response.error({\n                    error\n                });\n            }\n            /**\n             * Otherwise just continue and create a new log.\n             */\n        }\n\n        const currentIteration = taskLog?.iteration || 0;\n\n        try {\n            return await this.context.tasks.createLog(task, {\n                executionName: this.response.event.executionName,\n                iteration: currentIteration + 1\n            });\n        } catch (error) {\n            throw this.response.error({\n                error\n            });\n        }\n    }\n}\n"],"mappings":";;;;;;AAEA,IAAAA,MAAA,GAAAC,OAAA;AAEA,IAAAC,YAAA,GAAAD,OAAA;AAEA,IAAAE,SAAA,GAAAF,OAAA;AACA,IAAAG,iBAAA,GAAAH,OAAA;AACA,IAAAI,eAAA,GAAAJ,OAAA;AACA,IAAAK,mBAAA,GAAAL,OAAA;AAEO,MAAMM,WAAW,CAAyB;EAKtCC,WAAWA,CAACC,MAAmB,EAAEC,QAAmB,EAAEC,OAAgB,EAAE;IAC3E,IAAI,CAACF,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACE,OAAO,GAAGA,OAAO;IACtB,IAAI,CAACD,QAAQ,GAAGA,QAAQ;EAC5B;EAEA,MAAaE,GAAGA,CAACC,KAAuC,EAA4B;IAChF,MAAMC,MAAM,GAAGD,KAAK,CAACE,YAAY;IACjC;AACR;AACA;AACA;AACA;AACA;IACQ,IAAIC,IAA2B;IAC/B,IAAI;MACAA,IAAI,GAAG,MAAM,IAAI,CAACC,OAAO,CAACH,MAAM,CAAC;MACjC,IAAI,CAACH,OAAO,CAACO,QAAQ,CAACC,WAAW,CAACH,IAAI,CAACI,SAAS,CAAC;IACrD,CAAC,CAAC,OAAOC,KAAK,EAAE;MACZ;AACZ;AACA;MACY;MACA,OAAO,IAAI,CAACX,QAAQ,CAACW,KAAK,CAAC;QACvB,GAAG,IAAAC,sCAAkB,EAACD,KAAK;MAC/B,CAAC,CAAC;IACN;IACA;AACR;AACA;IACQ,MAAME,UAAU,GAAG,IAAI,CAACZ,OAAO,CAACa,KAAK,CAACC,aAAa,CAACT,IAAI,CAACU,YAAY,CAAC;IACtE,IAAI,CAACH,UAAU,EAAE;MACb,OAAO,IAAI,CAACb,QAAQ,CAACW,KAAK,CAAC;QACvBA,KAAK,EAAE;UACHM,OAAO,EAAE,SAASX,IAAI,CAACY,EAAE,6CAA6CZ,IAAI,CAACU,YAAY,sBAAsB;UAC7GG,IAAI,EAAE,uBAAuB;UAC7BC,IAAI,EAAE;YACFJ,YAAY,EAAEV,IAAI,CAACU;UACvB;QACJ;MACJ,CAAC,CAAC;IACN;IACA,MAAMK,mBAAmB,GAAG,CAAC,CAACR,UAAU,CAACQ,mBAAmB;;IAE5D;AACR;AACA;IACQ,IAAIC,OAAiB;IACrB,IAAI;MACAA,OAAO,GAAG,MAAM,IAAI,CAACC,UAAU,CAACjB,IAAI,EAAEe,mBAAmB,CAAC;IAC9D,CAAC,CAAC,OAAOV,KAAK,EAAE;MACZ,OAAO,IAAI,CAACX,QAAQ,CAACW,KAAK,CAAC;QACvBA;MACJ,CAAC,CAAC;IACN;IACA;AACR;AACA;AACA;IACQ,IAAIL,IAAI,CAACkB,UAAU,KAAKC,qBAAc,CAACC,OAAO,EAAE;MAC5C,OAAO,IAAI,CAAC1B,QAAQ,CAAC2B,OAAO,CAAC,CAAC;IAClC;IACA;AACR;AACA;IACQ;IAAA,KACK,IAAIrB,IAAI,CAACkB,UAAU,KAAKC,qBAAc,CAACG,OAAO,EAAE;MACjD,OAAO,IAAI,CAAC5B,QAAQ,CAACW,KAAK,CAAC;QACvBA,KAAK,EAAE;UACHM,OAAO,EAAE;QACb;MACJ,CAAC,CAAC;IACN;IACA;AACR;AACA;IACQ;IAAA,KACK,IAAIX,IAAI,CAACkB,UAAU,KAAKC,qBAAc,CAACI,MAAM,EAAE;MAChD,OAAO,IAAI,CAAC7B,QAAQ,CAACW,KAAK,CAAC;QACvBA,KAAK,EAAE;UACHM,OAAO,EAAE;QACb;MACJ,CAAC,CAAC;IACN;IAEA,MAAMa,YAAY,GAAG,IAAIC,sBAAY,CAAC,IAAI,CAAC/B,QAAQ,CAAC;IACpD,MAAMgC,KAAK,GAAG,IAAIC,kCAAgB,CAAC;MAC/BhC,OAAO,EAAE,IAAI,CAACA,OAAO;MACrBK,IAAI;MACJ4B,GAAG,EAAEZ,OAAO;MACZD;IACJ,CAAC,CAAC;IAEF,MAAMc,OAAO,GAAG,IAAIC,wBAAW,CAC3B,IAAI,CAACrC,MAAM,EACX,IAAI,CAACE,OAAO,EACZ,IAAI,CAACD,QAAQ,EACb8B,YAAY,EACZE,KACJ,CAAC;IAED,MAAMK,gBAAgB,GAAG,IAAIC,0BAAgB,CAAC,IAAI,CAACtC,QAAQ,EAAEgC,KAAK,CAAC;IAEnE,IAAI;MACA,MAAMO,MAAM,GAAG,MAAMJ,OAAO,CAACjC,GAAG,CAACW,UAAU,CAAC;MAE5C,MAAM,IAAI,CAAC2B,SAAS,CAACD,MAAM,EAAE1B,UAAU,EAAEP,IAAI,CAAC;MAE9C,OAAO,MAAM+B,gBAAgB,CAACI,IAAI,CAACF,MAAM,CAAC;IAC9C,CAAC,CAAC,OAAOG,EAAE,EAAE;MACT;AACZ;AACA;MACY,OAAO,MAAML,gBAAgB,CAACI,IAAI,CAC9B,IAAI,CAACzC,QAAQ,CAACW,KAAK,CAAC;QAChBA,KAAK,EAAE;UACHM,OAAO,EAAEyB,EAAE,CAACzB,OAAO;UACnBE,IAAI,EAAEuB,EAAE,CAACvB,IAAI,IAAI,YAAY;UAC7BwB,KAAK,EAAED,EAAE,CAACC,KAAK;UACfvB,IAAI,EAAE;YACF,GAAGsB,EAAE,CAACtB,IAAI;YACVwB,KAAK,EAAEtC,IAAI,CAACsC;UAChB;QACJ;MACJ,CAAC,CACL,CAAC;IACL;EACJ;EAEA,MAAcJ,SAASA,CACnBD,MAAuB,EACvB1B,UAA2B,EAC3BP,IAAW,EACE;IACb,IAAIiC,MAAM,CAACM,MAAM,KAAKC,yBAAkB,CAACC,KAAK,IAAIlC,UAAU,CAACmC,OAAO,EAAE;MAClE,IAAI;QACA,MAAMnC,UAAU,CAACmC,OAAO,CAAC;UACrB1C,IAAI;UACJL,OAAO,EAAE,IAAI,CAACA;QAClB,CAAC,CAAC;MACN,CAAC,CAAC,OAAOyC,EAAE,EAAE;QACTO,OAAO,CAACtC,KAAK,CAAC,0CAA0CL,IAAI,CAACY,EAAE,IAAI,CAAC;QACpE+B,OAAO,CAACf,GAAG,CAAC,IAAAtB,sCAAkB,EAAC8B,EAAE,CAAC,CAAC;MACvC;IACJ,CAAC,MAAM,IAAIH,MAAM,CAACM,MAAM,KAAKC,yBAAkB,CAACI,IAAI,IAAIrC,UAAU,CAACsC,MAAM,EAAE;MACvE,IAAI;QACA,MAAMtC,UAAU,CAACsC,MAAM,CAAC;UACpB7C,IAAI;UACJL,OAAO,EAAE,IAAI,CAACA;QAClB,CAAC,CAAC;MACN,CAAC,CAAC,OAAOyC,EAAE,EAAE;QACTO,OAAO,CAACtC,KAAK,CAAC,yCAAyCL,IAAI,CAACY,EAAE,IAAI,CAAC;QACnE+B,OAAO,CAACf,GAAG,CAAC,IAAAtB,sCAAkB,EAAC8B,EAAE,CAAC,CAAC;MACvC;IACJ;EACJ;EAEA,MAAcnC,OAAOA,CAAUW,EAAU,EAAqB;IAC1D,IAAI;MACA,MAAMZ,IAAI,GAAG,MAAM,IAAI,CAACP,MAAM,CAACE,OAAO,CAACa,KAAK,CAACP,OAAO,CAAIW,EAAE,CAAC;MAC3D,IAAIZ,IAAI,EAAE;QACN,OAAOA,IAAI;MACf;IACJ,CAAC,CAAC,OAAOoC,EAAE,EAAE;MACT,MAAM,IAAI,CAAC1C,QAAQ,CAACW,KAAK,CAAC;QACtBA,KAAK,EAAE;UACHM,OAAO,EAAEyB,EAAE,CAACzB,OAAO;UACnBE,IAAI,EAAEuB,EAAE,CAACvB,IAAI,IAAI,YAAY;UAC7BwB,KAAK,EAAED,EAAE,CAACC,KAAK;UACfvB,IAAI,EAAEsB,EAAE,CAACtB;QACb;MACJ,CAAC,CAAC;IACN;IACA,MAAM,IAAI,CAACpB,QAAQ,CAACW,KAAK,CAAC;MACtBA,KAAK,EAAE;QACHM,OAAO,EAAE,SAASC,EAAE,iDAAiD;QACrEC,IAAI,EAAE;MACV;IACJ,CAAC,CAAC;EACN;EAEA,MAAcI,UAAUA,CAACjB,IAAW,EAAEe,mBAA4B,EAAqB;IACnF;AACR;AACA;IACQ,IAAIA,mBAAmB,EAAE;MACrB,OAAO;QACHH,EAAE,EAAE,GAAGZ,IAAI,CAACY,EAAE,MAAM;QACpBkC,SAAS,EAAE9C,IAAI,CAAC8C,SAAS;QACzB1C,SAAS,EAAEJ,IAAI,CAACI,SAAS;QACzB2C,aAAa,EAAE/C,IAAI,CAAC+C,aAAa;QACjC/C,IAAI,EAAEA,IAAI,CAACY,EAAE;QACboC,SAAS,EAAEhD,IAAI,CAACiD,UAAU;QAC1BC,KAAK,EAAE;MACX,CAAC;IACL;IACA,IAAIlC,OAAwB,GAAG,IAAI;IACnC;AACR;AACA;IACQ,IAAI;MACAA,OAAO,GAAG,MAAM,IAAI,CAACrB,OAAO,CAACa,KAAK,CAAC2C,YAAY,CAACnD,IAAI,CAACY,EAAE,CAAC;IAC5D,CAAC,CAAC,OAAOP,KAAK,EAAE;MACZ;AACZ;AACA;MACY,IAAIA,KAAK,YAAY+C,6BAAa,KAAK,KAAK,EAAE;QAC1C,MAAM,IAAI,CAAC1D,QAAQ,CAACW,KAAK,CAAC;UACtBA;QACJ,CAAC,CAAC;MACN;MACA;AACZ;AACA;IACQ;IAEA,MAAMgD,gBAAgB,GAAGrC,OAAO,EAAEgC,SAAS,IAAI,CAAC;IAEhD,IAAI;MACA,OAAO,MAAM,IAAI,CAACrD,OAAO,CAACa,KAAK,CAAC8C,SAAS,CAACtD,IAAI,EAAE;QAC5C+C,aAAa,EAAE,IAAI,CAACrD,QAAQ,CAACG,KAAK,CAACkD,aAAa;QAChDC,SAAS,EAAEK,gBAAgB,GAAG;MAClC,CAAC,CAAC;IACN,CAAC,CAAC,OAAOhD,KAAK,EAAE;MACZ,MAAM,IAAI,CAACX,QAAQ,CAACW,KAAK,CAAC;QACtBA;MACJ,CAAC,CAAC;IACN;EACJ;AACJ;AAACkD,OAAA,CAAAhE,WAAA,GAAAA,WAAA","ignoreList":[]}
+{"version":3,"names":["TaskDataStatus","TaskManager","DatabaseResponse","TaskResponse","TaskManagerStore","getErrorProperties","AuthenticatedIdentity","TaskExecutionContext","TaskResultStatus","TaskControl","constructor","runner","response","context","run","event","taskId","webinyTaskId","task","getTask","security","setIdentity","id","createdBy","type","displayName","canAccessTenant","error","definition","tasks","getDefinition","definitionId","message","code","data","databaseLogs","taskLog","getTaskLog","taskStatus","ABORTED","aborted","SUCCESS","FAILED","store","log","executionContext","container","resolve","setStore","setRunner","setTimer","timer","setResponse","manager","databaseResponse","result","responseResult","from","runEvents","ex","stack","input","clear","status","ERROR","onError","console","DONE","onDone","params","createdOn","executionName","iteration","iterations","items","getLatestLog","currentIteration","createLog"],"sources":["TaskControl.ts"],"sourcesContent":["import type { ITaskEvent } from \"~/handler/types.js\";\nimport type { Context, ITask, ITaskDataInput, ITaskLog } from \"~/types.js\";\nimport { TaskDataStatus } from \"~/types.js\";\nimport type { ITaskControl, ITaskRunner } from \"./abstractions/index.js\";\nimport { TaskManager } from \"./TaskManager.js\";\nimport type { IResponse, IResponseResult } from \"~/response/abstractions/index.js\";\nimport { DatabaseResponse, TaskResponse } from \"~/response/index.js\";\nimport { TaskManagerStore } from \"./TaskManagerStore.js\";\nimport { getErrorProperties } from \"~/utils/getErrorProperties.js\";\nimport { AuthenticatedIdentity } from \"@webiny/api-core/features/security/IdentityContext/index.js\";\nimport { TaskExecutionContext } from \"~/features/TaskExecutionContext/index.js\";\nimport {\n    TaskDefinition,\n    TaskResultStatus\n} from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\n\ninterface IGetTaskLogParams {\n    task: ITask;\n    databaseLogs: boolean;\n}\n\nexport class TaskControl implements ITaskControl {\n    public readonly runner: ITaskRunner;\n    public readonly response: IResponse;\n    public readonly context: Context;\n\n    public constructor(runner: ITaskRunner, response: IResponse, context: Context) {\n        this.runner = runner;\n        this.context = context;\n        this.response = response;\n    }\n\n    public async run(event: Pick<ITaskEvent, \"webinyTaskId\">): Promise<IResponseResult> {\n        const taskId = event.webinyTaskId;\n        /**\n         * This is the initial getTask idea.\n         * We will need to take care of child tasks:\n         * * child tasks can be in multiple levels (child task creates a child task, etc...).\n         * * child tasks could be executed in parallel.\n         */\n        let task: ITask<ITaskDataInput>;\n        try {\n            task = await this.getTask(taskId);\n            this.context.security.setIdentity(\n                new AuthenticatedIdentity({\n                    id: task.createdBy.id,\n                    type: task.createdBy.type,\n                    displayName: task.createdBy.displayName ?? \"\",\n                    context: {\n                        canAccessTenant: true\n                    }\n                })\n            );\n        } catch (error) {\n            /**\n             * TODO Refactor error handling.\n             */\n            // @ts-expect-error\n            return this.response.error({\n                ...getErrorProperties(error)\n            });\n        }\n        /**\n         * Let's get the task definition.\n         */\n        const definition = this.context.tasks.getDefinition(task.definitionId);\n        if (!definition) {\n            return this.response.error({\n                error: {\n                    message: `Task \"${task.id}\" cannot be executed because there is no \"${task.definitionId}\" definition plugin.`,\n                    code: \"TASK_DEFINITION_ERROR\",\n                    data: {\n                        definitionId: task.definitionId\n                    }\n                }\n            });\n        }\n        /**\n         * Only enable logs if definition explicitly allows them.\n         */\n        const databaseLogs = definition.databaseLogs === true;\n\n        /**\n         * As this as a run of the task, we need to create a new log entry.\n         */\n        let taskLog: ITaskLog;\n        try {\n            taskLog = await this.getTaskLog({\n                task,\n                databaseLogs\n            });\n        } catch (error) {\n            return this.response.error({\n                error\n            });\n        }\n        /**\n         * Make sure that task does not run if it is aborted.\n         * This will effectively end the Step Function execution with a \"success\" status.\n         */\n        if (task.taskStatus === TaskDataStatus.ABORTED) {\n            return this.response.aborted();\n        }\n        /**\n         * Do not run if already a success (done).\n         */\n        //\n        else if (task.taskStatus === TaskDataStatus.SUCCESS) {\n            return this.response.error({\n                error: {\n                    message: \"Task is already done, cannot run it again.\"\n                }\n            });\n        }\n        /**\n         * Do not run if already failed.\n         */\n        //\n        else if (task.taskStatus === TaskDataStatus.FAILED) {\n            return this.response.error({\n                error: {\n                    message: \"Task has failed, cannot run it again.\"\n                }\n            });\n        }\n\n        const store = new TaskManagerStore({\n            context: this.context,\n            task,\n            log: taskLog,\n            databaseLogs\n        });\n\n        // Populate TaskExecutionContext BEFORE executing task\n        const executionContext = this.context.container.resolve(TaskExecutionContext);\n        executionContext.setStore(store);\n        executionContext.setRunner(this.runner);\n        executionContext.setTimer(this.runner.timer);\n        executionContext.setResponse(new TaskResponse(this.response));\n\n        const manager = new TaskManager(this.context, this.response, store);\n\n        const databaseResponse = new DatabaseResponse(this.response, store);\n\n        try {\n            const result = await manager.run(definition);\n\n            const responseResult = await databaseResponse.from(result);\n\n            // Get the updated task from store (no database read needed - store maintains local cache)\n            await this.runEvents(result, definition, store.getTask());\n\n            return responseResult;\n        } catch (ex) {\n            /**\n             * We always want to store the error in the task log.\n             */\n            return await databaseResponse.from(\n                this.response.error({\n                    error: {\n                        message: ex.message,\n                        code: ex.code || \"TASK_ERROR\",\n                        stack: ex.stack,\n                        data: {\n                            ...ex.data,\n                            input: task.input\n                        }\n                    }\n                })\n            );\n        } finally {\n            // Clear execution context after task completes\n            executionContext.clear();\n        }\n    }\n\n    private async runEvents(\n        result: IResponseResult,\n        definition: TaskDefinition.Runnable,\n        task: ITask\n    ): Promise<void> {\n        if (result.status === TaskResultStatus.ERROR && definition.onError) {\n            try {\n                await definition.onError({ task });\n            } catch (ex) {\n                console.error(`Error executing onError hook for task \"${task.id}\".`);\n                console.log(getErrorProperties(ex));\n            }\n        } else if (result.status === TaskResultStatus.DONE && definition.onDone) {\n            try {\n                await definition.onDone({ task });\n            } catch (ex) {\n                console.error(`Error executing onDone hook for task \"${task.id}\".`);\n                console.log(getErrorProperties(ex));\n            }\n        }\n    }\n\n    private async getTask<T extends TaskDefinition.TaskInput>(id: string): Promise<ITask<T>> {\n        try {\n            const task = await this.runner.context.tasks.getTask<T>(id);\n            if (task) {\n                return task;\n            }\n        } catch (ex) {\n            throw this.response.error({\n                error: {\n                    message: ex.message,\n                    code: ex.code || \"TASK_ERROR\",\n                    stack: ex.stack,\n                    data: ex.data\n                }\n            });\n        }\n        throw this.response.error({\n            error: {\n                message: `Task \"${id}\" cannot be executed because it does not exist.`,\n                code: \"TASK_NOT_FOUND\"\n            }\n        });\n    }\n\n    private async getTaskLog(params: IGetTaskLogParams): Promise<ITaskLog> {\n        const { task, databaseLogs } = params;\n        /**\n         * If logs are disabled, let's return a mocked one.\n         */\n        if (!databaseLogs) {\n            return {\n                id: `${task.id}-log`,\n                createdOn: task.createdOn,\n                createdBy: task.createdBy,\n                executionName: task.executionName,\n                task: task.id,\n                iteration: task.iterations,\n                items: []\n            };\n        }\n        let taskLog: ITaskLog | null = null;\n        /**\n         * First we are trying to get existing latest log.\n         */\n        try {\n            taskLog = await this.context.tasks.getLatestLog(task.id);\n        } catch (error) {\n            /**\n             * If error is not the NotFoundError, we need to throw it.\n             */\n            if (error.code !== \"NOT_FOUND\") {\n                throw this.response.error({\n                    error\n                });\n            }\n            /**\n             * Otherwise just continue and create a new log.\n             */\n        }\n\n        const currentIteration = taskLog?.iteration || 0;\n\n        try {\n            return await this.context.tasks.createLog(task, {\n                executionName: this.response.event.executionName,\n                iteration: currentIteration + 1\n            });\n        } catch (error) {\n            throw this.response.error({\n                error\n            });\n        }\n    }\n}\n"],"mappings":"AAEA,SAASA,cAAc;AAEvB,SAASC,WAAW;AAEpB,SAASC,gBAAgB,EAAEC,YAAY;AACvC,SAASC,gBAAgB;AACzB,SAASC,kBAAkB;AAC3B,SAASC,qBAAqB,QAAQ,6DAA6D;AACnG,SAASC,oBAAoB;AAC7B,SAEIC,gBAAgB,QACb,wDAAwD;AAO/D,OAAO,MAAMC,WAAW,CAAyB;EAKtCC,WAAWA,CAACC,MAAmB,EAAEC,QAAmB,EAAEC,OAAgB,EAAE;IAC3E,IAAI,CAACF,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACE,OAAO,GAAGA,OAAO;IACtB,IAAI,CAACD,QAAQ,GAAGA,QAAQ;EAC5B;EAEA,MAAaE,GAAGA,CAACC,KAAuC,EAA4B;IAChF,MAAMC,MAAM,GAAGD,KAAK,CAACE,YAAY;IACjC;AACR;AACA;AACA;AACA;AACA;IACQ,IAAIC,IAA2B;IAC/B,IAAI;MACAA,IAAI,GAAG,MAAM,IAAI,CAACC,OAAO,CAACH,MAAM,CAAC;MACjC,IAAI,CAACH,OAAO,CAACO,QAAQ,CAACC,WAAW,CAC7B,IAAIf,qBAAqB,CAAC;QACtBgB,EAAE,EAAEJ,IAAI,CAACK,SAAS,CAACD,EAAE;QACrBE,IAAI,EAAEN,IAAI,CAACK,SAAS,CAACC,IAAI;QACzBC,WAAW,EAAEP,IAAI,CAACK,SAAS,CAACE,WAAW,IAAI,EAAE;QAC7CZ,OAAO,EAAE;UACLa,eAAe,EAAE;QACrB;MACJ,CAAC,CACL,CAAC;IACL,CAAC,CAAC,OAAOC,KAAK,EAAE;MACZ;AACZ;AACA;MACY;MACA,OAAO,IAAI,CAACf,QAAQ,CAACe,KAAK,CAAC;QACvB,GAAGtB,kBAAkB,CAACsB,KAAK;MAC/B,CAAC,CAAC;IACN;IACA;AACR;AACA;IACQ,MAAMC,UAAU,GAAG,IAAI,CAACf,OAAO,CAACgB,KAAK,CAACC,aAAa,CAACZ,IAAI,CAACa,YAAY,CAAC;IACtE,IAAI,CAACH,UAAU,EAAE;MACb,OAAO,IAAI,CAAChB,QAAQ,CAACe,KAAK,CAAC;QACvBA,KAAK,EAAE;UACHK,OAAO,EAAE,SAASd,IAAI,CAACI,EAAE,6CAA6CJ,IAAI,CAACa,YAAY,sBAAsB;UAC7GE,IAAI,EAAE,uBAAuB;UAC7BC,IAAI,EAAE;YACFH,YAAY,EAAEb,IAAI,CAACa;UACvB;QACJ;MACJ,CAAC,CAAC;IACN;IACA;AACR;AACA;IACQ,MAAMI,YAAY,GAAGP,UAAU,CAACO,YAAY,KAAK,IAAI;;IAErD;AACR;AACA;IACQ,IAAIC,OAAiB;IACrB,IAAI;MACAA,OAAO,GAAG,MAAM,IAAI,CAACC,UAAU,CAAC;QAC5BnB,IAAI;QACJiB;MACJ,CAAC,CAAC;IACN,CAAC,CAAC,OAAOR,KAAK,EAAE;MACZ,OAAO,IAAI,CAACf,QAAQ,CAACe,KAAK,CAAC;QACvBA;MACJ,CAAC,CAAC;IACN;IACA;AACR;AACA;AACA;IACQ,IAAIT,IAAI,CAACoB,UAAU,KAAKtC,cAAc,CAACuC,OAAO,EAAE;MAC5C,OAAO,IAAI,CAAC3B,QAAQ,CAAC4B,OAAO,CAAC,CAAC;IAClC;IACA;AACR;AACA;IACQ;IAAA,KACK,IAAItB,IAAI,CAACoB,UAAU,KAAKtC,cAAc,CAACyC,OAAO,EAAE;MACjD,OAAO,IAAI,CAAC7B,QAAQ,CAACe,KAAK,CAAC;QACvBA,KAAK,EAAE;UACHK,OAAO,EAAE;QACb;MACJ,CAAC,CAAC;IACN;IACA;AACR;AACA;IACQ;IAAA,KACK,IAAId,IAAI,CAACoB,UAAU,KAAKtC,cAAc,CAAC0C,MAAM,EAAE;MAChD,OAAO,IAAI,CAAC9B,QAAQ,CAACe,KAAK,CAAC;QACvBA,KAAK,EAAE;UACHK,OAAO,EAAE;QACb;MACJ,CAAC,CAAC;IACN;IAEA,MAAMW,KAAK,GAAG,IAAIvC,gBAAgB,CAAC;MAC/BS,OAAO,EAAE,IAAI,CAACA,OAAO;MACrBK,IAAI;MACJ0B,GAAG,EAAER,OAAO;MACZD;IACJ,CAAC,CAAC;;IAEF;IACA,MAAMU,gBAAgB,GAAG,IAAI,CAAChC,OAAO,CAACiC,SAAS,CAACC,OAAO,CAACxC,oBAAoB,CAAC;IAC7EsC,gBAAgB,CAACG,QAAQ,CAACL,KAAK,CAAC;IAChCE,gBAAgB,CAACI,SAAS,CAAC,IAAI,CAACtC,MAAM,CAAC;IACvCkC,gBAAgB,CAACK,QAAQ,CAAC,IAAI,CAACvC,MAAM,CAACwC,KAAK,CAAC;IAC5CN,gBAAgB,CAACO,WAAW,CAAC,IAAIjD,YAAY,CAAC,IAAI,CAACS,QAAQ,CAAC,CAAC;IAE7D,MAAMyC,OAAO,GAAG,IAAIpD,WAAW,CAAC,IAAI,CAACY,OAAO,EAAE,IAAI,CAACD,QAAQ,EAAE+B,KAAK,CAAC;IAEnE,MAAMW,gBAAgB,GAAG,IAAIpD,gBAAgB,CAAC,IAAI,CAACU,QAAQ,EAAE+B,KAAK,CAAC;IAEnE,IAAI;MACA,MAAMY,MAAM,GAAG,MAAMF,OAAO,CAACvC,GAAG,CAACc,UAAU,CAAC;MAE5C,MAAM4B,cAAc,GAAG,MAAMF,gBAAgB,CAACG,IAAI,CAACF,MAAM,CAAC;;MAE1D;MACA,MAAM,IAAI,CAACG,SAAS,CAACH,MAAM,EAAE3B,UAAU,EAAEe,KAAK,CAACxB,OAAO,CAAC,CAAC,CAAC;MAEzD,OAAOqC,cAAc;IACzB,CAAC,CAAC,OAAOG,EAAE,EAAE;MACT;AACZ;AACA;MACY,OAAO,MAAML,gBAAgB,CAACG,IAAI,CAC9B,IAAI,CAAC7C,QAAQ,CAACe,KAAK,CAAC;QAChBA,KAAK,EAAE;UACHK,OAAO,EAAE2B,EAAE,CAAC3B,OAAO;UACnBC,IAAI,EAAE0B,EAAE,CAAC1B,IAAI,IAAI,YAAY;UAC7B2B,KAAK,EAAED,EAAE,CAACC,KAAK;UACf1B,IAAI,EAAE;YACF,GAAGyB,EAAE,CAACzB,IAAI;YACV2B,KAAK,EAAE3C,IAAI,CAAC2C;UAChB;QACJ;MACJ,CAAC,CACL,CAAC;IACL,CAAC,SAAS;MACN;MACAhB,gBAAgB,CAACiB,KAAK,CAAC,CAAC;IAC5B;EACJ;EAEA,MAAcJ,SAASA,CACnBH,MAAuB,EACvB3B,UAAmC,EACnCV,IAAW,EACE;IACb,IAAIqC,MAAM,CAACQ,MAAM,KAAKvD,gBAAgB,CAACwD,KAAK,IAAIpC,UAAU,CAACqC,OAAO,EAAE;MAChE,IAAI;QACA,MAAMrC,UAAU,CAACqC,OAAO,CAAC;UAAE/C;QAAK,CAAC,CAAC;MACtC,CAAC,CAAC,OAAOyC,EAAE,EAAE;QACTO,OAAO,CAACvC,KAAK,CAAC,0CAA0CT,IAAI,CAACI,EAAE,IAAI,CAAC;QACpE4C,OAAO,CAACtB,GAAG,CAACvC,kBAAkB,CAACsD,EAAE,CAAC,CAAC;MACvC;IACJ,CAAC,MAAM,IAAIJ,MAAM,CAACQ,MAAM,KAAKvD,gBAAgB,CAAC2D,IAAI,IAAIvC,UAAU,CAACwC,MAAM,EAAE;MACrE,IAAI;QACA,MAAMxC,UAAU,CAACwC,MAAM,CAAC;UAAElD;QAAK,CAAC,CAAC;MACrC,CAAC,CAAC,OAAOyC,EAAE,EAAE;QACTO,OAAO,CAACvC,KAAK,CAAC,yCAAyCT,IAAI,CAACI,EAAE,IAAI,CAAC;QACnE4C,OAAO,CAACtB,GAAG,CAACvC,kBAAkB,CAACsD,EAAE,CAAC,CAAC;MACvC;IACJ;EACJ;EAEA,MAAcxC,OAAOA,CAAqCG,EAAU,EAAqB;IACrF,IAAI;MACA,MAAMJ,IAAI,GAAG,MAAM,IAAI,CAACP,MAAM,CAACE,OAAO,CAACgB,KAAK,CAACV,OAAO,CAAIG,EAAE,CAAC;MAC3D,IAAIJ,IAAI,EAAE;QACN,OAAOA,IAAI;MACf;IACJ,CAAC,CAAC,OAAOyC,EAAE,EAAE;MACT,MAAM,IAAI,CAAC/C,QAAQ,CAACe,KAAK,CAAC;QACtBA,KAAK,EAAE;UACHK,OAAO,EAAE2B,EAAE,CAAC3B,OAAO;UACnBC,IAAI,EAAE0B,EAAE,CAAC1B,IAAI,IAAI,YAAY;UAC7B2B,KAAK,EAAED,EAAE,CAACC,KAAK;UACf1B,IAAI,EAAEyB,EAAE,CAACzB;QACb;MACJ,CAAC,CAAC;IACN;IACA,MAAM,IAAI,CAACtB,QAAQ,CAACe,KAAK,CAAC;MACtBA,KAAK,EAAE;QACHK,OAAO,EAAE,SAASV,EAAE,iDAAiD;QACrEW,IAAI,EAAE;MACV;IACJ,CAAC,CAAC;EACN;EAEA,MAAcI,UAAUA,CAACgC,MAAyB,EAAqB;IACnE,MAAM;MAAEnD,IAAI;MAAEiB;IAAa,CAAC,GAAGkC,MAAM;IACrC;AACR;AACA;IACQ,IAAI,CAAClC,YAAY,EAAE;MACf,OAAO;QACHb,EAAE,EAAE,GAAGJ,IAAI,CAACI,EAAE,MAAM;QACpBgD,SAAS,EAAEpD,IAAI,CAACoD,SAAS;QACzB/C,SAAS,EAAEL,IAAI,CAACK,SAAS;QACzBgD,aAAa,EAAErD,IAAI,CAACqD,aAAa;QACjCrD,IAAI,EAAEA,IAAI,CAACI,EAAE;QACbkD,SAAS,EAAEtD,IAAI,CAACuD,UAAU;QAC1BC,KAAK,EAAE;MACX,CAAC;IACL;IACA,IAAItC,OAAwB,GAAG,IAAI;IACnC;AACR;AACA;IACQ,IAAI;MACAA,OAAO,GAAG,MAAM,IAAI,CAACvB,OAAO,CAACgB,KAAK,CAAC8C,YAAY,CAACzD,IAAI,CAACI,EAAE,CAAC;IAC5D,CAAC,CAAC,OAAOK,KAAK,EAAE;MACZ;AACZ;AACA;MACY,IAAIA,KAAK,CAACM,IAAI,KAAK,WAAW,EAAE;QAC5B,MAAM,IAAI,CAACrB,QAAQ,CAACe,KAAK,CAAC;UACtBA;QACJ,CAAC,CAAC;MACN;MACA;AACZ;AACA;IACQ;IAEA,MAAMiD,gBAAgB,GAAGxC,OAAO,EAAEoC,SAAS,IAAI,CAAC;IAEhD,IAAI;MACA,OAAO,MAAM,IAAI,CAAC3D,OAAO,CAACgB,KAAK,CAACgD,SAAS,CAAC3D,IAAI,EAAE;QAC5CqD,aAAa,EAAE,IAAI,CAAC3D,QAAQ,CAACG,KAAK,CAACwD,aAAa;QAChDC,SAAS,EAAEI,gBAAgB,GAAG;MAClC,CAAC,CAAC;IACN,CAAC,CAAC,OAAOjD,KAAK,EAAE;MACZ,MAAM,IAAI,CAACf,QAAQ,CAACe,KAAK,CAAC;QACtBA;MACJ,CAAC,CAAC;IACN;EACJ;AACJ","ignoreList":[]}

package/runner/TaskEventValidation.d.ts

@@ -1,5 +1,5 @@
-import type { ITaskEventValidation, ITaskEventValidationResult } from "./abstractions";
-import type { ITaskEvent } from "../handler/types";
+import type { ITaskEventValidation, ITaskEventValidationResult } from "./abstractions/index.js";
+import type { ITaskEvent } from "../handler/types.js";
 export declare class TaskEventValidation implements ITaskEventValidation {
     validate(event: Partial<ITaskEvent>): ITaskEventValidationResult;
 }

package/runner/TaskEventValidation.js

@@ -1,30 +1,21 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.TaskEventValidation = void 0;
-var _zod = _interopRequireDefault(require("zod"));
-var _utils = require("@webiny/utils");
-const validation = _zod.default.object({
-  webinyTaskId: _zod.default.string(),
-  webinyTaskDefinitionId: _zod.default.string(),
-  endpoint: _zod.default.string(),
-  tenant: _zod.default.string(),
-  locale: _zod.default.string(),
-  executionName: _zod.default.string(),
-  stateMachineId: _zod.default.string()
+import zod from "zod";
+import { createZodError } from "@webiny/utils";
+const validation = zod.object({
+  webinyTaskId: zod.string(),
+  webinyTaskDefinitionId: zod.string(),
+  endpoint: zod.string(),
+  tenant: zod.string(),
+  executionName: zod.string(),
+  stateMachineId: zod.string()
 }).required();
-class TaskEventValidation {
+export class TaskEventValidation {
   validate(event) {
     const result = validation.safeParse(event);
     if (result.success) {
       return result.data;
     }
-    throw (0, _utils.createZodError)(result.error);
+    throw createZodError(result.error);
   }
 }
-exports.TaskEventValidation = TaskEventValidation;
 
 //# sourceMappingURL=TaskEventValidation.js.map

package/runner/TaskEventValidation.js.map

@@ -1 +1 @@
-{"version":3,"names":["_zod","_interopRequireDefault","require","_utils","validation","zod","object","webinyTaskId","string","webinyTaskDefinitionId","endpoint","tenant","locale","executionName","stateMachineId","required","TaskEventValidation","validate","event","result","safeParse","success","data","createZodError","error","exports"],"sources":["TaskEventValidation.ts"],"sourcesContent":["import zod from \"zod\";\nimport { createZodError } from \"@webiny/utils\";\nimport type { ITaskEventValidation, ITaskEventValidationResult } from \"./abstractions\";\nimport type { ITaskEvent } from \"~/handler/types\";\n\nconst validation = zod\n    .object({\n        webinyTaskId: zod.string(),\n        webinyTaskDefinitionId: zod.string(),\n        endpoint: zod.string(),\n        tenant: zod.string(),\n        locale: zod.string(),\n        executionName: zod.string(),\n        stateMachineId: zod.string()\n    })\n    .required();\n\nexport class TaskEventValidation implements ITaskEventValidation {\n    public validate(event: Partial<ITaskEvent>): ITaskEventValidationResult {\n        const result = validation.safeParse(event);\n        if (result.success) {\n            return result.data;\n        }\n        throw createZodError(result.error);\n    }\n}\n"],"mappings":";;;;;;;AAAA,IAAAA,IAAA,GAAAC,sBAAA,CAAAC,OAAA;AACA,IAAAC,MAAA,GAAAD,OAAA;AAIA,MAAME,UAAU,GAAGC,YAAG,CACjBC,MAAM,CAAC;EACJC,YAAY,EAAEF,YAAG,CAACG,MAAM,CAAC,CAAC;EAC1BC,sBAAsB,EAAEJ,YAAG,CAACG,MAAM,CAAC,CAAC;EACpCE,QAAQ,EAAEL,YAAG,CAACG,MAAM,CAAC,CAAC;EACtBG,MAAM,EAAEN,YAAG,CAACG,MAAM,CAAC,CAAC;EACpBI,MAAM,EAAEP,YAAG,CAACG,MAAM,CAAC,CAAC;EACpBK,aAAa,EAAER,YAAG,CAACG,MAAM,CAAC,CAAC;EAC3BM,cAAc,EAAET,YAAG,CAACG,MAAM,CAAC;AAC/B,CAAC,CAAC,CACDO,QAAQ,CAAC,CAAC;AAER,MAAMC,mBAAmB,CAAiC;EACtDC,QAAQA,CAACC,KAA0B,EAA8B;IACpE,MAAMC,MAAM,GAAGf,UAAU,CAACgB,SAAS,CAACF,KAAK,CAAC;IAC1C,IAAIC,MAAM,CAACE,OAAO,EAAE;MAChB,OAAOF,MAAM,CAACG,IAAI;IACtB;IACA,MAAM,IAAAC,qBAAc,EAACJ,MAAM,CAACK,KAAK,CAAC;EACtC;AACJ;AAACC,OAAA,CAAAT,mBAAA,GAAAA,mBAAA","ignoreList":[]}
+{"version":3,"names":["zod","createZodError","validation","object","webinyTaskId","string","webinyTaskDefinitionId","endpoint","tenant","executionName","stateMachineId","required","TaskEventValidation","validate","event","result","safeParse","success","data","error"],"sources":["TaskEventValidation.ts"],"sourcesContent":["import zod from \"zod\";\nimport { createZodError } from \"@webiny/utils\";\nimport type { ITaskEventValidation, ITaskEventValidationResult } from \"./abstractions/index.js\";\nimport type { ITaskEvent } from \"~/handler/types.js\";\n\nconst validation = zod\n    .object({\n        webinyTaskId: zod.string(),\n        webinyTaskDefinitionId: zod.string(),\n        endpoint: zod.string(),\n        tenant: zod.string(),\n        executionName: zod.string(),\n        stateMachineId: zod.string()\n    })\n    .required();\n\nexport class TaskEventValidation implements ITaskEventValidation {\n    public validate(event: Partial<ITaskEvent>): ITaskEventValidationResult {\n        const result = validation.safeParse(event);\n        if (result.success) {\n            return result.data;\n        }\n        throw createZodError(result.error);\n    }\n}\n"],"mappings":"AAAA,OAAOA,GAAG,MAAM,KAAK;AACrB,SAASC,cAAc,QAAQ,eAAe;AAI9C,MAAMC,UAAU,GAAGF,GAAG,CACjBG,MAAM,CAAC;EACJC,YAAY,EAAEJ,GAAG,CAACK,MAAM,CAAC,CAAC;EAC1BC,sBAAsB,EAAEN,GAAG,CAACK,MAAM,CAAC,CAAC;EACpCE,QAAQ,EAAEP,GAAG,CAACK,MAAM,CAAC,CAAC;EACtBG,MAAM,EAAER,GAAG,CAACK,MAAM,CAAC,CAAC;EACpBI,aAAa,EAAET,GAAG,CAACK,MAAM,CAAC,CAAC;EAC3BK,cAAc,EAAEV,GAAG,CAACK,MAAM,CAAC;AAC/B,CAAC,CAAC,CACDM,QAAQ,CAAC,CAAC;AAEf,OAAO,MAAMC,mBAAmB,CAAiC;EACtDC,QAAQA,CAACC,KAA0B,EAA8B;IACpE,MAAMC,MAAM,GAAGb,UAAU,CAACc,SAAS,CAACF,KAAK,CAAC;IAC1C,IAAIC,MAAM,CAACE,OAAO,EAAE;MAChB,OAAOF,MAAM,CAACG,IAAI;IACtB;IACA,MAAMjB,cAAc,CAACc,MAAM,CAACI,KAAK,CAAC;EACtC;AACJ","ignoreList":[]}

package/runner/TaskManager.d.ts

@@ -1,14 +1,11 @@
-import type { ITaskManager, ITaskManagerStorePrivate, ITaskRunner } from "./abstractions";
-import type { Context, ITaskDataInput, ITaskDefinition } from "../types";
-import type { IResponse, IResponseResult, ITaskResponse } from "../response/abstractions";
-type ITaskManagerRunner = Pick<ITaskRunner, "isCloseToTimeout" | "timer">;
-export declare class TaskManager<T = ITaskDataInput> implements ITaskManager<T> {
-    private readonly runner;
+import type { ITaskManager, ITaskManagerStorePrivate } from "./abstractions/index.js";
+import type { Context } from "../types.js";
+import type { IResponse, IResponseResult } from "../response/abstractions/index.js";
+import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+export declare class TaskManager implements ITaskManager {
     private readonly context;
     private readonly response;
-    private readonly taskResponse;
     private readonly store;
-    constructor(runner: ITaskManagerRunner, context: Context, response: IResponse, taskResponse: ITaskResponse, store: ITaskManagerStorePrivate);
-    run(definition: ITaskDefinition): Promise<IResponseResult>;
+    constructor(context: Context, response: IResponse, store: ITaskManagerStorePrivate);
+    run(definition: TaskDefinition.Runnable): Promise<IResponseResult>;
 }
-export {};

package/runner/TaskManager.js

@@ -1,34 +1,28 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.TaskManager = void 0;
-var _types = require("../types");
-var _getErrorProperties = require("../utils/getErrorProperties");
-class TaskManager {
-  constructor(runner, context, response, taskResponse, store) {
-    this.runner = runner;
+import { TaskDataStatus } from "../types.js";
+import { getErrorProperties } from "../utils/getErrorProperties.js";
+import { TaskResultStatus } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { TaskController } from "@webiny/api-core/features/task/TaskController/abstractions.js";
+export class TaskManager {
+  constructor(context, response, store) {
     this.context = context;
     this.response = response;
-    this.taskResponse = taskResponse;
     this.store = store;
   }
   async run(definition) {
     /**
      * If task was aborted, do not run it again, return as it was done.
      */
-    if (this.store.getStatus() === _types.TaskDataStatus.ABORTED) {
+    if (this.store.getStatus() === TaskDataStatus.ABORTED) {
       return this.response.aborted();
     }
     /**
      * If the task status is pending, update it to running and add a log.
      */
     //
-    else if (this.store.getStatus() === _types.TaskDataStatus.PENDING) {
+    else if (this.store.getStatus() === TaskDataStatus.PENDING) {
       try {
         await this.store.updateTask({
-          taskStatus: _types.TaskDataStatus.RUNNING,
+          taskStatus: TaskDataStatus.RUNNING,
           startedOn: new Date().toISOString(),
           executionName: this.response.event.executionName,
           iterations: 1
@@ -49,8 +43,7 @@ class TaskManager {
       try {
         if (definition.onMaxIterations) {
           await definition.onMaxIterations({
-            task: this.store.getTask(),
-            context: this.context
+            task: this.store.getTask()
           });
         }
         return this.response.error({
@@ -62,7 +55,7 @@ class TaskManager {
         return this.response.error({
           error: {
             message: "Failed to execute onMaxIterations handler.",
-            data: (0, _getErrorProperties.getErrorProperties)(ex)
+            data: getErrorProperties(ex)
           }
         });
       }
@@ -87,45 +80,33 @@ class TaskManager {
     let result;
     try {
       const input = structuredClone(this.store.getInput());
+      const controller = this.context.container.resolve(TaskController);
       /**
        * We always run the task without authorization because we are running a task without a user - nothing to authorize against.
        */
       result = await this.context.security.withoutAuthorization(async () => {
         return await definition.run({
           input,
-          context: this.context,
-          response: this.taskResponse,
-          isCloseToTimeout: seconds => {
-            return this.runner.isCloseToTimeout(seconds);
-          },
-          isAborted: () => {
-            return this.store.getStatus() === _types.TaskDataStatus.ABORTED;
-          },
-          store: this.store,
-          trigger: async params => {
-            return this.context.tasks.trigger({
-              ...params,
-              parent: this.store.getTask()
-            });
-          },
-          timer: this.runner.timer
+          controller
         });
       });
     } catch (ex) {
       return this.response.error({
-        error: (0, _getErrorProperties.getErrorProperties)(ex)
+        error: getErrorProperties(ex)
       });
     }
-    if (result.status === _types.TaskResponseStatus.CONTINUE) {
+
+    // Convert TaskDefinition.Result to IResponseResult
+    if (result.status === TaskResultStatus.CONTINUE) {
       return this.response.continue({
         input: result.input,
         wait: result.wait
       });
-    } else if (result.status === _types.TaskResponseStatus.ERROR) {
+    } else if (result.status === TaskResultStatus.ERROR) {
       return this.response.error({
         error: result.error
       });
-    } else if (result.status === _types.TaskResponseStatus.ABORTED) {
+    } else if (result.status === TaskResultStatus.ABORTED) {
       return this.response.aborted();
     }
     return this.response.done({
@@ -134,6 +115,5 @@ class TaskManager {
     });
   }
 }
-exports.TaskManager = TaskManager;
 
 //# sourceMappingURL=TaskManager.js.map

package/runner/TaskManager.js.map

@@ -1 +1 @@
-{"version":3,"names":["_types","require","_getErrorProperties","TaskManager","constructor","runner","context","response","taskResponse","store","run","definition","getStatus","TaskDataStatus","ABORTED","aborted","PENDING","updateTask","taskStatus","RUNNING","startedOn","Date","toISOString","executionName","event","iterations","save","error","getTask","maxIterations","onMaxIterations","task","message","ex","data","getErrorProperties","result","input","structuredClone","getInput","security","withoutAuthorization","isCloseToTimeout","seconds","isAborted","trigger","params","tasks","parent","timer","status","TaskResponseStatus","CONTINUE","continue","wait","ERROR","done","output","exports"],"sources":["TaskManager.ts"],"sourcesContent":["import type { ITaskManager, ITaskManagerStorePrivate, ITaskRunner } from \"./abstractions\";\nimport type { Context, ITask, ITaskDataInput, ITaskDefinition, ITaskTriggerParams } from \"~/types\";\nimport { TaskDataStatus, TaskResponseStatus } from \"~/types\";\nimport type {\n    IResponse,\n    IResponseResult,\n    ITaskResponse,\n    ITaskResponseResult\n} from \"~/response/abstractions\";\nimport { getErrorProperties } from \"~/utils/getErrorProperties\";\n\ntype ITaskManagerRunner = Pick<ITaskRunner, \"isCloseToTimeout\" | \"timer\">;\n\nexport class TaskManager<T = ITaskDataInput> implements ITaskManager<T> {\n    private readonly runner: ITaskManagerRunner;\n    private readonly context: Context;\n    private readonly response: IResponse;\n    private readonly taskResponse: ITaskResponse;\n    private readonly store: ITaskManagerStorePrivate;\n\n    public constructor(\n        runner: ITaskManagerRunner,\n        context: Context,\n        response: IResponse,\n        taskResponse: ITaskResponse,\n        store: ITaskManagerStorePrivate\n    ) {\n        this.runner = runner;\n        this.context = context;\n        this.response = response;\n        this.taskResponse = taskResponse;\n        this.store = store;\n    }\n\n    public async run(definition: ITaskDefinition): Promise<IResponseResult> {\n        /**\n         * If task was aborted, do not run it again, return as it was done.\n         */\n        if (this.store.getStatus() === TaskDataStatus.ABORTED) {\n            return this.response.aborted();\n        }\n        /**\n         * If the task status is pending, update it to running and add a log.\n         */\n        //\n        else if (this.store.getStatus() === TaskDataStatus.PENDING) {\n            try {\n                await this.store.updateTask({\n                    taskStatus: TaskDataStatus.RUNNING,\n                    startedOn: new Date().toISOString(),\n                    executionName: this.response.event.executionName,\n                    iterations: 1\n                });\n                await this.store.save();\n            } catch (error) {\n                return this.response.error({\n                    error\n                });\n            }\n        }\n        /**\n         * We do not want to run the task indefinitely.\n         * If the task has reached the max iterations, we will stop it and execute the onMaxIterations handler, if any.\n         */\n        //\n        else if (this.store.getTask().iterations >= definition.maxIterations) {\n            try {\n                if (definition.onMaxIterations) {\n                    await definition.onMaxIterations({\n                        task: this.store.getTask(),\n                        context: this.context\n                    });\n                }\n                return this.response.error({\n                    error: {\n                        message: \"Task reached max iterations.\"\n                    }\n                });\n            } catch (ex) {\n                return this.response.error({\n                    error: {\n                        message: \"Failed to execute onMaxIterations handler.\",\n                        data: getErrorProperties(ex)\n                    }\n                });\n            }\n        }\n        /**\n         * Always update the task iteration.\n         */\n        //\n        else {\n            try {\n                await this.store.updateTask(task => {\n                    return {\n                        iterations: task.iterations + 1\n                    };\n                });\n            } catch (error) {\n                return this.response.error({\n                    error\n                });\n            }\n        }\n\n        let result: ITaskResponseResult;\n\n        try {\n            const input = structuredClone(this.store.getInput());\n            /**\n             * We always run the task without authorization because we are running a task without a user - nothing to authorize against.\n             */\n            result = await this.context.security.withoutAuthorization(async () => {\n                return await definition.run({\n                    input,\n                    context: this.context,\n                    response: this.taskResponse,\n                    isCloseToTimeout: seconds => {\n                        return this.runner.isCloseToTimeout(seconds);\n                    },\n                    isAborted: () => {\n                        return this.store.getStatus() === TaskDataStatus.ABORTED;\n                    },\n                    store: this.store,\n                    trigger: async <I = ITaskDataInput>(\n                        params: Omit<ITaskTriggerParams<I>, \"parent\">\n                    ): Promise<ITask<I>> => {\n                        return this.context.tasks.trigger({\n                            ...params,\n                            parent: this.store.getTask()\n                        });\n                    },\n                    timer: this.runner.timer\n                });\n            });\n        } catch (ex) {\n            return this.response.error({\n                error: getErrorProperties(ex)\n            });\n        }\n\n        if (result.status === TaskResponseStatus.CONTINUE) {\n            return this.response.continue({\n                input: result.input,\n                wait: result.wait\n            });\n        } else if (result.status === TaskResponseStatus.ERROR) {\n            return this.response.error({\n                error: result.error\n            });\n        } else if (result.status === TaskResponseStatus.ABORTED) {\n            return this.response.aborted();\n        }\n        return this.response.done({\n            message: result.message,\n            output: result.output\n        });\n    }\n}\n"],"mappings":";;;;;;AAEA,IAAAA,MAAA,GAAAC,OAAA;AAOA,IAAAC,mBAAA,GAAAD,OAAA;AAIO,MAAME,WAAW,CAAgD;EAO7DC,WAAWA,CACdC,MAA0B,EAC1BC,OAAgB,EAChBC,QAAmB,EACnBC,YAA2B,EAC3BC,KAA+B,EACjC;IACE,IAAI,CAACJ,MAAM,GAAGA,MAAM;IACpB,IAAI,CAACC,OAAO,GAAGA,OAAO;IACtB,IAAI,CAACC,QAAQ,GAAGA,QAAQ;IACxB,IAAI,CAACC,YAAY,GAAGA,YAAY;IAChC,IAAI,CAACC,KAAK,GAAGA,KAAK;EACtB;EAEA,MAAaC,GAAGA,CAACC,UAA2B,EAA4B;IACpE;AACR;AACA;IACQ,IAAI,IAAI,CAACF,KAAK,CAACG,SAAS,CAAC,CAAC,KAAKC,qBAAc,CAACC,OAAO,EAAE;MACnD,OAAO,IAAI,CAACP,QAAQ,CAACQ,OAAO,CAAC,CAAC;IAClC;IACA;AACR;AACA;IACQ;IAAA,KACK,IAAI,IAAI,CAACN,KAAK,CAACG,SAAS,CAAC,CAAC,KAAKC,qBAAc,CAACG,OAAO,EAAE;MACxD,IAAI;QACA,MAAM,IAAI,CAACP,KAAK,CAACQ,UAAU,CAAC;UACxBC,UAAU,EAAEL,qBAAc,CAACM,OAAO;UAClCC,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;UACnCC,aAAa,EAAE,IAAI,CAAChB,QAAQ,CAACiB,KAAK,CAACD,aAAa;UAChDE,UAAU,EAAE;QAChB,CAAC,CAAC;QACF,MAAM,IAAI,CAAChB,KAAK,CAACiB,IAAI,CAAC,CAAC;MAC3B,CAAC,CAAC,OAAOC,KAAK,EAAE;QACZ,OAAO,IAAI,CAACpB,QAAQ,CAACoB,KAAK,CAAC;UACvBA;QACJ,CAAC,CAAC;MACN;IACJ;IACA;AACR;AACA;AACA;IACQ;IAAA,KACK,IAAI,IAAI,CAAClB,KAAK,CAACmB,OAAO,CAAC,CAAC,CAACH,UAAU,IAAId,UAAU,CAACkB,aAAa,EAAE;MAClE,IAAI;QACA,IAAIlB,UAAU,CAACmB,eAAe,EAAE;UAC5B,MAAMnB,UAAU,CAACmB,eAAe,CAAC;YAC7BC,IAAI,EAAE,IAAI,CAACtB,KAAK,CAACmB,OAAO,CAAC,CAAC;YAC1BtB,OAAO,EAAE,IAAI,CAACA;UAClB,CAAC,CAAC;QACN;QACA,OAAO,IAAI,CAACC,QAAQ,CAACoB,KAAK,CAAC;UACvBA,KAAK,EAAE;YACHK,OAAO,EAAE;UACb;QACJ,CAAC,CAAC;MACN,CAAC,CAAC,OAAOC,EAAE,EAAE;QACT,OAAO,IAAI,CAAC1B,QAAQ,CAACoB,KAAK,CAAC;UACvBA,KAAK,EAAE;YACHK,OAAO,EAAE,4CAA4C;YACrDE,IAAI,EAAE,IAAAC,sCAAkB,EAACF,EAAE;UAC/B;QACJ,CAAC,CAAC;MACN;IACJ;IACA;AACR;AACA;IACQ;IAAA,KACK;MACD,IAAI;QACA,MAAM,IAAI,CAACxB,KAAK,CAACQ,UAAU,CAACc,IAAI,IAAI;UAChC,OAAO;YACHN,UAAU,EAAEM,IAAI,CAACN,UAAU,GAAG;UAClC,CAAC;QACL,CAAC,CAAC;MACN,CAAC,CAAC,OAAOE,KAAK,EAAE;QACZ,OAAO,IAAI,CAACpB,QAAQ,CAACoB,KAAK,CAAC;UACvBA;QACJ,CAAC,CAAC;MACN;IACJ;IAEA,IAAIS,MAA2B;IAE/B,IAAI;MACA,MAAMC,KAAK,GAAGC,eAAe,CAAC,IAAI,CAAC7B,KAAK,CAAC8B,QAAQ,CAAC,CAAC,CAAC;MACpD;AACZ;AACA;MACYH,MAAM,GAAG,MAAM,IAAI,CAAC9B,OAAO,CAACkC,QAAQ,CAACC,oBAAoB,CAAC,YAAY;QAClE,OAAO,MAAM9B,UAAU,CAACD,GAAG,CAAC;UACxB2B,KAAK;UACL/B,OAAO,EAAE,IAAI,CAACA,OAAO;UACrBC,QAAQ,EAAE,IAAI,CAACC,YAAY;UAC3BkC,gBAAgB,EAAEC,OAAO,IAAI;YACzB,OAAO,IAAI,CAACtC,MAAM,CAACqC,gBAAgB,CAACC,OAAO,CAAC;UAChD,CAAC;UACDC,SAAS,EAAEA,CAAA,KAAM;YACb,OAAO,IAAI,CAACnC,KAAK,CAACG,SAAS,CAAC,CAAC,KAAKC,qBAAc,CAACC,OAAO;UAC5D,CAAC;UACDL,KAAK,EAAE,IAAI,CAACA,KAAK;UACjBoC,OAAO,EAAE,MACLC,MAA6C,IACzB;YACpB,OAAO,IAAI,CAACxC,OAAO,CAACyC,KAAK,CAACF,OAAO,CAAC;cAC9B,GAAGC,MAAM;cACTE,MAAM,EAAE,IAAI,CAACvC,KAAK,CAACmB,OAAO,CAAC;YAC/B,CAAC,CAAC;UACN,CAAC;UACDqB,KAAK,EAAE,IAAI,CAAC5C,MAAM,CAAC4C;QACvB,CAAC,CAAC;MACN,CAAC,CAAC;IACN,CAAC,CAAC,OAAOhB,EAAE,EAAE;MACT,OAAO,IAAI,CAAC1B,QAAQ,CAACoB,KAAK,CAAC;QACvBA,KAAK,EAAE,IAAAQ,sCAAkB,EAACF,EAAE;MAChC,CAAC,CAAC;IACN;IAEA,IAAIG,MAAM,CAACc,MAAM,KAAKC,yBAAkB,CAACC,QAAQ,EAAE;MAC/C,OAAO,IAAI,CAAC7C,QAAQ,CAAC8C,QAAQ,CAAC;QAC1BhB,KAAK,EAAED,MAAM,CAACC,KAAK;QACnBiB,IAAI,EAAElB,MAAM,CAACkB;MACjB,CAAC,CAAC;IACN,CAAC,MAAM,IAAIlB,MAAM,CAACc,MAAM,KAAKC,yBAAkB,CAACI,KAAK,EAAE;MACnD,OAAO,IAAI,CAAChD,QAAQ,CAACoB,KAAK,CAAC;QACvBA,KAAK,EAAES,MAAM,CAACT;MAClB,CAAC,CAAC;IACN,CAAC,MAAM,IAAIS,MAAM,CAACc,MAAM,KAAKC,yBAAkB,CAACrC,OAAO,EAAE;MACrD,OAAO,IAAI,CAACP,QAAQ,CAACQ,OAAO,CAAC,CAAC;IAClC;IACA,OAAO,IAAI,CAACR,QAAQ,CAACiD,IAAI,CAAC;MACtBxB,OAAO,EAAEI,MAAM,CAACJ,OAAO;MACvByB,MAAM,EAAErB,MAAM,CAACqB;IACnB,CAAC,CAAC;EACN;AACJ;AAACC,OAAA,CAAAvD,WAAA,GAAAA,WAAA","ignoreList":[]}
+{"version":3,"names":["TaskDataStatus","getErrorProperties","TaskResultStatus","TaskController","TaskManager","constructor","context","response","store","run","definition","getStatus","ABORTED","aborted","PENDING","updateTask","taskStatus","RUNNING","startedOn","Date","toISOString","executionName","event","iterations","save","error","getTask","maxIterations","onMaxIterations","task","message","ex","data","result","input","structuredClone","getInput","controller","container","resolve","security","withoutAuthorization","status","CONTINUE","continue","wait","ERROR","done","output"],"sources":["TaskManager.ts"],"sourcesContent":["import type { ITaskManager, ITaskManagerStorePrivate } from \"./abstractions/index.js\";\nimport type { Context } from \"~/types.js\";\nimport { TaskDataStatus } from \"~/types.js\";\nimport type { IResponse, IResponseResult } from \"~/response/abstractions/index.js\";\nimport { getErrorProperties } from \"~/utils/getErrorProperties.js\";\nimport {\n    TaskDefinition,\n    TaskResultStatus\n} from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { TaskController } from \"@webiny/api-core/features/task/TaskController/abstractions.js\";\n\nexport class TaskManager implements ITaskManager {\n    private readonly context: Context;\n    private readonly response: IResponse;\n    private readonly store: ITaskManagerStorePrivate;\n\n    public constructor(context: Context, response: IResponse, store: ITaskManagerStorePrivate) {\n        this.context = context;\n        this.response = response;\n        this.store = store;\n    }\n\n    public async run(definition: TaskDefinition.Runnable): Promise<IResponseResult> {\n        /**\n         * If task was aborted, do not run it again, return as it was done.\n         */\n        if (this.store.getStatus() === TaskDataStatus.ABORTED) {\n            return this.response.aborted();\n        }\n        /**\n         * If the task status is pending, update it to running and add a log.\n         */\n        //\n        else if (this.store.getStatus() === TaskDataStatus.PENDING) {\n            try {\n                await this.store.updateTask({\n                    taskStatus: TaskDataStatus.RUNNING,\n                    startedOn: new Date().toISOString(),\n                    executionName: this.response.event.executionName,\n                    iterations: 1\n                });\n                await this.store.save();\n            } catch (error) {\n                return this.response.error({\n                    error\n                });\n            }\n        }\n        /**\n         * We do not want to run the task indefinitely.\n         * If the task has reached the max iterations, we will stop it and execute the onMaxIterations handler, if any.\n         */\n        //\n        else if (this.store.getTask().iterations >= definition.maxIterations) {\n            try {\n                if (definition.onMaxIterations) {\n                    await definition.onMaxIterations({\n                        task: this.store.getTask()\n                    });\n                }\n                return this.response.error({\n                    error: {\n                        message: \"Task reached max iterations.\"\n                    }\n                });\n            } catch (ex) {\n                return this.response.error({\n                    error: {\n                        message: \"Failed to execute onMaxIterations handler.\",\n                        data: getErrorProperties(ex)\n                    }\n                });\n            }\n        }\n        /**\n         * Always update the task iteration.\n         */\n        //\n        else {\n            try {\n                await this.store.updateTask(task => {\n                    return {\n                        iterations: task.iterations + 1\n                    };\n                });\n            } catch (error) {\n                return this.response.error({\n                    error\n                });\n            }\n        }\n\n        let result: TaskDefinition.Result;\n\n        try {\n            const input = structuredClone(this.store.getInput());\n            const controller = this.context.container.resolve(TaskController);\n            /**\n             * We always run the task without authorization because we are running a task without a user - nothing to authorize against.\n             */\n            result = await this.context.security.withoutAuthorization(async () => {\n                return await definition.run({\n                    input,\n                    controller\n                });\n            });\n        } catch (ex) {\n            return this.response.error({\n                error: getErrorProperties(ex)\n            });\n        }\n\n        // Convert TaskDefinition.Result to IResponseResult\n        if (result.status === TaskResultStatus.CONTINUE) {\n            return this.response.continue({\n                input: result.input,\n                wait: result.wait\n            });\n        } else if (result.status === TaskResultStatus.ERROR) {\n            return this.response.error({\n                error: result.error\n            });\n        } else if (result.status === TaskResultStatus.ABORTED) {\n            return this.response.aborted();\n        }\n        return this.response.done({\n            message: result.message,\n            output: result.output\n        });\n    }\n}\n"],"mappings":"AAEA,SAASA,cAAc;AAEvB,SAASC,kBAAkB;AAC3B,SAEIC,gBAAgB,QACb,wDAAwD;AAC/D,SAASC,cAAc,QAAQ,+DAA+D;AAE9F,OAAO,MAAMC,WAAW,CAAyB;EAKtCC,WAAWA,CAACC,OAAgB,EAAEC,QAAmB,EAAEC,KAA+B,EAAE;IACvF,IAAI,CAACF,OAAO,GAAGA,OAAO;IACtB,IAAI,CAACC,QAAQ,GAAGA,QAAQ;IACxB,IAAI,CAACC,KAAK,GAAGA,KAAK;EACtB;EAEA,MAAaC,GAAGA,CAACC,UAAmC,EAA4B;IAC5E;AACR;AACA;IACQ,IAAI,IAAI,CAACF,KAAK,CAACG,SAAS,CAAC,CAAC,KAAKX,cAAc,CAACY,OAAO,EAAE;MACnD,OAAO,IAAI,CAACL,QAAQ,CAACM,OAAO,CAAC,CAAC;IAClC;IACA;AACR;AACA;IACQ;IAAA,KACK,IAAI,IAAI,CAACL,KAAK,CAACG,SAAS,CAAC,CAAC,KAAKX,cAAc,CAACc,OAAO,EAAE;MACxD,IAAI;QACA,MAAM,IAAI,CAACN,KAAK,CAACO,UAAU,CAAC;UACxBC,UAAU,EAAEhB,cAAc,CAACiB,OAAO;UAClCC,SAAS,EAAE,IAAIC,IAAI,CAAC,CAAC,CAACC,WAAW,CAAC,CAAC;UACnCC,aAAa,EAAE,IAAI,CAACd,QAAQ,CAACe,KAAK,CAACD,aAAa;UAChDE,UAAU,EAAE;QAChB,CAAC,CAAC;QACF,MAAM,IAAI,CAACf,KAAK,CAACgB,IAAI,CAAC,CAAC;MAC3B,CAAC,CAAC,OAAOC,KAAK,EAAE;QACZ,OAAO,IAAI,CAAClB,QAAQ,CAACkB,KAAK,CAAC;UACvBA;QACJ,CAAC,CAAC;MACN;IACJ;IACA;AACR;AACA;AACA;IACQ;IAAA,KACK,IAAI,IAAI,CAACjB,KAAK,CAACkB,OAAO,CAAC,CAAC,CAACH,UAAU,IAAIb,UAAU,CAACiB,aAAa,EAAE;MAClE,IAAI;QACA,IAAIjB,UAAU,CAACkB,eAAe,EAAE;UAC5B,MAAMlB,UAAU,CAACkB,eAAe,CAAC;YAC7BC,IAAI,EAAE,IAAI,CAACrB,KAAK,CAACkB,OAAO,CAAC;UAC7B,CAAC,CAAC;QACN;QACA,OAAO,IAAI,CAACnB,QAAQ,CAACkB,KAAK,CAAC;UACvBA,KAAK,EAAE;YACHK,OAAO,EAAE;UACb;QACJ,CAAC,CAAC;MACN,CAAC,CAAC,OAAOC,EAAE,EAAE;QACT,OAAO,IAAI,CAACxB,QAAQ,CAACkB,KAAK,CAAC;UACvBA,KAAK,EAAE;YACHK,OAAO,EAAE,4CAA4C;YACrDE,IAAI,EAAE/B,kBAAkB,CAAC8B,EAAE;UAC/B;QACJ,CAAC,CAAC;MACN;IACJ;IACA;AACR;AACA;IACQ;IAAA,KACK;MACD,IAAI;QACA,MAAM,IAAI,CAACvB,KAAK,CAACO,UAAU,CAACc,IAAI,IAAI;UAChC,OAAO;YACHN,UAAU,EAAEM,IAAI,CAACN,UAAU,GAAG;UAClC,CAAC;QACL,CAAC,CAAC;MACN,CAAC,CAAC,OAAOE,KAAK,EAAE;QACZ,OAAO,IAAI,CAAClB,QAAQ,CAACkB,KAAK,CAAC;UACvBA;QACJ,CAAC,CAAC;MACN;IACJ;IAEA,IAAIQ,MAA6B;IAEjC,IAAI;MACA,MAAMC,KAAK,GAAGC,eAAe,CAAC,IAAI,CAAC3B,KAAK,CAAC4B,QAAQ,CAAC,CAAC,CAAC;MACpD,MAAMC,UAAU,GAAG,IAAI,CAAC/B,OAAO,CAACgC,SAAS,CAACC,OAAO,CAACpC,cAAc,CAAC;MACjE;AACZ;AACA;MACY8B,MAAM,GAAG,MAAM,IAAI,CAAC3B,OAAO,CAACkC,QAAQ,CAACC,oBAAoB,CAAC,YAAY;QAClE,OAAO,MAAM/B,UAAU,CAACD,GAAG,CAAC;UACxByB,KAAK;UACLG;QACJ,CAAC,CAAC;MACN,CAAC,CAAC;IACN,CAAC,CAAC,OAAON,EAAE,EAAE;MACT,OAAO,IAAI,CAACxB,QAAQ,CAACkB,KAAK,CAAC;QACvBA,KAAK,EAAExB,kBAAkB,CAAC8B,EAAE;MAChC,CAAC,CAAC;IACN;;IAEA;IACA,IAAIE,MAAM,CAACS,MAAM,KAAKxC,gBAAgB,CAACyC,QAAQ,EAAE;MAC7C,OAAO,IAAI,CAACpC,QAAQ,CAACqC,QAAQ,CAAC;QAC1BV,KAAK,EAAED,MAAM,CAACC,KAAK;QACnBW,IAAI,EAAEZ,MAAM,CAACY;MACjB,CAAC,CAAC;IACN,CAAC,MAAM,IAAIZ,MAAM,CAACS,MAAM,KAAKxC,gBAAgB,CAAC4C,KAAK,EAAE;MACjD,OAAO,IAAI,CAACvC,QAAQ,CAACkB,KAAK,CAAC;QACvBA,KAAK,EAAEQ,MAAM,CAACR;MAClB,CAAC,CAAC;IACN,CAAC,MAAM,IAAIQ,MAAM,CAACS,MAAM,KAAKxC,gBAAgB,CAACU,OAAO,EAAE;MACnD,OAAO,IAAI,CAACL,QAAQ,CAACM,OAAO,CAAC,CAAC;IAClC;IACA,OAAO,IAAI,CAACN,QAAQ,CAACwC,IAAI,CAAC;MACtBjB,OAAO,EAAEG,MAAM,CAACH,OAAO;MACvBkB,MAAM,EAAEf,MAAM,CAACe;IACnB,CAAC,CAAC;EACN;AACJ","ignoreList":[]}

package/runner/TaskManagerStore.d.ts

@@ -1,6 +1,6 @@
-import type { ITask, ITaskDataInput, ITaskLog, ITaskManagerStoreInfoLog, ITaskManagerStorePrivate, ITaskManagerStoreSetOutputOptions, ITaskManagerStoreUpdateTaskInputOptions, ITaskManagerStoreUpdateTaskOptions, ITaskResponseDoneResultOutput, ITasksContextObject, TaskDataStatus } from "../types";
-import type { ITaskManagerStoreAddLogOptions, ITaskManagerStoreErrorLog, ITaskManagerStoreUpdateTaskInputParam, ITaskManagerStoreUpdateTaskParams } from "./abstractions";
-import type { GenericRecord } from "@webiny/api/types";
+import type { ITask, ITaskDataInput, ITaskLog, ITaskManagerStoreInfoLog, ITaskManagerStorePrivate, ITaskManagerStoreSetOutputOptions, ITaskManagerStoreUpdateTaskInputOptions, ITaskManagerStoreUpdateTaskOptions, ITasksContextObject, TaskDataStatus } from "../types.js";
+import type { ITaskManagerStoreAddLogOptions, ITaskManagerStoreErrorLog, ITaskManagerStoreUpdateTaskInputParam, ITaskManagerStoreUpdateTaskParams } from "./abstractions/index.js";
+import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 export interface TaskManagerStoreContext {
     tasks: Pick<ITasksContextObject, "updateTask" | "updateLog" | "listTasks">;
 }
@@ -8,19 +8,19 @@ export interface ITaskManagerStoreParams {
     context: TaskManagerStoreContext;
     task: ITask;
     log: ITaskLog;
-    disableDatabaseLogs?: boolean;
+    databaseLogs: boolean;
 }
-export declare class TaskManagerStore<T extends ITaskDataInput = ITaskDataInput, O extends ITaskResponseDoneResultOutput = ITaskResponseDoneResultOutput> implements ITaskManagerStorePrivate<T, O> {
+export declare class TaskManagerStore<T extends ITaskDataInput = ITaskDataInput, O extends TaskDefinition.TaskOutput = TaskDefinition.TaskOutput> implements ITaskManagerStorePrivate<T, O> {
     private readonly context;
     private task;
     private taskLog;
-    private readonly disableDatabaseLogs;
+    private readonly databaseLogs;
     private readonly taskUpdater;
     private readonly taskLogUpdater;
     constructor(params: ITaskManagerStoreParams);
     getStatus(): TaskDataStatus;
     getTask(): ITask<T, O>;
-    listChildTasks<I = GenericRecord, O extends ITaskResponseDoneResultOutput = ITaskResponseDoneResultOutput>(definitionId?: string): Promise<ITask<I, O>[]>;
+    listChildTasks<I extends TaskDefinition.TaskInput = TaskDefinition.TaskInput, O extends TaskDefinition.TaskOutput = TaskDefinition.TaskOutput>(definitionId?: string): Promise<ITask<I, O>[]>;
     updateTask(param: ITaskManagerStoreUpdateTaskParams<T, O>, options?: ITaskManagerStoreUpdateTaskOptions): Promise<void>;
     updateInput(param: ITaskManagerStoreUpdateTaskInputParam<T>, options?: ITaskManagerStoreUpdateTaskInputOptions): Promise<void>;
     getInput(): T;