Preflight/Preflight.psm1

Import-Module -Name $PSScriptRoot\..\Util\Util
Import-Module -Name $PSScriptRoot\..\CloudAccount\CloudAccount
Import-Module -Name $PSScriptRoot\..\MigrationProfile\MigrationProfile
function Get-RMCloudAttribute {
    param(
        [System.Object] $CloudAccount
    )

    $Response = Start-RMCloudAttributesPreflight -CloudAccount $CloudAccount
    Write-Output "Waiting for cloud attribute collection to complete..." | Out-Host
    $PreflightResult = Watch-RMPreflightStatus -PreflightId $Response.preflights[0].id `
        -TimeOutMessage "Timed out waiting for collection to complete"
    if ($PreflightResult.state -eq "success" -or ($PreflightResult.state -eq "error" -and $PreflightResult.error_type -eq "warning")) {
        return Get-CachedCloudAttribute -CloudAccount $CloudAccount
    } else {
        $PreflightID = $Response.preflights[0].id
        throw "Failed to collect attributes, preflight ID: $PreflightID"
    }
}

function Start-RMCloudAttributesPreflight {
    param(
        [System.Object] $CloudAccount
    )

    $RequestAttributes = @{
        "resource_id"= $CloudAccount.id
        "cloud_type" = $CloudAccount.type
        "type" = "target_cloud"
    }

    $CloudAttributesRequest = @($RequestAttributes)
    $CloudAttributesRequestJson = ConvertTo-Json $CloudAttributesRequest

    $RMLoginResult = Get-Variable -Name "RMContext-UserLogin" -ValueOnly
    $Uri = Get-Variable -Name "RMContext-ReactorURI" -ValueOnly

    $Headers = @{
        Accept = "application/rm+json"
        "X-Auth-Token" = $RMLoginResult.token
    }

    $Params = @{
        Method = "Post"
        Uri = $Uri + "/preflights"
        Body = $CloudAttributesRequestJson
        ContentType = "application/json"
        Headers = $Headers
    }

    Write-Output "Starting target cloud attribute collection..." | Out-Host
    return Invoke-RMRestMethod -Params $Params
}

function Start-RMMigrationPreflight {
    param(
        [string] $MigrationProfileId,
        [bool] $IgnoreValidationErrors
    )
    $MigrationProfileResponse = Get-RMMigrationProfileById -MigrationProfileId $MigrationProfileId
    $MigrationPreflightRequest = @(
        @{
            "type" = "source_config"
            "resource_id" = $MigrationProfileResponse.sources[0].id
            "overrides" = @{
                "ignore_validation_errors" = $IgnoreValidationErrors
            }
        }, @{
            "type" = "migration_profile"
            "resource_id" = $MigrationProfileId
            "overrides" = @{
                "ignore_validation_errors" = $IgnoreValidationErrors
            }
        }
    )

    $MigrationPreflightRequestJson = ConvertTo-Json $MigrationPreflightRequest

    $RMLoginResult = Get-Variable -Name "RMContext-UserLogin" -ValueOnly
    $Uri = Get-Variable -Name "RMContext-ReactorURI" -ValueOnly

    $Headers = @{
        Accept = "application/rm+json"
        "X-Auth-Token" = $RMLoginResult.token
    }

    $Params = @{
        Method = "Post"
        Uri = $Uri + "/preflights"
        Body = $MigrationPreflightRequestJson
        ContentType = "application/json"
        Headers = $Headers
    }

    Write-Output "Starting migration preflight..." | Out-Host
    $Response = Invoke-RMRestMethod -Params $Params
    Write-Output "Waiting for migration preflight to complete..." | Out-Host
    foreach ($Preflight in $Response.preflights) {
        $PreflightResult = Watch-RMPreflightStatus -PreflightId $Preflight.id `
            -TimeOutMessage "Timed out waiting for migration preflight to complete"
        Out-RMPreflight -PreflightResult $PreflightResult -IgnoreValidationErrors $IgnoreValidationErrors            
    }

}

function Out-RMPreflight {
    param(
        [System.Object] $PreflightResult,
        [bool] $IgnoreValidationErrors
    )

    $HasErrors = $false
    foreach ($Preflight in $PreflightResult.content) {
        if ($Preflight.state -eq "success") {
            continue
        }
        Write-Output $Preflight.label | Out-Host
        foreach ($Error in $Preflight.errors) {
            if ($Preflight.error_type -eq "warning") {
                Write-Warning -Message $Error.message
            }
            if ($Preflight.error_type -eq "fatal" -or $Preflight.error_type -eq "blocker") {
                Write-RMError -Message $Error.message
                $HasErrors = $true
            }
        }
    }

    if ($HasErrors -and !$IgnoreValidationErrors) {
        Exit 1
    }
}

Export-ModuleMember -Function Get-RMCloudAttribute, Start-RMMigrationPreflight, Out-RMPreflight