Org preference enablement added to disable rollback flags

Author: RedisBot

src/commands/omnistudio/migration/assess.ts

@@ -11,7 +11,7 @@ import { DebugTimer } from '../../../utils';
 import { Logger } from '../../../utils/logger';
 import OmnistudioRelatedObjectMigrationFacade from '../../../migration/related/OmnistudioRelatedObjectMigrationFacade';
 import { OmnistudioOrgDetails, OrgUtils } from '../../../utils/orgUtils';
-import { Constants } from '../../../utils/constants/stringContants';
+import { OrgPreferences } from '../../../utils/orgpreferences';
 
 Messages.importMessagesDirectory(__dirname);
 const messages = Messages.loadMessages('@salesforce/plugin-omnistudio-migration-tool', 'assess');
@@ -108,8 +108,8 @@ export default class Assess extends OmniStudioBaseCommand {
       assesmentInfo.lwcAssessmentInfos = relatedObjectAssessmentResult.lwcAssessmentInfos;
       assesmentInfo.apexAssessmentInfos = relatedObjectAssessmentResult.apexAssessmentInfos;
     }
-
-    await AssessmentReporter.generate(assesmentInfo, conn.instanceUrl, orgs, assessOnly, objectsToProcess);
+    orgs.rollbackFlags = await OrgPreferences.checkRollbackFlags(conn);
+    await AssessmentReporter.generate(assesmentInfo, conn.instanceUrl, orgs);
     return assesmentInfo;
   }
 

src/commands/omnistudio/migration/migrate.ts

@@ -91,12 +91,10 @@ export default class Migrate extends OmniStudioBaseCommand {
 
     // Enable Omni preferences
     try {
-      this.ux.log('Enabling Omni preferences...');
       await OrgPreferences.enableOmniPreferences(conn);
-      this.ux.log('Omni preferences enabled successfully.');
     } catch (error) {
       const errMsg = error instanceof Error ? error.message : String(error);
-      this.ux.log(`Error: Could not enable Omni preferences: ${errMsg}`);
+      this.ux.error(`Could not enable Omni preferences: ${errMsg}`);
       throw error;
     }
 

src/utils/interfaces.ts

@@ -144,3 +144,23 @@ export interface oldNew {
   old: string;
   new: string;
 }
+
+export interface OmniStudioSettingsMetadata {
+  fullName: string;
+  disableRollbackFlagsPref: boolean;
+}
+
+export interface QueryResult {
+  DeveloperName: string;
+  Value: string;
+  totalSize: number;
+  done: boolean;
+  records: Array<{
+    attributes: {
+      type: string;
+      url: string;
+    };
+    DeveloperName: string;
+    Value: string;
+  }>;
+}

src/utils/orgpreferences.ts

@@ -1,41 +1,5 @@
 import { Connection } from '@salesforce/core';
-
-/**
- * Interface representing the metadata structure for OmniStudio settings
- *
- * @interface OmniStudioSettingsMetadata
- * @property {string} fullName - The full name of the OmniStudio settings
- * @property {boolean} disableRollbackFlagsPref - Flag to disable rollback preferences
- */
-interface OmniStudioSettingsMetadata {
-  fullName: string;
-  disableRollbackFlagsPref: boolean;
-}
-
-/**
- * Interface representing the structure of query results from OmniInteractionConfig
- *
- * @interface QueryResult
- * @property {string} DeveloperName - Name of the configuration
- * @property {string} Value - Value of the configuration
- * @property {number} totalSize - Total number of records returned
- * @property {boolean} done - Whether the query is complete
- * @property {Array<{attributes: {type: string, url: string}, DeveloperName: string, Value: string}>} records - Array of query result records
- */
-interface QueryResult {
-  DeveloperName: string;
-  Value: string;
-  totalSize: number;
-  done: boolean;
-  records: Array<{
-    attributes: {
-      type: string;
-      url: string;
-    };
-    DeveloperName: string;
-    Value: string;
-  }>;
-}
+import { OmniStudioSettingsMetadata, QueryResult } from './interfaces';
 
 /**
  * Class to manage OmniStudio organization preferences
@@ -107,4 +71,4 @@ export class OrgPreferences {
       throw new Error(`Failed to check rollback flags: ${error instanceof Error ? error.message : String(error)}`);
     }
   }
-}
+}

src/utils/resultsbuilder/assessmentReporter.ts

@@ -29,6 +29,7 @@ export class AssessmentReporter {
     const integrationProcedureAssessmentFilePath = basePath + '/integration_procedure_assessment.html';
     const dataMapperAssessmentFilePath = basePath + '/datamapper_assessment.html';
     const apexAssessmentFilePath = basePath + '/apex_assessment.html';
+    const rollbackFlagsReportPath = basePath + '/rollback_flags_report.html';
     // TODO: Uncomment code once MVP for migration is completed
     // const lwcAssessmentFilePath = basePath + '/lwc_assessment.html';
     const orgDetails: ReportHeaderFormat[] = this.formattedOrgDetails(omnistudioOrgDetails);
@@ -145,8 +146,7 @@ export class AssessmentReporter {
       //   location: 'lwc_assessment.html',
       // },
     ];
-    
-    // Check rollback flags
+        // Check rollback flags
     const enabledFlags = omnistudioOrgDetails.rollbackFlags || [];
     if (enabledFlags.length > 0) {
       const rollbackFlagsReportPath = basePath + '/rollback_flags_report.html';
@@ -302,13 +302,18 @@ export class AssessmentReporter {
   }
   private static generateRollbackFlagsReport(enabledFlags: string[]): string {
     return `
-      <div class="slds-box slds-theme_warning" style="margin-bottom: 20px;">
-        <div class="slds-text-heading_medium slds-m-bottom_small">⚠️ Warning: Rollback Flags Enabled</div>
+      <div class="slds-box rollback-flags-warning">
+        <div class="slds-text-heading_medium">
+          <svg class="slds-icon slds-icon_small slds-m-right_x-small" aria-hidden="true">
+            <use xlink:href="/assets/icons/utility-sprite/svg/symbols.svg#warning"></use>
+          </svg>
+          Warning: Rollback Flags Enabled
+        </div>
         <p>The following rollback flags are currently enabled and will be disabled during migration:</p>
-        <ul class="slds-m-top_small">
+        <ul>
           ${enabledFlags.map((flag) => `<li>${flag}</li>`).join('')}
         </ul>
-        <p class="slds-m-top_small">
+        <p>
           <strong>Note:</strong> These flags will not be supported after migration. For assistance, please contact support.
         </p>
       </div>