HyperDeploy

0.1

Private/Initialize-VM.ps1


function Wait-ForOKHeartbeat {
    Param
    (
        [Parameter(Mandatory)]
        [VM]$VM,
        [Parameter(Mandatory)]
        [HyperVServer]$HyperVServer
    )

    while ((Get-VM -Name $VM.Name -ComputerName $HyperVServer.Name).HeartBeat -ne 'OkApplicationsUnknown') {
        Start-Sleep -Seconds 1
        $name = $VM.Name
        #Probs only want to do this when -verbose
        Write-Verbose "$name not ready. Waiting"

    }

}
function Wait-ForResponsiveVM {
    Param
    (
        [Parameter(Mandatory)]
        [VM]$VM,
        [Parameter(Mandatory)]
        [HyperVServer]$HyperVServer
    )

    Wait-ForOKHeartbeat -VM $VM -HyperVServer $HyperVServer

    #Wait for DHCP to assign IP
    while ((Get-VMNetworkAdapter -VMName $VM.Name -ComputerName $HyperVServer.Name).IpAddresses.Count -eq 0) {
        Start-Sleep -Seconds 1

        $name = $VM.Name
        Write-Verbose "$name has no IP yet. Waiting"

    }

    #Get VM IP Address
    $IP = (Get-VMNetworkAdapter -VMName $VM.Name -ComputerName $HyperVServer.Name).IpAddresses[0] 

    write-Verbose "IP is $IP"
    Write-Verbose "Waiting for WinRM"

    #Wait for WINRM to be responsive
    $failedWSMan = $false
    $Future5 = (Get-Date).AddMinutes(5)

    do { 
        $failedWSMan = $false
        try {
            test-wsman $IP -ErrorAction Stop
        }
        catch { 
            $failedWSMan = $true
        }


    }while ($failedWSMan -and (Get-Date) -lt ($Future5))

    return $IP
}

function Initialize-VM {
    Param
    (
        [Parameter(Mandatory)]
        [VM]$VM,
        [Parameter(Mandatory)]
        [HyperVServer]$HyperVServer

    )

    Set-Item wsman:\localhost\Client\TrustedHosts -value * -Force

    $VMName = $VM.Name
   
    $ip = Wait-ForResponsiveVM -VM $VM -HyperVServer $HyperVServer
    if ($ip) {
        Write-Verbose "WinRM connection established"
    }
    else {
        throw "Timed out waiting for WinRM"
    }

    $newCred = $null

    foreach ($script in $VM.Provisioning.Scripts) {


        if ($script.EndsWith(".Set-Credential.ps1")) {
            
            Write-Verbose "Credential script detected, getting credentials"

            $credObject = invoke-expression -Command $script

            if ($credObject -and $credObject.GetType().Name -eq 'PSCredential') {
                Write-Verbose "New credentials set" 

                $newCred = $credObject
            }
        }
        else {

            Write-Verbose "Provisioning $VMName using $script" 

            $InvokeParams = @{ 
                FilePath     = $script
                ComputerName = $ip[1]
            }
           
            if ($newCred) {
                $InvokeParams.Add("Credential", $newCred)
            }
            
            invoke-command @InvokeParams
    
            if ($VM.Provisioning.RebootAfterEachScript) {
                Stop-VM -Name  $VMName -ComputerName $HyperVServer.Name -Force
                Start-VM -Name  $VMName -ComputerName $HyperVServer.Name
        
                Wait-ForResponsiveVM -VM $VM -HyperVServer $HyperVServer
            }
        }

        

    }


}