@webiny/data-migration 5.37.3-beta.1 → 5.37.3-beta.2

package/repository/migrations.repository.js

@@ -105,6 +105,4 @@ class MigrationRepositoryImpl {
   }
 }
 exports.MigrationRepositoryImpl = MigrationRepositoryImpl;
-(0, _ioc.makeInjectable)(MigrationRepositoryImpl, [(0, _ioc.inject)(_symbols.PrimaryDynamoTableSymbol)]);
-
-//# sourceMappingURL=migrations.repository.js.map
+(0, _ioc.makeInjectable)(MigrationRepositoryImpl, [(0, _ioc.inject)(_symbols.PrimaryDynamoTableSymbol)]);

package/symbols.js

@@ -17,6 +17,4 @@ exports.PrimaryDynamoTableSymbol = PrimaryDynamoTableSymbol;
 const ElasticsearchDynamoTableSymbol = Symbol.for("ElasticsearchDynamoTable");
 exports.ElasticsearchDynamoTableSymbol = ElasticsearchDynamoTableSymbol;
 const ExecutionTimeLimiterSymbol = Symbol.for("ExecutionTimeLimiter");
-exports.ExecutionTimeLimiterSymbol = ExecutionTimeLimiterSymbol;
-
-//# sourceMappingURL=symbols.js.map
+exports.ExecutionTimeLimiterSymbol = ExecutionTimeLimiterSymbol;

package/types.d.ts

@@ -13,6 +13,7 @@ export interface MigrationRun {
     finishedOn: string;
     status: "init" | "running" | "pending" | "done" | "error";
     migrations: MigrationRunItem[];
+    context?: Record<string, any>;
     error?: {
         message: string;
         name?: string;
@@ -42,6 +43,7 @@ export interface DataMigrationContext<TCheckpoint = any> {
     projectVersion: string;
     logger: Logger;
     checkpoint?: TCheckpoint;
+    forceExecute: boolean;
     runningOutOfTime: () => boolean;
     createCheckpoint: (data: TCheckpoint) => void;
     createCheckpointAndExit: (data: TCheckpoint) => void;

package/types.js

@@ -2,6 +2,4 @@
 
 Object.defineProperty(exports, "__esModule", {
   value: true
-});
-
-//# sourceMappingURL=types.js.map
+});

package/types.js.map

@@ -1 +1 @@
-{"version":3,"names":[],"sources":["types.ts"],"sourcesContent":["import { Logger } from \"@webiny/logger\";\n\nexport type { Logger };\n\nexport interface MigrationItem {\n    id: string;\n    description: string;\n    reason: string;\n    startedOn?: string;\n    finishedOn?: string;\n}\n\nexport interface MigrationRun {\n    id: string;\n    startedOn: string;\n    finishedOn: string;\n    status: \"init\" | \"running\" | \"pending\" | \"done\" | \"error\";\n    migrations: MigrationRunItem[];\n    error?: {\n        message: string;\n        name?: string;\n        code?: string;\n        data?: Record<string, any>;\n        stack?: string;\n    };\n}\n\nexport interface MigrationRunItem {\n    id: string;\n    status: \"done\" | \"running\" | \"skipped\" | \"pending\" | \"not-applicable\" | \"error\";\n    startedOn?: string;\n    finishedOn?: string;\n}\n\nexport interface MigrationRepository {\n    getLastRun(): Promise<MigrationRun | null>;\n    saveRun(run: MigrationRun): Promise<void>;\n    listMigrations(params?: { limit: number }): Promise<MigrationItem[]>;\n    logMigration(migration: MigrationItem): Promise<void>;\n    createCheckpoint(id: string, data: unknown): Promise<void>;\n    getCheckpoint(id: string): Promise<unknown>;\n    deleteCheckpoint(id: string): Promise<void>;\n}\n\nexport interface DataMigrationContext<TCheckpoint = any> {\n    projectVersion: string;\n    logger: Logger;\n    checkpoint?: TCheckpoint;\n    runningOutOfTime: () => boolean;\n    createCheckpoint: (data: TCheckpoint) => void;\n    createCheckpointAndExit: (data: TCheckpoint) => void;\n}\n\nexport interface DataMigration<TCheckpoint = any> {\n    getId(): string;\n    getDescription(): string;\n    // This function should check of the migration needs to apply some changes to the system.\n    // Returning `false` means \"everything is ok, mark this migration as executed\".\n    shouldExecute(context: DataMigrationContext<TCheckpoint>): Promise<boolean>;\n    execute(context: DataMigrationContext<TCheckpoint>): Promise<void>;\n}\n\n/**\n * Migration execution time limiter (in milliseconds).\n */\nexport type ExecutionTimeLimiter = () => number;\n\nexport interface MigrationEventPayload {\n    command: \"status\" | \"execute\";\n    version?: string;\n    pattern?: string;\n}\n\nexport type MigrationEventHandlerResponse =\n    // When migration is triggered (via `Event` invocation type), it simply gets invoked, and returns nothing.\n    | undefined\n    // Last migration run state.\n    | MigrationStatusResponse\n    // If an unhandled error is thrown, return the error object.\n    | MigrationInvocationErrorResponse;\n\nexport interface MigrationInvocationErrorResponse {\n    error: { message: string };\n    data?: undefined;\n}\n\nexport interface MigrationStatusRunItem extends MigrationRunItem {\n    description: string;\n}\n\nexport interface MigrationStatus extends MigrationRun {\n    migrations: MigrationStatusRunItem[];\n}\n\nexport interface MigrationStatusResponse {\n    data: MigrationStatus;\n    error?: undefined;\n}\n"],"mappings":""}
+{"version":3,"names":[],"sources":["types.ts"],"sourcesContent":["import { Logger } from \"@webiny/logger\";\n\nexport type { Logger };\n\nexport interface MigrationItem {\n    id: string;\n    description: string;\n    reason: string;\n    startedOn?: string;\n    finishedOn?: string;\n}\n\nexport interface MigrationRun {\n    id: string;\n    startedOn: string;\n    finishedOn: string;\n    status: \"init\" | \"running\" | \"pending\" | \"done\" | \"error\";\n    migrations: MigrationRunItem[];\n    context?: Record<string, any>;\n    error?: {\n        message: string;\n        name?: string;\n        code?: string;\n        data?: Record<string, any>;\n        stack?: string;\n    };\n}\n\nexport interface MigrationRunItem {\n    id: string;\n    status: \"done\" | \"running\" | \"skipped\" | \"pending\" | \"not-applicable\" | \"error\";\n    startedOn?: string;\n    finishedOn?: string;\n}\n\nexport interface MigrationRepository {\n    getLastRun(): Promise<MigrationRun | null>;\n    saveRun(run: MigrationRun): Promise<void>;\n    listMigrations(params?: { limit: number }): Promise<MigrationItem[]>;\n    logMigration(migration: MigrationItem): Promise<void>;\n    createCheckpoint(id: string, data: unknown): Promise<void>;\n    getCheckpoint(id: string): Promise<unknown>;\n    deleteCheckpoint(id: string): Promise<void>;\n}\n\nexport interface DataMigrationContext<TCheckpoint = any> {\n    projectVersion: string;\n    logger: Logger;\n    checkpoint?: TCheckpoint;\n    forceExecute: boolean;\n    runningOutOfTime: () => boolean;\n    createCheckpoint: (data: TCheckpoint) => void;\n    createCheckpointAndExit: (data: TCheckpoint) => void;\n}\n\nexport interface DataMigration<TCheckpoint = any> {\n    getId(): string;\n    getDescription(): string;\n    // This function should check of the migration needs to apply some changes to the system.\n    // Returning `false` means \"everything is ok, mark this migration as executed\".\n    shouldExecute(context: DataMigrationContext<TCheckpoint>): Promise<boolean>;\n    execute(context: DataMigrationContext<TCheckpoint>): Promise<void>;\n}\n\n/**\n * Migration execution time limiter (in milliseconds).\n */\nexport type ExecutionTimeLimiter = () => number;\n\nexport interface MigrationEventPayload {\n    command: \"status\" | \"execute\";\n    version?: string;\n    pattern?: string;\n}\n\nexport type MigrationEventHandlerResponse =\n    // When migration is triggered (via `Event` invocation type), it simply gets invoked, and returns nothing.\n    | undefined\n    // Last migration run state.\n    | MigrationStatusResponse\n    // If an unhandled error is thrown, return the error object.\n    | MigrationInvocationErrorResponse;\n\nexport interface MigrationInvocationErrorResponse {\n    error: { message: string };\n    data?: undefined;\n}\n\nexport interface MigrationStatusRunItem extends MigrationRunItem {\n    description: string;\n}\n\nexport interface MigrationStatus extends MigrationRun {\n    migrations: MigrationStatusRunItem[];\n}\n\nexport interface MigrationStatusResponse {\n    data: MigrationStatus;\n    error?: undefined;\n}\n"],"mappings":""}

package/cli/getMigrationStatus.d.ts

@@ -1,9 +0,0 @@
-import LambdaClient from "aws-sdk/clients/lambda";
-import { MigrationEventHandlerResponse } from "../types";
-interface GetMigrationStatusParams {
-    lambdaClient: LambdaClient;
-    functionName: string;
-    payload?: Record<string, any>;
-}
-export declare const getMigrationStatus: ({ payload, functionName, lambdaClient }: GetMigrationStatusParams) => Promise<MigrationEventHandlerResponse>;
-export {};

package/cli/getMigrationStatus.js

@@ -1,29 +0,0 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.getMigrationStatus = void 0;
-var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
-var _utils = require("@webiny/utils");
-const getMigrationStatus = async ({
-  payload,
-  functionName,
-  lambdaClient
-}) => {
-  const getStatus = () => {
-    return lambdaClient.invoke({
-      FunctionName: functionName,
-      InvocationType: "RequestResponse",
-      Payload: JSON.stringify((0, _objectSpread2.default)((0, _objectSpread2.default)({}, payload), {}, {
-        command: "status"
-      }))
-    }).promise();
-  };
-  const response = await (0, _utils.executeWithRetry)(getStatus);
-  return JSON.parse(response.Payload);
-};
-exports.getMigrationStatus = getMigrationStatus;
-
-//# sourceMappingURL=getMigrationStatus.js.map

package/cli/getMigrationStatus.js.map

@@ -1 +0,0 @@
-{"version":3,"names":["_utils","require","getMigrationStatus","payload","functionName","lambdaClient","getStatus","invoke","FunctionName","InvocationType","Payload","JSON","stringify","_objectSpread2","default","command","promise","response","executeWithRetry","parse","exports"],"sources":["getMigrationStatus.ts"],"sourcesContent":["import { executeWithRetry } from \"@webiny/utils\";\nimport LambdaClient from \"aws-sdk/clients/lambda\";\nimport { MigrationEventHandlerResponse } from \"~/types\";\n\ninterface GetMigrationStatusParams {\n    lambdaClient: LambdaClient;\n    functionName: string;\n    payload?: Record<string, any>;\n}\n\nexport const getMigrationStatus = async ({\n    payload,\n    functionName,\n    lambdaClient\n}: GetMigrationStatusParams) => {\n    const getStatus = () => {\n        return lambdaClient\n            .invoke({\n                FunctionName: functionName,\n                InvocationType: \"RequestResponse\",\n                Payload: JSON.stringify({ ...payload, command: \"status\" })\n            })\n            .promise();\n    };\n\n    const response = await executeWithRetry(getStatus);\n\n    return JSON.parse(response.Payload as string) as MigrationEventHandlerResponse;\n};\n"],"mappings":";;;;;;;;AAAA,IAAAA,MAAA,GAAAC,OAAA;AAUO,MAAMC,kBAAkB,GAAG,MAAAA,CAAO;EACrCC,OAAO;EACPC,YAAY;EACZC;AACsB,CAAC,KAAK;EAC5B,MAAMC,SAAS,GAAGA,CAAA,KAAM;IACpB,OAAOD,YAAY,CACdE,MAAM,CAAC;MACJC,YAAY,EAAEJ,YAAY;MAC1BK,cAAc,EAAE,iBAAiB;MACjCC,OAAO,EAAEC,IAAI,CAACC,SAAS,KAAAC,cAAA,CAAAC,OAAA,MAAAD,cAAA,CAAAC,OAAA,MAAMX,OAAO;QAAEY,OAAO,EAAE;MAAQ,EAAE;IAC7D,CAAC,CAAC,CACDC,OAAO,CAAC,CAAC;EAClB,CAAC;EAED,MAAMC,QAAQ,GAAG,MAAM,IAAAC,uBAAgB,EAACZ,SAAS,CAAC;EAElD,OAAOK,IAAI,CAACQ,KAAK,CAACF,QAAQ,CAACP,OAAiB,CAAC;AACjD,CAAC;AAACU,OAAA,CAAAlB,kBAAA,GAAAA,kBAAA"}

package/cli/printReport.d.ts

@@ -1,9 +0,0 @@
-import { CliContext } from "@webiny/cli/types";
-import { MigrationEventHandlerResponse } from "../types";
-interface ReportParams {
-    response: MigrationEventHandlerResponse;
-    migrationLambdaArn: string;
-    context: CliContext;
-}
-export declare const printReport: ({ response, migrationLambdaArn, context }: ReportParams) => void;
-export {};

package/cli/printReport.js

@@ -1,59 +0,0 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.printReport = void 0;
-var _objectWithoutProperties2 = _interopRequireDefault(require("@babel/runtime/helpers/objectWithoutProperties"));
-var _centerAlign = _interopRequireDefault(require("center-align"));
-const _excluded = ["migrations"];
-const isError = response => {
-  if (!response) {
-    return false;
-  }
-  return "error" in response;
-};
-const makeEven = str => {
-  if (str.length % 2 > 0) {
-    return str + " ";
-  }
-  return str;
-};
-const printReport = ({
-  response,
-  migrationLambdaArn,
-  context
-}) => {
-  if (!response) {
-    return;
-  }
-  if (isError(response)) {
-    context.error(response.error.message);
-    return;
-  }
-  const functionName = migrationLambdaArn.split(":").pop();
-  context.success(`Data migration Lambda %s executed successfully!`, functionName);
-  const _response$data = response.data,
-    {
-      migrations
-    } = _response$data,
-    run = (0, _objectWithoutProperties2.default)(_response$data, _excluded);
-  if (!migrations.length) {
-    context.info(`No applicable migrations were found!`);
-    return;
-  }
-  const maxLength = Math.max(...migrations.map(mig => mig.status.length)) + 2;
-  context.info(`Migration run: %s`, run.id);
-  context.info(`Status: %s`, run.status);
-  context.info(`Started on: %s`, run.startedOn);
-  if (run.status === "done") {
-    context.info(`Finished on: %s`, run.finishedOn);
-  }
-  for (const migration of migrations) {
-    context.info(...[`[%s] %s: ${migration.description}`, (0, _centerAlign.default)(makeEven(migration.status), maxLength), migration.id]);
-  }
-};
-exports.printReport = printReport;
-
-//# sourceMappingURL=printReport.js.map

package/cli/printReport.js.map

@@ -1 +0,0 @@
-{"version":3,"names":["_centerAlign","_interopRequireDefault","require","_excluded","isError","response","makeEven","str","length","printReport","migrationLambdaArn","context","error","message","functionName","split","pop","success","_response$data","data","migrations","run","_objectWithoutProperties2","default","info","maxLength","Math","max","map","mig","status","id","startedOn","finishedOn","migration","description","center","exports"],"sources":["printReport.ts"],"sourcesContent":["import { CliContext } from \"@webiny/cli/types\";\nimport { MigrationEventHandlerResponse, MigrationInvocationErrorResponse } from \"~/types\";\nimport center from \"center-align\";\n\ninterface ReportParams {\n    response: MigrationEventHandlerResponse;\n    migrationLambdaArn: string;\n    context: CliContext;\n}\n\nconst isError = (\n    response: MigrationEventHandlerResponse\n): response is MigrationInvocationErrorResponse => {\n    if (!response) {\n        return false;\n    }\n\n    return \"error\" in response;\n};\n\nconst makeEven = (str: string) => {\n    if (str.length % 2 > 0) {\n        return str + \" \";\n    }\n    return str;\n};\n\nexport const printReport = ({ response, migrationLambdaArn, context }: ReportParams) => {\n    if (!response) {\n        return;\n    }\n\n    if (isError(response)) {\n        context.error(response.error.message);\n        return;\n    }\n\n    const functionName = migrationLambdaArn.split(\":\").pop();\n    context.success(`Data migration Lambda %s executed successfully!`, functionName);\n\n    const { migrations, ...run } = response.data;\n    if (!migrations.length) {\n        context.info(`No applicable migrations were found!`);\n        return;\n    }\n\n    const maxLength = Math.max(...migrations.map(mig => mig.status.length)) + 2;\n    context.info(`Migration run: %s`, run.id);\n    context.info(`Status: %s`, run.status);\n    context.info(`Started on: %s`, run.startedOn);\n    if (run.status === \"done\") {\n        context.info(`Finished on: %s`, run.finishedOn);\n    }\n    for (const migration of migrations) {\n        context.info(\n            ...[\n                `[%s] %s: ${migration.description}`,\n                center(makeEven(migration.status), maxLength),\n                migration.id\n            ]\n        );\n    }\n};\n"],"mappings":";;;;;;;;AAEA,IAAAA,YAAA,GAAAC,sBAAA,CAAAC,OAAA;AAAkC,MAAAC,SAAA;AAQlC,MAAMC,OAAO,GACTC,QAAuC,IACQ;EAC/C,IAAI,CAACA,QAAQ,EAAE;IACX,OAAO,KAAK;EAChB;EAEA,OAAO,OAAO,IAAIA,QAAQ;AAC9B,CAAC;AAED,MAAMC,QAAQ,GAAIC,GAAW,IAAK;EAC9B,IAAIA,GAAG,CAACC,MAAM,GAAG,CAAC,GAAG,CAAC,EAAE;IACpB,OAAOD,GAAG,GAAG,GAAG;EACpB;EACA,OAAOA,GAAG;AACd,CAAC;AAEM,MAAME,WAAW,GAAGA,CAAC;EAAEJ,QAAQ;EAAEK,kBAAkB;EAAEC;AAAsB,CAAC,KAAK;EACpF,IAAI,CAACN,QAAQ,EAAE;IACX;EACJ;EAEA,IAAID,OAAO,CAACC,QAAQ,CAAC,EAAE;IACnBM,OAAO,CAACC,KAAK,CAACP,QAAQ,CAACO,KAAK,CAACC,OAAO,CAAC;IACrC;EACJ;EAEA,MAAMC,YAAY,GAAGJ,kBAAkB,CAACK,KAAK,CAAC,GAAG,CAAC,CAACC,GAAG,CAAC,CAAC;EACxDL,OAAO,CAACM,OAAO,CAAE,iDAAgD,EAAEH,YAAY,CAAC;EAEhF,MAAAI,cAAA,GAA+Bb,QAAQ,CAACc,IAAI;IAAtC;MAAEC;IAAmB,CAAC,GAAAF,cAAA;IAALG,GAAG,OAAAC,yBAAA,CAAAC,OAAA,EAAAL,cAAA,EAAAf,SAAA;EAC1B,IAAI,CAACiB,UAAU,CAACZ,MAAM,EAAE;IACpBG,OAAO,CAACa,IAAI,CAAE,sCAAqC,CAAC;IACpD;EACJ;EAEA,MAAMC,SAAS,GAAGC,IAAI,CAACC,GAAG,CAAC,GAAGP,UAAU,CAACQ,GAAG,CAACC,GAAG,IAAIA,GAAG,CAACC,MAAM,CAACtB,MAAM,CAAC,CAAC,GAAG,CAAC;EAC3EG,OAAO,CAACa,IAAI,CAAE,mBAAkB,EAAEH,GAAG,CAACU,EAAE,CAAC;EACzCpB,OAAO,CAACa,IAAI,CAAE,YAAW,EAAEH,GAAG,CAACS,MAAM,CAAC;EACtCnB,OAAO,CAACa,IAAI,CAAE,gBAAe,EAAEH,GAAG,CAACW,SAAS,CAAC;EAC7C,IAAIX,GAAG,CAACS,MAAM,KAAK,MAAM,EAAE;IACvBnB,OAAO,CAACa,IAAI,CAAE,iBAAgB,EAAEH,GAAG,CAACY,UAAU,CAAC;EACnD;EACA,KAAK,MAAMC,SAAS,IAAId,UAAU,EAAE;IAChCT,OAAO,CAACa,IAAI,CACR,GAAG,CACE,YAAWU,SAAS,CAACC,WAAY,EAAC,EACnC,IAAAC,oBAAM,EAAC9B,QAAQ,CAAC4B,SAAS,CAACJ,MAAM,CAAC,EAAEL,SAAS,CAAC,EAC7CS,SAAS,CAACH,EAAE,CAEpB,CAAC;EACL;AACJ,CAAC;AAACM,OAAA,CAAA5B,WAAA,GAAAA,WAAA"}

package/cli/runMigration.d.ts

@@ -1,13 +0,0 @@
-import LambdaClient from "aws-sdk/clients/lambda";
-import { MigrationInvocationErrorResponse, MigrationRun, MigrationStatusResponse } from "../types";
-interface RunMigrationParams {
-    lambdaClient: LambdaClient;
-    functionName: string;
-    payload?: Record<string, any>;
-    statusCallback?: (status: MigrationRun) => void;
-}
-/**
- * Run the migration Lambda, and re-run when resuming is requested.
- */
-export declare const runMigration: ({ payload, functionName, lambdaClient, statusCallback }: RunMigrationParams) => Promise<MigrationStatusResponse | MigrationInvocationErrorResponse>;
-export {};

package/cli/runMigration.js

@@ -1,87 +0,0 @@
-"use strict";
-
-var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault").default;
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.runMigration = void 0;
-var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
-var _getMigrationStatus = require("./getMigrationStatus");
-const getMigrationStatusReportInterval = () => {
-  const envKey = "MIGRATION_STATUS_REPORT_INTERVAL";
-  if (envKey in process.env) {
-    return parseInt(String(process.env[envKey]));
-  }
-  return 2000;
-};
-
-/**
- * Run the migration Lambda, and re-run when resuming is requested.
- */
-const runMigration = async ({
-  payload,
-  functionName,
-  lambdaClient,
-  statusCallback
-}) => {
-  // We don't report status, if `stdout` is not TTY (usually in CIs, and child processes spawned programmatically).
-  const reportStatus = data => {
-    if (!process.stdout.isTTY || typeof statusCallback !== "function") {
-      return;
-    }
-    statusCallback(data);
-  };
-  const invokeMigration = async () => {
-    const response = await lambdaClient.invoke({
-      FunctionName: functionName,
-      InvocationType: "Event",
-      Payload: JSON.stringify((0, _objectSpread2.default)((0, _objectSpread2.default)({}, payload), {}, {
-        command: "execute"
-      }))
-    }).promise();
-    return response.StatusCode;
-  };
-
-  // Execute migration function.
-  await invokeMigration();
-
-  // Poll for status and re-execute when migration is in "pending" state.
-  let response;
-  while (true) {
-    await new Promise(resolve => setTimeout(resolve, getMigrationStatusReportInterval()));
-    response = await (0, _getMigrationStatus.getMigrationStatus)({
-      payload,
-      functionName,
-      lambdaClient
-    });
-    if (!response) {
-      continue;
-    }
-    const {
-      data,
-      error
-    } = response;
-
-    // If we received an error, it must be an unrecoverable error, and we don't retry.
-    if (error) {
-      return response;
-    }
-    switch (data.status) {
-      case "init":
-        reportStatus(data);
-        continue;
-      case "pending":
-        await invokeMigration();
-        break;
-      case "running":
-        reportStatus(data);
-        break;
-      case "done":
-      default:
-        return response;
-    }
-  }
-};
-exports.runMigration = runMigration;
-
-//# sourceMappingURL=runMigration.js.map

package/cli/runMigration.js.map

@@ -1 +0,0 @@
-{"version":3,"names":["_getMigrationStatus","require","getMigrationStatusReportInterval","envKey","process","env","parseInt","String","runMigration","payload","functionName","lambdaClient","statusCallback","reportStatus","data","stdout","isTTY","invokeMigration","response","invoke","FunctionName","InvocationType","Payload","JSON","stringify","_objectSpread2","default","command","promise","StatusCode","Promise","resolve","setTimeout","getMigrationStatus","error","status","exports"],"sources":["runMigration.ts"],"sourcesContent":["import LambdaClient from \"aws-sdk/clients/lambda\";\nimport {\n    MigrationEventHandlerResponse,\n    MigrationInvocationErrorResponse,\n    MigrationRun,\n    MigrationStatus,\n    MigrationStatusResponse\n} from \"~/types\";\nimport { getMigrationStatus } from \"./getMigrationStatus\";\n\ninterface RunMigrationParams {\n    lambdaClient: LambdaClient;\n    functionName: string;\n    payload?: Record<string, any>;\n    statusCallback?: (status: MigrationRun) => void;\n}\n\nconst getMigrationStatusReportInterval = () => {\n    const envKey = \"MIGRATION_STATUS_REPORT_INTERVAL\";\n    if (envKey in process.env) {\n        return parseInt(String(process.env[envKey]));\n    }\n    return 2000;\n};\n\n/**\n * Run the migration Lambda, and re-run when resuming is requested.\n */\nexport const runMigration = async ({\n    payload,\n    functionName,\n    lambdaClient,\n    statusCallback\n}: RunMigrationParams): Promise<MigrationStatusResponse | MigrationInvocationErrorResponse> => {\n    // We don't report status, if `stdout` is not TTY (usually in CIs, and child processes spawned programmatically).\n    const reportStatus = (data: MigrationStatus) => {\n        if (!process.stdout.isTTY || typeof statusCallback !== \"function\") {\n            return;\n        }\n\n        statusCallback(data);\n    };\n\n    const invokeMigration = async () => {\n        const response = await lambdaClient\n            .invoke({\n                FunctionName: functionName,\n                InvocationType: \"Event\",\n                Payload: JSON.stringify({ ...payload, command: \"execute\" })\n            })\n            .promise();\n\n        return response.StatusCode;\n    };\n\n    // Execute migration function.\n    await invokeMigration();\n\n    // Poll for status and re-execute when migration is in \"pending\" state.\n    let response: MigrationEventHandlerResponse;\n    while (true) {\n        await new Promise(resolve => setTimeout(resolve, getMigrationStatusReportInterval()));\n\n        response = await getMigrationStatus({\n            payload,\n            functionName,\n            lambdaClient\n        });\n\n        if (!response) {\n            continue;\n        }\n\n        const { data, error } = response;\n\n        // If we received an error, it must be an unrecoverable error, and we don't retry.\n        if (error) {\n            return response;\n        }\n\n        switch (data.status) {\n            case \"init\":\n                reportStatus(data);\n                continue;\n            case \"pending\":\n                await invokeMigration();\n                break;\n            case \"running\":\n                reportStatus(data);\n                break;\n            case \"done\":\n            default:\n                return response;\n        }\n    }\n};\n"],"mappings":";;;;;;;;AAQA,IAAAA,mBAAA,GAAAC,OAAA;AASA,MAAMC,gCAAgC,GAAGA,CAAA,KAAM;EAC3C,MAAMC,MAAM,GAAG,kCAAkC;EACjD,IAAIA,MAAM,IAAIC,OAAO,CAACC,GAAG,EAAE;IACvB,OAAOC,QAAQ,CAACC,MAAM,CAACH,OAAO,CAACC,GAAG,CAACF,MAAM,CAAC,CAAC,CAAC;EAChD;EACA,OAAO,IAAI;AACf,CAAC;;AAED;AACA;AACA;AACO,MAAMK,YAAY,GAAG,MAAAA,CAAO;EAC/BC,OAAO;EACPC,YAAY;EACZC,YAAY;EACZC;AACgB,CAAC,KAA0E;EAC3F;EACA,MAAMC,YAAY,GAAIC,IAAqB,IAAK;IAC5C,IAAI,CAACV,OAAO,CAACW,MAAM,CAACC,KAAK,IAAI,OAAOJ,cAAc,KAAK,UAAU,EAAE;MAC/D;IACJ;IAEAA,cAAc,CAACE,IAAI,CAAC;EACxB,CAAC;EAED,MAAMG,eAAe,GAAG,MAAAA,CAAA,KAAY;IAChC,MAAMC,QAAQ,GAAG,MAAMP,YAAY,CAC9BQ,MAAM,CAAC;MACJC,YAAY,EAAEV,YAAY;MAC1BW,cAAc,EAAE,OAAO;MACvBC,OAAO,EAAEC,IAAI,CAACC,SAAS,KAAAC,cAAA,CAAAC,OAAA,MAAAD,cAAA,CAAAC,OAAA,MAAMjB,OAAO;QAAEkB,OAAO,EAAE;MAAS,EAAE;IAC9D,CAAC,CAAC,CACDC,OAAO,CAAC,CAAC;IAEd,OAAOV,QAAQ,CAACW,UAAU;EAC9B,CAAC;;EAED;EACA,MAAMZ,eAAe,CAAC,CAAC;;EAEvB;EACA,IAAIC,QAAuC;EAC3C,OAAO,IAAI,EAAE;IACT,MAAM,IAAIY,OAAO,CAACC,OAAO,IAAIC,UAAU,CAACD,OAAO,EAAE7B,gCAAgC,CAAC,CAAC,CAAC,CAAC;IAErFgB,QAAQ,GAAG,MAAM,IAAAe,sCAAkB,EAAC;MAChCxB,OAAO;MACPC,YAAY;MACZC;IACJ,CAAC,CAAC;IAEF,IAAI,CAACO,QAAQ,EAAE;MACX;IACJ;IAEA,MAAM;MAAEJ,IAAI;MAAEoB;IAAM,CAAC,GAAGhB,QAAQ;;IAEhC;IACA,IAAIgB,KAAK,EAAE;MACP,OAAOhB,QAAQ;IACnB;IAEA,QAAQJ,IAAI,CAACqB,MAAM;MACf,KAAK,MAAM;QACPtB,YAAY,CAACC,IAAI,CAAC;QAClB;MACJ,KAAK,SAAS;QACV,MAAMG,eAAe,CAAC,CAAC;QACvB;MACJ,KAAK,SAAS;QACVJ,YAAY,CAACC,IAAI,CAAC;QAClB;MACJ,KAAK,MAAM;MACX;QACI,OAAOI,QAAQ;IACvB;EACJ;AACJ,CAAC;AAACkB,OAAA,CAAA5B,YAAA,GAAAA,YAAA"}