chore: saving the changes

Author: sf-aastha-paruthi

messages/migrate.json

@@ -263,5 +263,8 @@
   "errorEnablingOmniStudioSettingsMetadata": "We couldn't enable the Omnistudio Metadata setting: %s. Enable it manually.",
   "manuallyEnableOmniStudioSettingsMetadata": "Manually enable the Omnistudio Metadata setting in your org’s Omnistudio Settings page.",
   "omniStudioMetadataEnableConsentNotGiven": "You’ve not consented to proceed with enabling the Omnistudio Metadata setting. We’ll not be able to proceed with the migration.",
-  "enablingOmniStudioSettingsMetadataStatus": "Enabling Omnistudio Metadata setting…"
+  "enablingOmniStudioSettingsMetadataStatus": "Enabling Omnistudio Metadata setting…",
+  "omniStudioAllVersionsMigrationConsent": "All Versions of omnistudio components [Omniscript, Data Mapper, Integration Procedure, Flexcard] need to be migrated for metadata to be enabled as org is on Standard Data Model. Do you agree to migrate all versions? [y/n]",
+  "omniStudioAllVersionsMigrationConsentNotGiven": "You've not consented to proceed with the all versions migration. We'll not be able to proceed with the migration.",
+  "omniStudioAllVersionsMigrationConsentGiven": "You've consented to proceed with the all versions migration."
 }

src/commands/omnistudio/migration/migrate.ts

@@ -89,7 +89,7 @@ export default class Migrate extends OmniStudioBaseCommand {
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   public async runMigration(): Promise<any> {
     const migrateOnly = (this.flags.only || '') as string;
-    const allVersions = this.flags.allversions || (false as boolean);
+    let allVersions = this.flags.allversions || (false as boolean);
     const relatedObjects = (this.flags.relatedobjects || '') as string;
     // this.org is guaranteed because requiresUsername=true, as opposed to supportsUsername
     const conn = this.org.getConnection();
@@ -132,18 +132,8 @@ export default class Migrate extends OmniStudioBaseCommand {
     const isExperienceBundleMetadataAPIProgramaticallyEnabled: { value: boolean } = { value: false };
 
     if (isStandardDataModel()) {
-      // Get user consent to enable OmniStudio Metadata for standard data model migration
-      const omniStudioMetadataEnableConsent = await preMigrate.getOmniStudioMetadataEnableConsent();
-      if (!omniStudioMetadataEnableConsent) {
-        Logger.error(messages.getMessage('omniStudioMetadataEnableConsentNotGiven'));
-        return;
-      }
-
-      // Handle config tables cleanup for standard data model migration
-      const isMetadataCleanupSuccess = await preMigrate.handleOmniStudioMetadataCleanup();
-      if (!isMetadataCleanupSuccess) {
-        return;
-      }
+      allVersions = await preMigrate.handleAllVersionsPrerequisites(allVersions);
+      await preMigrate.handleOmnistudioMetadataPrerequisites();
     }
 
     let actionItems = [];

src/migration/premigrate.ts

@@ -17,6 +17,43 @@ export class PreMigrate extends BaseMigrationTool {
     super(namespace, connection, logger, messages, ux);
   }
 
+  /**
+   * Ensures all versions are migrated when on standard data model.
+   * If the -a flag was not provided, prompts user for consent.
+   *
+   * @param allVersionsFlagFromCLI - The allVersions flag value from CLI (-a flag)
+   * @returns true if all versions should be migrated, false otherwise
+   */
+  public async handleAllVersionsPrerequisites(allVersionsFlagFromCLI: boolean): Promise<boolean> {
+    if (allVersionsFlagFromCLI === false) {
+      // Get user consent to migrate allversions of OmniStudio components for standard data model migration
+      const omniStudioAllVersionsMigrationConsent = await this.getOmnistudioAllVersionsMigrationConsent();
+      if (!omniStudioAllVersionsMigrationConsent) {
+        Logger.error(this.messages.getMessage('omniStudioAllVersionsMigrationConsentNotGiven'));
+        process.exit(1);
+      }
+
+      Logger.logVerbose(this.messages.getMessage('omniStudioAllVersionsMigrationConsentGiven'));
+      return true;
+    }
+    return allVersionsFlagFromCLI;
+  }
+
+  public async handleOmnistudioMetadataPrerequisites(): Promise<void> {
+    // Get user consent to enable OmniStudio Metadata for standard data model migration
+    const omniStudioMetadataEnableConsent = await this.getOmniStudioMetadataEnableConsent();
+    if (!omniStudioMetadataEnableConsent) {
+      Logger.error(this.messages.getMessage('omniStudioMetadataEnableConsentNotGiven'));
+      process.exit(1);
+    }
+
+    // Handle config tables cleanup for standard data model migration
+    const isMetadataCleanupSuccess = await this.handleOmniStudioMetadataCleanup();
+    if (!isMetadataCleanupSuccess) {
+      process.exit(1);
+    }
+  }
+
   public async handleExperienceSitePrerequisites(
     objectsToProcess: string[],
     conn: Connection,
@@ -241,4 +278,39 @@ export class PreMigrate extends BaseMigrationTool {
     }
     return consent;
   }
+
+  /**
+   * Gets user consent for OmniStudio migrating all versions of Omnistudio Components
+   *
+   * @returns Promise<boolean> - true if user consents, false otherwise
+   */
+  private async getOmnistudioAllVersionsMigrationConsent(): Promise<boolean> {
+    const askWithTimeOut = PromptUtil.askWithTimeOut(this.messages);
+    let validResponse = false;
+    let consent = false;
+
+    while (!validResponse) {
+      try {
+        const resp = await askWithTimeOut(
+          Logger.prompt.bind(Logger),
+          this.messages.getMessage('omniStudioAllVersionsMigrationConsent')
+        );
+        const response = typeof resp === 'string' ? resp.trim().toLowerCase() : '';
+
+        if (response === YES_SHORT || response === YES_LONG) {
+          consent = true;
+          validResponse = true;
+        } else if (response === NO_SHORT || response === NO_LONG) {
+          consent = false;
+          validResponse = true;
+        } else {
+          Logger.error(this.messages.getMessage('invalidYesNoResponse'));
+        }
+      } catch (err) {
+        Logger.error(this.messages.getMessage('requestTimedOut'));
+        process.exit(1);
+      }
+    }
+    return consent;
+  }
 }