fix: save for later

Author: rajender-kumar-salesforce

src/commands/omnistudio/migration/assess.ts

@@ -5,6 +5,7 @@ import { SfCommand, Ux, Flags as flags } from '@salesforce/sf-plugins-core';
 import { AssessmentInfo } from '../../../utils/interfaces';
 import { AssessmentReporter } from '../../../utils/resultsbuilder/assessmentReporter';
 import { OmniScriptExportType, OmniScriptMigrationTool } from '../../../migration/omniscript';
+import { OmniScriptInstanceMigrationTool } from '../../../migration/omniscriptInstance';
 import { InvalidEntityTypeError } from '../../../migration/interfaces';
 import { CardMigrationTool } from '../../../migration/flexcard';
 import { DataRaptorMigrationTool } from '../../../migration/dataraptor';
@@ -183,6 +184,7 @@ export default class Assess extends SfCommand<AssessmentInfo> {
         warnings: 0,
         failed: 0,
       },
+      saveForLaterAssessmentInfos: [],
     };
 
     Logger.log(messages.getMessage('assessmentInitialization', [String(namespace)]));
@@ -280,6 +282,8 @@ export default class Assess extends SfCommand<AssessmentInfo> {
       await this.assessFlexCards(assesmentInfo, namespace, conn, allVersions, ux);
       await this.assessOmniScripts(assesmentInfo, namespace, conn, allVersions, OmniScriptExportType.OS, ux);
       await this.assessOmniScripts(assesmentInfo, namespace, conn, allVersions, OmniScriptExportType.IP, ux);
+      // Assess Save for Later after OmniScripts (depends on OmniScript assessment)
+      await this.assessSaveForLater(assesmentInfo, namespace, conn, ux);
       if (!isFoundationPackage()) {
         await this.assessGlobalAutoNumbers(assesmentInfo, namespace, conn, ux);
       }
@@ -395,6 +399,23 @@ export default class Assess extends SfCommand<AssessmentInfo> {
     }
   }
 
+  private async assessSaveForLater(
+    assesmentInfo: AssessmentInfo,
+    namespace: string,
+    conn: Connection,
+    ux: Ux
+  ): Promise<void> {
+    const saveForLaterMigrator = new OmniScriptInstanceMigrationTool(namespace, conn, Logger, messages, ux);
+    // Pass OmniScript assessment info to check dependencies
+    assesmentInfo.saveForLaterAssessmentInfos = await saveForLaterMigrator.assess(assesmentInfo.omniAssessmentInfo);
+    this.logAssessmentCompletionIfNeeded(
+      'assessedOmniScriptsCount',
+      'omniScriptAssessmentCompleted',
+      assesmentInfo.saveForLaterAssessmentInfos.length,
+      ['OmniScript Saved Sessions']
+    );
+  }
+
   private async assessGlobalAutoNumbers(
     assesmentInfo: AssessmentInfo,
     namespace: string,

src/commands/omnistudio/migration/migrate.ts

@@ -18,6 +18,7 @@ import { InvalidEntityTypeError, MigrationResult, MigrationTool } from '../../..
 import { ResultsBuilder } from '../../../utils/resultsbuilder';
 import { CardMigrationTool } from '../../../migration/flexcard';
 import { OmniScriptExportType, OmniScriptMigrationTool } from '../../../migration/omniscript';
+import { OmniScriptInstanceMigrationTool } from '../../../migration/omniscriptInstance';
 import { CustomLabelsMigrationTool } from '../../../migration/customLabels';
 import { Logger } from '../../../utils/logger';
 import OmnistudioRelatedObjectMigrationFacade from '../../../migration/related/OmnistudioRelatedObjectMigrationFacade';
@@ -249,6 +250,10 @@ export default class Migrate extends SfCommand<MigrateResult> {
     const relatedObjectMigrationResult = omnistudioRelatedObjectsMigration.migrateAll(objectsToProcess);
 
     // POST MIGRATION
+    // Note: Post migration executeTasks for Omniscript is commented out as per requirements
+    // Post migration tasks (enableDesignersToUseStandardDataModelIfNeeded, enableStandardRuntimeIfNeeded, etc.)
+    // are not executed for OmniScript migration
+    // However, deploy is still executed for related objects
 
     const postMigrate: PostMigrate = new PostMigrate(
       org,
@@ -262,9 +267,10 @@ export default class Migrate extends SfCommand<MigrateResult> {
       projectPath
     );
 
-    if (!migrateOnly) {
-      await postMigrate.executeTasks(namespace, actionItems);
-    }
+    // Commented out post migration tasks for OmniScript as per requirements
+    // if (!migrateOnly) {
+    //   await postMigrate.executeTasks(namespace, actionItems);
+    // }
 
     const migrationActionItems = this.collectActionItems(objectMigrationResults);
     actionItems = [...actionItems, ...migrationActionItems];
@@ -556,6 +562,8 @@ export default class Migrate extends SfCommand<MigrateResult> {
         new OmniScriptMigrationTool(OmniScriptExportType.IP, namespace, conn, logger, messages, ux, allVersions),
         // OmniScript
         new OmniScriptMigrationTool(OmniScriptExportType.OS, namespace, conn, logger, messages, ux, allVersions),
+        // Save for Later (must run after OmniScript migration to use NameMappingRegistry)
+        new OmniScriptInstanceMigrationTool(namespace, conn, logger, messages, ux),
         new CardMigrationTool(namespace, conn, logger, messages, ux, allVersions),
         new CustomLabelsMigrationTool(namespace, conn, logger, messages, ux),
       ];