@webiny/tasks 0.0.0-unstable.06b2ede40f

package/runner/TaskControl.js

@@ -0,0 +1,222 @@
+"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 {
+  constructor(runner, response, context) {
+    this.runner = runner;
+    this.context = context;
+    this.response = response;
+  }
+  async run(event) {
+    const taskId = event.webinyTaskId;
+    /**
+     * This is the initial getTask idea.
+     * We will need to take care of child tasks:
+     * * child tasks can be in multiple levels (child task creates a child task, etc...).
+     * * child tasks could be executed in parallel.
+     */
+    let task;
+    try {
+      task = await this.getTask(taskId);
+      this.context.security.setIdentity(task.createdBy);
+    } catch (error) {
+      /**
+       * TODO Refactor error handling.
+       */
+      // @ts-expect-error
+      return this.response.error({
+        ...(0, _getErrorProperties.getErrorProperties)(error)
+      });
+    }
+    /**
+     * Let's get the task definition.
+     */
+    const definition = this.context.tasks.getDefinition(task.definitionId);
+    if (!definition) {
+      return this.response.error({
+        error: {
+          message: `Task "${task.id}" cannot be executed because there is no "${task.definitionId}" definition plugin.`,
+          code: "TASK_DEFINITION_ERROR",
+          data: {
+            definitionId: task.definitionId
+          }
+        }
+      });
+    }
+    const disableDatabaseLogs = !!definition.disableDatabaseLogs;
+
+    /**
+     * 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);
+    } catch (error) {
+      return this.response.error({
+        error
+      });
+    }
+    /**
+     * 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) {
+      return this.response.aborted();
+    }
+    /**
+     * Do not run if already a success (done).
+     */
+    //
+    else if (task.taskStatus === _types.TaskDataStatus.SUCCESS) {
+      return this.response.error({
+        error: {
+          message: "Task is already done, cannot run it again."
+        }
+      });
+    }
+    /**
+     * Do not run if already failed.
+     */
+    //
+    else if (task.taskStatus === _types.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({
+      context: this.context,
+      task,
+      log: taskLog,
+      disableDatabaseLogs
+    });
+    const manager = new _TaskManager.TaskManager(this.runner, this.context, this.response, taskResponse, store);
+    const databaseResponse = new _response.DatabaseResponse(this.response, store);
+    try {
+      const result = await manager.run(definition);
+      await this.runEvents(result, definition, task);
+      return await databaseResponse.from(result);
+    } catch (ex) {
+      /**
+       * We always want to store the error in the task log.
+       */
+      return await databaseResponse.from(this.response.error({
+        error: {
+          message: ex.message,
+          code: ex.code || "TASK_ERROR",
+          stack: ex.stack,
+          data: {
+            ...ex.data,
+            input: task.input
+          }
+        }
+      }));
+    }
+  }
+  async runEvents(result, definition, task) {
+    if (result.status === _types.TaskResponseStatus.ERROR && definition.onError) {
+      try {
+        await definition.onError({
+          task,
+          context: this.context
+        });
+      } catch (ex) {
+        console.error(`Error executing onError hook for task "${task.id}".`);
+        console.log((0, _getErrorProperties.getErrorProperties)(ex));
+      }
+    } else if (result.status === _types.TaskResponseStatus.DONE && definition.onDone) {
+      try {
+        await definition.onDone({
+          task,
+          context: this.context
+        });
+      } catch (ex) {
+        console.error(`Error executing onDone hook for task "${task.id}".`);
+        console.log((0, _getErrorProperties.getErrorProperties)(ex));
+      }
+    }
+  }
+  async getTask(id) {
+    try {
+      const task = await this.runner.context.tasks.getTask(id);
+      if (task) {
+        return task;
+      }
+    } catch (ex) {
+      throw this.response.error({
+        error: {
+          message: ex.message,
+          code: ex.code || "TASK_ERROR",
+          stack: ex.stack,
+          data: ex.data
+        }
+      });
+    }
+    throw this.response.error({
+      error: {
+        message: `Task "${id}" cannot be executed because it does not exist.`,
+        code: "TASK_NOT_FOUND"
+      }
+    });
+  }
+  async getTaskLog(task, disableDatabaseLogs) {
+    /**
+     * If logs are disabled, let's return a mocked one.
+     */
+    if (disableDatabaseLogs) {
+      return {
+        id: `${task.id}-log`,
+        createdOn: task.createdOn,
+        createdBy: task.createdBy,
+        executionName: task.executionName,
+        task: task.id,
+        iteration: task.iterations,
+        items: []
+      };
+    }
+    let taskLog = null;
+    /**
+     * First we are trying to get existing latest log.
+     */
+    try {
+      taskLog = await this.context.tasks.getLatestLog(task.id);
+    } catch (error) {
+      /**
+       * If error is not the NotFoundError, we need to throw it.
+       */
+      if (error instanceof _handlerGraphql.NotFoundError === false) {
+        throw this.response.error({
+          error
+        });
+      }
+      /**
+       * Otherwise just continue and create a new log.
+       */
+    }
+    const currentIteration = taskLog?.iteration || 0;
+    try {
+      return await this.context.tasks.createLog(task, {
+        executionName: this.response.event.executionName,
+        iteration: currentIteration + 1
+      });
+    } catch (error) {
+      throw this.response.error({
+        error
+      });
+    }
+  }
+}
+exports.TaskControl = TaskControl;
+
+//# sourceMappingURL=TaskControl.js.map

package/runner/TaskControl.js.map

@@ -0,0 +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":[]}

package/runner/TaskEventValidation.d.ts

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

package/runner/TaskEventValidation.js

@@ -0,0 +1,30 @@
+"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()
+}).required();
+class TaskEventValidation {
+  validate(event) {
+    const result = validation.safeParse(event);
+    if (result.success) {
+      return result.data;
+    }
+    throw (0, _utils.createZodError)(result.error);
+  }
+}
+exports.TaskEventValidation = TaskEventValidation;
+
+//# sourceMappingURL=TaskEventValidation.js.map

package/runner/TaskEventValidation.js.map

@@ -0,0 +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":[]}

package/runner/TaskManager.d.ts

@@ -0,0 +1,14 @@
+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;
+    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>;
+}
+export {};

package/runner/TaskManager.js

@@ -0,0 +1,139 @@
+"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;
+    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) {
+      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) {
+      try {
+        await this.store.updateTask({
+          taskStatus: _types.TaskDataStatus.RUNNING,
+          startedOn: new Date().toISOString(),
+          executionName: this.response.event.executionName,
+          iterations: 1
+        });
+        await this.store.save();
+      } catch (error) {
+        return this.response.error({
+          error
+        });
+      }
+    }
+    /**
+     * We do not want to run the task indefinitely.
+     * If the task has reached the max iterations, we will stop it and execute the onMaxIterations handler, if any.
+     */
+    //
+    else if (this.store.getTask().iterations >= definition.maxIterations) {
+      try {
+        if (definition.onMaxIterations) {
+          await definition.onMaxIterations({
+            task: this.store.getTask(),
+            context: this.context
+          });
+        }
+        return this.response.error({
+          error: {
+            message: "Task reached max iterations."
+          }
+        });
+      } catch (ex) {
+        return this.response.error({
+          error: {
+            message: "Failed to execute onMaxIterations handler.",
+            data: (0, _getErrorProperties.getErrorProperties)(ex)
+          }
+        });
+      }
+    }
+    /**
+     * Always update the task iteration.
+     */
+    //
+    else {
+      try {
+        await this.store.updateTask(task => {
+          return {
+            iterations: task.iterations + 1
+          };
+        });
+      } catch (error) {
+        return this.response.error({
+          error
+        });
+      }
+    }
+    let result;
+    try {
+      const input = structuredClone(this.store.getInput());
+      /**
+       * 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
+        });
+      });
+    } catch (ex) {
+      return this.response.error({
+        error: (0, _getErrorProperties.getErrorProperties)(ex)
+      });
+    }
+    if (result.status === _types.TaskResponseStatus.CONTINUE) {
+      return this.response.continue({
+        input: result.input,
+        wait: result.wait
+      });
+    } else if (result.status === _types.TaskResponseStatus.ERROR) {
+      return this.response.error({
+        error: result.error
+      });
+    } else if (result.status === _types.TaskResponseStatus.ABORTED) {
+      return this.response.aborted();
+    }
+    return this.response.done({
+      message: result.message,
+      output: result.output
+    });
+  }
+}
+exports.TaskManager = TaskManager;
+
+//# sourceMappingURL=TaskManager.js.map

package/runner/TaskManager.js.map

@@ -0,0 +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":[]}

package/runner/TaskManagerStore.d.ts

@@ -0,0 +1,40 @@
+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";
+export interface TaskManagerStoreContext {
+    tasks: Pick<ITasksContextObject, "updateTask" | "updateLog" | "listTasks">;
+}
+export interface ITaskManagerStoreParams {
+    context: TaskManagerStoreContext;
+    task: ITask;
+    log: ITaskLog;
+    disableDatabaseLogs?: boolean;
+}
+export declare class TaskManagerStore<T extends ITaskDataInput = ITaskDataInput, O extends ITaskResponseDoneResultOutput = ITaskResponseDoneResultOutput> implements ITaskManagerStorePrivate<T, O> {
+    private readonly context;
+    private task;
+    private taskLog;
+    private readonly disableDatabaseLogs;
+    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>[]>;
+    updateTask(param: ITaskManagerStoreUpdateTaskParams<T, O>, options?: ITaskManagerStoreUpdateTaskOptions): Promise<void>;
+    updateInput(param: ITaskManagerStoreUpdateTaskInputParam<T>, options?: ITaskManagerStoreUpdateTaskInputOptions): Promise<void>;
+    getInput(): T;
+    updateOutput(values: Partial<O>, options?: ITaskManagerStoreSetOutputOptions): Promise<void>;
+    getOutput(): O;
+    /**
+     * Currently the methods throws an error if something goes wrong during the database update.
+     * TODO: Maybe we should wrap it into try/catch and return error if any?
+     */
+    addInfoLog(log: ITaskManagerStoreInfoLog, options?: ITaskManagerStoreAddLogOptions): Promise<void>;
+    /**
+     * Currently the methods throws an error if something goes wrong during the database update.
+     * TODO: Maybe we should wrap it into try/catch and return error if any?
+     */
+    addErrorLog(log: ITaskManagerStoreErrorLog, options?: ITaskManagerStoreAddLogOptions): Promise<void>;
+    save(): Promise<void>;
+}