feat: related objects migration enablement for standard data model

Author: shaurabh-tiwari-git

src/migration/flexcard.ts

@@ -234,6 +234,9 @@ export class CardMigrationTool extends BaseMigrationTool implements MigrationToo
         flexCardAssessmentInfo.warnings.push(
           this.messages.getMessage('needManualInterventionAsSpecialCharsInFlexcardName')
         );
+        flexCardAssessmentInfo.errors.push(
+          this.messages.getMessage('needManualInterventionAsSpecialCharsInFlexcardName')
+        );
         assessmentStatus = 'Needs Manual Intervention';
       }
     }
@@ -404,6 +407,9 @@ export class CardMigrationTool extends BaseMigrationTool implements MigrationToo
             flexCardAssessmentInfo.warnings.push(
               this.messages.getMessage('needManualInterventionAsSpecialCharsInChildFlexcardName')
             );
+            flexCardAssessmentInfo.errors.push(
+              this.messages.getMessage('needManualInterventionAsSpecialCharsInChildFlexcardName')
+            );
             flexCardAssessmentInfo.migrationStatus = 'Needs Manual Intervention';
           }
         }

src/migration/interfaces.ts

@@ -175,8 +175,15 @@ export interface ExpSiteComponentAttributes {
   richTextValue?: string;
 }
 
+export interface OmniScriptStandardKey {
+  type: string;
+  subtype: string;
+  language: string;
+}
+
 export interface MigrationStorage {
   osStorage: Map<string, OmniScriptStorage>;
+  osStandardStorage: Map<string, OmniScriptStorage>; // String keys (serialized OmniScriptStandardKey)
   fcStorage: Map<string, FlexcardStorage>;
 }
 

src/migration/omniscript.ts

@@ -19,6 +19,7 @@ import {
   MigrationResult,
   MigrationStorage,
   MigrationTool,
+  OmniScriptStandardKey,
   OmniScriptStorage,
   TransformData,
   UploadRecordResult,
@@ -657,6 +658,16 @@ export class OmniScriptMigrationTool extends BaseMigrationTool implements Migrat
           nameMapping.oldLanguage
         )}`;
 
+        if (this.IS_STANDARD_DATA_MODEL) {
+          // Create object key for new storage format
+          const keyObject: OmniScriptStandardKey = {
+            type: nameMapping.oldType,
+            subtype: nameMapping.oldSubtype,
+            language: this.cleanLanguageName(nameMapping.oldLanguage),
+          };
+          StorageUtil.addStandardOmniScriptToStorage(storage, keyObject, value);
+        }
+
         finalKey = finalKey.toLowerCase();
         if (storage.osStorage.has(finalKey)) {
           // Key already exists - handle accordingly
@@ -1175,6 +1186,16 @@ export class OmniScriptMigrationTool extends BaseMigrationTool implements Migrat
             oldrecord[this.getFieldKey('SubType__c')]
           }${this.cleanLanguageName(oldrecord[this.getFieldKey('Language__c')])}`;
 
+          if (this.IS_STANDARD_DATA_MODEL) {
+            // Create object key for new storage format
+            const keyObject: OmniScriptStandardKey = {
+              type: oldrecord[this.getFieldKey('Type__c')],
+              subtype: oldrecord[this.getFieldKey('SubType__c')],
+              language: this.cleanLanguageName(oldrecord[this.getFieldKey('Language__c')]),
+            };
+            StorageUtil.addStandardOmniScriptToStorage(storage, keyObject, value);
+          }
+
           finalKey = finalKey.toLowerCase();
           if (storage.osStorage.has(finalKey)) {
             // Key already exists - handle accordingly

src/migration/related/ExperienceSiteMigration.ts

@@ -9,6 +9,7 @@ import {
   ExpSiteComponent,
   ExpSiteComponentAttributes,
   MigrationStorage,
+  OmniScriptStandardKey,
   OmniScriptStorage,
   ExpSitePageJson,
   Storage,
@@ -19,19 +20,22 @@ import { FileDiffUtil } from '../../utils/lwcparser/fileutils/FileDiffUtil';
 import { ExperienceSiteAssessmentInfo, ExperienceSiteAssessmentPageInfo } from '../../utils';
 import { StorageUtil } from '../../utils/storageUtil';
 import { createProgressBar } from '../base';
+import { isStandardDataModel } from '../../utils/dataModelService';
 import { BaseRelatedObjectMigration } from './BaseRealtedObjectMigration';
 
 Messages.importMessagesDirectory(__dirname);
 
 const TARGET_COMPONENT_NAME_OS = 'runtime_omnistudio:omniscript';
 const TARGET_COMPONENT_NAME_FC = 'runtime_omnistudio:flexcard';
+const TARGET_COMPONENT_NAME_OS_EXP = 'runtime_omnistudio:omniscriptExperienceCloud';
 const FLEXCARD_PREFIX = 'cf';
 
 export class ExperienceSiteMigration extends BaseRelatedObjectMigration {
   private EXPERIENCE_SITES_PATH: string;
   private MIGRATE = 'Migrate';
   private ASSESS = 'Assess';
   private messages: Messages;
+  private IS_STANDARD_DATA_MODEL: boolean = isStandardDataModel();
 
   public constructor(projectPath: string, namespace: string, org: Org, messages: Messages) {
     super(projectPath, namespace, org);
@@ -235,6 +239,7 @@ export class ExperienceSiteMigration extends BaseRelatedObjectMigration {
       return;
     }
 
+    // Check for legacy wrapper component
     if (component.componentName === lookupComponentName) {
       Logger.logVerbose(this.messages.getMessage('omniWrapperFound'));
       experienceSiteAssessmentInfo.hasOmnistudioContent = true;
@@ -250,6 +255,18 @@ export class ExperienceSiteMigration extends BaseRelatedObjectMigration {
       return;
     }
 
+    if (this.IS_STANDARD_DATA_MODEL) {
+      // Check for new LWC components that need reference updates
+      if (this.isOmnistudioComponent(component.componentName)) {
+        Logger.logVerbose(`Found Omnistudio component: ${component.componentName}`);
+        experienceSiteAssessmentInfo.hasOmnistudioContent = true;
+
+        this.updateOmnistudioComponentReferences(component, experienceSiteAssessmentInfo, storage, type);
+
+        return;
+      }
+    }
+
     const regionsInsideComponent: ExpSiteRegion[] = component.regions;
 
     if (Array.isArray(regionsInsideComponent)) {
@@ -360,6 +377,106 @@ export class ExperienceSiteMigration extends BaseRelatedObjectMigration {
     return targetData === undefined || targetData.migrationSuccess === false || targetData.isDuplicate === true;
   }
 
+  /**
+   * Check if component is an Omnistudio LWC component
+   */
+  private isOmnistudioComponent(componentName: string): boolean {
+    return (
+      componentName === TARGET_COMPONENT_NAME_OS ||
+      componentName === TARGET_COMPONENT_NAME_FC ||
+      componentName === TARGET_COMPONENT_NAME_OS_EXP
+    );
+  }
+
+  /**
+   * Update references in Omnistudio LWC components
+   */
+  private updateOmnistudioComponentReferences(
+    component: ExpSiteComponent,
+    experienceSiteAssessmentInfo: ExperienceSiteAssessmentPageInfo,
+    storage: MigrationStorage,
+    type: string
+  ): void {
+    if (component === undefined || component.componentAttributes === undefined) {
+      return;
+    }
+    const componentName = component.componentName;
+    const attributes = component.componentAttributes;
+
+    if (componentName === TARGET_COMPONENT_NAME_OS || componentName === TARGET_COMPONENT_NAME_OS_EXP) {
+      this.updateOmniScriptComponentReferences(attributes, experienceSiteAssessmentInfo, storage, type);
+    } else if (componentName === TARGET_COMPONENT_NAME_FC) {
+      this.updateFlexCardComponentReferences(attributes, experienceSiteAssessmentInfo, storage, type);
+    }
+  }
+
+  /**
+   * Update OmniScript component references (for runtime_omnistudio:omniscript and runtime_omnistudio:omniscriptExperienceCloud)
+   */
+  private updateOmniScriptComponentReferences(
+    attributes: ExpSiteComponentAttributes,
+    experienceSiteAssessmentInfo: ExperienceSiteAssessmentPageInfo,
+    storage: MigrationStorage,
+    type: string
+  ): void {
+    const currentType = attributes['type'] as string;
+    const currentSubType = attributes['subType'] as string;
+    const currentLanguage = attributes['language'] as string;
+
+    if (!currentType || !currentSubType || !currentLanguage) {
+      return;
+    }
+
+    // Create the OmniScriptStandardKey object for lookup in osStandardStorage
+    const lookupKey: OmniScriptStandardKey = {
+      type: currentType,
+      subtype: currentSubType,
+      language: currentLanguage,
+    };
+    // Look up in osStandardStorage using the object key
+    const targetDataFromStorage: OmniScriptStorage = StorageUtil.getStandardOmniScript(storage, lookupKey);
+
+    if (this.shouldAddWarning(targetDataFromStorage)) {
+      const originalKey = `${currentType}_${currentSubType}_${currentLanguage}`;
+      const warningMsg: string = this.getWarningMessage(originalKey, targetDataFromStorage);
+      experienceSiteAssessmentInfo.warnings.push(warningMsg);
+      experienceSiteAssessmentInfo.status = type === this.ASSESS ? 'Needs Manual Intervention' : 'Skipped';
+    } else {
+      // Update the attributes with the new values from storage
+      attributes['type'] = targetDataFromStorage.type;
+      attributes['subType'] = targetDataFromStorage.subtype;
+      attributes['language'] = targetDataFromStorage.language;
+    }
+  }
+
+  /**
+   * Update FlexCard component references (for runtime_omnistudio:flexcard)
+   */
+  private updateFlexCardComponentReferences(
+    attributes: ExpSiteComponentAttributes,
+    experienceSiteAssessmentInfo: ExperienceSiteAssessmentPageInfo,
+    storage: MigrationStorage,
+    type: string
+  ): void {
+    const currentFlexCardName = attributes['flexcardName'] as string;
+
+    if (!currentFlexCardName) {
+      return;
+    }
+
+    // Look up in storage to see if this FlexCard was migrated
+    const targetDataFromStorageFC: FlexcardStorage = storage.fcStorage.get(currentFlexCardName.toLowerCase());
+
+    if (this.shouldAddWarning(targetDataFromStorageFC)) {
+      const warningMsg: string = this.getWarningMessage(currentFlexCardName, targetDataFromStorageFC);
+      experienceSiteAssessmentInfo.warnings.push(warningMsg);
+      experienceSiteAssessmentInfo.status = type === this.ASSESS ? 'Needs Manual Intervention' : 'Skipped';
+    } else {
+      // Update the flexcardName with the new value from storage
+      attributes['flexcardName'] = targetDataFromStorageFC.name;
+    }
+  }
+
   private getWarningMessage(oldTypeSubtypeLanguage: string, targetDataFromStorage: Storage): string {
     if (targetDataFromStorage === undefined) {
       // Add log verbose