fmt

0.2.7

Public/Build-FMTTestVM.ps1

function Build-FMTTestVM {
    param(
        [parameter()]
        [string] $rg,
        [parameter(Mandatory)]
        [string] $vmName,
        [parameter()]
        [string] $vnet,
        [parameter()]
        [string] $subNet,
        [parameter()]
        [string] $dataSubnet,
        [parameter()]
        [switch] $publicIP,
        [parameter()]
        [string] $subscriptionName,
        [parameter(Mandatory)]
        [System.Management.Automation.PSCredential] $adminCreds,
        [parameter()]
        [string] $size = 'Standard_D32s_v4',
        [parameter()]
        [switch] $Windows,
        [parameter()]
        [switch] $SQL,
        [parameter(Mandatory)]
        [int] $zone
    )

    # Check AZ context

    $session = New-FMTSession
    if (!$session) {
        $message = '-- stopping --'
        return $message
    }

    if ($subscriptionName) {
        try {
            Get-AzSubscription -SubscriptionName $subscriptionName | Set-AzContext
        } catch {
            $error[0]
            exit
        }
    }
    
    # Fill in the gaps in parameter list

    if (!$rg) {
        $rgArray = Get-AzResourceGroup | Select-Object resourcegroupname,location | Sort-Object resourcegroupname
        $rg = Build-MenuFromArray -array $rgArray -property resourcegroupname -message "Select the desired resource group"
    }

    if (!$vnet) {
        $vnetArray = Get-AzVirtualNetwork -ResourceGroupName $rg | Select-Object name,location | Sort-Object name
        $vnet = Build-MenuFromArray -array $vnetArray -property name -message "Select the desired Virtualnetwork"
    }

    if (!$subNet) {
        $subnetArray = Get-AzVirtualNetwork -Name $vnet | Get-AzVirtualNetworkSubnetConfig | Select-Object name
        $subnet = Build-MenuFromArray -array $subnetArray -property name -message "Select the desired management subnet"
    }

    if (!$dataSubnet) {
        $dataSubnetArray = Get-AzVirtualNetwork -Name $vnet | Get-AzVirtualNetworkSubnetConfig | Select-Object name
        $dataSubnet = Build-MenuFromArray -array $dataSubnetArray -property name -message "Select the desired data subnet. Leave blank for no data interface."
    }


    # Get location data
    $rgData = Get-AzResourceGroup -ResourceGroupName $rg

    # create a PIP
    if ($publicIP) {
        $pipName = $vmName + '-pip'
        $pip = Get-AzPublicIpAddress -ResourceGroupName $rg -Name $pipName -ErrorAction SilentlyContinue
        if (!$pip) {
            Write-Host -ForegroundColor yellow "-- Public IP not found, setting one up for use..."
            $pip = New-AzPublicIpAddress -Name $pipName -ResourceGroupName $rg -Location $rgData.Location -Sku Standard -AllocationMethod Static -Zone $zone
        }    
    }

    # Create Data interface(s)
    $nic1Name = $vmName + '-mgmt'
    $nic1subnet = Get-AzVirtualNetwork -ResourceGroupName $rg -Name $vnet | Get-AzVirtualNetworkSubnetConfig | where-object {$_.name -eq $subNet}
    if ($publicIP) {
        $mgmtNic = New-AzNetworkInterface -Subnet $nic1subnet -Name $nic1Name -ResourceGroupName $rg -Location $rgData.Location -EnableAcceleratedNetworking -PublicIpAddress $pip
    } else {
        $mgmtNic = New-AzNetworkInterface -Subnet $nic1subnet -Name $nic1Name -ResourceGroupName $rg -Location $rgData.Location -EnableAcceleratedNetworking 
    }

    if ($dataSubnet) {
        Write-Host '-- Adding Data interface --'
        $nic2Name = $vmName + '-data'
        $nic2subnet = Get-AzVirtualNetwork -ResourceGroupName $rg -Name $vnet | Get-AzVirtualNetworkSubnetConfig | where-object {$_.name -eq $dataSubnet}
        $dataNic = New-AzNetworkInterface -Subnet $nic2subnet -Name $nic2Name -ResourceGroupName $rg -Location $rgData.Location -EnableAcceleratedNetworking
    }

    # Create the user credential
    $cred = $adminCreds

    # create the VM config
    $vmConfig = New-AzVMConfig -VMName $vmName -VMSize $size -Zone $zone

    if ($Windows) {
        Write-Host '-- Setting OS for Windows --'
        $vmConfig | Set-AzVMOperatingSystem -Windows -ComputerName $vmName -Credential $cred 
        $vmConfig | Set-AzVMSourceImage -PublisherName "MicrosoftWindowsServer" -Offer "WindowsServer" -Skus "2019-datacenter-gensecond" -Version "latest"
    } elseif ($SQL) {
        Write-Host '-- Setting OS for Windows --'
        $vmConfig | Set-AzVMOperatingSystem -Windows -ComputerName $vmName -Credential $cred 
        $vmConfig | Set-AzVMSourceImage -PublisherName "microsoftsqlserver" -Offer "sql2019-ws2019" -Skus "enterprise" -Version "latest"
    } else {
        Write-Host '-- Setting OS for Linux --'
        $vmConfig | Set-AzVMOperatingSystem -Linux -ComputerName $vmName -Credential $cred
        $vmConfig | Set-AzVMSourceImage -PublisherName "Canonical" -Offer "0001-com-ubuntu-server-focal" -Skus "20_04-lts" -Version "latest"
    }
    $vmConfig | Add-AzVMNetworkInterface -Id $mgmtNic.Id

    if ($dataSubnet) {
        $vmConfig.NetworkProfile.NetworkInterfaces[0].Primary = $true
        $vmConfig | Add-AzVMNetworkInterface -Id $dataNic.Id
    }

    # Create a VM using the abov config
    New-AzVm -ResourceGroupName $rg -Location $rgData.Location -Zone $zone -VM $vmConfig
}