@salesforce/plugin-omnistudio-migration-tool 2.0.0-rc.6 → 2.0.0-rc.61

package/lib/commands/omnistudio/migration/migrate.js

@@ -15,19 +15,30 @@ var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (
 }) : function(o, v) {
     o["default"] = v;
 });
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 /* eslint-disable @typescript-eslint/no-unsafe-assignment */
 /* eslint-disable @typescript-eslint/no-unsafe-member-access */
+/* eslint-disable @typescript-eslint/no-unsafe-call */
 /* eslint-disable @typescript-eslint/no-explicit-any */
 /*
  * Copyright (c) 2020, salesforce.com, inc.
@@ -35,16 +46,17 @@ Object.defineProperty(exports, "__esModule", { value: true });
  * Licensed under the BSD 3-Clause license.
  * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
  */
+const path_1 = __importDefault(require("path"));
 const os = __importStar(require("os"));
-const command_1 = require("@salesforce/command");
 const core_1 = require("@salesforce/core");
-const basecommand_1 = __importDefault(require("../../basecommand"));
+const sf_plugins_core_1 = require("@salesforce/sf-plugins-core");
 const dataraptor_1 = require("../../../migration/dataraptor");
 const utils_1 = require("../../../utils");
 const interfaces_1 = require("../../../migration/interfaces");
 const resultsbuilder_1 = require("../../../utils/resultsbuilder");
 const flexcard_1 = require("../../../migration/flexcard");
 const omniscript_1 = require("../../../migration/omniscript");
+const customLabels_1 = require("../../../migration/customLabels");
 const logger_1 = require("../../../utils/logger");
 const OmnistudioRelatedObjectMigrationFacade_1 = __importDefault(require("../../../migration/related/OmnistudioRelatedObjectMigrationFacade"));
 const generatePackageXml_1 = require("../../../utils/generatePackageXml");
@@ -55,50 +67,61 @@ const projectPathUtil_1 = require("../../../utils/projectPathUtil");
 const promptUtil_1 = require("../../../utils/promptUtil");
 const projectPathUtil_2 = require("../../../utils/projectPathUtil");
 const postMigrate_1 = require("../../../migration/postMigrate");
+const premigrate_1 = require("../../../migration/premigrate");
 const globalautonumber_1 = require("../../../migration/globalautonumber");
+const dataModelService_1 = require("../../../utils/dataModelService");
+const NameMappingRegistry_1 = require("../../../migration/NameMappingRegistry");
+const ApexNamespaceRegistry_1 = require("../../../migration/ApexNamespaceRegistry");
+const validatorService_1 = require("../../../utils/validatorService");
+const omniscriptInstance_1 = require("../../../migration/omniscriptInstance");
 // Initialize Messages with the current plugin directory
 core_1.Messages.importMessagesDirectory(__dirname);
 // Load the specific messages for this file. Messages from @salesforce/command, @salesforce/core,
 // or any library that is using the messages framework can also be loaded this way.
 const messages = core_1.Messages.loadMessages('@salesforce/plugin-omnistudio-migration-tool', 'migrate');
-class Migrate extends basecommand_1.default {
+class Migrate extends sf_plugins_core_1.SfCommand {
+    constructor() {
+        super(...arguments);
+        // OmniStudio components that don't need migration logging when metadata API is enabled
+        this.OMNISTUDIO_COMPONENTS_FOR_LOGGING = [stringContants_1.Constants.CustomLabel, stringContants_1.Constants.GlobalAutoNumber];
+    }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async run() {
-        logger_1.Logger.initialiseLogger(this.ux, this.logger, 'migrate', this.flags.verbose);
+        const { flags: parsedFlags } = await this.parse(Migrate);
+        const ux = new sf_plugins_core_1.Ux();
+        const logger = await core_1.Logger.child(this.constructor.name);
+        logger_1.Logger.initialiseLogger(ux, logger, 'migrate', parsedFlags.verbose);
         try {
-            // eslint-disable-next-line @typescript-eslint/no-unsafe-return
-            return await this.runMigration();
+            return await this.runMigration(parsedFlags, ux, logger);
         }
         catch (e) {
             const error = e;
-            logger_1.Logger.error(messages.getMessage('errorRunningMigrate'), error);
+            logger_1.Logger.error(messages.getMessage('errorRunningMigrate', [error.message]));
             process.exit(1);
         }
     }
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    async runMigration() {
-        let apiVersion = this.flags.apiversion;
-        const migrateOnly = (this.flags.only || '');
-        const allVersions = this.flags.allversions || false;
-        const relatedObjects = (this.flags.relatedobjects || '');
-        // this.org is guaranteed because requiresUsername=true, as opposed to supportsUsername
-        const conn = this.org.getConnection();
-        if (apiVersion) {
-            conn.setApiVersion(apiVersion);
-        }
-        else {
-            apiVersion = conn.getApiVersion();
-        }
-        const orgs = await orgUtils_1.OrgUtils.getOrgDetails(conn, this.flags.namespace);
-        if (!orgs.hasValidNamespace && this.flags.namespace) {
-            logger_1.Logger.warn(messages.getMessage('invalidNamespace') + orgs.packageDetails.namespace);
-        }
-        if (!orgs.packageDetails) {
-            logger_1.Logger.error(messages.getMessage('noPackageInstalled'));
-            return;
+    // eslint-disable-next-line complexity
+    async runMigration(parsedFlags, ux, logger) {
+        var _a;
+        const migrateOnly = parsedFlags.only || '';
+        let allVersions = parsedFlags.allversions || false;
+        const relatedObjects = parsedFlags.relatedobjects || '';
+        // target-org is required by flag definition, so it will always be present
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        const org = parsedFlags['target-org'];
+        const conn = org.getConnection();
+        const apiVersion = conn.getApiVersion();
+        // Validate that --only and --relatedobjects flags are not used together
+        if (migrateOnly && relatedObjects) {
+            logger_1.Logger.error(messages.getMessage('relatedFlagsNotSupportedWithOnly'));
+            process.exit(1);
         }
-        if (orgs.omniStudioOrgPermissionEnabled) {
-            logger_1.Logger.error(messages.getMessage('alreadyStandardModel'));
+        const orgs = await orgUtils_1.OrgUtils.getOrgDetails(conn);
+        // Initialize global data model service
+        (0, dataModelService_1.initializeDataModelService)(orgs);
+        const validator = new validatorService_1.ValidatorService(orgs, messages, conn);
+        const isValidationPassed = await validator.validate();
+        if (!isValidationPassed) {
             return;
         }
         // Enable Omni preferences
@@ -122,61 +145,114 @@ class Migrate extends basecommand_1.default {
         let projectPath;
         let objectsToProcess = [];
         let targetApexNamespace;
+        const preMigrate = new premigrate_1.PreMigrate(namespace, conn, logger, messages, ux);
         const isExperienceBundleMetadataAPIProgramaticallyEnabled = { value: false };
-        if (relatedObjects) {
-            // To-Do: Add LWC to valid options when GA is released
-            const validOptions = [stringContants_1.Constants.Apex, stringContants_1.Constants.ExpSites, stringContants_1.Constants.FlexiPage];
-            objectsToProcess = relatedObjects.split(',').map((obj) => obj.trim());
-            // Validate input
-            for (const obj of objectsToProcess) {
-                if (!validOptions.includes(obj)) {
-                    logger_1.Logger.error(messages.getMessage('invalidRelatedObjectsOption', [obj]));
-                    process.exit(1);
-                }
+        if ((0, dataModelService_1.isStandardDataModel)()) {
+            if (!(0, dataModelService_1.isStandardDataModelWithMetadataAPIEnabled)()) {
+                // Only if metadata API is off we need the allVersions Consent and Omnistudio Metadata Cleanup
+                allVersions = await preMigrate.handleAllVersionsPrerequisites(allVersions);
+                await preMigrate.handleOmnistudioMetadataPrerequisites();
             }
-            // Check for general consent to make modifications with OMT
-            const generalConsent = await this.getGeneralConsent();
-            if (generalConsent) {
-                // Use ProjectPathUtil for APEX project folder selection (matches assess.ts logic)
-                projectPath = await projectPathUtil_1.ProjectPathUtil.getProjectPath(messages, true);
-                targetApexNamespace = await this.getTargetApexNamespace(objectsToProcess, targetApexNamespace);
-                await this.handleExperienceSitePrerequisites(objectsToProcess, conn, isExperienceBundleMetadataAPIProgramaticallyEnabled);
-                logger_1.Logger.logVerbose('The objects to process after handleExpSitePrerequisite are ' + JSON.stringify(objectsToProcess));
-            } // TODO - What if general consent is no
+        }
+        let actionItems = [];
+        let deploymentConfig = { autoDeploy: false, authKey: undefined };
+        if (relatedObjects) {
+            const relatedObjectProcessResult = await this.processRelatedObjects(relatedObjects, preMigrate, conn, isExperienceBundleMetadataAPIProgramaticallyEnabled, actionItems);
+            objectsToProcess = relatedObjectProcessResult.objectsToProcess;
+            projectPath = relatedObjectProcessResult.projectPath;
+            targetApexNamespace = relatedObjectProcessResult.targetApexNamespace;
+            deploymentConfig = relatedObjectProcessResult.deploymentConfig;
         }
         logger_1.Logger.log(messages.getMessage('migrationInitialization', [String(namespace)]));
         logger_1.Logger.log(messages.getMessage('apiVersionInfo', [apiVersion]));
         logger_1.Logger.logVerbose(messages.getMessage('migrationTargets', [migrateOnly || 'all']));
         logger_1.Logger.logVerbose(messages.getMessage('relatedObjectsInfo', [relatedObjects || 'none']));
         logger_1.Logger.logVerbose(messages.getMessage('allVersionsFlagInfo', [String(allVersions)]));
-        // const includeLwc = this.flags.lwc ? await this.ux.confirm('Do you want to include LWC migration? (yes/no)') : false;
-        // Register the migration objects
+        // Initialize the name mapping registry and pre-process all components
+        const nameRegistry = NameMappingRegistry_1.NameMappingRegistry.getInstance();
+        nameRegistry.clear(); // Clear any previous mappings
+        logger_1.Logger.log(messages.getMessage('startingComponentPreProcessing'));
+        await ApexNamespaceRegistry_1.ApexNamespaceRegistry.getInstance().initialize(conn, namespace);
+        await this.preProcessAllComponents(namespace, conn, migrateOnly);
+        // Register the migration objects with CORRECTED ORDER
         let migrationObjects = [];
-        migrationObjects = this.getMigrationObjects(migrateOnly, migrationObjects, namespace, conn, allVersions);
+        migrationObjects = this.getMigrationObjectsInCorrectOrder(migrateOnly, migrationObjects, namespace, conn, allVersions, logger, ux);
         // Migrate individual objects
         const debugTimer = utils_1.DebugTimer.getInstance();
-        // We need to truncate the standard objects first
-        let objectMigrationResults = await this.truncateObjects(migrationObjects, debugTimer);
-        objectMigrationResults = objectMigrationResults.filter((result) => result.name !== stringContants_1.Constants.GlobalAutoNumberComponentName);
+        // We need to truncate the standard objects first (in reverse order for cleanup)
+        let objectMigrationResults;
+        objectMigrationResults = await this.truncateObjects([...migrationObjects].reverse(), debugTimer);
         const allTruncateComplete = objectMigrationResults.length === 0;
-        if (allTruncateComplete) {
-            objectMigrationResults = await this.migrateObjects(migrationObjects, debugTimer);
+        // Log truncation errors if any exist
+        if (!allTruncateComplete) {
+            this.logTruncationErrors(objectMigrationResults);
+            return;
         }
-        const omnistudioRelatedObjectsMigration = new OmnistudioRelatedObjectMigrationFacade_1.default(namespace, migrateOnly, allVersions, this.org, projectPath, targetApexNamespace);
+        objectMigrationResults = await this.migrateObjects(migrationObjects, debugTimer, namespace);
+        const omnistudioRelatedObjectsMigration = new OmnistudioRelatedObjectMigrationFacade_1.default(namespace, migrateOnly, allVersions, org, projectPath, targetApexNamespace);
         const relatedObjectMigrationResult = omnistudioRelatedObjectsMigration.migrateAll(objectsToProcess);
-        generatePackageXml_1.generatePackageXml.createChangeList(relatedObjectMigrationResult.apexAssessmentInfos, relatedObjectMigrationResult.lwcAssessmentInfos, relatedObjectMigrationResult.flexipageAssessmentInfos);
         // POST MIGRATION
-        let actionItems = [];
-        const postMigrate = new postMigrate_1.PostMigrate(this.org, namespace, conn, this.logger, messages, this.ux, objectsToProcess);
+        const postMigrate = new postMigrate_1.PostMigrate(org, namespace, conn, logger, messages, ux, objectsToProcess, deploymentConfig, projectPath);
         if (!migrateOnly) {
-            actionItems = await postMigrate.setDesignersToUseStandardDataModel(namespace);
+            await postMigrate.executeTasks(namespace, actionItems);
         }
-        await postMigrate.restoreExperienceAPIMetadataSettings(isExperienceBundleMetadataAPIProgramaticallyEnabled);
         const migrationActionItems = this.collectActionItems(objectMigrationResults);
         actionItems = [...actionItems, ...migrationActionItems];
-        await resultsbuilder_1.ResultsBuilder.generateReport(objectMigrationResults, relatedObjectMigrationResult, conn.instanceUrl, orgs, messages, actionItems, objectsToProcess);
+        generatePackageXml_1.generatePackageXml.createChangeList(relatedObjectMigrationResult.apexAssessmentInfos, deploymentConfig.autoDeploy && deploymentConfig.authKey ? relatedObjectMigrationResult.lwcAssessmentInfos : [], relatedObjectMigrationResult.experienceSiteAssessmentInfos, relatedObjectMigrationResult.flexipageAssessmentInfos, org.getConnection().version, messages);
+        try {
+            generatePackageXml_1.generatePackageXml.createOmnistudioDeploymentXml(relatedObjectMigrationResult.apexAssessmentInfos, deploymentConfig.autoDeploy && deploymentConfig.authKey ? relatedObjectMigrationResult.lwcAssessmentInfos : [], relatedObjectMigrationResult.experienceSiteAssessmentInfos, relatedObjectMigrationResult.flexipageAssessmentInfos, org.getConnection().version);
+        }
+        catch (error) {
+            logger_1.Logger.error(messages.getMessage('unexpectedError'), error);
+            logger_1.Logger.logVerbose(error);
+        }
+        let deploymentFailed = false;
+        try {
+            await postMigrate.deploy(actionItems);
+        }
+        catch (error) {
+            deploymentFailed = true;
+            logger_1.Logger.error(messages.getMessage('errorDeployingComponents'), error);
+            logger_1.Logger.logVerbose(error);
+            // Even if deployment fails completely, continue with report generation
+            // Action items are already added to the array by the deploy method
+        }
+        await resultsbuilder_1.ResultsBuilder.generateReport(objectMigrationResults, relatedObjectMigrationResult, conn.instanceUrl, orgs, messages, actionItems, objectsToProcess, migrateOnly, deploymentFailed);
+        logger_1.Logger.log(messages.getMessage('migrationSuccessfulMessage', [
+            (_a = orgs.orgDetails) === null || _a === void 0 ? void 0 : _a.Id,
+            path_1.default.join(process.cwd(), stringContants_1.Constants.MigrationReportsFolderName),
+        ]));
         // Return results needed for --json flag
-        return { objectMigrationResults };
+        return { objectMigrationResults: [] };
+    }
+    async processRelatedObjects(relatedObjects, preMigrate, conn, isExperienceBundleMetadataAPIProgramaticallyEnabled, actionItems) {
+        const validOptions = [stringContants_1.Constants.Apex, stringContants_1.Constants.ExpSites, stringContants_1.Constants.FlexiPage, stringContants_1.Constants.LWC];
+        const objectsToProcess = relatedObjects.split(',').map((obj) => obj.trim());
+        // Validate input
+        for (const obj of objectsToProcess) {
+            if (!validOptions.includes(obj)) {
+                logger_1.Logger.error(messages.getMessage('invalidRelatedObjectsOption', [obj]));
+                process.exit(1);
+            }
+        }
+        let deploymentConfig = { autoDeploy: false, authKey: undefined };
+        let projectPath;
+        let targetApexNamespace;
+        // Check for general consent to make modifications with OMT
+        const generalConsent = await this.getGeneralConsent();
+        if (generalConsent) {
+            // Use ProjectPathUtil for APEX project folder selection (matches assess.ts logic)
+            projectPath = await projectPathUtil_1.ProjectPathUtil.getProjectPath(messages, true);
+            targetApexNamespace = await this.getTargetApexNamespace(objectsToProcess, targetApexNamespace);
+            await preMigrate.handleExperienceSitePrerequisites(objectsToProcess, conn, isExperienceBundleMetadataAPIProgramaticallyEnabled);
+            deploymentConfig = await preMigrate.getAutoDeployConsent(objectsToProcess.includes(stringContants_1.Constants.LWC), actionItems);
+        }
+        else {
+            objectsToProcess.length = 0;
+            logger_1.Logger.warn(messages.getMessage('relatedObjectsConsentNotGiven'));
+            logger_1.Logger.logVerbose(messages.getMessage('relatedObjectsToProcess', [JSON.stringify(objectsToProcess)]));
+        }
+        return { objectsToProcess, projectPath, targetApexNamespace, deploymentConfig };
     }
     async getMigrationConsent() {
         const askWithTimeOut = promptUtil_1.PromptUtil.askWithTimeOut(messages);
@@ -205,30 +281,6 @@ class Migrate extends basecommand_1.default {
         }
         return consent;
     }
-    async handleExperienceSitePrerequisites(objectsToProcess, conn, isExperienceBundleMetadataAPIProgramaticallyEnabled) {
-        if (objectsToProcess.includes(stringContants_1.Constants.ExpSites)) {
-            const expMetadataApiConsent = await this.getExpSiteMetadataEnableConsent();
-            logger_1.Logger.logVerbose(`The consent for exp site is  ${expMetadataApiConsent}`);
-            if (expMetadataApiConsent === false) {
-                logger_1.Logger.warn('Consent for experience sites is not provided. Experience sites will not be processed');
-                this.removeKeyFromRelatedObjectsToProcess(stringContants_1.Constants.ExpSites, objectsToProcess);
-                logger_1.Logger.logVerbose(`Objects to process after removing expsite are ${JSON.stringify(objectsToProcess)}`);
-                return;
-            }
-            const isMetadataAPIPreEnabled = await orgPreferences_1.OrgPreferences.isExperienceBundleMetadataAPIEnabled(conn);
-            if (isMetadataAPIPreEnabled === true) {
-                logger_1.Logger.logVerbose('ExperienceBundle metadata api is already enabled');
-                return;
-            }
-            logger_1.Logger.logVerbose('ExperienceBundle metadata api needs to be programatically enabled');
-            isExperienceBundleMetadataAPIProgramaticallyEnabled.value = await orgPreferences_1.OrgPreferences.setExperienceBundleMetadataAPI(conn, true);
-            if (isExperienceBundleMetadataAPIProgramaticallyEnabled.value === false) {
-                this.removeKeyFromRelatedObjectsToProcess(stringContants_1.Constants.ExpSites, objectsToProcess);
-                logger_1.Logger.warn('Since the api could not able enabled the experience sites would not be processed');
-            }
-            logger_1.Logger.logVerbose(`Objects to process are ${JSON.stringify(objectsToProcess)}`);
-        }
-    }
     collectActionItems(objectMigrationResults) {
         const actionItems = [];
         // Collect errors from migration results and add them to action items
@@ -239,20 +291,15 @@ class Migrate extends basecommand_1.default {
         }
         return actionItems;
     }
-    removeKeyFromRelatedObjectsToProcess(keyToRemove, relatedObjects) {
-        const index = relatedObjects.indexOf(stringContants_1.Constants.ExpSites);
-        if (index > -1) {
-            relatedObjects.splice(index, 1);
-        }
-    }
     async truncateObjects(migrationObjects, debugTimer) {
         const objectMigrationResults = [];
-        for (const cls of migrationObjects.reverse()) {
+        // Truncate in reverse order (highest dependencies first) - this is correct for cleanup
+        for (const cls of migrationObjects) {
             try {
-                logger_1.Logger.log(messages.getMessage('cleaningComponent', [cls.getName()]));
-                debugTimer.lap('Cleaning: ' + cls.getName());
+                const componentName = cls.getName();
+                this.logTruncationStart(componentName, debugTimer);
                 await cls.truncate();
-                logger_1.Logger.log(messages.getMessage('cleaningDone', [cls.getName()]));
+                this.logTruncationComplete(componentName);
             }
             catch (ex) {
                 objectMigrationResults.push({
@@ -266,30 +313,28 @@ class Migrate extends basecommand_1.default {
         }
         return objectMigrationResults;
     }
-    async migrateObjects(migrationObjects, debugTimer) {
+    async migrateObjects(migrationObjects, debugTimer, namespace) {
         let objectMigrationResults = [];
-        for (const cls of migrationObjects.reverse()) {
+        // Migrate in correct dependency order
+        for (const cls of migrationObjects) {
             try {
-                logger_1.Logger.log(messages.getMessage('migratingComponent', [cls.getName()]));
-                debugTimer.lap('Migrating: ' + cls.getName());
+                const componentName = cls.getName();
+                this.logMigrationStart(componentName, debugTimer);
                 const results = await cls.migrate();
-                if (results.some((result) => { var _a; return ((_a = result === null || result === void 0 ? void 0 : result.errors) === null || _a === void 0 ? void 0 : _a.length) > 0; })) {
-                    logger_1.Logger.error(messages.getMessage('migrationFailed', [cls.getName()]));
-                }
-                else {
-                    logger_1.Logger.log(messages.getMessage('migrationCompleted', [cls.getName()]));
-                }
+                this.logMigrationComplete(componentName, results);
                 objectMigrationResults = objectMigrationResults.concat(results.map((r) => {
                     return {
                         name: r.name,
                         data: this.mergeRecordAndUploadResults(r, cls),
                         errors: r.errors,
+                        totalCount: r.totalCount, // Preserve totalCount for custom labels
+                        allRecords: r.allRecords, // All records for CSV export
                     };
                 }));
             }
             catch (ex) {
                 if (ex instanceof interfaces_1.InvalidEntityTypeError) {
-                    logger_1.Logger.error(ex.message);
+                    logger_1.Logger.error(messages.getMessage('invalidTypeMigrateErrorMessage', [namespace]));
                     process.exit(1);
                 }
                 const errMsg = ex instanceof Error ? ex.message : String(ex);
@@ -304,31 +349,123 @@ class Migrate extends basecommand_1.default {
         }
         return objectMigrationResults;
     }
-    getMigrationObjects(migrateOnly, migrationObjects, namespace, conn, allVersions) {
+    /**
+     * Log the start of component migration and track timing
+     */
+    logMigrationStart(componentName, debugTimer) {
+        const shouldSkipLogging = this.shouldSkipLogging(componentName);
+        if (!shouldSkipLogging) {
+            logger_1.Logger.log(messages.getMessage('migratingComponent', [componentName]));
+            debugTimer.lap('Migrating: ' + componentName);
+        }
+    }
+    /**
+     * Log the completion or failure of component migration
+     */
+    logMigrationComplete(componentName, results) {
+        const shouldSkipLogging = this.shouldSkipLogging(componentName);
+        if (!shouldSkipLogging) {
+            if (results.some((result) => { var _a; return ((_a = result === null || result === void 0 ? void 0 : result.errors) === null || _a === void 0 ? void 0 : _a.length) > 0; }) && !(0, dataModelService_1.isStandardDataModelWithMetadataAPIEnabled)()) {
+                logger_1.Logger.error(messages.getMessage('migrationFailed', [componentName]));
+            }
+            else {
+                logger_1.Logger.log(messages.getMessage('migrationCompleted', [componentName]));
+            }
+        }
+    }
+    /**
+     * Check if migration logging should be skipped for a component
+     * Returns true if Metadata API is enabled and the component doesn't need migration
+     */
+    shouldSkipTruncationLogging(componentName) {
+        if ((0, dataModelService_1.isStandardDataModel)()) {
+            return true;
+        }
+        return this.shouldSkipLogging(componentName);
+    }
+    /**
+     * Check if migration logging should be skipped for a component
+     * Returns true if Metadata API is enabled and the component doesn't need migration
+     */
+    shouldSkipLogging(componentName) {
+        if (!(0, dataModelService_1.isStandardDataModelWithMetadataAPIEnabled)()) {
+            return false;
+        }
+        // Skip logging for OmniStudio components that don't need migration when metadata API is enabled
+        return !this.OMNISTUDIO_COMPONENTS_FOR_LOGGING.some((component) => componentName.toLowerCase().includes(component.toLowerCase()));
+    }
+    /**
+     * Log the start of component truncation and track timing
+     */
+    logTruncationStart(componentName, debugTimer) {
+        const shouldSkipLogging = this.shouldSkipTruncationLogging(componentName);
+        if (!shouldSkipLogging) {
+            logger_1.Logger.log(messages.getMessage('cleaningComponent', [componentName]));
+            debugTimer.lap('Cleaning: ' + componentName);
+        }
+    }
+    /**
+     * Log the completion of component truncation
+     */
+    logTruncationComplete(componentName) {
+        const shouldSkipLogging = this.shouldSkipTruncationLogging(componentName);
+        if (!shouldSkipLogging) {
+            logger_1.Logger.log(messages.getMessage('cleaningDone', [componentName]));
+        }
+    }
+    /**
+     * Get migration objects in the correct dependency order:
+     * 1. Data Mappers (lowest dependencies)
+     * 2. Integration Procedures/ OmniScripts
+     * 3. FlexCards (highest dependencies)
+     * 4. GlobalAutoNumbers (independent)
+     */
+    getMigrationObjectsInCorrectOrder(migrateOnly, migrationObjects, namespace, conn, allVersions, logger, ux) {
         if (!migrateOnly) {
             migrationObjects = [
-                new dataraptor_1.DataRaptorMigrationTool(namespace, conn, this.logger, messages, this.ux),
-                new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.All, namespace, conn, this.logger, messages, this.ux, allVersions),
-                new flexcard_1.CardMigrationTool(namespace, conn, this.logger, messages, this.ux, allVersions),
-                new globalautonumber_1.GlobalAutoNumberMigrationTool(namespace, conn, this.logger, messages, this.ux),
+                new dataraptor_1.DataRaptorMigrationTool(namespace, conn, logger, messages, ux),
+                // Integration Procedure
+                new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.IP, namespace, conn, logger, messages, ux, allVersions),
+                // OmniScript
+                new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.OS, namespace, conn, logger, messages, ux, allVersions),
+                new flexcard_1.CardMigrationTool(namespace, conn, logger, messages, ux, allVersions),
+                new customLabels_1.CustomLabelsMigrationTool(namespace, conn, logger, messages, ux),
             ];
+            if (!(0, dataModelService_1.isFoundationPackage)()) {
+                migrationObjects.push(new globalautonumber_1.GlobalAutoNumberMigrationTool(namespace, conn, logger, messages, ux));
+                migrationObjects.push(new omniscriptInstance_1.OmniScriptInstanceMigrationTool(namespace, conn, logger, messages, ux));
+            }
         }
         else {
+            // For single component migration, the order doesn't matter as much
+            // but we still maintain consistency
             switch (migrateOnly) {
                 case stringContants_1.Constants.Omniscript:
-                    migrationObjects.push(new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.OS, namespace, conn, this.logger, messages, this.ux, allVersions));
+                    migrationObjects.push(new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.OS, namespace, conn, logger, messages, ux, allVersions));
                     break;
                 case stringContants_1.Constants.IntegrationProcedure:
-                    migrationObjects.push(new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.IP, namespace, conn, this.logger, messages, this.ux, allVersions));
+                    migrationObjects.push(new omniscript_1.OmniScriptMigrationTool(omniscript_1.OmniScriptExportType.IP, namespace, conn, logger, messages, ux, allVersions));
                     break;
                 case stringContants_1.Constants.Flexcard:
-                    migrationObjects.push(new flexcard_1.CardMigrationTool(namespace, conn, this.logger, messages, this.ux, allVersions));
+                    migrationObjects.push(new flexcard_1.CardMigrationTool(namespace, conn, logger, messages, ux, allVersions));
                     break;
                 case stringContants_1.Constants.DataMapper:
-                    migrationObjects.push(new dataraptor_1.DataRaptorMigrationTool(namespace, conn, this.logger, messages, this.ux));
+                    migrationObjects.push(new dataraptor_1.DataRaptorMigrationTool(namespace, conn, logger, messages, ux));
                     break;
                 case stringContants_1.Constants.GlobalAutoNumber:
-                    migrationObjects.push(new globalautonumber_1.GlobalAutoNumberMigrationTool(namespace, conn, this.logger, messages, this.ux));
+                    if ((0, dataModelService_1.isFoundationPackage)()) {
+                        logger_1.Logger.warn(messages.getMessage('globalAutoNumberUnSupportedInOmnistudioPackage'));
+                    }
+                    migrationObjects.push(new globalautonumber_1.GlobalAutoNumberMigrationTool(namespace, conn, logger, messages, ux));
+                    break;
+                case stringContants_1.Constants.CustomLabel:
+                    migrationObjects.push(new customLabels_1.CustomLabelsMigrationTool(namespace, conn, logger, messages, ux));
+                    break;
+                case stringContants_1.Constants.SaveForLater:
+                    if ((0, dataModelService_1.isFoundationPackage)()) {
+                        logger_1.Logger.warn(messages.getMessage('globalAutoNumberUnSupportedInOmnistudioPackage'));
+                    }
+                    migrationObjects.push(new omniscriptInstance_1.OmniScriptInstanceMigrationTool(namespace, conn, logger, messages, ux));
                     break;
                 default:
                     throw new Error(messages.getMessage('invalidOnlyFlag'));
@@ -336,9 +473,127 @@ class Migrate extends basecommand_1.default {
         }
         return migrationObjects;
     }
+    /**
+     * Pre-process all components to register their name mappings
+     */
+    async preProcessAllComponents(namespace, conn, migrateOnly) {
+        try {
+            if ((0, dataModelService_1.isStandardDataModelWithMetadataAPIEnabled)()) {
+                return;
+            }
+            const nameRegistry = NameMappingRegistry_1.NameMappingRegistry.getInstance();
+            // Query all components that will be migrated
+            const dataMappers = await this.queryDataMappers(conn, namespace);
+            const allOmniScripts = await this.queryOmniScripts(conn, namespace, false); // All OmniScripts (LWC + Angular)
+            const integrationProcedures = await this.queryOmniScripts(conn, namespace, true); // Integration Procedures only
+            const flexCards = await this.queryFlexCards(conn, namespace);
+            // Separate OmniScripts into LWC and Angular types
+            const { lwc: lwcOmniScripts, angular: angularOmniScripts } = this.separateOmniScriptsByType(allOmniScripts, namespace);
+            // Filter based on migrateOnly flag if specified
+            const filteredData = this.filterComponentsByMigrateOnly(migrateOnly, dataMappers, lwcOmniScripts, // Only LWC OmniScripts for migration
+            integrationProcedures, flexCards);
+            // Register all name mappings (including Angular OmniScripts for tracking)
+            nameRegistry.preProcessComponents(filteredData.dataMappers, filteredData.omniScripts, // LWC OmniScripts
+            angularOmniScripts, // Angular OmniScripts (for tracking)
+            filteredData.integrationProcedures, filteredData.flexCards);
+            const allMappings = nameRegistry.getAllNameMappings();
+            logger_1.Logger.log(messages.getMessage('completeComponentMappingMessage', [allMappings.length]));
+        }
+        catch (error) {
+            logger_1.Logger.error(messages.getMessage('errorComponentMapping'), error);
+        }
+    }
+    /**
+     * Query DataMappers from the org
+     */
+    async queryDataMappers(conn, namespace) {
+        let query;
+        if ((0, dataModelService_1.isStandardDataModel)()) {
+            query = "SELECT Id, Name FROM OmniDataTransform WHERE Type != 'Migration'";
+        }
+        else {
+            query = `SELECT Id, Name FROM ${namespace}__DRBundle__c WHERE ${namespace}__Type__c != 'Migration'`;
+        }
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+        const result = await conn.query(query);
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-return
+        return result.records || [];
+    }
+    /**
+     * Query OmniScripts from the org (both LWC and Angular)
+     */
+    async queryOmniScripts(conn, namespace, isProcedure) {
+        const procedureFilter = isProcedure ? 'true' : 'false';
+        let query;
+        // Query all OmniScripts (both LWC and Angular)
+        if ((0, dataModelService_1.isStandardDataModel)()) {
+            query = `SELECT Id, Name, Type, SubType, Language, IsIntegrationProcedure, IsWebCompEnabled FROM OmniProcess WHERE IsIntegrationProcedure = ${procedureFilter} and IsActive = true`;
+        }
+        else {
+            query = `SELECT Id, Name, ${namespace}__Type__c, ${namespace}__SubType__c, ${namespace}__Language__c, ${namespace}__IsProcedure__c, ${namespace}__IsLwcEnabled__c FROM ${namespace}__OmniScript__c WHERE ${namespace}__IsProcedure__c = ${procedureFilter} and ${namespace}__IsActive__c = true`;
+        }
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+        const result = await conn.query(query);
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-return
+        return result.records || [];
+    }
+    /**
+     * Separate OmniScripts into LWC and Angular based on IsLwcEnabled__c field
+     */
+    separateOmniScriptsByType(omniscripts, namespace) {
+        const lwc = [];
+        const angular = [];
+        for (const omniscript of omniscripts) {
+            const isLwcEnabled = omniscript[(0, dataModelService_1.getFieldKeyForOmniscript)(namespace, 'IsLwcEnabled__c')];
+            if (isLwcEnabled) {
+                lwc.push(omniscript);
+            }
+            else {
+                angular.push(omniscript);
+            }
+        }
+        return { lwc, angular };
+    }
+    /**
+     * Query FlexCards from the org
+     */
+    async queryFlexCards(conn, namespace) {
+        let query;
+        if ((0, dataModelService_1.isStandardDataModel)()) {
+            query = 'SELECT Id, Name FROM OmniUiCard WHERE IsActive = true';
+        }
+        else {
+            query = `SELECT Id, Name FROM ${namespace}__VlocityCard__c WHERE ${namespace}__CardType__c = 'flex' AND ${namespace}__Active__c = true`;
+        }
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-call
+        const result = await conn.query(query);
+        // eslint-disable-next-line @typescript-eslint/no-unsafe-return
+        return result.records || [];
+    }
+    /**
+     * Filter components based on the migrateOnly flag
+     */
+    filterComponentsByMigrateOnly(migrateOnly, dataMappers, omniScripts, integrationProcedures, flexCards) {
+        if (!migrateOnly) {
+            return { dataMappers, omniScripts, integrationProcedures, flexCards };
+        }
+        // Return only the components that match the migrateOnly filter
+        switch (migrateOnly) {
+            case stringContants_1.Constants.DataMapper:
+                return { dataMappers, omniScripts: [], integrationProcedures: [], flexCards: [] };
+            case stringContants_1.Constants.Omniscript:
+                return { dataMappers: [], omniScripts, integrationProcedures: [], flexCards: [] };
+            case stringContants_1.Constants.IntegrationProcedure:
+                return { dataMappers: [], omniScripts: [], integrationProcedures, flexCards: [] };
+            case stringContants_1.Constants.Flexcard:
+                return { dataMappers: [], omniScripts: [], integrationProcedures: [], flexCards };
+            default:
+                return { dataMappers: [], omniScripts: [], integrationProcedures: [], flexCards: [] };
+        }
+    }
     async getTargetApexNamespace(objectsToProcess, targetApexNamespace) {
         if (objectsToProcess.includes(stringContants_1.Constants.Apex)) {
-            targetApexNamespace = await this.ux.prompt(messages.getMessage('enterTargetNamespace'));
+            targetApexNamespace = await (0, promptUtil_1.askQuestion)(messages.getMessage('enterTargetNamespace'));
             logger_1.Logger.log(messages.getMessage('usingTargetNamespace', [targetApexNamespace]));
         }
         return targetApexNamespace;
@@ -356,75 +611,106 @@ class Migrate extends basecommand_1.default {
         }
         return consent;
     }
-    async getExpSiteMetadataEnableConsent() {
-        let consent = null;
-        while (consent === null) {
-            try {
-                consent = await logger_1.Logger.confirm('By proceeding further, you hereby consent to enable digital experience metadata api(y/n). If y sites will be processed, if n expsites will not be processed');
-            }
-            catch (error) {
-                logger_1.Logger.log(messages.getMessage('invalidYesNoResponse'));
-                consent = null;
-            }
-        }
-        return consent;
-    }
     mergeRecordAndUploadResults(migrationResults, migrationTool) {
         const mergedResults = [];
         for (const record of Array.from(migrationResults.records.values())) {
-            const obj = {
-                id: record['Id'],
-                name: migrationTool.getRecordName(record),
-                status: messages.getMessage('labelStatusSkipped'),
-                errors: record['errors'],
-                migratedId: undefined,
-                warnings: [],
-                migratedName: '',
-            };
-            if (migrationResults.results.has(record['Id'])) {
-                const recordResults = migrationResults.results.get(record['Id']);
-                let errors = obj.errors || [];
-                errors = errors.concat(recordResults.errors || []);
-                obj.status = (recordResults === null || recordResults === void 0 ? void 0 : recordResults.skipped)
-                    ? messages.getMessage('labelStatusSkipped')
-                    : !recordResults || recordResults.hasErrors
-                        ? messages.getMessage('labelStatusFailed')
-                        : messages.getMessage('labelStatusComplete');
-                obj.errors = errors;
-                obj.migratedId = recordResults.id;
-                obj.warnings = recordResults.warnings;
-                obj.migratedName = recordResults.newName;
+            // For custom labels, preserve all the custom fields
+            if (migrationTool.getName().toLowerCase().includes(stringContants_1.Constants.CustomLabelPluralName.toLowerCase())) {
+                // Use the record as-is for custom labels since it already has all the needed fields
+                const customLabelRecord = record;
+                mergedResults.push({
+                    id: customLabelRecord.id || customLabelRecord.name,
+                    name: customLabelRecord.name || customLabelRecord.labelName,
+                    status: customLabelRecord.status || 'Skipped',
+                    errors: customLabelRecord.errors || [],
+                    migratedId: customLabelRecord.migratedId || customLabelRecord.id,
+                    warnings: customLabelRecord.warnings || [],
+                    migratedName: customLabelRecord.migratedName || customLabelRecord.name,
+                    // Preserve custom fields for custom labels
+                    coreInfo: customLabelRecord.coreInfo,
+                    packageInfo: customLabelRecord.packageInfo,
+                    message: customLabelRecord.message,
+                    cloneStatus: customLabelRecord.cloneStatus,
+                    labelName: customLabelRecord.labelName,
+                });
+            }
+            else {
+                // Original logic for other components
+                const obj = {
+                    id: record['Id'],
+                    name: migrationTool.getRecordName(record),
+                    status: 'Skipped',
+                    errors: record['errors'],
+                    migratedId: undefined,
+                    warnings: [],
+                    migratedName: '',
+                };
+                if (migrationResults.results.has(record['Id'])) {
+                    const recordResults = migrationResults.results.get(record['Id']);
+                    let errors = obj.errors || [];
+                    errors = errors.concat(recordResults.errors || []);
+                    if (recordResults === null || recordResults === void 0 ? void 0 : recordResults.skipped) {
+                        obj.status = 'Skipped';
+                    }
+                    else if (!recordResults || recordResults.hasErrors) {
+                        obj.status = 'Failed';
+                    }
+                    else {
+                        obj.status = 'Successfully migrated';
+                    }
+                    obj.errors = errors;
+                    obj.migratedId = recordResults.id;
+                    obj.warnings = recordResults.warnings;
+                    obj.migratedName = recordResults.newName;
+                }
+                mergedResults.push(obj);
             }
-            mergedResults.push(obj);
         }
         return mergedResults;
     }
+    logTruncationErrors(objectMigrationResults) {
+        objectMigrationResults.forEach((result) => {
+            if (result.errors && result.errors.length > 0) {
+                logger_1.Logger.error(messages.getMessage('truncationFailed', [result.name, result.errors.join(', ')]));
+            }
+        });
+    }
 }
-exports.default = Migrate;
 Migrate.description = messages.getMessage('commandDescription');
 Migrate.examples = messages.getMessage('examples').split(os.EOL);
-Migrate.args = [{ name: 'file' }];
-Migrate.flagsConfig = {
-    namespace: command_1.flags.string({
-        char: 'n',
-        description: messages.getMessage('namespaceFlagDescription'),
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+Migrate.args = [];
+Migrate.flags = {
+    'target-org': sf_plugins_core_1.Flags.optionalOrg({
+        summary: 'Target org username or alias',
+        char: 'u',
+        required: true,
+        aliases: ['targetusername'],
+        deprecateAliases: true,
+        makeDefault: false, // Prevent auto-resolution during command-reference generation
     }),
-    only: command_1.flags.string({
+    only: sf_plugins_core_1.Flags.string({
         char: 'o',
         description: messages.getMessage('onlyFlagDescription'),
     }),
-    allversions: command_1.flags.boolean({
+    allversions: sf_plugins_core_1.Flags.boolean({
         char: 'a',
         description: messages.getMessage('allVersionsDescription'),
         required: false,
     }),
-    relatedobjects: command_1.flags.string({
+    relatedobjects: sf_plugins_core_1.Flags.string({
         char: 'r',
         description: messages.getMessage('relatedObjectGA'),
     }),
-    verbose: command_1.flags.builtin({
-        type: 'builtin',
+    verbose: sf_plugins_core_1.Flags.boolean({
         description: 'Enable verbose output',
     }),
+    loglevel: sf_plugins_core_1.Flags.string({
+        description: 'Logging level (deprecated, use --verbose instead)',
+        deprecated: {
+            message: messages.getMessage('loglevelFlagDeprecated'),
+        },
+    }),
 };
+exports.default = Migrate;
 //# sourceMappingURL=migrate.js.map