01-Setup-ConversionEnvironment.ps1

#Requires -RunAsAdministrator
#Requires -Modules Az.Accounts, Az.Resources, Az.StackHCI

<#
.SYNOPSIS
    Sets up the environment for Gen 1 → Gen 2 VM conversion on Azure Local (Azure Stack HCI).

.DESCRIPTION
    This script:
    - Validates Azure Local cluster health and connectivity
    - Checks required PowerShell modules
    - Creates a working directory for conversion artifacts
    - Validates Azure Arc connectivity
    - Creates a conversion tracking log

.PARAMETER ClusterName
    Name of the Azure Local (HCI) cluster.

.PARAMETER WorkingDirectory
    Path for conversion artifacts (VHDX copies, logs, configs). Ideally a CSV or shared volume.

.PARAMETER SubscriptionId
    Azure subscription ID where the Arc-enabled VMs are registered.

.PARAMETER ResourceGroup
    Azure resource group containing the Arc-enabled VMs.

.EXAMPLE
    .\01-Setup-ConversionEnvironment.ps1 -ClusterName "AzSHCI-Cluster01" -WorkingDirectory "C:\ClusterStorage\Volume01\Gen2Conversion" -SubscriptionId "xxxxxxxx-xxxx-xxxx-xxxx-xxxxxxxxxxxx" -ResourceGroup "rg-azurelocal-prod"
#>

[CmdletBinding()]
param(
    [Parameter(Mandatory = $true)]
    [string]$ClusterName,

    [Parameter(Mandatory = $true)]
    [string]$WorkingDirectory,

    [Parameter(Mandatory = $true)]
    [string]$SubscriptionId,

    [Parameter(Mandatory = $true)]
    [string]$ResourceGroup
)

# ── Global Settings ──────────────────────────────────────────────────────────
$ErrorActionPreference = 'Stop'
$LogFile = Join-Path $WorkingDirectory "ConversionSetup_$(Get-Date -Format 'yyyyMMdd_HHmmss').log"

function Write-Log {
    param([string]$Message, [string]$Level = "INFO")
    $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
    $entry = "[$timestamp] [$Level] $Message"
    Write-Host $entry -ForegroundColor $(switch ($Level) { "ERROR" { "Red" } "WARN" { "Yellow" } "SUCCESS" { "Green" } default { "Cyan" } })
    if (Test-Path (Split-Path $LogFile -Parent)) {
        Add-Content -Path $LogFile -Value $entry
    }
}

# ── Step 1: Create Working Directory ─────────────────────────────────────────
Write-Log "═══════════════════════════════════════════════════════════════"
Write-Log "  Azure Local Gen1 → Gen2 Conversion - Environment Setup"
Write-Log "═══════════════════════════════════════════════════════════════"

Write-Log "Creating working directory: $WorkingDirectory"
$subDirs = @("Backups", "Configs", "Logs", "Temp", "Scripts")
foreach ($dir in $subDirs) {
    $path = Join-Path $WorkingDirectory $dir
    if (-not (Test-Path $path)) {
        New-Item -Path $path -ItemType Directory -Force | Out-Null
        Write-Log "  Created: $path"
    }
}
# Re-set log file now that directory exists
$LogFile = Join-Path $WorkingDirectory "Logs\ConversionSetup_$(Get-Date -Format 'yyyyMMdd_HHmmss').log"

# ── Step 2: Validate Required PowerShell Modules ─────────────────────────────
Write-Log "Checking required PowerShell modules..."

$requiredModules = @(
    @{ Name = "Az.Accounts";       MinVersion = "2.12.0" },
    @{ Name = "Az.Resources";      MinVersion = "6.0.0"  },
    @{ Name = "Az.StackHCI";       MinVersion = "1.0.0"  },
    @{ Name = "Az.ConnectedMachine"; MinVersion = "0.5.0" },
    @{ Name = "Hyper-V";           MinVersion = $null     },
    @{ Name = "FailoverClusters";  MinVersion = $null     }
)

$missingModules = @()
foreach ($mod in $requiredModules) {
    $installed = Get-Module -ListAvailable -Name $mod.Name | Sort-Object Version -Descending | Select-Object -First 1
    if (-not $installed) {
        Write-Log "  MISSING: $($mod.Name)" -Level "WARN"
        $missingModules += $mod.Name
    }
    else {
        Write-Log "  OK: $($mod.Name) v$($installed.Version)"
    }
}

if ($missingModules.Count -gt 0) {
    Write-Log "Installing missing modules: $($missingModules -join ', ')" -Level "WARN"
    foreach ($modName in $missingModules) {
        try {
            Install-Module -Name $modName -Force -AllowClobber -Scope CurrentUser
            Write-Log "  Installed: $modName" -Level "SUCCESS"
        }
        catch {
            Write-Log "  Failed to install $modName : $_" -Level "ERROR"
        }
    }
}

# ── Step 3: Validate Cluster Health ──────────────────────────────────────────
Write-Log "Validating Azure Local cluster: $ClusterName"

try {
    $cluster = Get-Cluster -Name $ClusterName
    Write-Log "  Cluster found: $($cluster.Name)" -Level "SUCCESS"

    # Check cluster nodes
    $nodes = Get-ClusterNode -Cluster $ClusterName
    foreach ($node in $nodes) {
        $status = if ($node.State -eq "Up") { "SUCCESS" } else { "WARN" }
        Write-Log "  Node: $($node.Name) - State: $($node.State)" -Level $status
    }

    # Check CSV health
    $csvs = Get-ClusterSharedVolume -Cluster $ClusterName
    foreach ($csv in $csvs) {
        $state = $csv.SharedVolumeInfo.FaultState
        $status = if ($state -eq "NoFaults") { "SUCCESS" } else { "WARN" }
        Write-Log "  CSV: $($csv.Name) - Fault State: $state" -Level $status
    }
}
catch {
    Write-Log "Failed to connect to cluster '$ClusterName': $_" -Level "ERROR"
    throw
}

# ── Step 4: Validate Azure Connectivity ──────────────────────────────────────
Write-Log "Validating Azure connectivity..."

try {
    $context = Get-AzContext
    if (-not $context) {
        Write-Log "Not logged into Azure. Initiating login..." -Level "WARN"
        Connect-AzAccount -SubscriptionId $SubscriptionId
        $context = Get-AzContext
    }

    if ($context.Subscription.Id -ne $SubscriptionId) {
        Set-AzContext -SubscriptionId $SubscriptionId | Out-Null
    }

    Write-Log "  Azure Account: $($context.Account.Id)" -Level "SUCCESS"
    Write-Log "  Subscription:  $SubscriptionId" -Level "SUCCESS"

    # Validate resource group exists
    $rg = Get-AzResourceGroup -Name $ResourceGroup -ErrorAction Stop
    Write-Log "  Resource Group: $($rg.ResourceGroupName) ($($rg.Location))" -Level "SUCCESS"
}
catch {
    Write-Log "Azure connectivity issue: $_" -Level "ERROR"
    throw
}

# ── Step 5: Validate Azure Local (HCI) Registration ─────────────────────────
Write-Log "Validating Azure Local cluster registration..."

try {
    $hciCluster = Get-AzStackHciCluster -ResourceGroupName $ResourceGroup -ErrorAction SilentlyContinue
    if ($hciCluster) {
        Write-Log "  HCI Cluster: $($hciCluster.Name)" -Level "SUCCESS"
        Write-Log "  Status:      $($hciCluster.Status)" -Level "SUCCESS"
        Write-Log "  Arc Status:  $($hciCluster.ArcServerResourceGroupName)" -Level "INFO"
    }
    else {
        Write-Log "  No Azure Stack HCI cluster found in resource group. Checking by name..." -Level "WARN"
    }
}
catch {
    Write-Log "Could not query HCI registration: $_" -Level "WARN"
}

# ── Step 6: Inventory Current Gen 1 VMs ─────────────────────────────────────
Write-Log "Inventorying Gen 1 VMs across cluster nodes..."

$allVMs = @()
foreach ($node in $nodes) {
    try {
        $vms = Get-VM -ComputerName $node.Name | Where-Object { $_.Generation -eq 1 }
        foreach ($vm in $vms) {
            $vmInfo = [PSCustomObject]@{
                VMName          = $vm.Name
                VMId            = $vm.VMId
                Generation      = $vm.Generation
                State           = $vm.State
                Host            = $node.Name
                MemoryMB        = $vm.MemoryAssigned / 1MB
                ProcessorCount  = $vm.ProcessorCount
                DynamicMemory   = $vm.DynamicMemoryEnabled
                VHDs            = ($vm | Get-VMHardDiskDrive | Select-Object -ExpandProperty Path) -join "; "
                NICs            = ($vm | Get-VMNetworkAdapter | Select-Object -ExpandProperty SwitchName) -join "; "
                CheckpointsExist = ($vm | Get-VMCheckpoint).Count -gt 0
            }
            $allVMs += $vmInfo
        }
    }
    catch {
        Write-Log "  Could not query VMs on node $($node.Name): $_" -Level "WARN"
    }
}

Write-Log "Found $($allVMs.Count) Gen 1 VMs across the cluster"

# Export inventory
$inventoryPath = Join-Path $WorkingDirectory "Configs\Gen1_VM_Inventory_$(Get-Date -Format 'yyyyMMdd').csv"
$allVMs | Export-Csv -Path $inventoryPath -NoTypeInformation
Write-Log "  Inventory exported: $inventoryPath" -Level "SUCCESS"

# Display summary
Write-Log ""
Write-Log "── Gen 1 VM Inventory Summary ──"
foreach ($vm in $allVMs) {
    $checkpointWarn = if ($vm.CheckpointsExist) { " [HAS CHECKPOINTS - Must remove first!]" } else { "" }
    Write-Log "  $($vm.VMName) | $($vm.State) | Host: $($vm.Host) | VHDs: $($vm.VHDs)$checkpointWarn"
}

# ── Step 7: Check Arc-enabled VM Resources ───────────────────────────────────
Write-Log ""
Write-Log "Checking for Arc-enabled VM resources in Azure..."

try {
    # Query for Arc-enabled HCI VMs
    $arcVMs = Get-AzResource -ResourceGroupName $ResourceGroup -ResourceType "Microsoft.AzureStackHCI/virtualMachineInstances" -ErrorAction SilentlyContinue

    if ($arcVMs) {
        Write-Log "  Found $($arcVMs.Count) Azure Local VM resources" -Level "SUCCESS"

        # Cross-reference with Gen 1 inventory
        $arcLookup = @{}
        foreach ($arcVM in $arcVMs) {
            $arcLookup[$arcVM.Name] = $arcVM.ResourceId
        }

        foreach ($vm in $allVMs) {
            if ($arcLookup.ContainsKey($vm.VMName)) {
                Write-Log "  Arc-managed: $($vm.VMName) → $($arcLookup[$vm.VMName])" -Level "SUCCESS"
            }
            else {
                Write-Log "  NOT Arc-managed: $($vm.VMName)" -Level "WARN"
            }
        }
    }
}
catch {
    Write-Log "Could not query Arc resources: $_" -Level "WARN"
}

# ── Step 8: Generate Configuration Export ────────────────────────────────────
Write-Log ""
Write-Log "Exporting detailed VM configurations for conversion..."

$configExportPath = Join-Path $WorkingDirectory "Configs"

foreach ($vm in $allVMs) {
    try {
        $vmDetail = Get-VM -Name $vm.VMName -ComputerName $vm.Host

        $config = [PSCustomObject]@{
            VMName             = $vmDetail.Name
            VMId               = $vmDetail.VMId.ToString()
            Generation         = $vmDetail.Generation
            Host               = $vm.Host
            State              = $vmDetail.State.ToString()
            ProcessorCount     = $vmDetail.ProcessorCount
            MemoryStartupMB    = $vmDetail.MemoryStartup / 1MB
            MemoryMinimumMB    = $vmDetail.MemoryMinimum / 1MB
            MemoryMaximumMB    = $vmDetail.MemoryMaximum / 1MB
            DynamicMemory      = $vmDetail.DynamicMemoryEnabled
            AutomaticStartAction = $vmDetail.AutomaticStartAction.ToString()
            AutomaticStopAction  = $vmDetail.AutomaticStopAction.ToString()
            Notes              = $vmDetail.Notes
        }

        # Get disk details
        $disks = Get-VMHardDiskDrive -VM $vmDetail
        $diskConfigs = @()
        foreach ($disk in $disks) {
            $vhd = Get-VHD -Path $disk.Path -ComputerName $vm.Host -ErrorAction SilentlyContinue
            $diskConfigs += [PSCustomObject]@{
                ControllerType     = $disk.ControllerType.ToString()
                ControllerNumber   = $disk.ControllerNumber
                ControllerLocation = $disk.ControllerLocation
                Path               = $disk.Path
                VhdType            = if ($vhd) { $vhd.VhdType.ToString() } else { "Unknown" }
                SizeGB             = if ($vhd) { [math]::Round($vhd.Size / 1GB, 2) } else { 0 }
                FileSizeGB         = if ($vhd) { [math]::Round($vhd.FileSize / 1GB, 2) } else { 0 }
                VhdFormat          = if ($vhd) { $vhd.VhdFormat.ToString() } else { "Unknown" }
            }
        }

        # Get NIC details
        $nics = Get-VMNetworkAdapter -VM $vmDetail
        $nicConfigs = @()
        foreach ($nic in $nics) {
            $nicConfigs += [PSCustomObject]@{
                Name         = $nic.Name
                SwitchName   = $nic.SwitchName
                MacAddress   = $nic.MacAddress
                VlanId       = (Get-VMNetworkAdapterVlan -VMNetworkAdapter $nic).AccessVlanId
                IsLegacy     = $nic.IsLegacy
                IPAddresses  = ($nic.IPAddresses -join ", ")
            }
        }

        # Save full config as JSON
        $fullConfig = @{
            VM    = $config
            Disks = $diskConfigs
            NICs  = $nicConfigs
        }

        $jsonPath = Join-Path $configExportPath "$($vm.VMName)_config.json"
        $fullConfig | ConvertTo-Json -Depth 5 | Out-File -FilePath $jsonPath -Encoding UTF8
        Write-Log "  Config exported: $jsonPath"
    }
    catch {
        Write-Log "  Failed to export config for $($vm.VMName): $_" -Level "ERROR"
    }
}

# ── Summary ──────────────────────────────────────────────────────────────────
Write-Log ""
Write-Log "═══════════════════════════════════════════════════════════════"
Write-Log "  Environment Setup Complete"
Write-Log "═══════════════════════════════════════════════════════════════"
Write-Log "  Working Directory:  $WorkingDirectory"
Write-Log "  Gen 1 VMs Found:   $($allVMs.Count)"
Write-Log "  Inventory CSV:     $inventoryPath"
Write-Log "  VM Configs:        $configExportPath"
Write-Log "  Log File:          $LogFile"
Write-Log ""
Write-Log "  Next Step: Run 02-Convert-MBRtoGPT.ps1 to prepare disks"
Write-Log "═══════════════════════════════════════════════════════════════"