@webiny/background-tasks 0.0.0-unstable.7be00a75a9 → 0.0.0-unstable.b6d7105cee

package/api/response/abstractions/TaskResponse.d.ts

@@ -0,0 +1,16 @@
+import type { IResponseError } from "./ResponseErrorResult.js";
+import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+export interface ITaskResponseContinueOptionsUntil {
+    date: Date;
+}
+export interface ITaskResponseContinueOptionsSeconds {
+    seconds: number;
+}
+export type ITaskResponseContinueOptions = ITaskResponseContinueOptionsUntil | ITaskResponseContinueOptionsSeconds;
+export interface ITaskResponse<I extends TaskDefinition.TaskInput = TaskDefinition.TaskInput, O extends TaskDefinition.TaskOutput = TaskDefinition.TaskOutput> {
+    done(output?: O): TaskDefinition.ResultDone<O>;
+    done(message?: string, output?: O): TaskDefinition.ResultDone<O>;
+    continue(data: I, options?: ITaskResponseContinueOptions): TaskDefinition.ResultContinue<I>;
+    error(error: IResponseError | Error | string): TaskDefinition.ResultError;
+    aborted(): TaskDefinition.ResultAborted;
+}

package/api/response/abstractions/index.d.ts

@@ -0,0 +1,7 @@
+export * from "./Response.js";
+export * from "./ResponseContinueResult.js";
+export * from "./ResponseDoneResult.js";
+export * from "./ResponseErrorResult.js";
+export * from "./ResponseAbortedResult.js";
+export * from "./ResponseBaseResult.js";
+export * from "./TaskResponse.js";

package/api/response/abstractions/index.js

@@ -0,0 +1,7 @@
+export * from "./Response.js";
+export * from "./ResponseContinueResult.js";
+export * from "./ResponseDoneResult.js";
+export * from "./ResponseErrorResult.js";
+export * from "./ResponseAbortedResult.js";
+export * from "./ResponseBaseResult.js";
+export * from "./TaskResponse.js";

package/api/response/index.d.ts

@@ -0,0 +1,7 @@
+export * from "./DatabaseResponse.js";
+export * from "./Response.js";
+export * from "./ResponseContinueResult.js";
+export * from "./ResponseDoneResult.js";
+export * from "./ResponseAbortedResult.js";
+export * from "./ResponseErrorResult.js";
+export * from "./TaskResponse.js";

package/api/response/index.js

@@ -0,0 +1,7 @@
+export * from "./DatabaseResponse.js";
+export * from "./Response.js";
+export * from "./ResponseContinueResult.js";
+export * from "./ResponseDoneResult.js";
+export * from "./ResponseAbortedResult.js";
+export * from "./ResponseErrorResult.js";
+export * from "./TaskResponse.js";

package/api/runner/TaskControl.d.ts

@@ -0,0 +1,14 @@
+import type { ITaskEvent } from "../../api/handler/types.js";
+import type { Context } from "../../api/types.js";
+import type { ITaskControl, ITaskRunner } from "./abstractions/index.js";
+import type { IResponse, IResponseResult } from "../../api/response/abstractions/index.js";
+export declare class TaskControl implements ITaskControl {
+    readonly runner: ITaskRunner;
+    readonly response: IResponse;
+    readonly context: Context;
+    constructor(runner: ITaskRunner, response: IResponse, context: Context);
+    run(event: Pick<ITaskEvent, "webinyTaskId">): Promise<IResponseResult>;
+    private runEvents;
+    private getTask;
+    private getTaskLog;
+}

package/api/runner/TaskControl.js

@@ -0,0 +1,173 @@
+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";
+class TaskControl {
+    constructor(runner, response, context){
+        this.runner = runner;
+        this.context = context;
+        this.response = response;
+    }
+    async run(event) {
+        const taskId = event.webinyTaskId;
+        let task;
+        try {
+            task = await this.getTask(taskId);
+            this.context.security.setIdentity(new AuthenticatedIdentity({
+                id: task.createdBy.id,
+                type: task.createdBy.type,
+                displayName: task.createdBy.displayName ?? "",
+                context: {
+                    canAccessTenant: true
+                }
+            }));
+        } catch (error) {
+            return this.response.error({
+                ...getErrorProperties(error)
+            });
+        }
+        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 databaseLogs = true === definition.databaseLogs;
+        let taskLog;
+        try {
+            taskLog = await this.getTaskLog({
+                task,
+                databaseLogs
+            });
+        } catch (error) {
+            return this.response.error({
+                error
+            });
+        }
+        if (task.taskStatus === TaskDataStatus.ABORTED) return this.response.aborted();
+        if (task.taskStatus === TaskDataStatus.SUCCESS) return this.response.error({
+            error: {
+                message: "Task is already done, cannot run it again."
+            }
+        });
+        if (task.taskStatus === TaskDataStatus.FAILED) return this.response.error({
+            error: {
+                message: "Task has failed, cannot run it again."
+            }
+        });
+        const store = new TaskManagerStore({
+            context: this.context,
+            task,
+            log: taskLog,
+            databaseLogs
+        });
+        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);
+            const responseResult = await databaseResponse.from(result);
+            await this.runEvents(result, definition, store.getTask());
+            return responseResult;
+        } catch (ex) {
+            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
+                    }
+                }
+            }));
+        } finally{
+            executionContext.clear();
+        }
+    }
+    async runEvents(result, definition, task) {
+        if (result.status === TaskResultStatus.ERROR && definition.onError) try {
+            await definition.onError({
+                task
+            });
+        } catch (ex) {
+            console.error(`Error executing onError hook for task "${task.id}".`);
+            console.log(getErrorProperties(ex));
+        }
+        else if (result.status === TaskResultStatus.DONE && definition.onDone) try {
+            await definition.onDone({
+                task
+            });
+        } catch (ex) {
+            console.error(`Error executing onDone hook for task "${task.id}".`);
+            console.log(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(params) {
+        const { task, databaseLogs } = params;
+        if (!databaseLogs) return {
+            id: `${task.id}-log`,
+            createdOn: task.createdOn,
+            createdBy: task.createdBy,
+            executionName: task.executionName,
+            task: task.id,
+            iteration: task.iterations,
+            items: []
+        };
+        let taskLog = null;
+        try {
+            taskLog = await this.context.tasks.getLatestLog(task.id);
+        } catch (error) {
+            if ("NOT_FOUND" !== error.code) throw this.response.error({
+                error
+            });
+        }
+        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
+            });
+        }
+    }
+}
+export { TaskControl };
+
+//# sourceMappingURL=TaskControl.js.map

package/api/runner/TaskControl.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/runner/TaskControl.js","sources":["../../../src/api/runner/TaskControl.ts"],"sourcesContent":["import type { ITaskEvent } from \"~/api/handler/types.js\";\nimport type { Context, ITask, ITaskDataInput, ITaskLog } from \"~/api/types.js\";\nimport { TaskDataStatus } from \"~/api/types.js\";\nimport type { ITaskControl, ITaskRunner } from \"./abstractions/index.js\";\nimport { TaskManager } from \"./TaskManager.js\";\nimport type { IResponse, IResponseResult } from \"~/api/response/abstractions/index.js\";\nimport { DatabaseResponse, TaskResponse } from \"~/api/response/index.js\";\nimport { TaskManagerStore } from \"./TaskManagerStore.js\";\nimport { getErrorProperties } from \"~/api/utils/getErrorProperties.js\";\nimport { AuthenticatedIdentity } from \"@webiny/api-core/features/security/IdentityContext/index.js\";\nimport { TaskExecutionContext } from \"~/api/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"],"names":["TaskControl","runner","response","context","event","taskId","task","AuthenticatedIdentity","error","getErrorProperties","definition","databaseLogs","taskLog","TaskDataStatus","store","TaskManagerStore","executionContext","TaskExecutionContext","TaskResponse","manager","TaskManager","databaseResponse","DatabaseResponse","result","responseResult","ex","TaskResultStatus","console","id","params","currentIteration"],"mappings":";;;;;;;;AAqBO,MAAMA;IAKT,YAAmBC,MAAmB,EAAEC,QAAmB,EAAEC,OAAgB,CAAE;QAC3E,IAAI,CAAC,MAAM,GAAGF;QACd,IAAI,CAAC,OAAO,GAAGE;QACf,IAAI,CAAC,QAAQ,GAAGD;IACpB;IAEA,MAAa,IAAIE,KAAuC,EAA4B;QAChF,MAAMC,SAASD,MAAM,YAAY;QAOjC,IAAIE;QACJ,IAAI;YACAA,OAAO,MAAM,IAAI,CAAC,OAAO,CAACD;YAC1B,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAC7B,IAAIE,sBAAsB;gBACtB,IAAID,KAAK,SAAS,CAAC,EAAE;gBACrB,MAAMA,KAAK,SAAS,CAAC,IAAI;gBACzB,aAAaA,KAAK,SAAS,CAAC,WAAW,IAAI;gBAC3C,SAAS;oBACL,iBAAiB;gBACrB;YACJ;QAER,EAAE,OAAOE,OAAO;YAKZ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvB,GAAGC,mBAAmBD,MAAM;YAChC;QACJ;QAIA,MAAME,aAAa,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,aAAa,CAACJ,KAAK,YAAY;QACrE,IAAI,CAACI,YACD,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YACvB,OAAO;gBACH,SAAS,CAAC,MAAM,EAAEJ,KAAK,EAAE,CAAC,0CAA0C,EAAEA,KAAK,YAAY,CAAC,oBAAoB,CAAC;gBAC7G,MAAM;gBACN,MAAM;oBACF,cAAcA,KAAK,YAAY;gBACnC;YACJ;QACJ;QAKJ,MAAMK,eAAeD,AAA4B,SAA5BA,WAAW,YAAY;QAK5C,IAAIE;QACJ,IAAI;YACAA,UAAU,MAAM,IAAI,CAAC,UAAU,CAAC;gBAC5BN;gBACAK;YACJ;QACJ,EAAE,OAAOH,OAAO;YACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvBA;YACJ;QACJ;QAKA,IAAIF,KAAK,UAAU,KAAKO,eAAe,OAAO,EAC1C,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO;QAM3B,IAAIP,KAAK,UAAU,KAAKO,eAAe,OAAO,EAC/C,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YACvB,OAAO;gBACH,SAAS;YACb;QACJ;QAMC,IAAIP,KAAK,UAAU,KAAKO,eAAe,MAAM,EAC9C,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YACvB,OAAO;gBACH,SAAS;YACb;QACJ;QAGJ,MAAMC,QAAQ,IAAIC,iBAAiB;YAC/B,SAAS,IAAI,CAAC,OAAO;YACrBT;YACA,KAAKM;YACLD;QACJ;QAGA,MAAMK,mBAAmB,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACC;QACxDD,iBAAiB,QAAQ,CAACF;QAC1BE,iBAAiB,SAAS,CAAC,IAAI,CAAC,MAAM;QACtCA,iBAAiB,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK;QAC3CA,iBAAiB,WAAW,CAAC,IAAIE,aAAa,IAAI,CAAC,QAAQ;QAE3D,MAAMC,UAAU,IAAIC,YAAY,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAEN;QAE7D,MAAMO,mBAAmB,IAAIC,iBAAiB,IAAI,CAAC,QAAQ,EAAER;QAE7D,IAAI;YACA,MAAMS,SAAS,MAAMJ,QAAQ,GAAG,CAACT;YAEjC,MAAMc,iBAAiB,MAAMH,iBAAiB,IAAI,CAACE;YAGnD,MAAM,IAAI,CAAC,SAAS,CAACA,QAAQb,YAAYI,MAAM,OAAO;YAEtD,OAAOU;QACX,EAAE,OAAOC,IAAI;YAIT,OAAO,MAAMJ,iBAAiB,IAAI,CAC9B,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBAChB,OAAO;oBACH,SAASI,GAAG,OAAO;oBACnB,MAAMA,GAAG,IAAI,IAAI;oBACjB,OAAOA,GAAG,KAAK;oBACf,MAAM;wBACF,GAAGA,GAAG,IAAI;wBACV,OAAOnB,KAAK,KAAK;oBACrB;gBACJ;YACJ;QAER,SAAU;YAENU,iBAAiB,KAAK;QAC1B;IACJ;IAEA,MAAc,UACVO,MAAuB,EACvBb,UAAmC,EACnCJ,IAAW,EACE;QACb,IAAIiB,OAAO,MAAM,KAAKG,iBAAiB,KAAK,IAAIhB,WAAW,OAAO,EAC9D,IAAI;YACA,MAAMA,WAAW,OAAO,CAAC;gBAAEJ;YAAK;QACpC,EAAE,OAAOmB,IAAI;YACTE,QAAQ,KAAK,CAAC,CAAC,uCAAuC,EAAErB,KAAK,EAAE,CAAC,EAAE,CAAC;YACnEqB,QAAQ,GAAG,CAAClB,mBAAmBgB;QACnC;aACG,IAAIF,OAAO,MAAM,KAAKG,iBAAiB,IAAI,IAAIhB,WAAW,MAAM,EACnE,IAAI;YACA,MAAMA,WAAW,MAAM,CAAC;gBAAEJ;YAAK;QACnC,EAAE,OAAOmB,IAAI;YACTE,QAAQ,KAAK,CAAC,CAAC,sCAAsC,EAAErB,KAAK,EAAE,CAAC,EAAE,CAAC;YAClEqB,QAAQ,GAAG,CAAClB,mBAAmBgB;QACnC;IAER;IAEA,MAAc,QAA4CG,EAAU,EAAqB;QACrF,IAAI;YACA,MAAMtB,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAIsB;YACxD,IAAItB,MACA,OAAOA;QAEf,EAAE,OAAOmB,IAAI;YACT,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACtB,OAAO;oBACH,SAASA,GAAG,OAAO;oBACnB,MAAMA,GAAG,IAAI,IAAI;oBACjB,OAAOA,GAAG,KAAK;oBACf,MAAMA,GAAG,IAAI;gBACjB;YACJ;QACJ;QACA,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YACtB,OAAO;gBACH,SAAS,CAAC,MAAM,EAAEG,GAAG,+CAA+C,CAAC;gBACrE,MAAM;YACV;QACJ;IACJ;IAEA,MAAc,WAAWC,MAAyB,EAAqB;QACnE,MAAM,EAAEvB,IAAI,EAAEK,YAAY,EAAE,GAAGkB;QAI/B,IAAI,CAAClB,cACD,OAAO;YACH,IAAI,GAAGL,KAAK,EAAE,CAAC,IAAI,CAAC;YACpB,WAAWA,KAAK,SAAS;YACzB,WAAWA,KAAK,SAAS;YACzB,eAAeA,KAAK,aAAa;YACjC,MAAMA,KAAK,EAAE;YACb,WAAWA,KAAK,UAAU;YAC1B,OAAO,EAAE;QACb;QAEJ,IAAIM,UAA2B;QAI/B,IAAI;YACAA,UAAU,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,YAAY,CAACN,KAAK,EAAE;QAC3D,EAAE,OAAOE,OAAO;YAIZ,IAAIA,AAAe,gBAAfA,MAAM,IAAI,EACV,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACtBA;YACJ;QAKR;QAEA,MAAMsB,mBAAmBlB,SAAS,aAAa;QAE/C,IAAI;YACA,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAACN,MAAM;gBAC5C,eAAe,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,aAAa;gBAChD,WAAWwB,mBAAmB;YAClC;QACJ,EAAE,OAAOtB,OAAO;YACZ,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACtBA;YACJ;QACJ;IACJ;AACJ"}

package/api/runner/TaskEventValidation.d.ts

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

package/api/runner/TaskEventValidation.js

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

package/api/runner/TaskEventValidation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/runner/TaskEventValidation.js","sources":["../../../src/api/runner/TaskEventValidation.ts"],"sourcesContent":["import zod from \"zod\";\nimport { createZodError } from \"@webiny/utils\";\nimport type { ITaskEventValidation, ITaskEventValidationResult } from \"./abstractions/index.js\";\nimport type { ITaskEvent } from \"~/api/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"],"names":["validation","zod","TaskEventValidation","event","result","createZodError"],"mappings":";;AAKA,MAAMA,aAAaC,IAAAA,MACR,CAAC;IACJ,cAAcA,IAAI,MAAM;IACxB,wBAAwBA,IAAI,MAAM;IAClC,UAAUA,IAAI,MAAM;IACpB,QAAQA,IAAI,MAAM;IAClB,eAAeA,IAAI,MAAM;IACzB,gBAAgBA,IAAI,MAAM;AAC9B,GACC,QAAQ;AAEN,MAAMC;IACF,SAASC,KAA0B,EAA8B;QACpE,MAAMC,SAASJ,WAAW,SAAS,CAACG;QACpC,IAAIC,OAAO,OAAO,EACd,OAAOA,OAAO,IAAI;QAEtB,MAAMC,eAAeD,OAAO,KAAK;IACrC;AACJ"}

package/api/runner/TaskManager.d.ts

@@ -0,0 +1,11 @@
+import type { ITaskManager, ITaskManagerStorePrivate } from "./abstractions/index.js";
+import type { Context } from "../../api/types.js";
+import type { IResponse, IResponseResult } from "../../api/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 store;
+    constructor(context: Context, response: IResponse, store: ITaskManagerStorePrivate);
+    run(definition: TaskDefinition.Runnable): Promise<IResponseResult>;
+}

package/api/runner/TaskManager.js

@@ -0,0 +1,81 @@
+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";
+class TaskManager {
+    constructor(context, response, store){
+        this.context = context;
+        this.response = response;
+        this.store = store;
+    }
+    async run(definition) {
+        if (this.store.getStatus() === TaskDataStatus.ABORTED) return this.response.aborted();
+        if (this.store.getStatus() === TaskDataStatus.PENDING) try {
+            await this.store.updateTask({
+                taskStatus: TaskDataStatus.RUNNING,
+                startedOn: new Date().toISOString(),
+                executionName: this.response.event.executionName,
+                iterations: 1
+            });
+            await this.store.save();
+        } catch (error) {
+            return this.response.error({
+                error
+            });
+        }
+        else if (this.store.getTask().iterations >= definition.maxIterations) try {
+            if (definition.onMaxIterations) await definition.onMaxIterations({
+                task: this.store.getTask()
+            });
+            return this.response.error({
+                error: {
+                    message: "Task reached max iterations."
+                }
+            });
+        } catch (ex) {
+            return this.response.error({
+                error: {
+                    message: "Failed to execute onMaxIterations handler.",
+                    data: getErrorProperties(ex)
+                }
+            });
+        }
+        else try {
+            await this.store.updateTask((task)=>({
+                    iterations: task.iterations + 1
+                }));
+        } catch (error) {
+            return this.response.error({
+                error
+            });
+        }
+        let result;
+        try {
+            const input = structuredClone(this.store.getInput());
+            const controller = this.context.container.resolve(TaskController);
+            result = await this.context.security.withoutAuthorization(async ()=>await definition.run({
+                    input,
+                    controller
+                }));
+        } catch (ex) {
+            return this.response.error({
+                error: getErrorProperties(ex)
+            });
+        }
+        if (result.status === TaskResultStatus.CONTINUE) return this.response.continue({
+            input: result.input,
+            wait: result.wait
+        });
+        if (result.status === TaskResultStatus.ERROR) return this.response.error({
+            error: result.error
+        });
+        if (result.status === TaskResultStatus.ABORTED) return this.response.aborted();
+        return this.response.done({
+            message: result.message,
+            output: result.output
+        });
+    }
+}
+export { TaskManager };
+
+//# sourceMappingURL=TaskManager.js.map

package/api/runner/TaskManager.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/runner/TaskManager.js","sources":["../../../src/api/runner/TaskManager.ts"],"sourcesContent":["import type { ITaskManager, ITaskManagerStorePrivate } from \"./abstractions/index.js\";\nimport type { Context } from \"~/api/types.js\";\nimport { TaskDataStatus } from \"~/api/types.js\";\nimport type { IResponse, IResponseResult } from \"~/api/response/abstractions/index.js\";\nimport { getErrorProperties } from \"~/api/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"],"names":["TaskManager","context","response","store","definition","TaskDataStatus","Date","error","ex","getErrorProperties","task","result","input","structuredClone","controller","TaskController","TaskResultStatus"],"mappings":";;;;AAWO,MAAMA;IAKT,YAAmBC,OAAgB,EAAEC,QAAmB,EAAEC,KAA+B,CAAE;QACvF,IAAI,CAAC,OAAO,GAAGF;QACf,IAAI,CAAC,QAAQ,GAAGC;QAChB,IAAI,CAAC,KAAK,GAAGC;IACjB;IAEA,MAAa,IAAIC,UAAmC,EAA4B;QAI5E,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,OAAOC,eAAe,OAAO,EACjD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO;QAM3B,IAAI,IAAI,CAAC,KAAK,CAAC,SAAS,OAAOA,eAAe,OAAO,EACtD,IAAI;YACA,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC;gBACxB,YAAYA,eAAe,OAAO;gBAClC,WAAW,IAAIC,OAAO,WAAW;gBACjC,eAAe,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,aAAa;gBAChD,YAAY;YAChB;YACA,MAAM,IAAI,CAAC,KAAK,CAAC,IAAI;QACzB,EAAE,OAAOC,OAAO;YACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvBA;YACJ;QACJ;aAOC,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,GAAG,UAAU,IAAIH,WAAW,aAAa,EAChE,IAAI;YACA,IAAIA,WAAW,eAAe,EAC1B,MAAMA,WAAW,eAAe,CAAC;gBAC7B,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO;YAC5B;YAEJ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvB,OAAO;oBACH,SAAS;gBACb;YACJ;QACJ,EAAE,OAAOI,IAAI;YACT,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvB,OAAO;oBACH,SAAS;oBACT,MAAMC,mBAAmBD;gBAC7B;YACJ;QACJ;aAOA,IAAI;YACA,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,CAACE,CAAAA,OACjB;oBACH,YAAYA,KAAK,UAAU,GAAG;gBAClC;QAER,EAAE,OAAOH,OAAO;YACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvBA;YACJ;QACJ;QAGJ,IAAII;QAEJ,IAAI;YACA,MAAMC,QAAQC,gBAAgB,IAAI,CAAC,KAAK,CAAC,QAAQ;YACjD,MAAMC,aAAa,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACC;YAIlDJ,SAAS,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAC,UAC/C,MAAMP,WAAW,GAAG,CAAC;oBACxBQ;oBACAE;gBACJ;QAER,EAAE,OAAON,IAAI;YACT,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;gBACvB,OAAOC,mBAAmBD;YAC9B;QACJ;QAGA,IAAIG,OAAO,MAAM,KAAKK,iBAAiB,QAAQ,EAC3C,OAAO,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC1B,OAAOL,OAAO,KAAK;YACnB,MAAMA,OAAO,IAAI;QACrB;QACG,IAAIA,OAAO,MAAM,KAAKK,iBAAiB,KAAK,EAC/C,OAAO,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;YACvB,OAAOL,OAAO,KAAK;QACvB;QACG,IAAIA,OAAO,MAAM,KAAKK,iBAAiB,OAAO,EACjD,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO;QAEhC,OAAO,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;YACtB,SAASL,OAAO,OAAO;YACvB,QAAQA,OAAO,MAAM;QACzB;IACJ;AACJ"}

package/api/runner/TaskManagerStore.d.ts

@@ -0,0 +1,40 @@
+import type { ITask, ITaskDataInput, ITaskLog, ITaskManagerStoreInfoLog, ITaskManagerStorePrivate, ITaskManagerStoreSetOutputOptions, ITaskManagerStoreUpdateTaskInputOptions, ITaskManagerStoreUpdateTaskOptions, ITasksContextObject, TaskDataStatus } from "../../api/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">;
+}
+export interface ITaskManagerStoreParams {
+    context: TaskManagerStoreContext;
+    task: ITask;
+    log: ITaskLog;
+    databaseLogs: boolean;
+}
+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 databaseLogs;
+    private readonly taskUpdater;
+    private readonly taskLogUpdater;
+    constructor(params: ITaskManagerStoreParams);
+    getStatus(): TaskDataStatus;
+    getTask(): ITask<T, 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;
+    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>;
+}

package/api/runner/TaskManagerStore.js

@@ -0,0 +1,109 @@
+import { TaskLogItemType } from "../types.js";
+import deep_equal from "deep-equal";
+import { getObjectProperties } from "../utils/getObjectProperties.js";
+import { ObjectUpdater } from "../utils/ObjectUpdater.js";
+const getInput = (originalInput, input)=>{
+    if ("function" == typeof input) return input(originalInput);
+    return {
+        ...originalInput,
+        ...input
+    };
+};
+class TaskManagerStore {
+    constructor(params){
+        this.taskUpdater = new ObjectUpdater();
+        this.taskLogUpdater = new ObjectUpdater();
+        this.context = params.context;
+        this.task = params.task;
+        this.taskLog = params.log;
+        this.databaseLogs = true === params.databaseLogs;
+    }
+    getStatus() {
+        return this.task.taskStatus;
+    }
+    getTask() {
+        return this.task;
+    }
+    async listChildTasks(definitionId) {
+        const where = {
+            parentId: this.task.id
+        };
+        if (definitionId) where.definitionId = definitionId;
+        const result = await this.context.tasks.listTasks({
+            where,
+            sort: [
+                "createdOn_ASC"
+            ],
+            limit: 1000000
+        });
+        return result.items;
+    }
+    async updateTask(param, options) {
+        const data = "function" == typeof param ? param(this.task) : param;
+        if (deep_equal(data, this.task)) return;
+        this.taskUpdater.update(data);
+        if (options?.save === false) return;
+        await this.save();
+    }
+    async updateInput(param, options) {
+        const input = getInput(this.task.input, param);
+        if (deep_equal(input, this.task.input)) return;
+        this.taskUpdater.update({
+            input: input
+        });
+        if (options?.save === false) return;
+        await this.save();
+    }
+    getInput() {
+        return this.task.input;
+    }
+    async updateOutput(values, options = {}) {
+        this.taskUpdater.update({
+            output: values
+        });
+        if (options?.save === false) return;
+        await this.save();
+    }
+    getOutput() {
+        return this.task.output;
+    }
+    async addInfoLog(log, options) {
+        if (!this.databaseLogs) return;
+        this.taskLogUpdater.update({
+            items: [
+                {
+                    message: log.message,
+                    data: log.data,
+                    type: TaskLogItemType.INFO,
+                    createdOn: new Date().toISOString()
+                }
+            ]
+        });
+        if (options?.save === false) return;
+        await this.save();
+    }
+    async addErrorLog(log, options) {
+        if (!this.databaseLogs) return;
+        console.error(log.error);
+        this.taskLogUpdater.update({
+            items: [
+                {
+                    message: log.message,
+                    error: log.error instanceof Error ? getObjectProperties(log.error) : log.error,
+                    type: TaskLogItemType.ERROR,
+                    createdOn: new Date().toISOString()
+                }
+            ]
+        });
+        if (options?.save === false) return;
+        await this.save();
+    }
+    async save() {
+        if (this.taskUpdater.isDirty()) this.task = await this.context.tasks.updateTask(this.task.id, this.taskUpdater.fetch());
+        if (!this.databaseLogs) return;
+        if (this.taskLogUpdater.isDirty()) this.taskLog = await this.context.tasks.updateLog(this.taskLog.id, this.taskLogUpdater.fetch());
+    }
+}
+export { TaskManagerStore };
+
+//# sourceMappingURL=TaskManagerStore.js.map

package/api/runner/TaskManagerStore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api/runner/TaskManagerStore.js","sources":["../../../src/api/runner/TaskManagerStore.ts"],"sourcesContent":["import type {\n    IListTaskParamsWhere,\n    ITask,\n    ITaskDataInput,\n    ITaskLog,\n    ITaskManagerStoreInfoLog,\n    ITaskManagerStorePrivate,\n    ITaskManagerStoreSetOutputOptions,\n    ITaskManagerStoreUpdateTaskInputOptions,\n    ITaskManagerStoreUpdateTaskOptions,\n    ITasksContextObject,\n    TaskDataStatus\n} from \"~/api/types.js\";\nimport { TaskLogItemType } from \"~/api/types.js\";\nimport type {\n    ITaskManagerStoreAddLogOptions,\n    ITaskManagerStoreErrorLog,\n    ITaskManagerStoreUpdateTaskInputParam,\n    ITaskManagerStoreUpdateTaskParams\n} from \"./abstractions/index.js\";\n/**\n * Package deep-equal does not have types.\n */\nimport deepEqual from \"deep-equal\";\nimport { getObjectProperties } from \"~/api/utils/getObjectProperties.js\";\nimport { ObjectUpdater } from \"~/api/utils/ObjectUpdater.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\n\nconst getInput = <T extends ITaskDataInput = ITaskDataInput>(\n    originalInput: T,\n    input: ITaskManagerStoreUpdateTaskInputParam<T>\n): T => {\n    if (typeof input === \"function\") {\n        return input(originalInput);\n    }\n    return {\n        ...originalInput,\n        ...input\n    };\n};\n\nexport interface TaskManagerStoreContext {\n    tasks: Pick<ITasksContextObject, \"updateTask\" | \"updateLog\" | \"listTasks\">;\n}\n\nexport interface ITaskManagerStoreParams {\n    context: TaskManagerStoreContext;\n    task: ITask;\n    log: ITaskLog;\n    databaseLogs: boolean;\n}\n\nexport class TaskManagerStore<\n    T extends ITaskDataInput = ITaskDataInput,\n    O extends TaskDefinition.TaskOutput = TaskDefinition.TaskOutput\n> implements ITaskManagerStorePrivate<T, O> {\n    private readonly context: TaskManagerStoreContext;\n    private task: ITask<T, O>;\n    private taskLog: ITaskLog;\n    private readonly databaseLogs: boolean;\n\n    private readonly taskUpdater = new ObjectUpdater<ITask<T, O>>();\n    private readonly taskLogUpdater = new ObjectUpdater<ITaskLog>();\n\n    public constructor(params: ITaskManagerStoreParams) {\n        this.context = params.context;\n        this.task = params.task as ITask<T, O>;\n        this.taskLog = params.log;\n        this.databaseLogs = params.databaseLogs === true;\n    }\n\n    public getStatus(): TaskDataStatus {\n        return this.task.taskStatus;\n    }\n\n    public getTask(): ITask<T, O> {\n        return this.task as ITask<T, O>;\n    }\n\n    public async listChildTasks<\n        I extends TaskDefinition.TaskInput = TaskDefinition.TaskInput,\n        O extends TaskDefinition.TaskOutput = TaskDefinition.TaskOutput\n    >(definitionId?: string): Promise<ITask<I, O>[]> {\n        const where: IListTaskParamsWhere = {\n            parentId: this.task.id\n        };\n        if (definitionId) {\n            where.definitionId = definitionId;\n        }\n        const result = await this.context.tasks.listTasks<I, O>({\n            where,\n            sort: [\"createdOn_ASC\"],\n            limit: 1000000\n        });\n        return result.items;\n    }\n\n    public async updateTask(\n        param: ITaskManagerStoreUpdateTaskParams<T, O>,\n        options?: ITaskManagerStoreUpdateTaskOptions\n    ): Promise<void> {\n        const data = typeof param === \"function\" ? param(this.task) : param;\n\n        /**\n         * No need to update if nothing changed.\n         */\n        if (deepEqual(data, this.task)) {\n            return;\n        }\n\n        this.taskUpdater.update(data);\n\n        if (options?.save === false) {\n            return;\n        }\n        await this.save();\n    }\n\n    public async updateInput(\n        param: ITaskManagerStoreUpdateTaskInputParam<T>,\n        options?: ITaskManagerStoreUpdateTaskInputOptions\n    ): Promise<void> {\n        const input = getInput<T>(this.task.input, param);\n\n        /**\n         * No need to update if nothing changed.\n         */\n        if (deepEqual(input, this.task.input)) {\n            return;\n        }\n        this.taskUpdater.update({\n            input: input as T\n        });\n        if (options?.save === false) {\n            return;\n        }\n        await this.save();\n    }\n\n    public getInput(): T {\n        return this.task.input as T;\n    }\n\n    public async updateOutput(\n        values: Partial<O>,\n        options: ITaskManagerStoreSetOutputOptions = {}\n    ): Promise<void> {\n        this.taskUpdater.update({\n            output: values as O\n        });\n        if (options?.save === false) {\n            return;\n        }\n        await this.save();\n    }\n\n    public getOutput(): O {\n        return this.task.output as O;\n    }\n    /**\n     * Currently the methods throws an error if something goes wrong during the database update.\n     * TODO: Maybe we should wrap it into try/catch and return error if any?\n     */\n    public async addInfoLog(\n        log: ITaskManagerStoreInfoLog,\n        options?: ITaskManagerStoreAddLogOptions\n    ): Promise<void> {\n        if (!this.databaseLogs) {\n            return;\n        }\n        this.taskLogUpdater.update({\n            items: [\n                {\n                    message: log.message,\n                    data: log.data,\n                    type: TaskLogItemType.INFO,\n                    createdOn: new Date().toISOString()\n                }\n            ]\n        });\n        if (options?.save === false) {\n            return;\n        }\n\n        await this.save();\n    }\n    /**\n     * Currently the methods throws an error if something goes wrong during the database update.\n     * TODO: Maybe we should wrap it into try/catch and return error if any?\n     */\n    public async addErrorLog(\n        log: ITaskManagerStoreErrorLog,\n        options?: ITaskManagerStoreAddLogOptions\n    ): Promise<void> {\n        if (!this.databaseLogs) {\n            return;\n        }\n        /**\n         * Let's log the error to the console as well.\n         */\n        console.error(log.error);\n        /**\n         * Then update the log object.\n         */\n        this.taskLogUpdater.update({\n            items: [\n                {\n                    message: log.message,\n                    error: log.error instanceof Error ? getObjectProperties(log.error) : log.error,\n                    type: TaskLogItemType.ERROR,\n                    createdOn: new Date().toISOString()\n                }\n            ]\n        });\n        if (options?.save === false) {\n            return;\n        }\n        await this.save();\n    }\n\n    public async save(): Promise<void> {\n        /**\n         * Update both task and the log, if anything to update.\n         */\n        if (this.taskUpdater.isDirty()) {\n            this.task = await this.context.tasks.updateTask<T, O>(\n                this.task.id,\n                this.taskUpdater.fetch()\n            );\n        }\n        if (!this.databaseLogs) {\n            return;\n        }\n        if (this.taskLogUpdater.isDirty()) {\n            this.taskLog = await this.context.tasks.updateLog(\n                this.taskLog.id,\n                this.taskLogUpdater.fetch()\n            );\n        }\n    }\n}\n"],"names":["getInput","originalInput","input","TaskManagerStore","params","ObjectUpdater","definitionId","where","result","param","options","data","deepEqual","values","log","TaskLogItemType","Date","console","Error","getObjectProperties"],"mappings":";;;;AA4BA,MAAMA,WAAW,CACbC,eACAC;IAEA,IAAI,AAAiB,cAAjB,OAAOA,OACP,OAAOA,MAAMD;IAEjB,OAAO;QACH,GAAGA,aAAa;QAChB,GAAGC,KAAK;IACZ;AACJ;AAaO,MAAMC;IAYT,YAAmBC,MAA+B,CAAE;aAHnC,WAAW,GAAG,IAAIC;aAClB,cAAc,GAAG,IAAIA;QAGlC,IAAI,CAAC,OAAO,GAAGD,OAAO,OAAO;QAC7B,IAAI,CAAC,IAAI,GAAGA,OAAO,IAAI;QACvB,IAAI,CAAC,OAAO,GAAGA,OAAO,GAAG;QACzB,IAAI,CAAC,YAAY,GAAGA,AAAwB,SAAxBA,OAAO,YAAY;IAC3C;IAEO,YAA4B;QAC/B,OAAO,IAAI,CAAC,IAAI,CAAC,UAAU;IAC/B;IAEO,UAAuB;QAC1B,OAAO,IAAI,CAAC,IAAI;IACpB;IAEA,MAAa,eAGXE,YAAqB,EAA0B;QAC7C,MAAMC,QAA8B;YAChC,UAAU,IAAI,CAAC,IAAI,CAAC,EAAE;QAC1B;QACA,IAAID,cACAC,MAAM,YAAY,GAAGD;QAEzB,MAAME,SAAS,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAO;YACpDD;YACA,MAAM;gBAAC;aAAgB;YACvB,OAAO;QACX;QACA,OAAOC,OAAO,KAAK;IACvB;IAEA,MAAa,WACTC,KAA8C,EAC9CC,OAA4C,EAC/B;QACb,MAAMC,OAAO,AAAiB,cAAjB,OAAOF,QAAuBA,MAAM,IAAI,CAAC,IAAI,IAAIA;QAK9D,IAAIG,WAAUD,MAAM,IAAI,CAAC,IAAI,GACzB;QAGJ,IAAI,CAAC,WAAW,CAAC,MAAM,CAACA;QAExB,IAAID,SAAS,SAAS,OAClB;QAEJ,MAAM,IAAI,CAAC,IAAI;IACnB;IAEA,MAAa,YACTD,KAA+C,EAC/CC,OAAiD,EACpC;QACb,MAAMR,QAAQF,SAAY,IAAI,CAAC,IAAI,CAAC,KAAK,EAAES;QAK3C,IAAIG,WAAUV,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,GAChC;QAEJ,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACpB,OAAOA;QACX;QACA,IAAIQ,SAAS,SAAS,OAClB;QAEJ,MAAM,IAAI,CAAC,IAAI;IACnB;IAEO,WAAc;QACjB,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK;IAC1B;IAEA,MAAa,aACTG,MAAkB,EAClBH,UAA6C,CAAC,CAAC,EAClC;QACb,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC;YACpB,QAAQG;QACZ;QACA,IAAIH,SAAS,SAAS,OAClB;QAEJ,MAAM,IAAI,CAAC,IAAI;IACnB;IAEO,YAAe;QAClB,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM;IAC3B;IAKA,MAAa,WACTI,GAA6B,EAC7BJ,OAAwC,EAC3B;QACb,IAAI,CAAC,IAAI,CAAC,YAAY,EAClB;QAEJ,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;YACvB,OAAO;gBACH;oBACI,SAASI,IAAI,OAAO;oBACpB,MAAMA,IAAI,IAAI;oBACd,MAAMC,gBAAgB,IAAI;oBAC1B,WAAW,IAAIC,OAAO,WAAW;gBACrC;aACH;QACL;QACA,IAAIN,SAAS,SAAS,OAClB;QAGJ,MAAM,IAAI,CAAC,IAAI;IACnB;IAKA,MAAa,YACTI,GAA8B,EAC9BJ,OAAwC,EAC3B;QACb,IAAI,CAAC,IAAI,CAAC,YAAY,EAClB;QAKJO,QAAQ,KAAK,CAACH,IAAI,KAAK;QAIvB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC;YACvB,OAAO;gBACH;oBACI,SAASA,IAAI,OAAO;oBACpB,OAAOA,IAAI,KAAK,YAAYI,QAAQC,oBAAoBL,IAAI,KAAK,IAAIA,IAAI,KAAK;oBAC9E,MAAMC,gBAAgB,KAAK;oBAC3B,WAAW,IAAIC,OAAO,WAAW;gBACrC;aACH;QACL;QACA,IAAIN,SAAS,SAAS,OAClB;QAEJ,MAAM,IAAI,CAAC,IAAI;IACnB;IAEA,MAAa,OAAsB;QAI/B,IAAI,IAAI,CAAC,WAAW,CAAC,OAAO,IACxB,IAAI,CAAC,IAAI,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAC3C,IAAI,CAAC,IAAI,CAAC,EAAE,EACZ,IAAI,CAAC,WAAW,CAAC,KAAK;QAG9B,IAAI,CAAC,IAAI,CAAC,YAAY,EAClB;QAEJ,IAAI,IAAI,CAAC,cAAc,CAAC,OAAO,IAC3B,IAAI,CAAC,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAC7C,IAAI,CAAC,OAAO,CAAC,EAAE,EACf,IAAI,CAAC,cAAc,CAAC,KAAK;IAGrC;AACJ"}

package/api/runner/TaskRunner.d.ts

@@ -0,0 +1,25 @@
+import type { ITaskRawEvent } from "../../api/handler/types.js";
+import type { ITaskEventValidation, ITaskRunner } from "./abstractions/index.js";
+import type { Context } from "../../api/types.js";
+import type { IResponseResult } from "../../api/response/abstractions/index.js";
+import type { ITimer } from "@webiny/handler-aws/utils/index.js";
+export declare class TaskRunner<C extends Context = Context> implements ITaskRunner<C> {
+    /**
+     * When DI is introduced, these will get injected.
+     *
+     * container.bind<Request>("Request").toConstantValue(request);
+     * @inject("Request") public readonly request: Request;
+     *
+     * Follow the same example for the rest of the properties.
+     */
+    readonly context: C;
+    readonly timer: ITimer;
+    private readonly validation;
+    /**
+     * We take all required variables separately because they will get injected via DI - so less refactoring is required in the future.
+     */
+    constructor(context: C, timer: ITimer, validation: ITaskEventValidation);
+    isCloseToTimeout(seconds?: number): boolean;
+    run(rawEvent: ITaskRawEvent): Promise<IResponseResult>;
+    private getIsCloseToTimeoutMilliseconds;
+}