Migrate Migrate from generation to main (#25087)

* Move Migrate to main

* Update ChangeLog.md

---------

Co-authored-by: NoriZC <[email protected]>

Author: azure-powershell-bot

src/Migrate/Migrate.Autorest/build-module.ps1

@@ -12,7 +12,7 @@
 # Code generated by Microsoft (R) AutoRest Code Generator.Changes may cause incorrect behavior and will be lost if the code
 # is regenerated.
 # ----------------------------------------------------------------------------------
-param([switch]$Isolated, [switch]$Run, [switch]$Test, [switch]$Docs, [switch]$Pack, [switch]$Code, [switch]$Release, [switch]$Debugger, [switch]$NoDocs)
+param([switch]$Isolated, [switch]$Run, [switch]$Test, [switch]$Docs, [switch]$Pack, [switch]$Code, [switch]$Release, [switch]$Debugger, [switch]$NoDocs, [Switch]$DisableAfterBuildTasks)
 $ErrorActionPreference = 'Stop'
 
 if($PSEdition -ne 'Core') {
@@ -169,4 +169,14 @@ if (Test-Path (Join-Path $PSScriptRoot 'generate-portal-ux.ps1'))
   . (Join-Path $PSScriptRoot 'generate-portal-ux.ps1')
 }
 
+if (-not $DisableAfterBuildTasks){
+  $afterBuildTasksPath = Join-Path $PSScriptRoot ''
+  $afterBuildTasksArgs = ConvertFrom-Json 'true' -AsHashtable
+  if(Test-Path -Path $afterBuildTasksPath -PathType leaf){
+    Write-Host -ForegroundColor Green 'Running after build tasks...'
+    . $afterBuildTasksPath @afterBuildTasksArgs
+  }
+}
+
+
 Write-Host -ForegroundColor Green '-------------Done-------------'

src/Migrate/Migrate.Autorest/custom/Get-AzMigrateHCIServerReplication.ps1

@@ -162,58 +162,45 @@ function Get-AzMigrateHCIServerReplication {
 
             # Occasionally, Get Machine Site will not return machine site even when the site exist,
             # hence retry get machine site.
-            $attempts = 4
-            for ($i = 1; $i -le $attempts; $i++) {
-                try {
-                    if ($siteType -eq $SiteTypes.VMwareSites) {
-                        $siteObject = Az.Migrate\Get-AzMigrateSite @PSBoundParameters -ErrorVariable notPresent -ErrorAction SilentlyContinue
-                    }
-                    elseif ($siteType -eq $SiteTypes.HyperVSites) {
-                        $siteObject = Az.Migrate.Internal\Get-AzMigrateHyperVSite @PSBoundParameters -ErrorVariable notPresent -ErrorAction SilentlyContinue
-                    } 
+            if ($siteType -eq $SiteTypes.VMwareSites) {
+                $siteObject = InvokeAzMigrateGetCommandWithRetries `
+                    -CommandName 'Az.Migrate\Get-AzMigrateSite' `
+                    -Parameters $PSBoundParameters `
+                    -ErrorMessage "Machine site '$siteName' with Type '$siteType' not found."
+            } elseif ($siteType -eq $SiteTypes.HyperVSites) {
+                $siteObject = InvokeAzMigrateGetCommandWithRetries `
+                    -CommandName 'Az.Migrate.Internal\Get-AzMigrateHyperVSite' `
+                    -Parameters $PSBoundParameters `
+                    -ErrorMessage "Machine site '$siteName' with Type '$siteType' not found."
+            }
 
-                    if ($null -eq $siteObject) {
-                        throw "Machine site not found."
-                    }
-                    else {
-                        $ProjectName = $siteObject.DiscoverySolutionId.Split("/")[8]
-                    }
+            # $siteObject is not null or exception would have been thrown
+            $ProjectName = $siteObject.DiscoverySolutionId.Split("/")[8]
 
-                    break;
-                }
-                catch {
-                    if ($i -lt $attempts)
-                    {
-                        Write-Host "Machine site not found. Retrying in 30 seconds..."
-                        Start-Sleep -Seconds 30
-                    }
-                    else
-                    {
-                        throw "Machine site '$siteName' with Type '$siteType' not found."
-                    }
-                }
-            }
-                
             $null = $PSBoundParameters.Remove('SiteName')
 
-            $null = $PSBoundParameters.Add("Name", "Servers-Migration-ServerMigration_DataReplication")
+            # Get the migrate solution.
+            $amhSolutionName = "Servers-Migration-ServerMigration_DataReplication"
+            $null = $PSBoundParameters.Add("Name", $amhSolutionName)
             $null = $PSBoundParameters.Add("MigrateProjectName", $ProjectName)
-                    
-            $solution = Az.Migrate\Get-AzMigrateSolution @PSBoundParameters
-            if ($solution -and ($solution.Count -ge 1)) {
-                $VaultName = $solution.DetailExtendedDetail.AdditionalProperties.vaultId.Split("/")[8]
-            }
-            else {
-                throw "Solution not found."
-            }
+
+            $solution = InvokeAzMigrateGetCommandWithRetries `
+                -CommandName 'Az.Migrate\Get-AzMigrateSolution' `
+                -Parameters $PSBoundParameters `
+                -ErrorMessage "No Data Replication Service Solution '$amhSolutionName' found in resource group '$ResourceGroupName' and project '$ProjectName'. Please verify your appliance setup."
 
             $null = $PSBoundParameters.Remove("Name")
             $null = $PSBoundParameters.Remove("MigrateProjectName")
+
+            $VaultName = $solution.DetailExtendedDetail.AdditionalProperties.vaultId.Split("/")[8]
+            if ([string]::IsNullOrEmpty($VaultName)) {
+                throw "Azure Migrate Project not configured: missing replication vault. Setup Azure Migrate Project and run the Initialize-AzMigrateHCIReplicationInfrastructure script before proceeding."
+            }
 
             $null = $PSBoundParameters.Add("VaultName", $VaultName)
             $null = $PSBoundParameters.Add("Name", $ProtectedItemName)
 
-            return Az.Migrate.Internal\Get-AzMigrateProtectedItem @PSBoundParameters -ErrorVariable notPresent -ErrorAction SilentlyContinue
+            return Az.Migrate.Internal\Get-AzMigrateProtectedItem @PSBoundParameters
         }
 
         if (($parameterSet -match 'List') -or ($parameterSet -eq 'GetByMachineName')) {
@@ -234,23 +221,27 @@ function Get-AzMigrateHCIServerReplication {
                 $ProjectName = $projectIdArray[8]
             }
 
+            $amhSolutionName = "Servers-Migration-ServerMigration_DataReplication"
             $null = $PSBoundParameters.Add("ResourceGroupName", $ResourceGroupName)
-            $null = $PSBoundParameters.Add("Name", "Servers-Migration-ServerMigration_DataReplication")
+            $null = $PSBoundParameters.Add("Name", $amhSolutionName)
             $null = $PSBoundParameters.Add("MigrateProjectName", $ProjectName)
-                
-            $solution = Az.Migrate\Get-AzMigrateSolution @PSBoundParameters
-            if ($solution -and ($solution.Count -ge 1)) {
-                $VaultName = $solution.DetailExtendedDetail.AdditionalProperties.vaultId.Split("/")[8]
-            }
-            else {
-                throw "Solution not found."
-            }
+
+            $solution = InvokeAzMigrateGetCommandWithRetries `
+                -CommandName 'Az.Migrate\Get-AzMigrateSolution' `
+                -Parameters $PSBoundParameters `
+                -ErrorMessage "No Data Replication Service Solution '$amhSolutionName' found in resource group '$ResourceGroupName' and project '$ProjectName'. Please verify your appliance setup."
 
             $null = $PSBoundParameters.Remove("Name")
             $null = $PSBoundParameters.Remove("MigrateProjectName")
+
+            $VaultName = $solution.DetailExtendedDetail.AdditionalProperties.vaultId.Split("/")[8]
+            if ([string]::IsNullOrEmpty($VaultName)) {
+                throw "Azure Migrate Project not configured: missing replication vault. Setup Azure Migrate Project and run the Initialize-AzMigrateHCIReplicationInfrastructure script before proceeding."
+            }
+
             $null = $PSBoundParameters.Add("VaultName", $VaultName)
 
-            $replicatingItems = Az.Migrate.Internal\Get-AzMigrateProtectedItem @PSBoundParameters -ErrorVariable notPresent -ErrorAction SilentlyContinue
+            $replicatingItems = Az.Migrate.Internal\Get-AzMigrateProtectedItem @PSBoundParameters
 
             if ($parameterSet -eq "GetByMachineName") {
                 $replicatingItems = $replicatingItems | Where-Object { $_.Property.FabricObjectName -eq $MachineName }
@@ -274,7 +265,7 @@ function Get-AzMigrateHCIServerReplication {
             $null = $PSBoundParameters.Add("VaultName", $VaultName)
             $null = $PSBoundParameters.Add("Name", $ProtectedItemName)
 
-            return Az.Migrate.Internal\Get-AzMigrateProtectedItem @PSBoundParameters -ErrorVariable notPresent -ErrorAction SilentlyContinue
+            return Az.Migrate.Internal\Get-AzMigrateProtectedItem @PSBoundParameters
         }
     }
 }

src/Migrate/Migrate.Autorest/custom/Helper/AzStackHCICommonSettings.ps1

@@ -75,4 +75,10 @@ enum StorageAccountProvisioningState
     Creating
     ResolvingDNS
     Succeeded
+}
+
+# Selection state for VM NIC
+$VMNicSelection = @{
+    SelectedByUser = "SelectedByUser";
+    NotSelected    = "NotSelected";
 }

src/Migrate/Migrate.Autorest/custom/Helper/CommonHelper.ps1

@@ -176,4 +176,58 @@ function GenerateHashForArtifact {
     else {
         return $hashCode
     }
+}
+
+function InvokeAzMigrateGetCommandWithRetries {
+    [Microsoft.Azure.PowerShell.Cmdlets.Migrate.DoNotExportAttribute()]
+    param(
+        [Parameter(Mandatory)]
+        [System.String]
+        # Specifies the name of Az.Migrate command.
+        ${CommandName},
+
+        [Parameter(Mandatory)]
+        [System.Collections.Hashtable]
+        # Specifies the parameters for Az.Migrate command.
+        ${Parameters},
+
+        [Parameter()]
+        [System.Int32]
+        # Specifies the maximum number of retries.
+        ${MaxRetryCount} = 3,
+
+        [Parameter()]
+        [System.Int32]
+        # Specifies the delay between retries in seconds.
+        ${RetryDelayInSeconds} = 30,
+
+        [Parameter()]
+        [System.String]
+        # Specifies the error message to throw if command fails.
+        ${ErrorMessage} = "Internal Az.Migrate commands failed to execute."
+    )
+
+    process {
+        for ($i = 0; $i -le $MaxRetryCount; $i++) {
+            try {
+                $result = & $CommandName @Parameters -ErrorVariable notPresent -ErrorAction SilentlyContinue
+
+                if ($null -eq $result) {
+                    throw $ErrorMessage
+                }
+
+                break
+            }
+            catch {
+                if ($i -lt $MaxRetryCount) {
+                    Start-Sleep -Seconds $RetryDelayInSeconds
+                }
+                else {
+                    throw $ErrorMessage
+                }
+            }
+        }
+
+        return $result
+    }
 }