@salesforce/plugin-omnistudio-migration-tool 2.0.0-rc.2 → 2.0.0-rc.20

package/lib/migration/omniscript.js

@@ -9,6 +9,7 @@ const Element_1 = __importDefault(require("../mappings/Element"));
 const OmniScriptDefinition_1 = __importDefault(require("../mappings/OmniScriptDefinition"));
 const utils_1 = require("../utils");
 const base_1 = require("./base");
+const interfaces_1 = require("./interfaces");
 const net_1 = require("../utils/net");
 const FormulaUtil_1 = require("../utils/formula/FormulaUtil");
 const stringval_1 = require("../utils/StringValue/stringval");
@@ -19,11 +20,22 @@ const storageUtil_1 = require("../utils/storageUtil");
 class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
     constructor(exportType, namespace, connection, logger, messages, ux, allVersions) {
         super(namespace, connection, logger, messages, ux);
+        // Reserved keys that should not be used for storing output
+        this.reservedKeys = new Set(['Request', 'Response', 'Condition']);
+        // Tags to validate in PropertySet for reserved key usage
+        this.tagsToValidate = new Set(['additionalOutput']);
+        // constants
+        this.OMNISCRIPT = 'OmniScript';
         this.exportType = exportType;
         this.allVersions = allVersions;
     }
     getName() {
-        return 'OmniScript / Integration Procedures';
+        if (this.exportType === OmniScriptExportType.IP) {
+            return 'Integration Procedures';
+        }
+        else if (this.exportType === OmniScriptExportType.OS) {
+            return 'OmniScripts';
+        }
     }
     getRecordName(record) {
         return (record[this.namespacePrefix + 'Type__c'] +
@@ -116,16 +128,19 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
     }
     async assess(dataRaptorAssessmentInfos, flexCardAssessmentInfos) {
         try {
-            logger_1.Logger.log(this.messages.getMessage('startingOmniScriptAssessment'));
+            const exportComponentType = this.getName();
+            logger_1.Logger.log(this.messages.getMessage('startingOmniScriptAssessment', [exportComponentType]));
             const omniscripts = await this.getAllOmniScripts();
-            logger_1.Logger.log(this.messages.getMessage('foundOmniScriptsToAssess', [omniscripts.length]));
+            logger_1.Logger.log(this.messages.getMessage('foundOmniScriptsToAssess', [omniscripts.length, exportComponentType]));
             const omniAssessmentInfos = await this.processOmniComponents(omniscripts, dataRaptorAssessmentInfos, flexCardAssessmentInfos);
+            await this.updateStorageForOmniscriptAssessment(omniAssessmentInfos === null || omniAssessmentInfos === void 0 ? void 0 : omniAssessmentInfos.osAssessmentInfos);
             return omniAssessmentInfos;
         }
         catch (err) {
-            logger_1.Logger.error(this.messages.getMessage('errorDuringOmniScriptAssessment'));
-            logger_1.Logger.error(JSON.stringify(err));
-            logger_1.Logger.error(err.stack);
+            if (err instanceof interfaces_1.InvalidEntityTypeError) {
+                throw err;
+            }
+            logger_1.Logger.error(this.messages.getMessage('errorDuringOmniScriptAssessment'), err);
         }
     }
     async processOmniComponents(omniscripts, dataRaptorAssessmentInfos, flexCardAssessmentInfos) {
@@ -135,7 +150,8 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
         const existingOmniscriptNames = new Set();
         const existingDataRaptorNames = new Set(dataRaptorAssessmentInfos.map((info) => info.name));
         const existingFlexCardNames = new Set(flexCardAssessmentInfos.map((info) => info.name));
-        const progressBar = (0, base_2.createProgressBar)('Assessing', 'Omniscript and Integration Procedure');
+        const progressBarType = this.getName();
+        const progressBar = (0, base_2.createProgressBar)('Assessing', progressBarType);
         let progressCounter = 0;
         progressBar.start(omniscripts.length, progressCounter);
         // First, collect all OmniScript names from the omniscripts array
@@ -163,7 +179,7 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                         infos: [],
                         warnings: [],
                         errors: [this.messages.getMessage('unexpectedError')],
-                        migrationStatus: 'Can be Automated',
+                        migrationStatus: 'Failed',
                         type: 'OmniScript',
                         missingIP: [],
                         missingDR: [],
@@ -183,20 +199,15 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                         warnings: [],
                         errors: [this.messages.getMessage('unexpectedError')],
                         path: '',
+                        migrationStatus: 'Failed',
                     });
                 }
                 const error = e;
-                logger_1.Logger.error(JSON.stringify(error));
-                logger_1.Logger.error(error.stack);
+                logger_1.Logger.error('Error processing omniscripts', error);
                 continue;
             }
             if (omniAssessmentInfo.type === 'OmniScript') {
                 const type = omniscript[this.namespacePrefix + 'IsLwcEnabled__c'] ? 'LWC' : 'Angular';
-                let migrationStatus = 'Can be Automated';
-                if (type === 'Angular') {
-                    omniAssessmentInfo.warnings.unshift(this.messages.getMessage('angularOSWarning'));
-                    migrationStatus = 'Need Manual Intervention';
-                }
                 const osAssessmentInfo = {
                     name: omniAssessmentInfo.name,
                     type: type,
@@ -213,7 +224,8 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                     infos: [],
                     warnings: omniAssessmentInfo.warnings,
                     errors: [],
-                    migrationStatus: migrationStatus,
+                    migrationStatus: omniAssessmentInfo.migrationStatus,
+                    nameMapping: omniAssessmentInfo.nameMapping,
                 };
                 osAssessmentInfos.push(osAssessmentInfo);
             }
@@ -228,6 +240,7 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                     dependenciesRemoteAction: omniAssessmentInfo.dependenciesRemoteAction,
                     infos: [],
                     warnings: omniAssessmentInfo.warnings,
+                    migrationStatus: omniAssessmentInfo.migrationStatus,
                     errors: [],
                     path: '',
                 };
@@ -253,10 +266,26 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
         const dependenciesRA = [];
         const dependenciesLWC = [];
         //const missingRA: string[] = [];
+        // Check for duplicate element names within the same OmniScript
+        const elementNames = new Set();
+        const duplicateElementNames = new Set();
+        // Track reserved keys found in PropertySet
+        const foundReservedKeys = new Set();
+        for (const elem of elements) {
+            const elemName = elem['Name'];
+            if (elementNames.has(elemName)) {
+                duplicateElementNames.add(elemName);
+            }
+            else {
+                elementNames.add(elemName);
+            }
+        }
         for (const elem of elements) {
             const type = elem[this.namespacePrefix + 'Type__c'];
             const elemName = `${elem['Name']}`;
             const propertySet = JSON.parse(elem[this.namespacePrefix + 'PropertySet__c'] || '{}');
+            // Collect reserved keys from PropertySet
+            this.collectReservedKeys(propertySet, foundReservedKeys);
             // Check for OmniScript dependencies
             if (type === 'OmniScript') {
                 const nameVal = `${elemName}`;
@@ -281,7 +310,12 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 }
             }
             // Check for DataRaptor dependencies
-            if (['DataRaptor Extract Action', 'DataRaptor Turbo Action', 'DataRaptor Post Action'].includes(type)) {
+            if ([
+                'DataRaptor Extract Action',
+                'DataRaptor Turbo Action',
+                'DataRaptor Transform Action',
+                'DataRaptor Post Action',
+            ].includes(type)) {
                 const nameVal = `${elemName}`;
                 dependencyDR.push({ name: propertySet['bundle'], location: nameVal });
                 if (!existingOmniscriptNames.has(nameVal) && !existingDataRaptorNames.has(nameVal)) {
@@ -321,12 +355,24 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
         const existingSubTypeVal = new stringval_1.StringVal(existingSubType, 'sub type');
         const omniScriptName = omniscript[this.namespacePrefix + 'Name'];
         const existingOmniScriptNameVal = new stringval_1.StringVal(omniScriptName, 'name');
+        let assessmentStatus = 'Ready for migration';
         const warnings = [];
-        if (omniProcessType === 'OmniScript') {
-            this.addToAssessmentStorage(existingTypeVal, existingSubTypeVal, omniscript[this.namespacePrefix + 'Language__c']);
+        // Check for Angular OmniScript dependencies
+        for (const osDep of dependencyOS) {
+            if (this.nameRegistry.isAngularOmniScript(osDep.name)) {
+                warnings.push(this.messages.getMessage('angularOmniScriptDependencyWarning', [osDep.location, osDep.name]));
+                assessmentStatus = 'Needs Manual Intervention';
+            }
         }
-        const recordName = `${existingTypeVal.cleanName()}_` +
-            `${existingSubTypeVal.cleanName()}` +
+        // This we need broken down, better create an object and propagate it
+        // Here break it and then combine it
+        const newType = existingTypeVal.cleanName();
+        const newSubType = existingSubTypeVal.cleanName();
+        const newLanguage = omniscript[this.namespacePrefix + 'Language__c']
+            ? `${omniscript[this.namespacePrefix + 'Language__c']}`
+            : '';
+        const recordName = `${newType}_` +
+            `${newSubType}` +
             (omniscript[this.namespacePrefix + 'Language__c'] ? `_${omniscript[this.namespacePrefix + 'Language__c']}` : '') +
             `_${omniscript[this.namespacePrefix + 'Version__c']}`;
         const oldName = `${existingTypeVal.val}_` +
@@ -339,6 +385,7 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 existingTypeVal.val,
                 existingTypeVal.cleanName(),
             ]));
+            assessmentStatus = 'Warnings';
         }
         if (!existingSubTypeVal.isNameCleaned()) {
             warnings.push(this.messages.getMessage('changeMessage', [
@@ -346,6 +393,7 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 existingSubTypeVal.val,
                 existingSubTypeVal.cleanName(),
             ]));
+            assessmentStatus = 'Warnings';
         }
         if (!existingOmniScriptNameVal.isNameCleaned()) {
             warnings.push(this.messages.getMessage('changeMessage', [
@@ -353,14 +401,35 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 existingOmniScriptNameVal.val,
                 existingOmniScriptNameVal.cleanName(),
             ]));
+            assessmentStatus = 'Warnings';
         }
         if (existingOmniscriptNames.has(recordName)) {
             warnings.push(this.messages.getMessage('duplicatedName') + '  ' + recordName);
+            assessmentStatus = 'Needs Manual Intervention';
         }
         else {
             existingOmniscriptNames.add(recordName);
         }
-        return {
+        // Add warning for duplicate element names within the same OmniScript
+        if (duplicateElementNames.size > 0) {
+            const duplicateNamesList = Array.from(duplicateElementNames).join(', ');
+            warnings.unshift(this.messages.getMessage('invalidOrRepeatingOmniscriptElementNames', [duplicateNamesList]));
+            assessmentStatus = 'Needs Manual Intervention';
+        }
+        // Add warning for reserved keys found in PropertySet
+        if (foundReservedKeys.size > 0) {
+            const reservedKeysList = Array.from(foundReservedKeys).join(', ');
+            warnings.unshift(this.messages.getMessage('reservedKeysFoundInPropertySet', [reservedKeysList]));
+            assessmentStatus = 'Needs Manual Intervention';
+        }
+        if (omniProcessType === this.OMNISCRIPT) {
+            const type = omniscript[this.namespacePrefix + 'IsLwcEnabled__c'] ? 'LWC' : 'Angular';
+            if (type === 'Angular') {
+                warnings.unshift(this.messages.getMessage('angularOSWarning'));
+                assessmentStatus = 'Needs Manual Intervention';
+            }
+        }
+        const result = {
             name: recordName,
             id: omniscript['Id'],
             oldName: oldName,
@@ -372,12 +441,70 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
             infos: [],
             warnings: warnings,
             errors: [],
-            migrationStatus: 'Can be Automated',
+            migrationStatus: assessmentStatus,
             type: omniProcessType,
             missingDR: missingDR,
             missingIP: missingIP,
             missingOS: missingOS,
         };
+        if (omniProcessType === this.OMNISCRIPT) {
+            const nameMapping = {
+                oldType: existingType,
+                oldSubtype: existingSubType,
+                oldLanguage: omniscript[this.namespacePrefix + 'Language__c'],
+                newType: newType,
+                newSubType: newSubType,
+                newLanguage: newLanguage,
+            };
+            result.nameMapping = nameMapping;
+        }
+        return result;
+    }
+    updateStorageForOmniscriptAssessment(osAssessmentInfo) {
+        if (osAssessmentInfo === undefined || osAssessmentInfo === null) {
+            logger_1.Logger.error(this.messages.getMessage('missingInfo'));
+            return;
+        }
+        let storage = storageUtil_1.StorageUtil.getOmnistudioAssessmentStorage();
+        logger_1.Logger.logVerbose(this.messages.getMessage('updatingStorageForOmniscipt', ['Assessment']));
+        for (let currentOsRecordInfo of osAssessmentInfo) {
+            try {
+                let nameMapping = currentOsRecordInfo.nameMapping;
+                if (nameMapping === undefined) {
+                    logger_1.Logger.logVerbose(this.messages.getMessage('nameMappingUndefined'));
+                    continue;
+                }
+                let value = {
+                    type: nameMapping.newType,
+                    subtype: nameMapping.newSubType,
+                    language: nameMapping.newLanguage,
+                    isDuplicate: false,
+                };
+                if (currentOsRecordInfo.errors && currentOsRecordInfo.errors.length > 0) {
+                    value.error = currentOsRecordInfo.errors;
+                    value.migrationSuccess = false;
+                }
+                else {
+                    value.migrationSuccess = true;
+                }
+                let finalKey = `${nameMapping.oldType}${nameMapping.oldSubtype}${nameMapping.oldLanguage}`;
+                finalKey = finalKey.toLowerCase();
+                if (storage.osStorage.has(finalKey)) {
+                    // Key already exists - handle accordingly
+                    logger_1.Logger.logVerbose(this.messages.getMessage('keyAlreadyInStorage', [finalKey]));
+                    value.isDuplicate = true;
+                    storage.osStorage.set(finalKey, value);
+                }
+                else {
+                    // Key doesn't exist - safe to set
+                    storage.osStorage.set(finalKey, value);
+                }
+            }
+            catch (error) {
+                logger_1.Logger.error(error);
+            }
+        }
+        storageUtil_1.StorageUtil.printAssessmentStorage();
     }
     async migrate() {
         // Get All Records from OmniScript__c (IP & OS Parent Records)
@@ -388,10 +515,14 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
         // Variables to be returned After Migration
         let originalOsRecords = new Map();
         let osUploadInfo = new Map();
-        logger_1.Logger.log(this.messages.getMessage('foundOmniScriptsToMigrate', [omniscripts.length]));
-        const progressBar = (0, base_2.createProgressBar)('Migrating', 'Omniscript and Integration Procedure');
+        const exportComponentType = this.getName();
+        logger_1.Logger.log(this.messages.getMessage('foundOmniScriptsToMigrate', [omniscripts.length, exportComponentType]));
+        const progressBarType = exportComponentType;
+        const progressBar = (0, base_2.createProgressBar)('Migrating', progressBarType);
         let progressCounter = 0;
         progressBar.start(omniscripts.length, progressCounter);
+        let foundAngularBasedOmniScripts = false;
+        const angularWarningMessage = this.messages.getMessage('angularOmniscriptWarningMessage');
         for (let omniscript of omniscripts) {
             const mappedRecords = [];
             // const originalRecords = new Map<string, AnyJson>();
@@ -408,20 +539,18 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 const type = omniscript[this.namespacePrefix + 'IsLwcEnabled__c'] ? 'LWC' : 'Angular';
                 if (type === 'Angular') {
                     // Skip Angular OmniScripts and add a warning record
-                    const warningMessage = this.messages.getMessage('angularOmniscriptWarningMessage');
                     const skippedResponse = {
                         referenceId: recordId,
                         id: '',
                         success: false,
-                        hasErrors: true,
-                        errors: [warningMessage],
-                        warnings: [],
+                        hasErrors: false,
+                        errors: [],
+                        warnings: [angularWarningMessage],
                         newName: '',
+                        skipped: true,
                     };
                     osUploadInfo.set(recordId, skippedResponse);
-                    progressBar.stop();
-                    logger_1.Logger.warn(warningMessage);
-                    progressBar.start(omniscripts.length, progressCounter);
+                    foundAngularBasedOmniScripts = true;
                     continue;
                 }
             }
@@ -429,24 +558,74 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
             omniscript[`${this.namespacePrefix}IsActive__c`] = false;
             // Get All elements for each OmniScript__c record(i.e IP/OS)
             const elements = await this.getAllElementsForOmniScript(recordId);
+            // Check for duplicate element names within the same OmniScript
+            const elementNames = new Set();
+            const duplicateElementNames = new Set();
+            for (const elem of elements) {
+                const elemName = elem['Name'];
+                if (elementNames.has(elemName)) {
+                    duplicateElementNames.add(elemName);
+                }
+                else {
+                    elementNames.add(elemName);
+                }
+            }
+            // If duplicate element names found, skip this OmniScript
+            if (duplicateElementNames.size > 0) {
+                const duplicateNamesList = Array.from(duplicateElementNames).join(', ');
+                const skippedResponse = {
+                    referenceId: recordId,
+                    id: '',
+                    success: false,
+                    hasErrors: false,
+                    errors: [],
+                    warnings: [this.messages.getMessage('invalidOrRepeatingOmniscriptElementNames', [duplicateNamesList])],
+                    newName: '',
+                    skipped: true,
+                };
+                osUploadInfo.set(recordId, skippedResponse);
+                originalOsRecords.set(recordId, omniscript);
+                continue;
+            }
             if (omniscript[`${this.namespacePrefix}IsProcedure__c`] === true) {
+                // Check for reserved keys in PropertySet for Integration Procedures
+                const foundReservedKeys = new Set();
                 // do the formula replacement from custom to standard notation
                 if (functionDefinitionMetadata.length > 0 && elements.length > 0) {
                     for (let ipElement of elements) {
                         if (ipElement[`${this.namespacePrefix}PropertySet__c`] != null) {
+                            // Check for reserved keys while processing the PropertySet
+                            const propertySet = JSON.parse(ipElement[`${this.namespacePrefix}PropertySet__c`] || '{}');
+                            this.collectReservedKeys(propertySet, foundReservedKeys);
                             var originalString = ipElement[`${this.namespacePrefix}PropertySet__c`];
                             try {
                                 originalString = (0, FormulaUtil_1.getReplacedString)(this.namespacePrefix, ipElement[`${this.namespacePrefix}PropertySet__c`], functionDefinitionMetadata);
                                 ipElement[`${this.namespacePrefix}PropertySet__c`] = originalString;
                             }
                             catch (ex) {
-                                logger_1.Logger.error(JSON.stringify(ex));
-                                logger_1.Logger.error(ex.stack);
+                                logger_1.Logger.error('Error processing formula for integration procedure', ex);
                                 logger_1.Logger.logVerbose(this.messages.getMessage('formulaSyntaxError', [ipElement[`${this.namespacePrefix}PropertySet__c`]]));
                             }
                         }
                     }
                 }
+                // If reserved keys found, skip this IP
+                if (foundReservedKeys.size > 0) {
+                    const reservedKeysList = Array.from(foundReservedKeys).join(', ');
+                    const skippedResponse = {
+                        referenceId: recordId,
+                        id: '',
+                        success: false,
+                        hasErrors: false,
+                        errors: [],
+                        warnings: [this.messages.getMessage('reservedKeysFoundInPropertySet', [reservedKeysList])],
+                        newName: '',
+                        skipped: true,
+                    };
+                    osUploadInfo.set(recordId, skippedResponse);
+                    originalOsRecords.set(recordId, omniscript);
+                    continue;
+                }
             }
             // Perform the transformation for OS/IP Parent Record from OmniScript__c
             const mappedOmniScript = this.mapOmniScriptRecord(omniscript);
@@ -509,13 +688,30 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 osUploadResponse.type = mappedOmniScript[OmniScript_1.default.Type__c];
                 osUploadResponse.subtype = mappedOmniScript[OmniScript_1.default.SubType__c];
                 osUploadResponse.language = mappedOmniScript[OmniScript_1.default.Language__c];
-                const originalOsName = omniscript[this.namespacePrefix + 'Type__c'] +
-                    '_' +
-                    omniscript[this.namespacePrefix + 'SubType__c'] +
-                    '_' +
-                    omniscript[this.namespacePrefix + 'Language__c'];
+                let originalOsName;
+                if (this.allVersions) {
+                    originalOsName =
+                        omniscript[this.namespacePrefix + 'Type__c'] +
+                            '_' +
+                            omniscript[this.namespacePrefix + 'SubType__c'] +
+                            '_' +
+                            omniscript[this.namespacePrefix + 'Language__c'] +
+                            '_' +
+                            (omniscript[this.namespacePrefix + 'Version__c'] || '1');
+                }
+                else {
+                    originalOsName =
+                        omniscript[this.namespacePrefix + 'Type__c'] +
+                            '_' +
+                            omniscript[this.namespacePrefix + 'SubType__c'] +
+                            '_' +
+                            omniscript[this.namespacePrefix + 'Language__c'] +
+                            '_1';
+                }
+                // Always set the new name to show the migrated name
+                osUploadResponse.newName = mappedOsName;
+                // Only add warning if the name was actually modified
                 if (originalOsName !== mappedOsName) {
-                    osUploadResponse.newName = mappedOsName;
                     osUploadResponse.warnings.unshift('WARNING: OmniScript name has been modified to fit naming rules: ' + mappedOsName);
                 }
                 try {
@@ -571,6 +767,9 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
             originalOsRecords.set(recordId, omniscript);
         }
         progressBar.stop();
+        if (foundAngularBasedOmniScripts) {
+            logger_1.Logger.warn(angularWarningMessage);
+        }
         this.updateStorageForOmniscript(osUploadInfo, originalOsRecords);
         const objectMigrationResults = [];
         if (this.exportType === OmniScriptExportType.All || this.exportType === OmniScriptExportType.IP) {
@@ -602,24 +801,9 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
             results: resultMap,
         };
     }
-    addToAssessmentStorage(type, subtype, language) {
-        let storage = storageUtil_1.StorageUtil.getOmnistudioAssessmentStorage();
-        const key = `${type.val}${subtype.val}${language}`;
-        if (storage.osStorage.has(key)) {
-            storage.osStorage.get(key).isDuplicate = true;
-        }
-        else {
-            storage.osStorage.set(key, {
-                type: type.cleanName(),
-                subtype: subtype.cleanName(),
-                language: language || 'English',
-                isDuplicate: false,
-            });
-        }
-    }
     updateStorageForOmniscript(osUploadInfo, originalOsRecords) {
         let storage = storageUtil_1.StorageUtil.getOmnistudioMigrationStorage();
-        logger_1.Logger.logVerbose('Started updating migration storage for omniscript');
+        logger_1.Logger.logVerbose(this.messages.getMessage('updatingStorageForOmniscipt', ['Migration']));
         for (let key of Array.from(originalOsRecords.keys())) {
             try {
                 let oldrecord = originalOsRecords.get(key);
@@ -631,17 +815,24 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                         language: newrecord['language'],
                         isDuplicate: false,
                     };
-                    if (newrecord.hasErrors) {
-                        value.error = newrecord.errors;
+                    // New record can be undefined
+                    if (newrecord === undefined) {
                         value.migrationSuccess = false;
                     }
                     else {
-                        value.migrationSuccess = true;
+                        if (newrecord.hasErrors) {
+                            value.error = newrecord.errors;
+                            value.migrationSuccess = false;
+                        }
+                        else {
+                            value.migrationSuccess = true;
+                        }
                     }
                     let finalKey = `${oldrecord[this.namespacePrefix + 'Type__c']}${oldrecord[this.namespacePrefix + 'SubType__c']}${oldrecord[this.namespacePrefix + 'Language__c']}`;
+                    finalKey = finalKey.toLowerCase();
                     if (storage.osStorage.has(finalKey)) {
                         // Key already exists - handle accordingly
-                        logger_1.Logger.logVerbose(`Key ${finalKey} already exists in storage`);
+                        logger_1.Logger.logVerbose(this.messages.getMessage('keyAlreadyInStorage', [finalKey]));
                         value.isDuplicate = true;
                         storage.osStorage.set(finalKey, value);
                     }
@@ -674,11 +865,21 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
                 { field: this.namespacePrefix + 'SubType__c', direction: utils_1.SortDirection.ASC },
                 { field: this.namespacePrefix + 'Version__c', direction: utils_1.SortDirection.ASC },
             ];
-            return await utils_1.QueryTools.queryWithFilterAndSort(this.connection, this.namespace, OmniScriptMigrationTool.OMNISCRIPT_NAME, this.getOmniScriptFields(), filters, sortFields);
+            return await utils_1.QueryTools.queryWithFilterAndSort(this.connection, this.namespace, OmniScriptMigrationTool.OMNISCRIPT_NAME, this.getOmniScriptFields(), filters, sortFields).catch((err) => {
+                if (err.errorCode === 'INVALID_TYPE') {
+                    throw new interfaces_1.InvalidEntityTypeError(`${OmniScriptMigrationTool.OMNISCRIPT_NAME} type is not found under this namespace`);
+                }
+                throw err;
+            });
         }
         else {
             filters.set(this.namespacePrefix + 'IsActive__c', true);
-            return await utils_1.QueryTools.queryWithFilter(this.connection, this.namespace, OmniScriptMigrationTool.OMNISCRIPT_NAME, this.getOmniScriptFields(), filters);
+            return await utils_1.QueryTools.queryWithFilter(this.connection, this.namespace, OmniScriptMigrationTool.OMNISCRIPT_NAME, this.getOmniScriptFields(), filters).catch((err) => {
+                if (err.errorCode === 'INVALID_TYPE') {
+                    throw new interfaces_1.InvalidEntityTypeError(`${OmniScriptMigrationTool.OMNISCRIPT_NAME} type is not found under this namespace`);
+                }
+                throw err;
+            });
         }
     }
     // Get All Elements w.r.t OmniScript__c i.e Elements tagged to passed in IP/OS
@@ -797,6 +998,7 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
     }
     // Maps an individual Element into an OmniProcessElement record
     mapElementData(elementRecord, omniProcessId, parentElementUploadResponse, invalidIpReferences) {
+        var _a, _b;
         // Transformed object
         const mappedObject = {};
         // Get the fields of the record
@@ -817,37 +1019,128 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
         // We need to fix the child references
         const elementType = mappedObject[Element_1.default.Type__c];
         const propertySet = JSON.parse(mappedObject[Element_1.default.PropertySet__c] || '{}');
+        // Use NameMappingRegistry to update all dependency references
+        const updatedPropertySet = this.nameRegistry.updateDependencyReferences(propertySet);
         switch (elementType) {
             case 'OmniScript':
-                propertySet['Type'] = this.cleanName(propertySet['Type']);
-                propertySet['Sub Type'] = this.cleanName(propertySet['Sub Type']);
+                // Use registry for OmniScript references with explicit fallback
+                const osType = propertySet['Type'] || '';
+                const osSubType = propertySet['Sub Type'] || '';
+                const osLanguage = propertySet['Language'] || 'English';
+                // Construct full OmniScript name to check registry
+                const fullOmniScriptName = `${osType}_${osSubType}_${osLanguage}`;
+                if (this.nameRegistry.isAngularOmniScript(fullOmniScriptName)) {
+                    // Referenced OmniScript is Angular - add warning and keep original reference
+                    logger_1.Logger.logVerbose(`\n${this.messages.getMessage('angularOmniScriptDependencyWarning', [
+                        'OmniScript element',
+                        fullOmniScriptName,
+                    ])}`);
+                    // Keep original reference as-is since Angular OmniScript won't be migrated
+                    updatedPropertySet['Type'] = osType;
+                    updatedPropertySet['Sub Type'] = osSubType;
+                    updatedPropertySet['Language'] = osLanguage;
+                }
+                else if (this.nameRegistry.hasOmniScriptMapping(fullOmniScriptName)) {
+                    // Registry has mapping for this LWC OmniScript - extract cleaned parts
+                    const cleanedFullName = this.nameRegistry.getCleanedName(fullOmniScriptName, 'OmniScript');
+                    const parts = cleanedFullName.split('_');
+                    if (parts.length >= 2) {
+                        updatedPropertySet['Type'] = parts[0];
+                        updatedPropertySet['Sub Type'] = parts[1];
+                        // Language doesn't typically change, but update if provided
+                        if (parts.length >= 3) {
+                            updatedPropertySet['Language'] = parts[2];
+                        }
+                    }
+                }
+                else {
+                    // No registry mapping - use original fallback approach
+                    logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['OmniScript', fullOmniScriptName])}`);
+                    updatedPropertySet['Type'] = this.cleanName(osType);
+                    updatedPropertySet['Sub Type'] = this.cleanName(osSubType);
+                }
                 break;
             case 'Integration Procedure Action':
-                const remoteOptions = propertySet['remoteOptions'] || {};
-                remoteOptions['preTransformBundle'] = this.cleanName(remoteOptions['preTransformBundle']);
-                remoteOptions['postTransformBundle'] = this.cleanName(remoteOptions['postTransformBundle']);
-                propertySet['remoteOptions'] = remoteOptions;
-                propertySet['preTransformBundle'] = this.cleanName(propertySet['preTransformBundle']);
-                propertySet['postTransformBundle'] = this.cleanName(propertySet['postTransformBundle']);
-                // We can't update the IP references, we need to let the user know
+                const remoteOptions = updatedPropertySet['remoteOptions'] || {};
+                // Use registry for DataMapper references with explicit fallback
+                const preTransformBundle = (_a = propertySet['remoteOptions']) === null || _a === void 0 ? void 0 : _a['preTransformBundle'];
+                if (preTransformBundle) {
+                    if (this.nameRegistry.hasDataMapperMapping(preTransformBundle)) {
+                        remoteOptions['preTransformBundle'] = this.nameRegistry.getDataMapperCleanedName(preTransformBundle);
+                    }
+                    else {
+                        logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['DataMapper', preTransformBundle])}`);
+                        remoteOptions['preTransformBundle'] = this.cleanName(preTransformBundle);
+                    }
+                }
+                const postTransformBundle = (_b = propertySet['remoteOptions']) === null || _b === void 0 ? void 0 : _b['postTransformBundle'];
+                if (postTransformBundle) {
+                    if (this.nameRegistry.hasDataMapperMapping(postTransformBundle)) {
+                        remoteOptions['postTransformBundle'] = this.nameRegistry.getDataMapperCleanedName(postTransformBundle);
+                    }
+                    else {
+                        logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['DataMapper', postTransformBundle])}`);
+                        remoteOptions['postTransformBundle'] = this.cleanName(postTransformBundle);
+                    }
+                }
+                updatedPropertySet['remoteOptions'] = remoteOptions;
+                const preBundle = propertySet['preTransformBundle'];
+                if (preBundle) {
+                    if (this.nameRegistry.hasDataMapperMapping(preBundle)) {
+                        updatedPropertySet['preTransformBundle'] = this.nameRegistry.getDataMapperCleanedName(preBundle);
+                    }
+                    else {
+                        logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['DataMapper', preBundle])}`);
+                        updatedPropertySet['preTransformBundle'] = this.cleanName(preBundle);
+                    }
+                }
+                const postBundle = propertySet['postTransformBundle'];
+                if (postBundle) {
+                    if (this.nameRegistry.hasDataMapperMapping(postBundle)) {
+                        updatedPropertySet['postTransformBundle'] = this.nameRegistry.getDataMapperCleanedName(postBundle);
+                    }
+                    else {
+                        logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['DataMapper', postBundle])}`);
+                        updatedPropertySet['postTransformBundle'] = this.cleanName(postBundle);
+                    }
+                }
+                // Use registry for Integration Procedure references
                 const key = propertySet['integrationProcedureKey'] || '';
                 if (key) {
-                    const parts = key.split('_');
-                    const newKey = parts.map((p) => this.cleanName(p, true)).join('_');
-                    if (parts.length > 2) {
-                        invalidIpReferences.set(mappedObject[Element_1.default.Name], key);
+                    const hasRegistryMapping = this.nameRegistry.hasIntegrationProcedureMapping(key);
+                    if (hasRegistryMapping) {
+                        const cleanedIpName = this.nameRegistry.getIntegrationProcedureCleanedName(key);
+                        updatedPropertySet['integrationProcedureKey'] = cleanedIpName;
+                    }
+                    else {
+                        logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['IntegrationProcedure', key])}`);
+                        const parts = key.split('_');
+                        const newKey = parts.map((p) => this.cleanName(p, true)).join('_');
+                        if (parts.length > 2) {
+                            invalidIpReferences.set(mappedObject[Element_1.default.Name], key);
+                        }
+                        updatedPropertySet['integrationProcedureKey'] = newKey;
                     }
-                    propertySet['integrationProcedureKey'] = newKey;
                 }
                 break;
             case 'DataRaptor Turbo Action':
             case 'DataRaptor Transform Action':
             case 'DataRaptor Post Action':
             case 'DataRaptor Extract Action':
-                propertySet['bundle'] = this.cleanName(propertySet['bundle']);
+                // Use registry for DataMapper references with explicit fallback
+                const bundleName = propertySet['bundle'];
+                if (bundleName) {
+                    if (this.nameRegistry.hasDataMapperMapping(bundleName)) {
+                        updatedPropertySet['bundle'] = this.nameRegistry.getDataMapperCleanedName(bundleName);
+                    }
+                    else {
+                        logger_1.Logger.logVerbose(`\n${this.messages.getMessage('componentMappingNotFound', ['DataMapper', bundleName])}`);
+                        updatedPropertySet['bundle'] = this.cleanName(bundleName);
+                    }
+                }
                 break;
         }
-        mappedObject[Element_1.default.PropertySet__c] = JSON.stringify(propertySet);
+        mappedObject[Element_1.default.PropertySet__c] = JSON.stringify(updatedPropertySet);
         // BATCH framework requires that each record has an "attributes" property
         mappedObject['attributes'] = {
             type: OmniScriptMigrationTool.OMNIPROCESSELEMENT_NAME,
@@ -900,6 +1193,34 @@ class OmniScriptMigrationTool extends base_1.BaseMigrationTool {
     getOmniScriptDefinitionFields() {
         return Object.keys(OmniScriptDefinition_1.default);
     }
+    /**
+     * Collects reserved keys found in PropertySet tagsToValidate
+     * @param propertySet - The PropertySet JSON object to validate
+     * @param foundReservedKeys - Set to collect found reserved keys
+     */
+    collectReservedKeys(propertySet, foundReservedKeys) {
+        // Iterate through each tag that needs validation
+        for (const tagToValidate of this.tagsToValidate) {
+            const tagValue = propertySet[tagToValidate];
+            if (tagValue) {
+                if (typeof tagValue === 'object' && tagValue !== null) {
+                    // If it's an object, check all its keys
+                    const keys = Object.keys(tagValue);
+                    for (const key of keys) {
+                        if (this.reservedKeys.has(key)) {
+                            foundReservedKeys.add(key);
+                        }
+                    }
+                }
+                else if (typeof tagValue === 'string') {
+                    // If it's a string, check if the value itself is a reserved key
+                    if (this.reservedKeys.has(tagValue)) {
+                        foundReservedKeys.add(tagValue);
+                    }
+                }
+            }
+        }
+    }
     sleep() {
         return new Promise((resolve) => {
             setTimeout(resolve, 5000);