@salesforce/plugin-data-seeding 1.1.3

package/lib/commands/data-seeding/migrate/index.js

@@ -0,0 +1,146 @@
+/*
+ * Copyright (c) 2023, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+import { SfCommand, Flags } from '@salesforce/sf-plugins-core';
+import { Duration } from '@salesforce/kit';
+import { Messages, PollingClient, SfError } from '@salesforce/core';
+import { initiateDataSeed, pollSeedStatus, initiateJWTMint } from '../../../utils/api.js';
+import { getSeedMigrateMso, getSeedMigrateStage as getStage } from '../../../utils/mso.js';
+import { MigrateRequestCache } from '../../../utils/cache.js';
+Messages.importMessagesDirectoryFromMetaUrl(import.meta.url);
+const messages = Messages.loadMessages('@salesforce/plugin-data-seeding', 'data-seeding.migrate');
+export default class DataSeedingMigrate extends SfCommand {
+    static summary = messages.getMessage('summary');
+    static description = messages.getMessage('description');
+    static examples = messages.getMessages('examples');
+    static flags = {
+        // TODO: The org flags will need to use Flags.requiredOrg() once auth is finalized
+        'target-org': Flags.requiredOrg({
+            summary: messages.getMessage('flags.target-org.summary'),
+            char: 'o',
+            required: true,
+        }),
+        'source-org': Flags.requiredOrg({
+            summary: messages.getMessage('flags.source-org.summary'),
+            char: 's',
+            required: true,
+        }),
+        'config-file': Flags.file({
+            summary: messages.getMessage('flags.config-file.summary'),
+            char: 'f',
+            required: true,
+            exists: true,
+        }),
+        wait: Flags.duration({
+            summary: messages.getMessage('flags.wait.summary'),
+            char: 'w',
+            unit: 'minutes',
+            defaultValue: 33,
+            min: 1,
+            exclusive: ['async'],
+        }),
+        async: Flags.boolean({
+            summary: messages.getMessage('flags.async.summary'),
+            exclusive: ['wait'],
+        }),
+    };
+    async run() {
+        const { flags } = await this.parse(DataSeedingMigrate);
+        const { async, 'config-file': configFile, 'source-org': sourceOrgObj, 'target-org': targetOrgObj, wait } = flags;
+        const sourceOrg = sourceOrgObj.getOrgId();
+        const srcAccessToken = sourceOrgObj.getConnection().accessToken;
+        const srcOrgInstUrl = sourceOrgObj.getConnection().instanceUrl;
+        const targetOrg = targetOrgObj.getOrgId();
+        const tgtAccessToken = targetOrgObj.getConnection().accessToken;
+        const tgtOrgInstUrl = targetOrgObj.getConnection().instanceUrl;
+        // Fetch Valid JWT with Data Seed Org Perm
+        const { jwt: jwtValue } = await initiateJWTMint(srcOrgInstUrl, srcAccessToken, tgtOrgInstUrl, tgtAccessToken);
+        const { request_id: jobId } = await initiateDataSeed(configFile, 'data-copy', jwtValue, srcOrgInstUrl, srcAccessToken, tgtOrgInstUrl, tgtAccessToken, sourceOrg);
+        if (!jobId)
+            throw new Error('Failed to receive job id');
+        const reportMessage = messages.getMessage('report.suggestion', [jobId]);
+        await (await MigrateRequestCache.create()).createCacheEntry(jobId);
+        const baseData = { jobId, sourceOrg, targetOrg };
+        const buildResponse = (response) => ({
+            dataSeedingJob: 'migrate',
+            startTime: response?.execution_start_time,
+            endTime: response?.execution_end_time,
+            status: response.status,
+            ...baseData,
+        });
+        if (wait && !async) {
+            const mso = getSeedMigrateMso({ jsonEnabled: this.jsonEnabled() });
+            mso.updateData(baseData);
+            const completedStatus = ['Completed', 'Partially Completed', 'Failed'];
+            const options = {
+                poll: async () => {
+                    const { jwt: jwtValueNew } = await initiateJWTMint(srcOrgInstUrl, srcAccessToken, tgtOrgInstUrl, tgtAccessToken);
+                    const response = await pollSeedStatus(jobId, jwtValueNew);
+                    mso.goto(getStage(response.step), {
+                        startTime: response.execution_start_time,
+                        endTime: response.execution_end_time,
+                        status: response.status,
+                    });
+                    return {
+                        completed: completedStatus.includes(response.status),
+                        payload: response,
+                    };
+                },
+                frequency: Duration.seconds(1),
+                timeout: wait,
+            };
+            try {
+                const client = await PollingClient.create(options);
+                const pollResult = await client.subscribe();
+                switch (pollResult.status) {
+                    case 'Completed':
+                        mso.stop();
+                        break;
+                    case 'Failed':
+                        mso.error();
+                        throw new SfError(`Data migration job failed on step: ${pollResult.step}\nLog Text: ${pollResult.log_text}`);
+                    case 'Partially Completed':
+                        mso.stop('warning');
+                        this.log(`Process partially completed: ${pollResult.log_text}`);
+                        break;
+                    default:
+                        mso.stop('current');
+                }
+                return buildResponse(pollResult);
+            }
+            catch (e) {
+                const err = SfError.wrap(e);
+                if (err.message.includes('The client has timed out')) {
+                    mso.updateData({ status: 'Client Timeout' });
+                    err.actions = [reportMessage];
+                    mso.stop('current');
+                }
+                else {
+                    mso.error();
+                }
+                throw err;
+            }
+        }
+        else {
+            const { jwt: jwtValueNew } = await initiateJWTMint(srcOrgInstUrl, srcAccessToken, tgtOrgInstUrl, tgtAccessToken);
+            const response = await pollSeedStatus(jobId, jwtValueNew);
+            const mso = getSeedMigrateMso({
+                jsonEnabled: this.jsonEnabled(),
+                showElapsedTime: false,
+                showStageTime: false,
+            });
+            mso.goto(getStage(response.step), {
+                ...baseData,
+                startTime: response.execution_start_time,
+                status: 'Initiated',
+            });
+            mso.stop('async');
+            this.log(reportMessage);
+            return buildResponse(response);
+        }
+    }
+}
+//# sourceMappingURL=index.js.map

package/lib/commands/data-seeding/migrate/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/commands/data-seeding/migrate/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAgB,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAClF,OAAO,EAAE,gBAAgB,EAAoB,cAAc,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAE5G,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,IAAI,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAC3F,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAE9D,QAAQ,CAAC,kCAAkC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC7D,MAAM,QAAQ,GAAG,QAAQ,CAAC,YAAY,CAAC,iCAAiC,EAAE,sBAAsB,CAAC,CAAC;AAElG,MAAM,CAAC,OAAO,OAAO,kBAAmB,SAAQ,SAAmC;IAC1E,MAAM,CAAU,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IACzD,MAAM,CAAU,WAAW,GAAG,QAAQ,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;IACjE,MAAM,CAAU,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAE5D,MAAM,CAAU,KAAK,GAAG;QAC7B,kFAAkF;QAClF,YAAY,EAAE,KAAK,CAAC,WAAW,CAAC;YAC9B,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,0BAA0B,CAAC;YACxD,IAAI,EAAE,GAAG;YACT,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,YAAY,EAAE,KAAK,CAAC,WAAW,CAAC;YAC9B,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,0BAA0B,CAAC;YACxD,IAAI,EAAE,GAAG;YACT,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,aAAa,EAAE,KAAK,CAAC,IAAI,CAAC;YACxB,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,2BAA2B,CAAC;YACzD,IAAI,EAAE,GAAG;YACT,QAAQ,EAAE,IAAI;YACd,MAAM,EAAE,IAAI;SACb,CAAC;QACF,IAAI,EAAE,KAAK,CAAC,QAAQ,CAAC;YACnB,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,oBAAoB,CAAC;YAClD,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,SAAS;YACf,YAAY,EAAE,EAAE;YAChB,GAAG,EAAE,CAAC;YACN,SAAS,EAAE,CAAC,OAAO,CAAC;SACrB,CAAC;QACF,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC;YACnB,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,qBAAqB,CAAC;YACnD,SAAS,EAAE,CAAC,MAAM,CAAC;SACpB,CAAC;KACH,CAAC;IAEK,KAAK,CAAC,GAAG;QACd,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;QACvD,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,EAAE,YAAY,EAAE,IAAI,EAAE,GAAG,KAAK,CAAC;QAEjH,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,cAAc,GAAG,YAAY,CAAC,aAAa,EAAE,CAAC,WAAqB,CAAC;QAC1E,MAAM,aAAa,GAAG,YAAY,CAAC,aAAa,EAAE,CAAC,WAAW,CAAC;QAE/D,MAAM,SAAS,GAAG,YAAY,CAAC,QAAQ,EAAE,CAAC;QAC1C,MAAM,cAAc,GAAG,YAAY,CAAC,aAAa,EAAE,CAAC,WAAqB,CAAC;QAC1E,MAAM,aAAa,GAAG,YAAY,CAAC,aAAa,EAAE,CAAC,WAAW,CAAC;QAE/D,0CAA0C;QAC1C,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,cAAc,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;QAC9G,MAAM,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,MAAM,gBAAgB,CAClD,UAAU,EACV,WAAW,EACX,QAAQ,EACR,aAAa,EACb,cAAc,EACd,aAAa,EACb,cAAc,EACd,SAAS,CACV,CAAC;QAEF,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAExD,MAAM,aAAa,GAAG,QAAQ,CAAC,UAAU,CAAC,mBAAmB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;QAExE,MAAM,CAAC,MAAM,mBAAmB,CAAC,MAAM,EAAE,CAAC,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QAEnE,MAAM,QAAQ,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;QAEjD,MAAM,aAAa,GAAG,CAAC,QAA0B,EAA4B,EAAE,CAAC,CAAC;YAC/E,cAAc,EAAE,SAAS;YACzB,SAAS,EAAE,QAAQ,EAAE,oBAAoB;YACzC,OAAO,EAAE,QAAQ,EAAE,kBAAkB;YACrC,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,GAAG,QAAQ;SACZ,CAAC,CAAC;QAEH,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,GAAG,GAAG,iBAAiB,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;YACnE,GAAG,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAEzB,MAAM,eAAe,GAAG,CAAC,WAAW,EAAE,qBAAqB,EAAE,QAAQ,CAAC,CAAC;YAEvE,MAAM,OAAO,GAA0B;gBACrC,IAAI,EAAE,KAAK,IAA2B,EAAE;oBACtC,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,MAAM,eAAe,CAChD,aAAa,EACb,cAAc,EACd,aAAa,EACb,cAAc,CACf,CAAC;oBACF,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;oBAE1D,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;wBAChC,SAAS,EAAE,QAAQ,CAAC,oBAAoB;wBACxC,OAAO,EAAE,QAAQ,CAAC,kBAAkB;wBACpC,MAAM,EAAE,QAAQ,CAAC,MAAM;qBACxB,CAAC,CAAC;oBAEH,OAAO;wBACL,SAAS,EAAE,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC;wBACpD,OAAO,EAAE,QAAQ;qBAClB,CAAC;gBACJ,CAAC;gBACD,SAAS,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC9B,OAAO,EAAE,IAAI;aACd,CAAC;YAEF,IAAI,CAAC;gBACH,MAAM,MAAM,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBACnD,MAAM,UAAU,GAAqB,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBAE9D,QAAQ,UAAU,CAAC,MAAM,EAAE,CAAC;oBAC1B,KAAK,WAAW;wBACd,GAAG,CAAC,IAAI,EAAE,CAAC;wBACX,MAAM;oBACR,KAAK,QAAQ;wBACX,GAAG,CAAC,KAAK,EAAE,CAAC;wBACZ,MAAM,IAAI,OAAO,CACf,sCAAsC,UAAU,CAAC,IAAI,eAAe,UAAU,CAAC,QAAQ,EAAE,CAC1F,CAAC;oBACJ,KAAK,qBAAqB;wBACxB,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;wBACpB,IAAI,CAAC,GAAG,CAAC,gCAAgC,UAAU,CAAC,QAAQ,EAAE,CAAC,CAAC;wBAChE,MAAM;oBACR;wBACE,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACxB,CAAC;gBAED,OAAO,aAAa,CAAC,UAAU,CAAC,CAAC;YACnC,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,MAAM,GAAG,GAAG,OAAO,CAAC,IAAI,CAAC,CAAU,CAAC,CAAC;gBAErC,IAAI,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,0BAA0B,CAAC,EAAE,CAAC;oBACrD,GAAG,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC;oBAC7C,GAAG,CAAC,OAAO,GAAG,CAAC,aAAa,CAAC,CAAC;oBAC9B,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACtB,CAAC;qBAAM,CAAC;oBACN,GAAG,CAAC,KAAK,EAAE,CAAC;gBACd,CAAC;gBAED,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,MAAM,eAAe,CAAC,aAAa,EAAE,cAAc,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;YACjH,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;YAE1D,MAAM,GAAG,GAAG,iBAAiB,CAAC;gBAC5B,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;gBAC/B,eAAe,EAAE,KAAK;gBACtB,aAAa,EAAE,KAAK;aACrB,CAAC,CAAC;YAEH,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE;gBAChC,GAAG,QAAQ;gBACX,SAAS,EAAE,QAAQ,CAAC,oBAAoB;gBACxC,MAAM,EAAE,WAAW;aACpB,CAAC,CAAC;YAEH,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAClB,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;YAExB,OAAO,aAAa,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC;IACH,CAAC"}

package/lib/commands/data-seeding/migrate/report.d.ts

@@ -0,0 +1,12 @@
+import { SfCommand } from '@salesforce/sf-plugins-core';
+import { DataSeedingReportResult } from '../../../utils/types.js';
+export default class DataSeedingMigrateReport extends SfCommand<DataSeedingReportResult> {
+    static readonly summary: string;
+    static readonly description: string;
+    static readonly examples: string[];
+    static readonly flags: {
+        'job-id': import("@oclif/core/interfaces").OptionFlag<string | undefined, import("@oclif/core/interfaces").CustomOptions>;
+        'use-most-recent': import("@oclif/core/interfaces").BooleanFlag<boolean>;
+    };
+    run(): Promise<DataSeedingReportResult>;
+}

package/lib/commands/data-seeding/migrate/report.js

@@ -0,0 +1,68 @@
+/*
+ * Copyright (c) 2023, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+import { SfCommand, Flags } from '@salesforce/sf-plugins-core';
+import { Messages, SfError } from '@salesforce/core';
+import { pollSeedStatus } from '../../../utils/api.js';
+import { getSeedMigrateMso, getSeedMigrateStage as getStage } from '../../../utils/mso.js';
+import { MigrateRequestCache } from '../../../utils/cache.js';
+Messages.importMessagesDirectoryFromMetaUrl(import.meta.url);
+const messages = Messages.loadMessages('@salesforce/plugin-data-seeding', 'data-seeding.migrate.report');
+export default class DataSeedingMigrateReport extends SfCommand {
+    static summary = messages.getMessage('summary');
+    static description = messages.getMessage('description');
+    static examples = messages.getMessages('examples');
+    static flags = {
+        'job-id': Flags.string({
+            summary: messages.getMessage('flags.job-id.summary'),
+            char: 'i',
+            exactlyOne: ['job-id', 'use-most-recent'],
+        }),
+        'use-most-recent': Flags.boolean({
+            summary: messages.getMessage('flags.use-most-recent.summary'),
+            char: 'r',
+            exactlyOne: ['job-id', 'use-most-recent'],
+        }),
+    };
+    async run() {
+        const { flags } = await this.parse(DataSeedingMigrateReport);
+        const jobId = flags['job-id'] ?? (await MigrateRequestCache.create()).resolveFromCache().jobId;
+        if (!jobId)
+            throw new SfError('No job ID provided or found in cache');
+        const response = await pollSeedStatus(jobId, '');
+        const data = {
+            jobId,
+            startTime: response.execution_start_time,
+            endTime: response.execution_end_time,
+            status: response.status,
+        };
+        const mso = getSeedMigrateMso({
+            jsonEnabled: this.jsonEnabled(),
+            showElapsedTime: false,
+            showStageTime: false,
+        });
+        mso.goto(getStage(response.step), data);
+        switch (response.status) {
+            case 'In Progress':
+                mso.stop('current');
+                break;
+            case 'Partially Completed':
+                mso.stop('warning');
+                this.log(`Process partially completed: ${response.log_text}`);
+                break;
+            case 'Failed':
+                mso.error();
+                throw new SfError(`Failed on step: ${response.step}\nLog Text: ${response.log_text}`);
+            default:
+                mso.stop();
+        }
+        return {
+            dataSeedingJob: 'migrate',
+            ...data,
+        };
+    }
+}
+//# sourceMappingURL=report.js.map

package/lib/commands/data-seeding/migrate/report.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"report.js","sourceRoot":"","sources":["../../../../src/commands/data-seeding/migrate/report.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,6BAA6B,CAAC;AAC/D,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,IAAI,QAAQ,EAAE,MAAM,uBAAuB,CAAC;AAE3F,OAAO,EAAE,mBAAmB,EAAE,MAAM,yBAAyB,CAAC;AAE9D,QAAQ,CAAC,kCAAkC,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC7D,MAAM,QAAQ,GAAG,QAAQ,CAAC,YAAY,CAAC,iCAAiC,EAAE,6BAA6B,CAAC,CAAC;AAEzG,MAAM,CAAC,OAAO,OAAO,wBAAyB,SAAQ,SAAkC;IAC/E,MAAM,CAAU,OAAO,GAAG,QAAQ,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IACzD,MAAM,CAAU,WAAW,GAAG,QAAQ,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC;IACjE,MAAM,CAAU,QAAQ,GAAG,QAAQ,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;IAE5D,MAAM,CAAU,KAAK,GAAG;QAC7B,QAAQ,EAAE,KAAK,CAAC,MAAM,CAAC;YACrB,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,sBAAsB,CAAC;YACpD,IAAI,EAAE,GAAG;YACT,UAAU,EAAE,CAAC,QAAQ,EAAE,iBAAiB,CAAC;SAC1C,CAAC;QACF,iBAAiB,EAAE,KAAK,CAAC,OAAO,CAAC;YAC/B,OAAO,EAAE,QAAQ,CAAC,UAAU,CAAC,+BAA+B,CAAC;YAC7D,IAAI,EAAE,GAAG;YACT,UAAU,EAAE,CAAC,QAAQ,EAAE,iBAAiB,CAAC;SAC1C,CAAC;KACH,CAAC;IAEK,KAAK,CAAC,GAAG;QACd,MAAM,EAAE,KAAK,EAAE,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;QAE7D,MAAM,KAAK,GAAG,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,mBAAmB,CAAC,MAAM,EAAE,CAAC,CAAC,gBAAgB,EAAE,CAAC,KAAK,CAAC;QAE/F,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,OAAO,CAAC,sCAAsC,CAAC,CAAC;QAEtE,MAAM,QAAQ,GAAG,MAAM,cAAc,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAEjD,MAAM,IAAI,GAAG;YACX,KAAK;YACL,SAAS,EAAE,QAAQ,CAAC,oBAAoB;YACxC,OAAO,EAAE,QAAQ,CAAC,kBAAkB;YACpC,MAAM,EAAE,QAAQ,CAAC,MAAM;SACxB,CAAC;QAEF,MAAM,GAAG,GAAG,iBAAiB,CAAC;YAC5B,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;YAC/B,eAAe,EAAE,KAAK;YACtB,aAAa,EAAE,KAAK;SACrB,CAAC,CAAC;QAEH,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC;QAExC,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC;YACxB,KAAK,aAAa;gBAChB,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACpB,MAAM;YACR,KAAK,qBAAqB;gBACxB,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACpB,IAAI,CAAC,GAAG,CAAC,gCAAgC,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAC9D,MAAM;YACR,KAAK,QAAQ;gBACX,GAAG,CAAC,KAAK,EAAE,CAAC;gBACZ,MAAM,IAAI,OAAO,CAAC,mBAAmB,QAAQ,CAAC,IAAI,eAAe,QAAQ,CAAC,QAAQ,EAAE,CAAC,CAAC;YACxF;gBACE,GAAG,CAAC,IAAI,EAAE,CAAC;QACf,CAAC;QAED,OAAO;YACL,cAAc,EAAE,SAAS;YACzB,GAAG,IAAI;SACR,CAAC;IACJ,CAAC"}

package/lib/index.d.ts

@@ -0,0 +1,2 @@
+declare const _default: {};
+export default _default;

package/lib/index.js

@@ -0,0 +1,8 @@
+/*
+ * Copyright (c) 2021, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+export default {};
+//# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,eAAe,EAAE,CAAC"}

package/lib/utils/api.d.ts

@@ -0,0 +1,22 @@
+export type SeedResponse = {
+    request_id: string;
+};
+export type ServletResponse = {
+    jwt: string;
+};
+export type AuthServletResponse = {
+    statusCode: string;
+    body: string;
+};
+export type PollSeedResponse = {
+    execution_end_time: string;
+    execution_start_time: string;
+    log_text: string;
+    request_id: string;
+    status: string;
+    step: string;
+};
+export type DataSeedingOperation = 'data-generation' | 'data-copy';
+export declare const initiateDataSeed: (config: string, operation: DataSeedingOperation, jwt: string, srcOrgUrl: string, srcAccessToken: string, tgtOrgUrl: string, tgtAccessToken: string, srcOrgId: string) => Promise<SeedResponse>;
+export declare const initiateJWTMint: (srcOrgUrl: string, srcAccessToken: string, tgtOrgUrl: string, tgtAccessToken: string) => Promise<ServletResponse>;
+export declare const pollSeedStatus: (jobId: string, jwt: string) => Promise<PollSeedResponse>;

package/lib/utils/api.js

@@ -0,0 +1,84 @@
+/*
+ * Copyright (c) 2024, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+import fs from 'node:fs';
+import got from 'got';
+import FormData from 'form-data';
+import { SfError, Logger } from '@salesforce/core';
+const baseUrl = 'https://api.salesforce.com/platform/data-seed/v1';
+const seedUrl = `${baseUrl}/data-seed`;
+const pollUrl = `${baseUrl}/status`;
+const sfRegion = 'us-east-1';
+export const initiateDataSeed = async (config, operation, jwt, srcOrgUrl, srcAccessToken, tgtOrgUrl, tgtAccessToken, srcOrgId) => {
+    const form = new FormData();
+    form.append('config_file', fs.createReadStream(config));
+    form.append('operation', operation);
+    form.append('source_access_token', srcAccessToken);
+    form.append('source_instance_url', srcOrgUrl);
+    form.append('target_access_token', tgtAccessToken);
+    form.append('target_instance_url', tgtOrgUrl);
+    form.append('source_org_id', srcOrgId);
+    // TODO: Update to use .json() instead of JSON.parse once the Error response is changed to be JSON
+    //       Update the return type as well
+    const response = await got.post(seedUrl, {
+        throwHttpErrors: false,
+        headers: {
+            ...form.getHeaders(),
+            Authorization: `Bearer ${jwt}`,
+            'x-salesforce-region': sfRegion,
+        },
+        body: form,
+    });
+    if (response.statusCode !== 200) {
+        throw new SfError(`Failed to initiate data-seeding operation (${operation}). Response:\n${response.body}`);
+    }
+    return JSON.parse(response.body);
+};
+export const initiateJWTMint = async (srcOrgUrl, srcAccessToken, tgtOrgUrl, tgtAccessToken) => {
+    const srcServletUrl = `${srcOrgUrl}/dataseed/auth`;
+    const tgtServletUrl = `${tgtOrgUrl}/dataseed/auth`;
+    const [responseSrc, responseTgt] = await Promise.all([
+        callAuthServlet(srcServletUrl, srcAccessToken),
+        callAuthServlet(tgtServletUrl, tgtAccessToken),
+    ]);
+    if (responseSrc.statusCode === '200') {
+        return JSON.parse(responseSrc.body);
+    }
+    if (responseTgt.statusCode === '200') {
+        return JSON.parse(responseTgt.body);
+    }
+    throw new SfError(`Org permission for data seed not found in either the source or target org.\nSource Response: Error Code : ${responseSrc.statusCode} - ${responseSrc.body}.  \nTarget Response: Error Code : ${responseTgt.statusCode} - ${responseTgt.body}`);
+};
+const callAuthServlet = async (url, accessToken) => {
+    const response = await got.post(url, {
+        throwHttpErrors: false,
+        headers: {
+            Authorization: `Bearer ${accessToken}`,
+        },
+    });
+    return {
+        statusCode: response.statusCode.toString(), // Convert to string
+        body: response.body,
+    };
+};
+export const pollSeedStatus = async (jobId, jwt) => {
+    const logger = await Logger.child('PollSeedStatus');
+    // TODO: Update to use .json() instead of JSON.parse once the Error response is changed to be JSON
+    //       Update the return type as well
+    const headers = {
+        Authorization: `Bearer ${jwt}`,
+        'x-salesforce-region': sfRegion,
+    };
+    const response = await got.get(`${pollUrl}/${jobId}`, { throwHttpErrors: false, headers });
+    if (response.statusCode !== 200) {
+        // TODO: Print error body once the Error response is changed to be JSON
+        throw new SfError(`Failed to poll data seeding status for ${jobId}`);
+    }
+    const json = JSON.parse(response.body);
+    logger.debug(json);
+    return json;
+};
+//# sourceMappingURL=api.js.map

package/lib/utils/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../src/utils/api.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,MAAM,SAAS,CAAC;AACzB,OAAO,GAAG,MAAM,KAAK,CAAC;AACtB,OAAO,QAAQ,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAwBnD,MAAM,OAAO,GAAG,kDAAkD,CAAC;AACnE,MAAM,OAAO,GAAG,GAAG,OAAO,YAAY,CAAC;AACvC,MAAM,OAAO,GAAG,GAAG,OAAO,SAAS,CAAC;AACpC,MAAM,QAAQ,GAAG,WAAW,CAAA;AAC5B,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,EACnC,MAAc,EACd,SAA+B,EAC/B,GAAW,EACX,SAAiB,EACjB,cAAsB,EACtB,SAAiB,EACjB,cAAsB,EACtB,QAAgB,EACO,EAAE;IACzB,MAAM,IAAI,GAAG,IAAI,QAAQ,EAAE,CAAC;IAC5B,IAAI,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,CAAC;IACxD,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IACpC,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;IACnD,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAC;IAC9C,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;IACnD,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,SAAS,CAAC,CAAC;IAC9C,IAAI,CAAC,MAAM,CAAC,eAAe,EAAC,QAAQ,CAAC,CAAC;IACtC,kGAAkG;IAClG,uCAAuC;IACvC,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE;QACvC,eAAe,EAAE,KAAK;QACtB,OAAO,EAAE;YACP,GAAG,IAAI,CAAC,UAAU,EAAE;YACpB,aAAa,EAAE,UAAU,GAAG,EAAE;YAC9B,qBAAqB,EAAC,QAAQ;SAC/B;QACD,IAAI,EAAE,IAAI;KACX,CAAC,CAAC;IAEH,IAAI,QAAQ,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;QAChC,MAAM,IAAI,OAAO,CAAC,8CAA8C,SAAS,iBAAiB,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;IAC7G,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAiB,CAAC;AACnD,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,eAAe,GAAG,KAAK,EAClC,SAAiB,EACjB,cAAsB,EACtB,SAAiB,EACjB,cAAsB,EACI,EAAE;IAC5B,MAAM,aAAa,GAAG,GAAG,SAAS,gBAAgB,CAAC;IACnD,MAAM,aAAa,GAAG,GAAG,SAAS,gBAAgB,CAAC;IAEnD,MAAM,CAAC,WAAW,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACnD,eAAe,CAAC,aAAa,EAAE,cAAc,CAAC;QAC9C,eAAe,CAAC,aAAa,EAAE,cAAc,CAAC;KAC/C,CAAC,CAAC;IAEH,IAAI,WAAW,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAoB,CAAC;IACzD,CAAC;IAED,IAAI,WAAW,CAAC,UAAU,KAAK,KAAK,EAAE,CAAC;QACrC,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,IAAI,CAAoB,CAAC;IACzD,CAAC;IAED,MAAM,IAAI,OAAO,CACf,6GAA6G,WAAW,CAAC,UAAU,MAAM,WAAW,CAAC,IAAI,sCAAsC,WAAW,CAAC,UAAU,MAAM,WAAW,CAAC,IAAI,EAAE,CAC9O,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,eAAe,GAAG,KAAK,EAAE,GAAW,EAAE,WAAmB,EAAgC,EAAE;IAC/F,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE;QACnC,eAAe,EAAE,KAAK;QACtB,OAAO,EAAE;YACP,aAAa,EAAE,UAAU,WAAW,EAAE;SACvC;KACF,CAAC,CAAC;IACH,OAAO;QACL,UAAU,EAAE,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,EAAE,oBAAoB;QAChE,IAAI,EAAE,QAAQ,CAAC,IAAI;KACpB,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,KAAK,EAAE,KAAa,EAAE,GAAW,EAA6B,EAAE;IAC5F,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;IAEpD,kGAAkG;IAClG,uCAAuC;IACvC,MAAM,OAAO,GAAG;QACd,aAAa,EAAE,UAAU,GAAG,EAAE;QAC9B,qBAAqB,EAAE,QAAQ;KAChC,CAAC;IACF,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,GAAG,CAAC,GAAG,OAAO,IAAI,KAAK,EAAE,EAAE,EAAE,eAAe,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;IAE3F,IAAI,QAAQ,CAAC,UAAU,KAAK,GAAG,EAAE,CAAC;QAChC,uEAAuE;QACvE,MAAM,IAAI,OAAO,CAAC,0CAA0C,KAAK,EAAE,CAAC,CAAC;IACvE,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAqB,CAAC;IAC3D,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAEnB,OAAO,IAAI,CAAC;AACd,CAAC,CAAC"}

package/lib/utils/cache.d.ts

@@ -0,0 +1,20 @@
+import { TTLConfig } from '@salesforce/core';
+export type SeedingOptions = {
+    jobId: string;
+};
+export type SeedingCacheConfig = {
+    jobId: string;
+};
+export declare abstract class DataSeedingRequestCache extends TTLConfig<TTLConfig.Options, SeedingCacheConfig> {
+    static getDefaultOptions(): TTLConfig.Options;
+    createCacheEntry(jobId: string): Promise<void>;
+    resolveFromCache(): SeedingOptions;
+}
+export declare class GenerateRequestCache extends DataSeedingRequestCache {
+    static getDefaultOptions(): TTLConfig.Options;
+    static unset(key: string): Promise<void>;
+}
+export declare class MigrateRequestCache extends DataSeedingRequestCache {
+    static getDefaultOptions(): TTLConfig.Options;
+    static unset(key: string): Promise<void>;
+}

package/lib/utils/cache.js

@@ -0,0 +1,65 @@
+/*
+ * Copyright (c) 2023, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+import { TTLConfig, Global, SfError } from '@salesforce/core';
+import { Duration } from '@salesforce/kit';
+export class DataSeedingRequestCache extends TTLConfig {
+    static getDefaultOptions() {
+        return {
+            isGlobal: true,
+            isState: true,
+            filename: DataSeedingRequestCache.getFileName(),
+            stateFolder: Global.SF_STATE_FOLDER,
+            ttl: Duration.days(14),
+        };
+    }
+    async createCacheEntry(jobId) {
+        if (!jobId)
+            throw new SfError('Job ID is required to create a cache entry');
+        this.set(jobId, { jobId });
+        await this.write();
+    }
+    resolveFromCache() {
+        const key = this.getLatestKey();
+        if (!key)
+            throw new SfError('Could not find a job ID to resume');
+        const { jobId } = this.get(key);
+        return { jobId };
+    }
+}
+export class GenerateRequestCache extends DataSeedingRequestCache {
+    static getDefaultOptions() {
+        return {
+            isGlobal: true,
+            isState: true,
+            filename: 'data-seeding-generate-request-cache.json',
+            stateFolder: Global.SF_STATE_FOLDER,
+            ttl: Duration.days(14),
+        };
+    }
+    static async unset(key) {
+        const cache = await GenerateRequestCache.create();
+        cache.unset(key);
+        await cache.write();
+    }
+}
+export class MigrateRequestCache extends DataSeedingRequestCache {
+    static getDefaultOptions() {
+        return {
+            isGlobal: true,
+            isState: true,
+            filename: 'data-seeding-migrate-request-cache.json',
+            stateFolder: Global.SF_STATE_FOLDER,
+            ttl: Duration.days(14),
+        };
+    }
+    static async unset(key) {
+        const cache = await MigrateRequestCache.create();
+        cache.unset(key);
+        await cache.write();
+    }
+}
+//# sourceMappingURL=cache.js.map

package/lib/utils/cache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../../src/utils/cache.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAC9D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAU3C,MAAM,OAAgB,uBAAwB,SAAQ,SAAgD;IAC7F,MAAM,CAAC,iBAAiB;QAC7B,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,uBAAuB,CAAC,WAAW,EAAE;YAC/C,WAAW,EAAE,MAAM,CAAC,eAAe;YACnC,GAAG,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;SACvB,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,KAAa;QACzC,IAAI,CAAC,KAAK;YAAE,MAAM,IAAI,OAAO,CAAC,4CAA4C,CAAC,CAAC;QAE5E,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3B,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAEM,gBAAgB;QACrB,MAAM,GAAG,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAChC,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,OAAO,CAAC,mCAAmC,CAAC,CAAC;QAEjE,MAAM,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAChC,OAAO,EAAE,KAAK,EAAE,CAAC;IACnB,CAAC;CACF;AAED,MAAM,OAAO,oBAAqB,SAAQ,uBAAuB;IACxD,MAAM,CAAC,iBAAiB;QAC7B,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,0CAA0C;YACpD,WAAW,EAAE,MAAM,CAAC,eAAe;YACnC,GAAG,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;SACvB,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAW;QACnC,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,MAAM,EAAE,CAAC;QAClD,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;CACF;AAED,MAAM,OAAO,mBAAoB,SAAQ,uBAAuB;IACvD,MAAM,CAAC,iBAAiB;QAC7B,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,IAAI;YACb,QAAQ,EAAE,yCAAyC;YACnD,WAAW,EAAE,MAAM,CAAC,eAAe;YACnC,GAAG,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;SACvB,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,GAAW;QACnC,MAAM,KAAK,GAAG,MAAM,mBAAmB,CAAC,MAAM,EAAE,CAAC;QACjD,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACjB,MAAM,KAAK,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;CACF"}

package/lib/utils/mso.d.ts

@@ -0,0 +1,15 @@
+import { MultiStageOutput, MultiStageOutputOptions } from '@oclif/multi-stage-output';
+type SeedGenerateMsoData = {
+    jobId: string;
+    startTime?: string;
+    endTime?: string;
+    sourceOrg?: string;
+    targetOrg?: string;
+    status?: string;
+};
+type SeedMsoOverrides = Partial<MultiStageOutputOptions<SeedGenerateMsoData>>;
+export declare const getSeedGenerateStage: (step: string) => string;
+export declare const getSeedGenerateMso: (overrides?: SeedMsoOverrides) => MultiStageOutput<SeedGenerateMsoData>;
+export declare const getSeedMigrateStage: (step: string) => string;
+export declare const getSeedMigrateMso: (overrides?: SeedMsoOverrides) => MultiStageOutput<SeedGenerateMsoData>;
+export {};

package/lib/utils/mso.js

@@ -0,0 +1,112 @@
+/*
+ * Copyright (c) 2024, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+import { MultiStageOutput } from '@oclif/multi-stage-output';
+import { StandardColors } from '@salesforce/sf-plugins-core';
+import { SfError } from '@salesforce/core';
+/* Data Seed Generation */
+// The keys in this Map are steps returned by the server
+//  - They have been converted to lowercase for later comparison
+// The values in this Map are used as the stage names in mso
+const seedGenerateStagesMap = new Map([
+    ['init', 'Initializing'],
+    ['querying source org', 'Querying Source Org'],
+    ['data generation', 'Data Generation'],
+    ['populating target org', 'Populating Target Org'],
+    ['pipeline finished', 'Finalizing'],
+]);
+export const getSeedGenerateStage = (step) => {
+    const stage = seedGenerateStagesMap.get(step.toLowerCase());
+    if (!stage)
+        throw new SfError(`Unable to map step (${step}) from server to multi-stage-output stage`);
+    return stage;
+};
+export const getSeedGenerateMso = (overrides = {}) => {
+    const defaults = {
+        title: 'Data Seed Generation',
+        stages: Array.from(seedGenerateStagesMap.values()),
+        ...overrides,
+    };
+    return getSeedMso(defaults);
+};
+/* Data Seed Migration */
+const seedMigrateStagesMap = new Map([
+    ['querying source org', 'Querying Source Org'],
+    ['populating target org', 'Populating Target Org'],
+    ['pipeline finished', 'Finalizing'],
+]);
+export const getSeedMigrateStage = (step) => {
+    const stage = seedMigrateStagesMap.get(step.toLowerCase());
+    if (!stage)
+        throw new SfError(`Unable to map step (${step}) from server to multi-stage-output stage`);
+    return stage;
+};
+export const getSeedMigrateMso = (overrides = {}) => {
+    const defaults = {
+        title: 'Data Seed Migration',
+        stages: Array.from(seedMigrateStagesMap.values()),
+        ...overrides,
+    };
+    return getSeedMso(defaults);
+};
+/* Shared MSO config */
+const getSeedMso = (overrides = {}) => new MultiStageOutput({
+    preStagesBlock: [
+        {
+            get: (data) => data?.jobId,
+            type: 'static-key-value',
+            label: 'Job ID',
+            neverCollapse: true,
+        },
+        {
+            get: (data) => data?.sourceOrg,
+            type: 'static-key-value',
+            label: 'Source Org',
+        },
+        {
+            get: (data) => data?.targetOrg,
+            type: 'static-key-value',
+            label: 'Target Org',
+        },
+    ],
+    postStagesBlock: [
+        {
+            get: (data) => {
+                const status = data?.status;
+                switch (status) {
+                    case 'Completed':
+                    case 'Initiated':
+                        return StandardColors.success(status);
+                    case 'Client Timeout':
+                    case 'Partially Completed':
+                        return StandardColors.warning(status);
+                    case 'Failed':
+                        return StandardColors.error(status);
+                    default:
+                        return status;
+                }
+            },
+            type: 'static-key-value',
+            label: 'Status',
+            neverCollapse: true,
+        },
+        {
+            get: (data) => data?.startTime,
+            type: 'static-key-value',
+            label: 'Start Time',
+        },
+        {
+            get: (data) => data?.endTime,
+            type: 'static-key-value',
+            label: 'End Time',
+        },
+    ],
+    jsonEnabled: false,
+    stages: [],
+    title: '',
+    ...overrides,
+});
+//# sourceMappingURL=mso.js.map

package/lib/utils/mso.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mso.js","sourceRoot":"","sources":["../../src/utils/mso.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,gBAAgB,EAA2B,MAAM,2BAA2B,CAAC;AACtF,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAe3C,0BAA0B;AAE1B,wDAAwD;AACxD,gEAAgE;AAChE,4DAA4D;AAC5D,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAiB;IACpD,CAAC,MAAM,EAAC,cAAc,CAAC;IACvB,CAAC,qBAAqB,EAAE,qBAAqB,CAAC;IAC9C,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;IACtC,CAAC,uBAAuB,EAAE,uBAAuB,CAAC;IAClD,CAAC,mBAAmB,EAAE,YAAY,CAAC;CACpC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,IAAY,EAAU,EAAE;IAC3D,MAAM,KAAK,GAAG,qBAAqB,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC5D,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,OAAO,CAAC,uBAAuB,IAAI,2CAA2C,CAAC,CAAC;IACtG,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAAC,YAA8B,EAAE,EAAyC,EAAE;IAC5G,MAAM,QAAQ,GAAqB;QACjC,KAAK,EAAE,sBAAsB;QAC7B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAAC;QAClD,GAAG,SAAS;KACb,CAAC;IAEF,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC;AAC9B,CAAC,CAAC;AAEF,yBAAyB;AAEzB,MAAM,oBAAoB,GAAG,IAAI,GAAG,CAAiB;IACnD,CAAC,qBAAqB,EAAE,qBAAqB,CAAC;IAC9C,CAAC,uBAAuB,EAAE,uBAAuB,CAAC;IAClD,CAAC,mBAAmB,EAAE,YAAY,CAAC;CACpC,CAAC,CAAC;AAEH,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,IAAY,EAAU,EAAE;IAC1D,MAAM,KAAK,GAAG,oBAAoB,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC3D,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,OAAO,CAAC,uBAAuB,IAAI,2CAA2C,CAAC,CAAC;IACtG,OAAO,KAAK,CAAC;AACf,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,YAA8B,EAAE,EAAyC,EAAE;IAC3G,MAAM,QAAQ,GAAqB;QACjC,KAAK,EAAE,qBAAqB;QAC5B,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC;QACjD,GAAG,SAAS;KACb,CAAC;IAEF,OAAO,UAAU,CAAC,QAAQ,CAAC,CAAC;AAC9B,CAAC,CAAC;AAEF,uBAAuB;AAEvB,MAAM,UAAU,GAAG,CAAC,YAA8B,EAAE,EAAyC,EAAE,CAC7F,IAAI,gBAAgB,CAAC;IACnB,cAAc,EAAE;QACd;YACE,GAAG,EAAE,CAAC,IAAI,EAAU,EAAE,CAAC,IAAI,EAAE,KAAK;YAClC,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,QAAQ;YACf,aAAa,EAAE,IAAI;SACpB;QACD;YACE,GAAG,EAAE,CAAC,IAAI,EAAU,EAAE,CAAC,IAAI,EAAE,SAAS;YACtC,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,YAAY;SACpB;QACD;YACE,GAAG,EAAE,CAAC,IAAI,EAAU,EAAE,CAAC,IAAI,EAAE,SAAS;YACtC,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,YAAY;SACpB;KACF;IACD,eAAe,EAAE;QACf;YACE,GAAG,EAAE,CAAC,IAAI,EAAU,EAAE;gBACpB,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,CAAC;gBAC5B,QAAQ,MAAM,EAAE,CAAC;oBACf,KAAK,WAAW,CAAC;oBACjB,KAAK,WAAW;wBACd,OAAO,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBACxC,KAAK,gBAAgB,CAAC;oBACtB,KAAK,qBAAqB;wBACxB,OAAO,cAAc,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;oBACxC,KAAK,QAAQ;wBACX,OAAO,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;oBACtC;wBACE,OAAO,MAAM,CAAC;gBAClB,CAAC;YACH,CAAC;YACD,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,QAAQ;YACf,aAAa,EAAE,IAAI;SACpB;QACD;YACE,GAAG,EAAE,CAAC,IAAI,EAAU,EAAE,CAAC,IAAI,EAAE,SAAS;YACtC,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,YAAY;SACpB;QACD;YACE,GAAG,EAAE,CAAC,IAAI,EAAU,EAAE,CAAC,IAAI,EAAE,OAAO;YACpC,IAAI,EAAE,kBAAkB;YACxB,KAAK,EAAE,UAAU;SAClB;KACF;IACD,WAAW,EAAE,KAAK;IAClB,MAAM,EAAE,EAAE;IACV,KAAK,EAAE,EAAE;IACT,GAAG,SAAS;CACb,CAAC,CAAC"}

package/lib/utils/types.d.ts

@@ -0,0 +1,16 @@
+export type DataSeedingJob = 'generate' | 'migrate';
+export type DataSeedingReportResult = {
+    dataSeedingJob: DataSeedingJob;
+    jobId: string;
+    startTime?: string;
+    endTime?: string;
+    status?: string;
+};
+export type DataSeedingGenerateResult = DataSeedingReportResult & {
+    sourceOrg: string;
+    targetOrg: string;
+};
+export type DataSeedingMigrateResult = DataSeedingReportResult & {
+    sourceOrg: string;
+    targetOrg: string;
+};

package/lib/utils/types.js

@@ -0,0 +1,8 @@
+/*
+ * Copyright (c) 2023, salesforce.com, inc.
+ * All rights reserved.
+ * Licensed under the BSD 3-Clause license.
+ * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
+ */
+export {};
+//# sourceMappingURL=types.js.map

package/lib/utils/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/utils/types.ts"],"names":[],"mappings":"AAAA;;;;;GAKG"}