@webiny/api-headless-cms-es-tasks 6.4.12-beta.5 → 6.5.0-beta.0

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@webiny/api-headless-cms-es-tasks",
-	"version": "6.4.12-beta.5",
+	"version": "6.5.0-beta.0",
 	"type": "module",
 	"exports": {
 		".": "./index.js",
@@ -17,24 +17,24 @@
 	},
 	"license": "MIT",
 	"dependencies": {
-		"@webiny/api-core": "6.4.12-beta.5",
-		"@webiny/api-headless-cms": "6.4.12-beta.5",
-		"@webiny/api-headless-cms-ddb-es": "6.4.12-beta.5",
-		"@webiny/api-opensearch": "6.4.12-beta.5",
-		"@webiny/background-tasks": "6.4.12-beta.5",
-		"@webiny/handler": "6.4.12-beta.5",
-		"@webiny/handler-aws": "6.4.12-beta.5",
-		"@webiny/utils": "6.4.12-beta.5"
+		"@webiny/api-core": "6.5.0-beta.0",
+		"@webiny/api-headless-cms": "6.5.0-beta.0",
+		"@webiny/api-headless-cms-ddb-es": "6.5.0-beta.0",
+		"@webiny/api-opensearch": "6.5.0-beta.0",
+		"@webiny/background-tasks": "6.5.0-beta.0",
+		"@webiny/handler": "6.5.0-beta.0",
+		"@webiny/handler-aws": "6.5.0-beta.0",
+		"@webiny/utils": "6.5.0-beta.0"
 	},
 	"devDependencies": {
-		"@webiny/api": "6.4.12-beta.5",
-		"@webiny/build-tools": "6.4.12-beta.5",
-		"@webiny/handler-graphql": "6.4.12-beta.5",
-		"@webiny/plugins": "6.4.12-beta.5",
-		"@webiny/project-utils": "6.4.12-beta.5",
-		"@webiny/wcp": "6.4.12-beta.5",
-		"typescript": "6.0.3",
-		"vitest": "5.0.1"
+		"@webiny/api": "6.5.0-beta.0",
+		"@webiny/build-tools": "6.5.0-beta.0",
+		"@webiny/handler-graphql": "6.5.0-beta.0",
+		"@webiny/plugins": "6.5.0-beta.0",
+		"@webiny/project-utils": "6.5.0-beta.0",
+		"@webiny/wcp": "6.5.0-beta.0",
+		"typescript": "7.0.2",
+		"vitest": "4.1.10"
 	},
 	"publishConfig": {
 		"access": "public"

package/tasks/MockDataCreator/MockDataCreator.d.ts

@@ -1,8 +1,10 @@
 import type { IMockDataCreatorInput, IMockDataCreatorOutput } from "./types.js";
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import type { Context } from "../../types.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
 export declare class MockDataCreator<I extends IMockDataCreatorInput, O extends IMockDataCreatorOutput> {
-    private context;
-    constructor(context: Context);
+    private readonly context;
+    private readonly openSearchClient;
+    constructor(context: Context, openSearchClient: OpenSearchClient.Interface);
     execute(params: TaskDefinition.RunParams<I, O>): Promise<TaskDefinition.Result<I, O>>;
 }

package/tasks/MockDataCreator/MockDataCreator.js

@@ -5,13 +5,14 @@ import { mdbid } from "@webiny/utils";
 import { GetModelUseCase } from "@webiny/api-headless-cms/features/contentModel/GetModel/index.js";
 import { CreateEntryUseCase } from "@webiny/api-headless-cms/features/contentEntry/CreateEntry/index.js";
 class MockDataCreator {
-    constructor(context){
+    constructor(context, openSearchClient){
         this.context = context;
+        this.openSearchClient = openSearchClient;
     }
     async execute(params) {
         const { input, controller } = params;
         if (controller.runtime.isAborted()) return controller.response.aborted();
-        else if (controller.runtime.isCloseToTimeout()) return controller.response.continue({
+        if (controller.runtime.isCloseToTimeout()) return controller.response.continue({
             ...input
         });
         const getModel = this.context.container.resolve(GetModelUseCase);
@@ -19,7 +20,7 @@ class MockDataCreator {
         const modelResult = await getModel.execute("cars");
         if (modelResult.isFail()) return controller.response.error(modelResult.error);
         const model = modelResult.value;
-        const healthCheck = createWaitUntilHealthy(this.context.opensearch, {
+        const healthCheck = createWaitUntilHealthy(this.openSearchClient.use(), {
             waitingTimeStep: 20,
             maxWaitingTime: 150,
             maxProcessorPercent: 80,
@@ -29,7 +30,7 @@ class MockDataCreator {
         let createdAmount = input.createdAmount;
         for(; createdAmount < input.totalAmount; createdAmount++){
             if (controller.runtime.isAborted()) return controller.response.aborted();
-            else if (controller.runtime.isCloseToTimeout()) return controller.response.continue({
+            if (controller.runtime.isCloseToTimeout()) return controller.response.continue({
                 ...input,
                 createdAmount
             });

package/tasks/MockDataCreator/MockDataCreator.js.map

@@ -1 +1 @@
-{"version":3,"file":"tasks/MockDataCreator/MockDataCreator.js","sources":["../../../src/tasks/MockDataCreator/MockDataCreator.ts"],"sourcesContent":["import type { IMockDataCreatorInput, IMockDataCreatorOutput } from \"./types.js\";\nimport { mockData } from \"./mockData.js\";\nimport { createWaitUntilHealthy } from \"@webiny/api-opensearch/utils/waitUntilHealthy/index.js\";\nimport { OpenSearchCatClusterHealthStatus } from \"@webiny/api-opensearch/operations/types.js\";\nimport { mdbid } from \"@webiny/utils\";\nimport { GetModelUseCase } from \"@webiny/api-headless-cms/features/contentModel/GetModel/index.js\";\nimport { CreateEntryUseCase } from \"@webiny/api-headless-cms/features/contentEntry/CreateEntry/index.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport type { Context } from \"~/types.js\";\n\nexport class MockDataCreator<I extends IMockDataCreatorInput, O extends IMockDataCreatorOutput> {\n    constructor(private context: Context) {}\n\n    public async execute(\n        params: TaskDefinition.RunParams<I, O>\n    ): Promise<TaskDefinition.Result<I, O>> {\n        const { input, controller } = params;\n\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        } else if (controller.runtime.isCloseToTimeout()) {\n            return controller.response.continue({\n                ...input\n            });\n        }\n\n        const getModel = this.context.container.resolve(GetModelUseCase);\n        const createEntry = this.context.container.resolve(CreateEntryUseCase);\n\n        const modelResult = await getModel.execute(\"cars\");\n        if (modelResult.isFail()) {\n            return controller.response.error(modelResult.error);\n        }\n\n        const model = modelResult.value;\n\n        const healthCheck = createWaitUntilHealthy(this.context.opensearch, {\n            waitingTimeStep: 20,\n            maxWaitingTime: 150,\n            maxProcessorPercent: 80,\n            minClusterHealthStatus: OpenSearchCatClusterHealthStatus.Yellow,\n            maxRamPercent: 101\n        });\n\n        let createdAmount = input.createdAmount;\n\n        for (; createdAmount < input.totalAmount; createdAmount++) {\n            if (controller.runtime.isAborted()) {\n                return controller.response.aborted();\n            } else if (controller.runtime.isCloseToTimeout()) {\n                return controller.response.continue({\n                    ...input,\n                    createdAmount\n                });\n            }\n            if (createdAmount % 50 === 0) {\n                try {\n                    await healthCheck.wait({\n                        async onUnhealthy({\n                            waitingTimeStep,\n                            startedAt,\n                            mustEndAt,\n                            runs,\n                            waitingReason\n                        }) {\n                            console.warn(`Cluster is unhealthy on run #${runs}.`, {\n                                startedAt,\n                                mustEndAt,\n                                waitingTimeStep,\n                                waitingReason\n                            });\n                        },\n                        async onTimeout({\n                            waitingTimeStep,\n                            startedAt,\n                            mustEndAt,\n                            runs,\n                            waitingReason\n                        }) {\n                            console.warn(`Cluster health check timed out on run #${runs}.`, {\n                                startedAt,\n                                mustEndAt,\n                                waitingTimeStep,\n                                waitingReason\n                            });\n                        }\n                    });\n                } catch {\n                    return controller.response.continue(\n                        {\n                            ...input,\n                            createdAmount\n                        },\n                        {\n                            seconds: 30\n                        }\n                    );\n                }\n            }\n            const taskId = controller.state.getTask().id;\n\n            const createResult = await createEntry.execute(model, {\n                id: `${taskId}${mdbid()}`,\n                location: mockData.wbyAco_location,\n                values: mockData\n            });\n\n            if (createResult.isFail()) {\n                return controller.response.error(createResult.error);\n            }\n        }\n\n        return params.controller.response.done(`Created ${input.totalAmount} records.`);\n    }\n}\n"],"names":["MockDataCreator","context","params","input","controller","getModel","GetModelUseCase","createEntry","CreateEntryUseCase","modelResult","model","healthCheck","createWaitUntilHealthy","OpenSearchCatClusterHealthStatus","createdAmount","waitingTimeStep","startedAt","mustEndAt","runs","waitingReason","console","taskId","createResult","mdbid","mockData"],"mappings":";;;;;;AAUO,MAAMA;IACT,YAAoBC,OAAgB,CAAE;aAAlBA,OAAO,GAAPA;IAAmB;IAEvC,MAAa,QACTC,MAAsC,EACF;QACpC,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGF;QAE9B,IAAIE,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;aAC/B,IAAIA,WAAW,OAAO,CAAC,gBAAgB,IAC1C,OAAOA,WAAW,QAAQ,CAAC,QAAQ,CAAC;YAChC,GAAGD,KAAK;QACZ;QAGJ,MAAME,WAAW,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACC;QAChD,MAAMC,cAAc,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACC;QAEnD,MAAMC,cAAc,MAAMJ,SAAS,OAAO,CAAC;QAC3C,IAAII,YAAY,MAAM,IAClB,OAAOL,WAAW,QAAQ,CAAC,KAAK,CAACK,YAAY,KAAK;QAGtD,MAAMC,QAAQD,YAAY,KAAK;QAE/B,MAAME,cAAcC,uBAAuB,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE;YAChE,iBAAiB;YACjB,gBAAgB;YAChB,qBAAqB;YACrB,wBAAwBC,iCAAiC,MAAM;YAC/D,eAAe;QACnB;QAEA,IAAIC,gBAAgBX,MAAM,aAAa;QAEvC,MAAOW,gBAAgBX,MAAM,WAAW,EAAEW,gBAAiB;YACvD,IAAIV,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;iBAC/B,IAAIA,WAAW,OAAO,CAAC,gBAAgB,IAC1C,OAAOA,WAAW,QAAQ,CAAC,QAAQ,CAAC;gBAChC,GAAGD,KAAK;gBACRW;YACJ;YAEJ,IAAIA,gBAAgB,OAAO,GACvB,IAAI;gBACA,MAAMH,YAAY,IAAI,CAAC;oBACnB,MAAM,aAAY,EACdI,eAAe,EACfC,SAAS,EACTC,SAAS,EACTC,IAAI,EACJC,aAAa,EAChB;wBACGC,QAAQ,IAAI,CAAC,CAAC,6BAA6B,EAAEF,KAAK,CAAC,CAAC,EAAE;4BAClDF;4BACAC;4BACAF;4BACAI;wBACJ;oBACJ;oBACA,MAAM,WAAU,EACZJ,eAAe,EACfC,SAAS,EACTC,SAAS,EACTC,IAAI,EACJC,aAAa,EAChB;wBACGC,QAAQ,IAAI,CAAC,CAAC,uCAAuC,EAAEF,KAAK,CAAC,CAAC,EAAE;4BAC5DF;4BACAC;4BACAF;4BACAI;wBACJ;oBACJ;gBACJ;YACJ,EAAE,OAAM;gBACJ,OAAOf,WAAW,QAAQ,CAAC,QAAQ,CAC/B;oBACI,GAAGD,KAAK;oBACRW;gBACJ,GACA;oBACI,SAAS;gBACb;YAER;YAEJ,MAAMO,SAASjB,WAAW,KAAK,CAAC,OAAO,GAAG,EAAE;YAE5C,MAAMkB,eAAe,MAAMf,YAAY,OAAO,CAACG,OAAO;gBAClD,IAAI,GAAGW,SAASE,SAAS;gBACzB,UAAUC,SAAS,eAAe;gBAClC,QAAQA;YACZ;YAEA,IAAIF,aAAa,MAAM,IACnB,OAAOlB,WAAW,QAAQ,CAAC,KAAK,CAACkB,aAAa,KAAK;QAE3D;QAEA,OAAOpB,OAAO,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAEC,MAAM,WAAW,CAAC,SAAS,CAAC;IAClF;AACJ"}
+{"version":3,"file":"tasks/MockDataCreator/MockDataCreator.js","sources":["../../../src/tasks/MockDataCreator/MockDataCreator.ts"],"sourcesContent":["import type { IMockDataCreatorInput, IMockDataCreatorOutput } from \"./types.js\";\nimport { mockData } from \"./mockData.js\";\nimport { createWaitUntilHealthy } from \"@webiny/api-opensearch/utils/waitUntilHealthy/index.js\";\nimport { OpenSearchCatClusterHealthStatus } from \"@webiny/api-opensearch/operations/types.js\";\nimport { mdbid } from \"@webiny/utils\";\nimport { GetModelUseCase } from \"@webiny/api-headless-cms/features/contentModel/GetModel/index.js\";\nimport { CreateEntryUseCase } from \"@webiny/api-headless-cms/features/contentEntry/CreateEntry/index.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport type { Context } from \"~/types.js\";\nimport { OpenSearchClient } from \"@webiny/api-opensearch/exports/api/opensearch.js\";\n\nexport class MockDataCreator<I extends IMockDataCreatorInput, O extends IMockDataCreatorOutput> {\n    constructor(\n        private readonly context: Context,\n        private readonly openSearchClient: OpenSearchClient.Interface\n    ) {}\n\n    public async execute(\n        params: TaskDefinition.RunParams<I, O>\n    ): Promise<TaskDefinition.Result<I, O>> {\n        const { input, controller } = params;\n\n        if (controller.runtime.isAborted()) {\n            return controller.response.aborted();\n        } else if (controller.runtime.isCloseToTimeout()) {\n            return controller.response.continue({\n                ...input\n            });\n        }\n\n        const getModel = this.context.container.resolve(GetModelUseCase);\n        const createEntry = this.context.container.resolve(CreateEntryUseCase);\n\n        const modelResult = await getModel.execute(\"cars\");\n        if (modelResult.isFail()) {\n            return controller.response.error(modelResult.error);\n        }\n\n        const model = modelResult.value;\n\n        const healthCheck = createWaitUntilHealthy(this.openSearchClient.use(), {\n            waitingTimeStep: 20,\n            maxWaitingTime: 150,\n            maxProcessorPercent: 80,\n            minClusterHealthStatus: OpenSearchCatClusterHealthStatus.Yellow,\n            maxRamPercent: 101\n        });\n\n        let createdAmount = input.createdAmount;\n\n        for (; createdAmount < input.totalAmount; createdAmount++) {\n            if (controller.runtime.isAborted()) {\n                return controller.response.aborted();\n            } else if (controller.runtime.isCloseToTimeout()) {\n                return controller.response.continue({\n                    ...input,\n                    createdAmount\n                });\n            }\n            if (createdAmount % 50 === 0) {\n                try {\n                    await healthCheck.wait({\n                        async onUnhealthy({\n                            waitingTimeStep,\n                            startedAt,\n                            mustEndAt,\n                            runs,\n                            waitingReason\n                        }) {\n                            console.warn(`Cluster is unhealthy on run #${runs}.`, {\n                                startedAt,\n                                mustEndAt,\n                                waitingTimeStep,\n                                waitingReason\n                            });\n                        },\n                        async onTimeout({\n                            waitingTimeStep,\n                            startedAt,\n                            mustEndAt,\n                            runs,\n                            waitingReason\n                        }) {\n                            console.warn(`Cluster health check timed out on run #${runs}.`, {\n                                startedAt,\n                                mustEndAt,\n                                waitingTimeStep,\n                                waitingReason\n                            });\n                        }\n                    });\n                } catch {\n                    return controller.response.continue(\n                        {\n                            ...input,\n                            createdAmount\n                        },\n                        {\n                            seconds: 30\n                        }\n                    );\n                }\n            }\n            const taskId = controller.state.getTask().id;\n\n            const createResult = await createEntry.execute(model, {\n                id: `${taskId}${mdbid()}`,\n                location: mockData.wbyAco_location,\n                values: mockData\n            });\n\n            if (createResult.isFail()) {\n                return controller.response.error(createResult.error);\n            }\n        }\n\n        return params.controller.response.done(`Created ${input.totalAmount} records.`);\n    }\n}\n"],"names":["MockDataCreator","context","openSearchClient","params","input","controller","getModel","GetModelUseCase","createEntry","CreateEntryUseCase","modelResult","model","healthCheck","createWaitUntilHealthy","OpenSearchCatClusterHealthStatus","createdAmount","waitingTimeStep","startedAt","mustEndAt","runs","waitingReason","console","taskId","createResult","mdbid","mockData"],"mappings":";;;;;;AAWO,MAAMA;IACT,YACqBC,OAAgB,EAChBC,gBAA4C,CAC/D;aAFmBD,OAAO,GAAPA;aACAC,gBAAgB,GAAhBA;IAClB;IAEH,MAAa,QACTC,MAAsC,EACF;QACpC,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGF;QAE9B,IAAIE,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;QAC/B,IAAIA,WAAW,OAAO,CAAC,gBAAgB,IAC1C,OAAOA,WAAW,QAAQ,CAAC,QAAQ,CAAC;YAChC,GAAGD,KAAK;QACZ;QAGJ,MAAME,WAAW,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACC;QAChD,MAAMC,cAAc,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACC;QAEnD,MAAMC,cAAc,MAAMJ,SAAS,OAAO,CAAC;QAC3C,IAAII,YAAY,MAAM,IAClB,OAAOL,WAAW,QAAQ,CAAC,KAAK,CAACK,YAAY,KAAK;QAGtD,MAAMC,QAAQD,YAAY,KAAK;QAE/B,MAAME,cAAcC,uBAAuB,IAAI,CAAC,gBAAgB,CAAC,GAAG,IAAI;YACpE,iBAAiB;YACjB,gBAAgB;YAChB,qBAAqB;YACrB,wBAAwBC,iCAAiC,MAAM;YAC/D,eAAe;QACnB;QAEA,IAAIC,gBAAgBX,MAAM,aAAa;QAEvC,MAAOW,gBAAgBX,MAAM,WAAW,EAAEW,gBAAiB;YACvD,IAAIV,WAAW,OAAO,CAAC,SAAS,IAC5B,OAAOA,WAAW,QAAQ,CAAC,OAAO;YAC/B,IAAIA,WAAW,OAAO,CAAC,gBAAgB,IAC1C,OAAOA,WAAW,QAAQ,CAAC,QAAQ,CAAC;gBAChC,GAAGD,KAAK;gBACRW;YACJ;YAEJ,IAAIA,gBAAgB,OAAO,GACvB,IAAI;gBACA,MAAMH,YAAY,IAAI,CAAC;oBACnB,MAAM,aAAY,EACdI,eAAe,EACfC,SAAS,EACTC,SAAS,EACTC,IAAI,EACJC,aAAa,EAChB;wBACGC,QAAQ,IAAI,CAAC,CAAC,6BAA6B,EAAEF,KAAK,CAAC,CAAC,EAAE;4BAClDF;4BACAC;4BACAF;4BACAI;wBACJ;oBACJ;oBACA,MAAM,WAAU,EACZJ,eAAe,EACfC,SAAS,EACTC,SAAS,EACTC,IAAI,EACJC,aAAa,EAChB;wBACGC,QAAQ,IAAI,CAAC,CAAC,uCAAuC,EAAEF,KAAK,CAAC,CAAC,EAAE;4BAC5DF;4BACAC;4BACAF;4BACAI;wBACJ;oBACJ;gBACJ;YACJ,EAAE,OAAM;gBACJ,OAAOf,WAAW,QAAQ,CAAC,QAAQ,CAC/B;oBACI,GAAGD,KAAK;oBACRW;gBACJ,GACA;oBACI,SAAS;gBACb;YAER;YAEJ,MAAMO,SAASjB,WAAW,KAAK,CAAC,OAAO,GAAG,EAAE;YAE5C,MAAMkB,eAAe,MAAMf,YAAY,OAAO,CAACG,OAAO;gBAClD,IAAI,GAAGW,SAASE,SAAS;gBACzB,UAAUC,SAAS,eAAe;gBAClC,QAAQA;YACZ;YAEA,IAAIF,aAAa,MAAM,IACnB,OAAOlB,WAAW,QAAQ,CAAC,KAAK,CAACkB,aAAa,KAAK;QAE3D;QAEA,OAAOpB,OAAO,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAEC,MAAM,WAAW,CAAC,SAAS,CAAC;IAClF;AACJ"}

package/tasks/MockDataCreatorTask.d.ts

@@ -1,14 +1,16 @@
 import type { IMockDataCreatorInput, IMockDataCreatorOutput } from "../tasks/MockDataCreator/types.js";
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { CmsContext } from "@webiny/api-headless-cms/features/shared/abstractions.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
 export declare const MOCK_DATA_CREATOR_TASK_ID = "mockDataCreator";
 declare class MockDataCreatorTask implements TaskDefinition.Interface<IMockDataCreatorInput, IMockDataCreatorOutput> {
-    private context;
+    private readonly context;
+    private readonly openSearchClient;
     id: string;
     title: string;
     maxIterations: number;
     selfCleanup: "always";
-    constructor(context: CmsContext.Interface);
+    constructor(context: CmsContext.Interface, openSearchClient: OpenSearchClient.Interface);
     run(params: TaskDefinition.RunParams<IMockDataCreatorInput, IMockDataCreatorOutput>): Promise<TaskDefinition.Result<IMockDataCreatorInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskOutput>>;
 }
 export declare const MockDataCreatorTaskDefinition: typeof MockDataCreatorTask & {

package/tasks/MockDataCreatorTask.js

@@ -1,17 +1,19 @@
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { CmsContext } from "@webiny/api-headless-cms/features/shared/abstractions.js";
+import { MockDataCreator } from "./MockDataCreator/MockDataCreator.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
 const MOCK_DATA_CREATOR_TASK_ID = "mockDataCreator";
 class MockDataCreatorTask {
-    constructor(context){
+    constructor(context, openSearchClient){
         this.context = context;
+        this.openSearchClient = openSearchClient;
         this.id = MOCK_DATA_CREATOR_TASK_ID;
         this.title = "Mock Data Creator";
         this.maxIterations = 500;
         this.selfCleanup = "always";
     }
     async run(params) {
-        const { MockDataCreator } = await import("./MockDataCreator/MockDataCreator.js");
-        const carsMock = new MockDataCreator(this.context);
+        const carsMock = new MockDataCreator(this.context, this.openSearchClient);
         try {
             return await carsMock.execute(params);
         } catch (ex) {
@@ -22,7 +24,8 @@ class MockDataCreatorTask {
 const MockDataCreatorTaskDefinition = TaskDefinition.createImplementation({
     implementation: MockDataCreatorTask,
     dependencies: [
-        CmsContext
+        CmsContext,
+        OpenSearchClient
     ]
 });
 export { MOCK_DATA_CREATOR_TASK_ID, MockDataCreatorTaskDefinition };

package/tasks/MockDataCreatorTask.js.map

@@ -1 +1 @@
-{"version":3,"file":"tasks/MockDataCreatorTask.js","sources":["../../src/tasks/MockDataCreatorTask.ts"],"sourcesContent":["import type { Context } from \"~/types.js\";\nimport type {\n    IMockDataCreatorInput,\n    IMockDataCreatorOutput\n} from \"~/tasks/MockDataCreator/types.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { CmsContext } from \"@webiny/api-headless-cms/features/shared/abstractions.js\";\n\nexport const MOCK_DATA_CREATOR_TASK_ID = \"mockDataCreator\";\n\nclass MockDataCreatorTask implements TaskDefinition.Interface<\n    IMockDataCreatorInput,\n    IMockDataCreatorOutput\n> {\n    id = MOCK_DATA_CREATOR_TASK_ID;\n    title = \"Mock Data Creator\";\n    maxIterations = 500;\n\n    selfCleanup = \"always\" as const;\n\n    constructor(private context: CmsContext.Interface) {}\n\n    async run(params: TaskDefinition.RunParams<IMockDataCreatorInput, IMockDataCreatorOutput>) {\n        const { MockDataCreator } = await import(\n            /* webpackChunkName: \"MockDataCreator\" */ \"./MockDataCreator/MockDataCreator.js\"\n        );\n\n        const carsMock = new MockDataCreator<IMockDataCreatorInput, IMockDataCreatorOutput>(\n            this.context as Context\n        );\n\n        try {\n            return await carsMock.execute(params);\n        } catch (ex) {\n            return params.controller.response.error(ex);\n        }\n    }\n}\n\nexport const MockDataCreatorTaskDefinition = TaskDefinition.createImplementation({\n    implementation: MockDataCreatorTask,\n    dependencies: [CmsContext]\n});\n"],"names":["MOCK_DATA_CREATOR_TASK_ID","MockDataCreatorTask","context","params","MockDataCreator","carsMock","ex","MockDataCreatorTaskDefinition","TaskDefinition","CmsContext"],"mappings":";;AAQO,MAAMA,4BAA4B;AAEzC,MAAMC;IAUF,YAAoBC,OAA6B,CAAE;aAA/BA,OAAO,GAAPA;aANpB,EAAE,GAAGF;aACL,KAAK,GAAG;aACR,aAAa,GAAG;aAEhB,WAAW,GAAG;IAEsC;IAEpD,MAAM,IAAIG,MAA+E,EAAE;QACvF,MAAM,EAAEC,eAAe,EAAE,GAAG,MAAM,MAAM,CAAN;QAIlC,MAAMC,WAAW,IAAID,gBACjB,IAAI,CAAC,OAAO;QAGhB,IAAI;YACA,OAAO,MAAMC,SAAS,OAAO,CAACF;QAClC,EAAE,OAAOG,IAAI;YACT,OAAOH,OAAO,UAAU,CAAC,QAAQ,CAAC,KAAK,CAACG;QAC5C;IACJ;AACJ;AAEO,MAAMC,gCAAgCC,eAAe,oBAAoB,CAAC;IAC7E,gBAAgBP;IAChB,cAAc;QAACQ;KAAW;AAC9B"}
+{"version":3,"file":"tasks/MockDataCreatorTask.js","sources":["../../src/tasks/MockDataCreatorTask.ts"],"sourcesContent":["import type { Context } from \"~/types.js\";\nimport type {\n    IMockDataCreatorInput,\n    IMockDataCreatorOutput\n} from \"~/tasks/MockDataCreator/types.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { CmsContext } from \"@webiny/api-headless-cms/features/shared/abstractions.js\";\nimport { MockDataCreator } from \"./MockDataCreator/MockDataCreator.js\";\nimport { OpenSearchClient } from \"@webiny/api-opensearch/exports/api/opensearch.js\";\n\nexport const MOCK_DATA_CREATOR_TASK_ID = \"mockDataCreator\";\n\nclass MockDataCreatorTask implements TaskDefinition.Interface<\n    IMockDataCreatorInput,\n    IMockDataCreatorOutput\n> {\n    id = MOCK_DATA_CREATOR_TASK_ID;\n    title = \"Mock Data Creator\";\n    maxIterations = 500;\n\n    selfCleanup = \"always\" as const;\n\n    constructor(\n        private readonly context: CmsContext.Interface,\n        private readonly openSearchClient: OpenSearchClient.Interface\n    ) {}\n\n    async run(params: TaskDefinition.RunParams<IMockDataCreatorInput, IMockDataCreatorOutput>) {\n        const carsMock = new MockDataCreator<IMockDataCreatorInput, IMockDataCreatorOutput>(\n            this.context as Context,\n            this.openSearchClient\n        );\n\n        try {\n            return await carsMock.execute(params);\n        } catch (ex) {\n            return params.controller.response.error(ex);\n        }\n    }\n}\n\nexport const MockDataCreatorTaskDefinition = TaskDefinition.createImplementation({\n    implementation: MockDataCreatorTask,\n    dependencies: [CmsContext, OpenSearchClient]\n});\n"],"names":["MOCK_DATA_CREATOR_TASK_ID","MockDataCreatorTask","context","openSearchClient","params","carsMock","MockDataCreator","ex","MockDataCreatorTaskDefinition","TaskDefinition","CmsContext","OpenSearchClient"],"mappings":";;;;AAUO,MAAMA,4BAA4B;AAEzC,MAAMC;IAUF,YACqBC,OAA6B,EAC7BC,gBAA4C,CAC/D;aAFmBD,OAAO,GAAPA;aACAC,gBAAgB,GAAhBA;aARrB,EAAE,GAAGH;aACL,KAAK,GAAG;aACR,aAAa,GAAG;aAEhB,WAAW,GAAG;IAKX;IAEH,MAAM,IAAII,MAA+E,EAAE;QACvF,MAAMC,WAAW,IAAIC,gBACjB,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,gBAAgB;QAGzB,IAAI;YACA,OAAO,MAAMD,SAAS,OAAO,CAACD;QAClC,EAAE,OAAOG,IAAI;YACT,OAAOH,OAAO,UAAU,CAAC,QAAQ,CAAC,KAAK,CAACG;QAC5C;IACJ;AACJ;AAEO,MAAMC,gCAAgCC,eAAe,oBAAoB,CAAC;IAC7E,gBAAgBR;IAChB,cAAc;QAACS;QAAYC;KAAiB;AAChD"}

package/tasks/MockDataManager/MockDataManager.d.ts

@@ -1,9 +1,13 @@
 import type { IMockDataManagerInput, IMockDataManagerOutput } from "../../tasks/MockDataManager/types.js";
 import type { Context } from "../../types.js";
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
+import type { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js";
 export declare class MockDataManager<I extends IMockDataManagerInput, O extends IMockDataManagerOutput> {
-    private context;
-    constructor(context: Context);
+    private readonly context;
+    private readonly openSearchClient;
+    private readonly indexProvider;
+    constructor(context: Context, openSearchClient: OpenSearchClient.Interface, indexProvider: CmsModelOpenSearchIndexProvider.Interface);
     execute(params: TaskDefinition.RunParams<I, O>): Promise<TaskDefinition.Result<I, O>>;
     private listChildTasksNotDone;
     private abortChildTasks;

package/tasks/MockDataManager/MockDataManager.js

@@ -3,11 +3,12 @@ import { calculateAmounts } from "./calculateAmounts.js";
 import { WAIT_MAX_SECONDS, calculateSeconds } from "./calculateSeconds.js";
 import { createModelAndGroup } from "./createModelAndGroup.js";
 import { disableIndexing, enableIndexing } from "../../utils/index.js";
-import { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js";
 import { MOCK_DATA_CREATOR_TASK_ID } from "../MockDataCreatorTask.js";
 class MockDataManager {
-    constructor(context){
+    constructor(context, openSearchClient, indexProvider){
         this.context = context;
+        this.openSearchClient = openSearchClient;
+        this.indexProvider = indexProvider;
     }
     async execute(params) {
         const { input, controller } = params;
@@ -15,18 +16,21 @@ class MockDataManager {
         if (controller.runtime.isAborted()) {
             await this.abortChildTasks(this.context, taskId);
             return controller.response.aborted();
-        } else if (input.seconds) {
+        }
+        if (input.seconds) {
             const items = await this.listChildTasksNotDone(this.context, taskId);
             if (items.length > 0) return controller.response.continue({
                 ...input
             }, {
                 seconds: input.seconds || WAIT_MAX_SECONDS
             });
-            const model = (await this.context.cms.listModels()).find((m)=>m.modelId === input.modelId);
-            if (model) await enableIndexing({
-                client: this.context.opensearch,
-                model,
-                indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)
+            await enableIndexing({
+                client: this.openSearchClient.use(),
+                model: {
+                    modelId: input.modelId,
+                    tenant: "root"
+                },
+                indexProvider: this.indexProvider
             });
             return controller.response.done();
         }
@@ -38,8 +42,8 @@ class MockDataManager {
         if ("string" == typeof result) return controller.response.done(result);
         await disableIndexing({
             model: result.model,
-            client: this.context.opensearch,
-            indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)
+            client: this.openSearchClient.use(),
+            indexProvider: this.indexProvider
         });
         const { amountOfTasks, amountOfRecords } = calculateAmounts(input.amount);
         const seconds = calculateSeconds(amountOfRecords);

package/tasks/MockDataManager/MockDataManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"tasks/MockDataManager/MockDataManager.js","sources":["../../../src/tasks/MockDataManager/MockDataManager.ts"],"sourcesContent":["import { TaskDataStatus } from \"@webiny/background-tasks/api\";\nimport type {\n    IMockDataManagerInput,\n    IMockDataManagerOutput\n} from \"~/tasks/MockDataManager/types.js\";\nimport { calculateAmounts } from \"./calculateAmounts.js\";\nimport type { IMockDataCreatorInput } from \"~/tasks/MockDataCreator/types.js\";\nimport { calculateSeconds, WAIT_MAX_SECONDS } from \"./calculateSeconds.js\";\nimport { createModelAndGroup } from \"~/tasks/MockDataManager/createModelAndGroup.js\";\nimport type { Context } from \"~/types.js\";\nimport { disableIndexing, enableIndexing } from \"~/utils/index.js\";\nimport { CmsModelOpenSearchIndexProvider } from \"@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { MOCK_DATA_CREATOR_TASK_ID } from \"~/tasks/MockDataCreatorTask.js\";\nimport { TaskService } from \"@webiny/api-core/features/task/TaskService/index.js\";\n\nexport class MockDataManager<I extends IMockDataManagerInput, O extends IMockDataManagerOutput> {\n    constructor(private context: Context) {}\n\n    public async execute(\n        params: TaskDefinition.RunParams<I, O>\n    ): Promise<TaskDefinition.Result<I, O>> {\n        const { input, controller } = params;\n\n        const taskId = controller.state.getTask().id;\n        if (controller.runtime.isAborted()) {\n            await this.abortChildTasks(this.context, taskId);\n            return controller.response.aborted();\n        } else if (input.seconds) {\n            const items = await this.listChildTasksNotDone(this.context, taskId);\n\n            if (items.length > 0) {\n                return controller.response.continue(\n                    {\n                        ...input\n                    },\n                    {\n                        seconds: input.seconds || WAIT_MAX_SECONDS\n                    }\n                );\n            }\n\n            const model = (await this.context.cms.listModels()).find(\n                m => m.modelId === input.modelId\n            );\n            if (model) {\n                await enableIndexing({\n                    client: this.context.opensearch,\n                    model,\n                    indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)\n                });\n            }\n            return controller.response.done();\n        }\n\n        const result = await createModelAndGroup({\n            context: this.context,\n            modelId: input.modelId,\n            overwrite: input.overwrite\n        });\n        if (typeof result === \"string\") {\n            return controller.response.done(result);\n        }\n\n        await disableIndexing({\n            model: result.model,\n            client: this.context.opensearch,\n            indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)\n        });\n\n        const { amountOfTasks, amountOfRecords } = calculateAmounts(input.amount);\n\n        const seconds = calculateSeconds(amountOfRecords);\n\n        for (let current = 0; current < amountOfTasks; current++) {\n            await controller.task.trigger<IMockDataCreatorInput>({\n                definition: MOCK_DATA_CREATOR_TASK_ID,\n                input: {\n                    totalAmount: amountOfRecords,\n                    createdAmount: 0\n                },\n                name: `Mock Data Creator Task #${current + 1} of ${amountOfTasks}`\n            });\n        }\n\n        return controller.response.continue(\n            {\n                ...input,\n                seconds,\n                amountOfTasks,\n                amountOfRecords\n            },\n            {\n                seconds\n            }\n        );\n    }\n\n    private async listChildTasksNotDone(context: Context, id: string): Promise<TaskService.Task[]> {\n        const { items } = await context.tasks.listTasks({\n            where: {\n                parentId: id,\n                taskStatus_in: [TaskDataStatus.PENDING, TaskDataStatus.RUNNING]\n            },\n            limit: 10000\n        });\n        return items;\n    }\n\n    private async abortChildTasks(context: Context, id: string): Promise<void> {\n        const items = await this.listChildTasksNotDone(context, id);\n        for (const item of items) {\n            await context.tasks.abort({\n                id: item.id,\n                message: \"Aborted by parent task.\"\n            });\n        }\n    }\n}\n"],"names":["MockDataManager","context","params","input","controller","taskId","items","WAIT_MAX_SECONDS","model","m","enableIndexing","CmsModelOpenSearchIndexProvider","result","createModelAndGroup","disableIndexing","amountOfTasks","amountOfRecords","calculateAmounts","seconds","calculateSeconds","current","MOCK_DATA_CREATOR_TASK_ID","id","TaskDataStatus","item"],"mappings":";;;;;;;AAgBO,MAAMA;IACT,YAAoBC,OAAgB,CAAE;aAAlBA,OAAO,GAAPA;IAAmB;IAEvC,MAAa,QACTC,MAAsC,EACF;QACpC,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGF;QAE9B,MAAMG,SAASD,WAAW,KAAK,CAAC,OAAO,GAAG,EAAE;QAC5C,IAAIA,WAAW,OAAO,CAAC,SAAS,IAAI;YAChC,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAEC;YACzC,OAAOD,WAAW,QAAQ,CAAC,OAAO;QACtC,OAAO,IAAID,MAAM,OAAO,EAAE;YACtB,MAAMG,QAAQ,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,OAAO,EAAED;YAE7D,IAAIC,MAAM,MAAM,GAAG,GACf,OAAOF,WAAW,QAAQ,CAAC,QAAQ,CAC/B;gBACI,GAAGD,KAAK;YACZ,GACA;gBACI,SAASA,MAAM,OAAO,IAAII;YAC9B;YAIR,MAAMC,QAAS,OAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,EAAC,EAAG,IAAI,CACpDC,CAAAA,IAAKA,EAAE,OAAO,KAAKN,MAAM,OAAO;YAEpC,IAAIK,OACA,MAAME,eAAe;gBACjB,QAAQ,IAAI,CAAC,OAAO,CAAC,UAAU;gBAC/BF;gBACA,eAAe,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACG;YAClD;YAEJ,OAAOP,WAAW,QAAQ,CAAC,IAAI;QACnC;QAEA,MAAMQ,SAAS,MAAMC,oBAAoB;YACrC,SAAS,IAAI,CAAC,OAAO;YACrB,SAASV,MAAM,OAAO;YACtB,WAAWA,MAAM,SAAS;QAC9B;QACA,IAAI,AAAkB,YAAlB,OAAOS,QACP,OAAOR,WAAW,QAAQ,CAAC,IAAI,CAACQ;QAGpC,MAAME,gBAAgB;YAClB,OAAOF,OAAO,KAAK;YACnB,QAAQ,IAAI,CAAC,OAAO,CAAC,UAAU;YAC/B,eAAe,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACD;QAClD;QAEA,MAAM,EAAEI,aAAa,EAAEC,eAAe,EAAE,GAAGC,iBAAiBd,MAAM,MAAM;QAExE,MAAMe,UAAUC,iBAAiBH;QAEjC,IAAK,IAAII,UAAU,GAAGA,UAAUL,eAAeK,UAC3C,MAAMhB,WAAW,IAAI,CAAC,OAAO,CAAwB;YACjD,YAAYiB;YACZ,OAAO;gBACH,aAAaL;gBACb,eAAe;YACnB;YACA,MAAM,CAAC,wBAAwB,EAAEI,UAAU,EAAE,IAAI,EAAEL,eAAe;QACtE;QAGJ,OAAOX,WAAW,QAAQ,CAAC,QAAQ,CAC/B;YACI,GAAGD,KAAK;YACRe;YACAH;YACAC;QACJ,GACA;YACIE;QACJ;IAER;IAEA,MAAc,sBAAsBjB,OAAgB,EAAEqB,EAAU,EAA+B;QAC3F,MAAM,EAAEhB,KAAK,EAAE,GAAG,MAAML,QAAQ,KAAK,CAAC,SAAS,CAAC;YAC5C,OAAO;gBACH,UAAUqB;gBACV,eAAe;oBAACC,eAAe,OAAO;oBAAEA,eAAe,OAAO;iBAAC;YACnE;YACA,OAAO;QACX;QACA,OAAOjB;IACX;IAEA,MAAc,gBAAgBL,OAAgB,EAAEqB,EAAU,EAAiB;QACvE,MAAMhB,QAAQ,MAAM,IAAI,CAAC,qBAAqB,CAACL,SAASqB;QACxD,KAAK,MAAME,QAAQlB,MACf,MAAML,QAAQ,KAAK,CAAC,KAAK,CAAC;YACtB,IAAIuB,KAAK,EAAE;YACX,SAAS;QACb;IAER;AACJ"}
+{"version":3,"file":"tasks/MockDataManager/MockDataManager.js","sources":["../../../src/tasks/MockDataManager/MockDataManager.ts"],"sourcesContent":["import type { StorageCmsModel } from \"@webiny/api-headless-cms/types/index.js\";\nimport { TaskDataStatus } from \"@webiny/background-tasks/api\";\nimport type {\n    IMockDataManagerInput,\n    IMockDataManagerOutput\n} from \"~/tasks/MockDataManager/types.js\";\nimport { calculateAmounts } from \"./calculateAmounts.js\";\nimport type { IMockDataCreatorInput } from \"~/tasks/MockDataCreator/types.js\";\nimport { calculateSeconds, WAIT_MAX_SECONDS } from \"./calculateSeconds.js\";\nimport { createModelAndGroup } from \"~/tasks/MockDataManager/createModelAndGroup.js\";\nimport type { Context } from \"~/types.js\";\nimport { disableIndexing, enableIndexing } from \"~/utils/index.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { MOCK_DATA_CREATOR_TASK_ID } from \"~/tasks/MockDataCreatorTask.js\";\nimport { TaskService } from \"@webiny/api-core/features/task/TaskService/index.js\";\nimport { OpenSearchClient } from \"@webiny/api-opensearch/exports/api/opensearch.js\";\nimport type { CmsModelOpenSearchIndexProvider } from \"@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js\";\n\nexport class MockDataManager<I extends IMockDataManagerInput, O extends IMockDataManagerOutput> {\n    constructor(\n        private readonly context: Context,\n        private readonly openSearchClient: OpenSearchClient.Interface,\n        private readonly indexProvider: CmsModelOpenSearchIndexProvider.Interface\n    ) {}\n\n    public async execute(\n        params: TaskDefinition.RunParams<I, O>\n    ): Promise<TaskDefinition.Result<I, O>> {\n        const { input, controller } = params;\n\n        const taskId = controller.state.getTask().id;\n        if (controller.runtime.isAborted()) {\n            await this.abortChildTasks(this.context, taskId);\n            return controller.response.aborted();\n        } else if (input.seconds) {\n            const items = await this.listChildTasksNotDone(this.context, taskId);\n\n            /**\n             * If there are still running creator tasks, we need to wait a bit more.\n             */\n            if (items.length > 0) {\n                return controller.response.continue(\n                    {\n                        ...input\n                    },\n                    {\n                        seconds: input.seconds || WAIT_MAX_SECONDS\n                    }\n                );\n            }\n            /**\n             * If there are no running tasks, we can enable indexing and finish the manager task.\n             */\n            await enableIndexing({\n                client: this.openSearchClient.use(),\n                model: {\n                    modelId: input.modelId,\n                    tenant: \"root\"\n                } as StorageCmsModel,\n                indexProvider: this.indexProvider\n            });\n            return controller.response.done();\n        }\n\n        const result = await createModelAndGroup({\n            context: this.context,\n            modelId: input.modelId,\n            overwrite: input.overwrite\n        });\n        if (typeof result === \"string\") {\n            return controller.response.done(result);\n        }\n\n        await disableIndexing({\n            model: result.model,\n            client: this.openSearchClient.use(),\n            indexProvider: this.indexProvider\n        });\n\n        const { amountOfTasks, amountOfRecords } = calculateAmounts(input.amount);\n\n        const seconds = calculateSeconds(amountOfRecords);\n\n        for (let current = 0; current < amountOfTasks; current++) {\n            await controller.task.trigger<IMockDataCreatorInput>({\n                definition: MOCK_DATA_CREATOR_TASK_ID,\n                input: {\n                    totalAmount: amountOfRecords,\n                    createdAmount: 0\n                },\n                name: `Mock Data Creator Task #${current + 1} of ${amountOfTasks}`\n            });\n        }\n\n        return controller.response.continue(\n            {\n                ...input,\n                seconds,\n                amountOfTasks,\n                amountOfRecords\n            },\n            {\n                seconds\n            }\n        );\n    }\n\n    private async listChildTasksNotDone(context: Context, id: string): Promise<TaskService.Task[]> {\n        const { items } = await context.tasks.listTasks({\n            where: {\n                parentId: id,\n                taskStatus_in: [TaskDataStatus.PENDING, TaskDataStatus.RUNNING]\n            },\n            limit: 10000\n        });\n        return items;\n    }\n\n    private async abortChildTasks(context: Context, id: string): Promise<void> {\n        const items = await this.listChildTasksNotDone(context, id);\n        for (const item of items) {\n            await context.tasks.abort({\n                id: item.id,\n                message: \"Aborted by parent task.\"\n            });\n        }\n    }\n}\n"],"names":["MockDataManager","context","openSearchClient","indexProvider","params","input","controller","taskId","items","WAIT_MAX_SECONDS","enableIndexing","result","createModelAndGroup","disableIndexing","amountOfTasks","amountOfRecords","calculateAmounts","seconds","calculateSeconds","current","MOCK_DATA_CREATOR_TASK_ID","id","TaskDataStatus","item"],"mappings":";;;;;;AAkBO,MAAMA;IACT,YACqBC,OAAgB,EAChBC,gBAA4C,EAC5CC,aAAwD,CAC3E;aAHmBF,OAAO,GAAPA;aACAC,gBAAgB,GAAhBA;aACAC,aAAa,GAAbA;IAClB;IAEH,MAAa,QACTC,MAAsC,EACF;QACpC,MAAM,EAAEC,KAAK,EAAEC,UAAU,EAAE,GAAGF;QAE9B,MAAMG,SAASD,WAAW,KAAK,CAAC,OAAO,GAAG,EAAE;QAC5C,IAAIA,WAAW,OAAO,CAAC,SAAS,IAAI;YAChC,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,OAAO,EAAEC;YACzC,OAAOD,WAAW,QAAQ,CAAC,OAAO;QACtC;QAAO,IAAID,MAAM,OAAO,EAAE;YACtB,MAAMG,QAAQ,MAAM,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,OAAO,EAAED;YAK7D,IAAIC,MAAM,MAAM,GAAG,GACf,OAAOF,WAAW,QAAQ,CAAC,QAAQ,CAC/B;gBACI,GAAGD,KAAK;YACZ,GACA;gBACI,SAASA,MAAM,OAAO,IAAII;YAC9B;YAMR,MAAMC,eAAe;gBACjB,QAAQ,IAAI,CAAC,gBAAgB,CAAC,GAAG;gBACjC,OAAO;oBACH,SAASL,MAAM,OAAO;oBACtB,QAAQ;gBACZ;gBACA,eAAe,IAAI,CAAC,aAAa;YACrC;YACA,OAAOC,WAAW,QAAQ,CAAC,IAAI;QACnC;QAEA,MAAMK,SAAS,MAAMC,oBAAoB;YACrC,SAAS,IAAI,CAAC,OAAO;YACrB,SAASP,MAAM,OAAO;YACtB,WAAWA,MAAM,SAAS;QAC9B;QACA,IAAI,AAAkB,YAAlB,OAAOM,QACP,OAAOL,WAAW,QAAQ,CAAC,IAAI,CAACK;QAGpC,MAAME,gBAAgB;YAClB,OAAOF,OAAO,KAAK;YACnB,QAAQ,IAAI,CAAC,gBAAgB,CAAC,GAAG;YACjC,eAAe,IAAI,CAAC,aAAa;QACrC;QAEA,MAAM,EAAEG,aAAa,EAAEC,eAAe,EAAE,GAAGC,iBAAiBX,MAAM,MAAM;QAExE,MAAMY,UAAUC,iBAAiBH;QAEjC,IAAK,IAAII,UAAU,GAAGA,UAAUL,eAAeK,UAC3C,MAAMb,WAAW,IAAI,CAAC,OAAO,CAAwB;YACjD,YAAYc;YACZ,OAAO;gBACH,aAAaL;gBACb,eAAe;YACnB;YACA,MAAM,CAAC,wBAAwB,EAAEI,UAAU,EAAE,IAAI,EAAEL,eAAe;QACtE;QAGJ,OAAOR,WAAW,QAAQ,CAAC,QAAQ,CAC/B;YACI,GAAGD,KAAK;YACRY;YACAH;YACAC;QACJ,GACA;YACIE;QACJ;IAER;IAEA,MAAc,sBAAsBhB,OAAgB,EAAEoB,EAAU,EAA+B;QAC3F,MAAM,EAAEb,KAAK,EAAE,GAAG,MAAMP,QAAQ,KAAK,CAAC,SAAS,CAAC;YAC5C,OAAO;gBACH,UAAUoB;gBACV,eAAe;oBAACC,eAAe,OAAO;oBAAEA,eAAe,OAAO;iBAAC;YACnE;YACA,OAAO;QACX;QACA,OAAOd;IACX;IAEA,MAAc,gBAAgBP,OAAgB,EAAEoB,EAAU,EAAiB;QACvE,MAAMb,QAAQ,MAAM,IAAI,CAAC,qBAAqB,CAACP,SAASoB;QACxD,KAAK,MAAME,QAAQf,MACf,MAAMP,QAAQ,KAAK,CAAC,KAAK,CAAC;YACtB,IAAIsB,KAAK,EAAE;YACX,SAAS;QACb;IAER;AACJ"}

package/tasks/MockDataManager/createModelAndGroup.js

@@ -1,13 +1,14 @@
 import { createGroupData } from "./group.js";
 import { createCarsModel } from "./model.js";
 import { createIndex } from "../../utils/index.js";
-import { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js";
+import { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/index.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
 const createModelAndGroup = async (params)=>{
     const { context, modelId, overwrite = false } = params;
     let model = (await context.cms.listModels()).find((m)=>m.modelId === modelId);
     let group;
     if (model && !overwrite) return `Model "${modelId}" already exists.`;
-    else if (!model) {
+    if (!model) {
         group = (await context.cms.listGroups()).find((group)=>"mocks" === group.slug);
         if (!group) {
             const groupData = createGroupData();
@@ -18,7 +19,7 @@ const createModelAndGroup = async (params)=>{
     }
     await createIndex({
         model,
-        client: context.opensearch,
+        client: context.container.resolve(OpenSearchClient).use(),
         indexProvider: context.container.resolve(CmsModelOpenSearchIndexProvider)
     });
     return {

package/tasks/MockDataManager/createModelAndGroup.js.map

@@ -1 +1 @@
-{"version":3,"file":"tasks/MockDataManager/createModelAndGroup.js","sources":["../../../src/tasks/MockDataManager/createModelAndGroup.ts"],"sourcesContent":["import type { CmsGroup, CmsModel } from \"@webiny/api-headless-cms/types/index.js\";\nimport type { Context } from \"~/types.js\";\nimport { createGroupData } from \"./group.js\";\nimport { createCarsModel } from \"./model.js\";\nimport { createIndex } from \"~/utils/index.js\";\nimport { CmsModelOpenSearchIndexProvider } from \"@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js\";\n\ninterface ICreateModelAndGroupParams {\n    context: Context;\n    modelId: string;\n    overwrite?: boolean;\n}\nexport interface ICreateModelAndGroupResultSuccess {\n    group: CmsGroup;\n    model: CmsModel;\n}\nexport type ICreateModelAndGroupResult = string | ICreateModelAndGroupResultSuccess;\n\nexport const createModelAndGroup = async (\n    params: ICreateModelAndGroupParams\n): Promise<ICreateModelAndGroupResult> => {\n    const { context, modelId, overwrite = false } = params;\n    /**\n     * First we need to check if the model already exists in the database. If not, we need to create it.\n     */\n    let model = (await context.cms.listModels()).find(m => m.modelId === modelId);\n    let group: CmsGroup | undefined;\n    if (model && !overwrite) {\n        return `Model \"${modelId}\" already exists.`;\n    } else if (!model) {\n        group = (await context.cms.listGroups()).find(group => group.slug === \"mocks\");\n        if (!group) {\n            const groupData = createGroupData();\n            group = await context.cms.createGroup(groupData);\n        }\n        /**\n         * Possibly we need to create the model.\n         */\n        const carsModel = createCarsModel(group);\n        model = await context.cms.createModel(carsModel);\n    }\n    await createIndex({\n        model,\n        client: context.opensearch,\n        indexProvider: context.container.resolve(CmsModelOpenSearchIndexProvider)\n    });\n\n    return {\n        group: group as CmsGroup,\n        model: model as CmsModel\n    };\n};\n"],"names":["createModelAndGroup","params","context","modelId","overwrite","model","m","group","groupData","createGroupData","carsModel","createCarsModel","createIndex","CmsModelOpenSearchIndexProvider"],"mappings":";;;;AAkBO,MAAMA,sBAAsB,OAC/BC;IAEA,MAAM,EAAEC,OAAO,EAAEC,OAAO,EAAEC,YAAY,KAAK,EAAE,GAAGH;IAIhD,IAAII,QAAS,OAAMH,QAAQ,GAAG,CAAC,UAAU,EAAC,EAAG,IAAI,CAACI,CAAAA,IAAKA,EAAE,OAAO,KAAKH;IACrE,IAAII;IACJ,IAAIF,SAAS,CAACD,WACV,OAAO,CAAC,OAAO,EAAED,QAAQ,iBAAiB,CAAC;SACxC,IAAI,CAACE,OAAO;QACfE,QAAS,OAAML,QAAQ,GAAG,CAAC,UAAU,EAAC,EAAG,IAAI,CAACK,CAAAA,QAASA,AAAe,YAAfA,MAAM,IAAI;QACjE,IAAI,CAACA,OAAO;YACR,MAAMC,YAAYC;YAClBF,QAAQ,MAAML,QAAQ,GAAG,CAAC,WAAW,CAACM;QAC1C;QAIA,MAAME,YAAYC,gBAAgBJ;QAClCF,QAAQ,MAAMH,QAAQ,GAAG,CAAC,WAAW,CAACQ;IAC1C;IACA,MAAME,YAAY;QACdP;QACA,QAAQH,QAAQ,UAAU;QAC1B,eAAeA,QAAQ,SAAS,CAAC,OAAO,CAACW;IAC7C;IAEA,OAAO;QACH,OAAON;QACP,OAAOF;IACX;AACJ"}
+{"version":3,"file":"tasks/MockDataManager/createModelAndGroup.js","sources":["../../../src/tasks/MockDataManager/createModelAndGroup.ts"],"sourcesContent":["import type { CmsGroup, CmsModel } from \"@webiny/api-headless-cms/types/index.js\";\nimport type { Context } from \"~/types.js\";\nimport { createGroupData } from \"./group.js\";\nimport { createCarsModel } from \"./model.js\";\nimport { createIndex } from \"~/utils/index.js\";\nimport { CmsModelOpenSearchIndexProvider } from \"@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/index.js\";\nimport { OpenSearchClient } from \"@webiny/api-opensearch/exports/api/opensearch.js\";\n\ninterface ICreateModelAndGroupParams {\n    context: Context;\n    modelId: string;\n    overwrite?: boolean;\n}\nexport interface ICreateModelAndGroupResultSuccess {\n    group: CmsGroup;\n    model: CmsModel;\n}\nexport type ICreateModelAndGroupResult = string | ICreateModelAndGroupResultSuccess;\n\nexport const createModelAndGroup = async (\n    params: ICreateModelAndGroupParams\n): Promise<ICreateModelAndGroupResult> => {\n    const { context, modelId, overwrite = false } = params;\n    /**\n     * First we need to check if the model already exists in the database. If not, we need to create it.\n     */\n    let model = (await context.cms.listModels()).find(m => m.modelId === modelId);\n    let group: CmsGroup | undefined;\n    if (model && !overwrite) {\n        return `Model \"${modelId}\" already exists.`;\n    } else if (!model) {\n        group = (await context.cms.listGroups()).find(group => group.slug === \"mocks\");\n        if (!group) {\n            const groupData = createGroupData();\n            group = await context.cms.createGroup(groupData);\n        }\n        /**\n         * Possibly we need to create the model.\n         */\n        const carsModel = createCarsModel(group);\n        model = await context.cms.createModel(carsModel);\n    }\n    await createIndex({\n        model,\n        client: context.container.resolve(OpenSearchClient).use(),\n        indexProvider: context.container.resolve(CmsModelOpenSearchIndexProvider)\n    });\n\n    return {\n        group: group as CmsGroup,\n        model: model as CmsModel\n    };\n};\n"],"names":["createModelAndGroup","params","context","modelId","overwrite","model","m","group","groupData","createGroupData","carsModel","createCarsModel","createIndex","OpenSearchClient","CmsModelOpenSearchIndexProvider"],"mappings":";;;;;AAmBO,MAAMA,sBAAsB,OAC/BC;IAEA,MAAM,EAAEC,OAAO,EAAEC,OAAO,EAAEC,YAAY,KAAK,EAAE,GAAGH;IAIhD,IAAII,QAAS,OAAMH,QAAQ,GAAG,CAAC,UAAU,EAAC,EAAG,IAAI,CAACI,CAAAA,IAAKA,EAAE,OAAO,KAAKH;IACrE,IAAII;IACJ,IAAIF,SAAS,CAACD,WACV,OAAO,CAAC,OAAO,EAAED,QAAQ,iBAAiB,CAAC;IACxC,IAAI,CAACE,OAAO;QACfE,QAAS,OAAML,QAAQ,GAAG,CAAC,UAAU,EAAC,EAAG,IAAI,CAACK,CAAAA,QAASA,AAAe,YAAfA,MAAM,IAAI;QACjE,IAAI,CAACA,OAAO;YACR,MAAMC,YAAYC;YAClBF,QAAQ,MAAML,QAAQ,GAAG,CAAC,WAAW,CAACM;QAC1C;QAIA,MAAME,YAAYC,gBAAgBJ;QAClCF,QAAQ,MAAMH,QAAQ,GAAG,CAAC,WAAW,CAACQ;IAC1C;IACA,MAAME,YAAY;QACdP;QACA,QAAQH,QAAQ,SAAS,CAAC,OAAO,CAACW,kBAAkB,GAAG;QACvD,eAAeX,QAAQ,SAAS,CAAC,OAAO,CAACY;IAC7C;IAEA,OAAO;QACH,OAAOP;QACP,OAAOF;IACX;AACJ"}

package/tasks/MockDataManagerTask.d.ts

@@ -1,16 +1,19 @@
 import type { IMockDataManagerInput, IMockDataManagerOutput } from "../tasks/MockDataManager/types.js";
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { CmsContext } from "@webiny/api-headless-cms/features/shared/abstractions.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
+import { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/index.js";
 export declare const MOCK_DATA_MANAGER_TASK_ID = "mockDataManager";
 declare class MockDataManagerTask implements TaskDefinition.Interface<IMockDataManagerInput, IMockDataManagerOutput> {
+    private readonly context;
+    private readonly openSearchClient;
+    private readonly indexProvider;
     id: string;
     title: string;
     maxIterations: number;
     selfCleanup: "always";
-    private readonly context;
-    constructor(context: CmsContext.Interface);
+    constructor(context: CmsContext.Interface, openSearchClient: OpenSearchClient.Interface, indexProvider: CmsModelOpenSearchIndexProvider.Interface);
     run(params: TaskDefinition.RunParams<IMockDataManagerInput, IMockDataManagerOutput>): Promise<TaskDefinition.Result<IMockDataManagerInput, import("@webiny/api-core/features/task/TaskDefinition/abstractions.js").ITaskOutput>>;
-    private findModel;
     onError(): Promise<void>;
     onAbort(): Promise<void>;
 }

package/tasks/MockDataManagerTask.js

@@ -1,20 +1,23 @@
 import { CARS_MODEL_ID } from "./MockDataManager/constants.js";
 import { enableIndexing } from "../utils/index.js";
-import { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js";
 import { TaskDefinition } from "@webiny/api-core/features/task/TaskDefinition/index.js";
 import { CmsContext } from "@webiny/api-headless-cms/features/shared/abstractions.js";
+import { OpenSearchClient } from "@webiny/api-opensearch/exports/api/opensearch.js";
+import { CmsModelOpenSearchIndexProvider } from "@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/index.js";
+import { MockDataManager } from "./MockDataManager/MockDataManager.js";
 const MOCK_DATA_MANAGER_TASK_ID = "mockDataManager";
 class MockDataManagerTask {
-    constructor(context){
+    constructor(context, openSearchClient, indexProvider){
+        this.context = context;
+        this.openSearchClient = openSearchClient;
+        this.indexProvider = indexProvider;
         this.id = MOCK_DATA_MANAGER_TASK_ID;
         this.title = "Mock Data Manager";
         this.maxIterations = 500;
         this.selfCleanup = "always";
-        this.context = context;
     }
     async run(params) {
-        const { MockDataManager } = await import("./MockDataManager/MockDataManager.js");
-        const carsMock = new MockDataManager(this.context);
+        const carsMock = new MockDataManager(this.context, this.openSearchClient, this.indexProvider);
         try {
             return await carsMock.execute({
                 ...params,
@@ -27,30 +30,33 @@ class MockDataManagerTask {
             return params.controller.response.error(ex);
         }
     }
-    async findModel() {
-        return (await this.context.cms.listModels()).find((m)=>m.modelId === CARS_MODEL_ID);
-    }
     async onError() {
-        const model = await this.findModel();
-        if (model) await enableIndexing({
-            client: this.context.opensearch,
-            model,
-            indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)
+        await enableIndexing({
+            client: this.openSearchClient.use(),
+            model: {
+                modelId: CARS_MODEL_ID,
+                tenant: "root"
+            },
+            indexProvider: this.indexProvider
         });
     }
     async onAbort() {
-        const model = await this.findModel();
-        if (model) await enableIndexing({
-            client: this.context.opensearch,
-            model,
-            indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)
+        await enableIndexing({
+            client: this.openSearchClient.use(),
+            model: {
+                modelId: CARS_MODEL_ID,
+                tenant: "root"
+            },
+            indexProvider: this.indexProvider
         });
     }
 }
 const MockDataManagerTaskDefinition = TaskDefinition.createImplementation({
     implementation: MockDataManagerTask,
     dependencies: [
-        CmsContext
+        CmsContext,
+        OpenSearchClient,
+        CmsModelOpenSearchIndexProvider
     ]
 });
 export { MOCK_DATA_MANAGER_TASK_ID, MockDataManagerTaskDefinition };

package/tasks/MockDataManagerTask.js.map

@@ -1 +1 @@
-{"version":3,"file":"tasks/MockDataManagerTask.js","sources":["../../src/tasks/MockDataManagerTask.ts"],"sourcesContent":["import type { Context } from \"~/types.js\";\nimport type {\n    IMockDataManagerInput,\n    IMockDataManagerOutput\n} from \"~/tasks/MockDataManager/types.js\";\nimport { CARS_MODEL_ID } from \"~/tasks/MockDataManager/constants.js\";\nimport { enableIndexing } from \"~/utils/index.js\";\nimport { CmsModelOpenSearchIndexProvider } from \"@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/CmsModelOpenSearchIndexProvider.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { CmsContext } from \"@webiny/api-headless-cms/features/shared/abstractions.js\";\n\nexport const MOCK_DATA_MANAGER_TASK_ID = \"mockDataManager\";\n\nclass MockDataManagerTask implements TaskDefinition.Interface<\n    IMockDataManagerInput,\n    IMockDataManagerOutput\n> {\n    id = MOCK_DATA_MANAGER_TASK_ID;\n    title = \"Mock Data Manager\";\n    maxIterations = 500;\n\n    selfCleanup = \"always\" as const;\n\n    private readonly context: Context;\n\n    constructor(context: CmsContext.Interface) {\n        this.context = context as Context;\n    }\n\n    async run(params: TaskDefinition.RunParams<IMockDataManagerInput, IMockDataManagerOutput>) {\n        const { MockDataManager } = await import(\n            /* webpackChunkName: \"MockDataManager\" */ \"./MockDataManager/MockDataManager.js\"\n        );\n\n        const carsMock = new MockDataManager<IMockDataManagerInput, IMockDataManagerOutput>(\n            this.context as Context\n        );\n\n        try {\n            return await carsMock.execute({\n                ...params,\n                input: {\n                    ...params.input,\n                    modelId: CARS_MODEL_ID\n                }\n            });\n        } catch (ex) {\n            return params.controller.response.error(ex);\n        }\n    }\n\n    private async findModel() {\n        return (await this.context.cms.listModels()).find(m => m.modelId === CARS_MODEL_ID);\n    }\n\n    async onError() {\n        const model = await this.findModel();\n        if (model) {\n            await enableIndexing({\n                client: this.context.opensearch,\n                model,\n                indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)\n            });\n        }\n    }\n\n    async onAbort() {\n        const model = await this.findModel();\n        if (model) {\n            await enableIndexing({\n                client: this.context.opensearch,\n                model,\n                indexProvider: this.context.container.resolve(CmsModelOpenSearchIndexProvider)\n            });\n        }\n    }\n}\n\nexport const MockDataManagerTaskDefinition = TaskDefinition.createImplementation({\n    implementation: MockDataManagerTask,\n    dependencies: [CmsContext]\n});\n"],"names":["MOCK_DATA_MANAGER_TASK_ID","MockDataManagerTask","context","params","MockDataManager","carsMock","CARS_MODEL_ID","ex","m","model","enableIndexing","CmsModelOpenSearchIndexProvider","MockDataManagerTaskDefinition","TaskDefinition","CmsContext"],"mappings":";;;;;AAWO,MAAMA,4BAA4B;AAEzC,MAAMC;IAYF,YAAYC,OAA6B,CAAE;aAR3C,EAAE,GAAGF;aACL,KAAK,GAAG;aACR,aAAa,GAAG;aAEhB,WAAW,GAAG;QAKV,IAAI,CAAC,OAAO,GAAGE;IACnB;IAEA,MAAM,IAAIC,MAA+E,EAAE;QACvF,MAAM,EAAEC,eAAe,EAAE,GAAG,MAAM,MAAM,CAAN;QAIlC,MAAMC,WAAW,IAAID,gBACjB,IAAI,CAAC,OAAO;QAGhB,IAAI;YACA,OAAO,MAAMC,SAAS,OAAO,CAAC;gBAC1B,GAAGF,MAAM;gBACT,OAAO;oBACH,GAAGA,OAAO,KAAK;oBACf,SAASG;gBACb;YACJ;QACJ,EAAE,OAAOC,IAAI;YACT,OAAOJ,OAAO,UAAU,CAAC,QAAQ,CAAC,KAAK,CAACI;QAC5C;IACJ;IAEA,MAAc,YAAY;QACtB,OAAQ,OAAM,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,EAAC,EAAG,IAAI,CAACC,CAAAA,IAAKA,EAAE,OAAO,KAAKF;IACzE;IAEA,MAAM,UAAU;QACZ,MAAMG,QAAQ,MAAM,IAAI,CAAC,SAAS;QAClC,IAAIA,OACA,MAAMC,eAAe;YACjB,QAAQ,IAAI,CAAC,OAAO,CAAC,UAAU;YAC/BD;YACA,eAAe,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACE;QAClD;IAER;IAEA,MAAM,UAAU;QACZ,MAAMF,QAAQ,MAAM,IAAI,CAAC,SAAS;QAClC,IAAIA,OACA,MAAMC,eAAe;YACjB,QAAQ,IAAI,CAAC,OAAO,CAAC,UAAU;YAC/BD;YACA,eAAe,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAACE;QAClD;IAER;AACJ;AAEO,MAAMC,gCAAgCC,eAAe,oBAAoB,CAAC;IAC7E,gBAAgBZ;IAChB,cAAc;QAACa;KAAW;AAC9B"}
+{"version":3,"file":"tasks/MockDataManagerTask.js","sources":["../../src/tasks/MockDataManagerTask.ts"],"sourcesContent":["import type { StorageCmsModel } from \"@webiny/api-headless-cms/types/index.js\";\nimport type { Context } from \"~/types.js\";\nimport type {\n    IMockDataManagerInput,\n    IMockDataManagerOutput\n} from \"~/tasks/MockDataManager/types.js\";\nimport { CARS_MODEL_ID } from \"~/tasks/MockDataManager/constants.js\";\nimport { enableIndexing } from \"~/utils/index.js\";\nimport { TaskDefinition } from \"@webiny/api-core/features/task/TaskDefinition/index.js\";\nimport { CmsContext } from \"@webiny/api-headless-cms/features/shared/abstractions.js\";\nimport { OpenSearchClient } from \"@webiny/api-opensearch/exports/api/opensearch.js\";\nimport { CmsModelOpenSearchIndexProvider } from \"@webiny/api-headless-cms-ddb-es/features/CmsModelOpenSearchIndex/index.js\";\nimport { MockDataManager } from \"./MockDataManager/MockDataManager.js\";\n\nexport const MOCK_DATA_MANAGER_TASK_ID = \"mockDataManager\";\n\nclass MockDataManagerTask implements TaskDefinition.Interface<\n    IMockDataManagerInput,\n    IMockDataManagerOutput\n> {\n    id = MOCK_DATA_MANAGER_TASK_ID;\n    title = \"Mock Data Manager\";\n    maxIterations = 500;\n\n    selfCleanup = \"always\" as const;\n\n    constructor(\n        private readonly context: CmsContext.Interface,\n        private readonly openSearchClient: OpenSearchClient.Interface,\n        private readonly indexProvider: CmsModelOpenSearchIndexProvider.Interface\n    ) {}\n\n    async run(params: TaskDefinition.RunParams<IMockDataManagerInput, IMockDataManagerOutput>) {\n        const carsMock = new MockDataManager<IMockDataManagerInput, IMockDataManagerOutput>(\n            this.context as Context,\n            this.openSearchClient,\n            this.indexProvider\n        );\n\n        try {\n            return await carsMock.execute({\n                ...params,\n                input: {\n                    ...params.input,\n                    modelId: CARS_MODEL_ID\n                }\n            });\n        } catch (ex) {\n            return params.controller.response.error(ex);\n        }\n    }\n    async onError() {\n        await enableIndexing({\n            client: this.openSearchClient.use(),\n            model: {\n                modelId: CARS_MODEL_ID,\n                tenant: \"root\"\n            } as StorageCmsModel,\n            indexProvider: this.indexProvider\n        });\n    }\n\n    async onAbort() {\n        await enableIndexing({\n            client: this.openSearchClient.use(),\n            model: {\n                modelId: CARS_MODEL_ID,\n                tenant: \"root\"\n            } as StorageCmsModel,\n            indexProvider: this.indexProvider\n        });\n    }\n}\n\nexport const MockDataManagerTaskDefinition = TaskDefinition.createImplementation({\n    implementation: MockDataManagerTask,\n    dependencies: [CmsContext, OpenSearchClient, CmsModelOpenSearchIndexProvider]\n});\n"],"names":["MOCK_DATA_MANAGER_TASK_ID","MockDataManagerTask","context","openSearchClient","indexProvider","params","carsMock","MockDataManager","CARS_MODEL_ID","ex","enableIndexing","MockDataManagerTaskDefinition","TaskDefinition","CmsContext","OpenSearchClient","CmsModelOpenSearchIndexProvider"],"mappings":";;;;;;;AAcO,MAAMA,4BAA4B;AAEzC,MAAMC;IAUF,YACqBC,OAA6B,EAC7BC,gBAA4C,EAC5CC,aAAwD,CAC3E;aAHmBF,OAAO,GAAPA;aACAC,gBAAgB,GAAhBA;aACAC,aAAa,GAAbA;aATrB,EAAE,GAAGJ;aACL,KAAK,GAAG;aACR,aAAa,GAAG;aAEhB,WAAW,GAAG;IAMX;IAEH,MAAM,IAAIK,MAA+E,EAAE;QACvF,MAAMC,WAAW,IAAIC,gBACjB,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,gBAAgB,EACrB,IAAI,CAAC,aAAa;QAGtB,IAAI;YACA,OAAO,MAAMD,SAAS,OAAO,CAAC;gBAC1B,GAAGD,MAAM;gBACT,OAAO;oBACH,GAAGA,OAAO,KAAK;oBACf,SAASG;gBACb;YACJ;QACJ,EAAE,OAAOC,IAAI;YACT,OAAOJ,OAAO,UAAU,CAAC,QAAQ,CAAC,KAAK,CAACI;QAC5C;IACJ;IACA,MAAM,UAAU;QACZ,MAAMC,eAAe;YACjB,QAAQ,IAAI,CAAC,gBAAgB,CAAC,GAAG;YACjC,OAAO;gBACH,SAASF;gBACT,QAAQ;YACZ;YACA,eAAe,IAAI,CAAC,aAAa;QACrC;IACJ;IAEA,MAAM,UAAU;QACZ,MAAME,eAAe;YACjB,QAAQ,IAAI,CAAC,gBAAgB,CAAC,GAAG;YACjC,OAAO;gBACH,SAASF;gBACT,QAAQ;YACZ;YACA,eAAe,IAAI,CAAC,aAAa;QACrC;IACJ;AACJ;AAEO,MAAMG,gCAAgCC,eAAe,oBAAoB,CAAC;IAC7E,gBAAgBX;IAChB,cAAc;QAACY;QAAYC;QAAkBC;KAAgC;AACjF"}

package/types.d.ts

@@ -1,6 +1,5 @@
-import type { OpenSearchContext } from "@webiny/api-opensearch/types.js";
 import type { CmsContext } from "@webiny/api-headless-cms/types/index.js";
 import type { Context as TasksContext } from "@webiny/background-tasks/api/types.js";
 export type * from "./tasks/MockDataManager/types.js";
-export interface Context extends CmsContext, OpenSearchContext, TasksContext {
+export interface Context extends CmsContext, TasksContext {
 }