Private/Submit-ScriptToVmAndExecute.ps1
|
function Global:Submit-ScriptToVmAndExecute { [CmdletBinding()] <# .SYNOPSIS Initializes the newly created VM .DESCRIPTION ... #> param( [Parameter(Mandatory = $true)] [string] $ResourceGroupName, [Parameter(Mandatory = $true)] [string] $ResourceLocation, [Parameter(Mandatory = $true)] [string] $VMName, [Parameter(Mandatory = $true)] [ValidateSet('InstallD365Module', 'InitVM', 'DownloadBC', 'InstallBC', 'GeneralizeVM', 'WriteProperties', 'CreateUpdateScheduledTask', 'InvokeAutoUpdate','UpdateNetworkSettings','EnableRemoting')] [string] $ScriptBlockName, [Parameter(Mandatory = $false)] $RunParameter, [Parameter(Mandatory = $true)] [string] $MsgBeforeExecuting, [Parameter(Mandatory = $false)] [switch] $ScaleSetExecution = $false ) process { . $PSScriptRoot\Scriptblocks.ps1 $tagName = "Command $ScriptBlockName" if (-not($ScaleSetExecution)) { $vm = Get-AzVM -ResourceGroupName $ResourceGroupName -Name $VMName if ($vm.Tags.ContainsKey($tagName)) { Write-CustomHost -Message "Command $ScriptBlockName was already executed." return } } if ($MsgBeforeExecuting) { Write-CustomHost -Message $MsgBeforeExecuting } $fullscriptpath = New-TemporaryFile | Rename-Item -NewName { $_ -replace 'tmp$', 'ps1' } -PassThru | Select-Object -ExpandProperty FullName $content = "" switch ($ScriptBlockName) { 'InstallD365Module' { $content = $installD365Module } 'InitVM' { $content = $initializeVm } 'DownloadBC' { $content = $downloadBCDVD } 'InstallBC' { $content = $installBC } 'GeneralizeVM' { $content = $generalizeVM } 'WriteProperties' { $content = $writeProperties } 'CreateUpdateScheduledTask' { $content = $createUpdateScheduledTask } 'InvokeAutoUpdate' { $content = $invokeAutoUpdate } 'UpdateNetworkSettings' { $content = $setNetConnectionProfile } 'EnableRemoting' { $content = $enableRemoting } } Set-Content -Path $fullscriptpath -Value $content Write-CustomHost -Message "Running command '$ScriptBlockName' on VM $VMName..." # Tag VM / VMSS $resource = Get-AzResource -ResourceGroupName $ResourceGroupName -Name $VMName $existingTags = $vm.Tags if ($existingTags.Count -ne 0) { $existingTags.Add($tagName, "Executed") Set-AzResource -ResourceId $resource.Id -Tag $existingTags -Force | Out-Null } else { Set-AzResource -ResourceId $resource.Id -Tag @{$tagName = "Executed" } -Force | Out-Null } if (-not($ScaleSetExecution)) { Invoke-AzVMRunCommand -ResourceGroupName $ResourceGroupName -VMName $VMName -CommandId 'RunPowerShellScript' -ScriptPath $fullscriptpath -Parameter $RunParameter | Out-Null } else { foreach ($instance in Get-AzVmssVM -ResourceGroupName $ResourceGroupName -VMScaleSetName $VMName){ $instanceId = $instance.InstanceId } Write-CustomHost -Message "on Instance $instanceId..." Invoke-AzVmssVMRunCommand -ResourceGroupName $ResourceGroupName -VMScaleSetName $VMName -InstanceId $instanceId -CommandId 'RunPowerShellScript' -ScriptPath $fullscriptpath -Parameter $RunParameter | Out-Null } Write-CustomHost -Message "Command completed." Remove-Item $fullscriptpath -Force } } |