Public/Invoke-ComputerMaintenance.ps1
function Invoke-ComputerMaintenance { [CmdletBinding()] Param ( [Parameter(Mandatory)] [string]$ComputerName, [int]$PreventiveLockTimeout = $ModuleWidePreventiveLockTimeout, [System.TimeSpan]$PreventiveLockThreshold = $ModuleWidePreventiveLockThreshold, [string]$SkipPreventivelyLockedExceptionTarget = $ModuleWideSkipPreventivelyLockedExceptionTarget, [switch]$SkipNotLockable = $ModuleWideSkipNotLockable, [switch]$SkipPreventivelyLocked = $ModuleWideSkipPreventivelyLocked, [switch]$EnableMaintenanceLog = $ModuleWideEnableMaintenanceLog ) $ErrorActionPreference = 'Stop' Write-Debug -Message ('ENTER {0}' -f $MyInvocation.MyCommand.Name) try { Write-Debug -Message ('ENTER TRY {0}' -f $MyInvocation.MyCommand.Name) Write-Debug -Message ('$ComputerName = ''{0}''' -f $ComputerName) Write-Debug -Message ('$PreventiveLockTimeout = {0}' -f $PreventiveLockTimeout) Write-Debug -Message ('$PreventiveLockThreshold: ''{0}''' -f [string]$PreventiveLockThreshold) Write-Debug -Message ('SkipPreventivelyLockedExceptionTarget = ''{0}''' -f $SkipPreventivelyLockedExceptionTarget) Write-Debug -Message ('$SkipNotLockable = ${0}' -f $SkipNotLockable) Write-Debug -Message ('$SkipPreventivelyLocked = ${0}' -f $SkipPreventivelyLocked) Write-Debug -Message ('$EnableMaintenanceLog = ${0}' -f $EnableMaintenanceLog) Write-Debug -Message ('$IsMaintenanceAllowed = Test-MaintenanceAllowed -ComputerName ''{0}''' -f $ComputerName) $IsMaintenanceAllowed = Test-MaintenanceAllowed -ComputerName $ComputerName Write-Debug -Message ('$IsMaintenanceAllowed = ${0}' -f $IsMaintenanceAllowed) Write-Debug -Message 'if ($IsMaintenanceAllowed)' if ($IsMaintenanceAllowed) { Write-Debug -Message '$HostLock = $null' $HostLock = $null Write-Debug -Message ('$HostLock: ''{0}''' -f $HostLock) Write-Debug -Message '$DestinationHostLock = $null' $DestinationHostLock = $null Write-Debug -Message ('$DestinationHostLock: ''{0}''' -f $DestinationHostLock) Write-Debug -Message ('$PendingReboot = Test-PendingReboot -ComputerName ''{0}'' -Detailed' -f $ComputerName) $PendingReboot = Test-PendingReboot -ComputerName $ComputerName -Detailed Write-Debug -Message ('$PendingReboot: ''{0}''' -f $PendingReboot) Write-Debug -Message ('$PendingReboot.PendingFileRenameOperationsValue: ''{0}''' -f [string]$PendingReboot.PendingFileRenameOperationsValue) Write-Debug -Message ('$PendingReboot.ComponentBasedServicing: ''{0}''' -f $PendingReboot.ComponentBasedServicing) Write-Debug -Message 'if ($PendingReboot.ComponentBasedServicing)' if ($PendingReboot.ComponentBasedServicing) { # Sometimes, when CBS requires reboot, search for updates hangs up. Therefore, we need to reboot the machine first. Write-Debug -Message ('$ComputerWorkload = Initialize-ComputerMaintenance -ComputerName ''{0}'' -HostLock ([ref]$HostLock) -DestinationHostLock ([ref]$DestinationHostLock) -SkipNotLockable:${1} -SkipPreventivelyLocked:${2} -SkipPreventivelyLockedExceptionTarget ''{3}''' -f $ComputerName, $SkipNotLockable, $SkipPreventivelyLocked, $SkipPreventivelyLockedExceptionTarget) $ComputerWorkload = Initialize-ComputerMaintenance -ComputerName $ComputerName -HostLock ([ref]$HostLock) -DestinationHostLock ([ref]$DestinationHostLock) -SkipNotLockable:$SkipNotLockable -SkipPreventivelyLocked:$SkipPreventivelyLocked -SkipPreventivelyLockedExceptionTarget $SkipPreventivelyLockedExceptionTarget Write-Debug -Message ('$ComputerWorkload: ''{0}''' -f [string]$ComputerWorkload) Write-Debug -Message ('$HostLock: ''{0}''' -f $HostLock) Write-Debug -Message 'if ($HostLock)' if ($HostLock) { Write-Debug -Message ('Invoke-ComputerRestart -ComputerName ''{0}''' -f $ComputerName) Invoke-ComputerRestart -ComputerName $ComputerName Write-Debug -Message ('$IsMaintenanceNeeded = Test-MaintenanceNeeded -ComputerName ''{0}''' -f $ComputerName) $IsMaintenanceNeeded = Test-MaintenanceNeeded -ComputerName $ComputerName Write-Debug -Message ('$IsMaintenanceNeeded = ${0}' -f $IsMaintenanceNeeded) if ($IsMaintenanceNeeded) { Write-Debug -Message ('Start-Maintenance -ComputerName ''{0}'' -EnableMaintenanceLog:${1}' -f $ComputerName, $EnableMaintenanceLog) Start-Maintenance -ComputerName $ComputerName -EnableMaintenanceLog:$EnableMaintenanceLog } Write-Debug -Message ('$ComputerWorkload: ''{0}''' -f [string]$ComputerWorkload) Write-Debug -Message ('$DestinationHostLock: ''{0}''' -f $DestinationHostLock) Write-Debug -Message ('Complete-ComputerMaintenance -ComputerName ''{0}'' -ComputerWorkload $ComputerWorkload -DestinationHostLock ([ref]$DestinationHostLock)' -f $ComputerName) Complete-ComputerMaintenance -ComputerName $ComputerName -ComputerWorkload $ComputerWorkload -DestinationHostLock ([ref]$DestinationHostLock) } } else { # Most frequent case: CBS does not prevent from searching for updates Write-Debug -Message ('$IsMaintenanceNeeded = Test-MaintenanceNeeded -ComputerName ''{0}''' -f $ComputerName) $IsMaintenanceNeeded = Test-MaintenanceNeeded -ComputerName $ComputerName Write-Debug -Message ('$IsMaintenanceNeeded = ${0}' -f $IsMaintenanceNeeded) if ($IsMaintenanceNeeded) { Write-Debug -Message ('$ComputerWorkload = Initialize-ComputerMaintenance -ComputerName ''{0}'' -HostLock ([ref]$HostLock) -DestinationHostLock ([ref]$DestinationHostLock) -SkipNotLockable:${1} -SkipPreventivelyLocked:${2} -SkipPreventivelyLockedExceptionTarget ''{3}''' -f $ComputerName, $SkipNotLockable, $SkipPreventivelyLocked, $SkipPreventivelyLockedExceptionTarget) $ComputerWorkload = Initialize-ComputerMaintenance -ComputerName $ComputerName -HostLock ([ref]$HostLock) -DestinationHostLock ([ref]$DestinationHostLock) -SkipNotLockable:$SkipNotLockable -SkipPreventivelyLocked:$SkipPreventivelyLocked -SkipPreventivelyLockedExceptionTarget $SkipPreventivelyLockedExceptionTarget Write-Debug -Message ('$ComputerWorkload: ''{0}''' -f [string]$ComputerWorkload) Write-Debug -Message ('$HostLock: ''{0}''' -f $HostLock) Write-Debug -Message 'if ($HostLock)' if ($HostLock) { Write-Debug -Message ('Start-Maintenance -ComputerName ''{0}'' -EnableMaintenanceLog:${1}' -f $ComputerName, $EnableMaintenanceLog) Start-Maintenance -ComputerName $ComputerName -EnableMaintenanceLog:$EnableMaintenanceLog Write-Debug -Message ('$ComputerWorkload: ''{0}''' -f [string]$ComputerWorkload) Write-Debug -Message ('$DestinationHostLock: ''{0}''' -f $DestinationHostLock) Write-Debug -Message ('Complete-ComputerMaintenance -ComputerName ''{0}'' -ComputerWorkload $ComputerWorkload -DestinationHostLock ([ref]$DestinationHostLock)' -f $ComputerName) Complete-ComputerMaintenance -ComputerName $ComputerName -ComputerWorkload $ComputerWorkload -DestinationHostLock ([ref]$DestinationHostLock) } } else { # If the machine does not need any updates, but just a reboot, let's reboot it. Write-Debug -Message ('$PendingReboot.IsRebootPending: ''{0}''' -f $PendingReboot.IsRebootPending) if ($PendingReboot.IsRebootPending) { Write-Debug -Message ('$ComputerWorkload = Initialize-ComputerMaintenance -ComputerName ''{0}'' -HostLock ([ref]$HostLock) -DestinationHostLock ([ref]$DestinationHostLock) -SkipNotLockable:${1} -SkipPreventivelyLocked:${2} -SkipPreventivelyLockedExceptionTarget ''{3}''' -f $ComputerName, $SkipNotLockable, $SkipPreventivelyLocked, $SkipPreventivelyLockedExceptionTarget) $ComputerWorkload = Initialize-ComputerMaintenance -ComputerName $ComputerName -HostLock ([ref]$HostLock) -DestinationHostLock ([ref]$DestinationHostLock) -SkipNotLockable:$SkipNotLockable -SkipPreventivelyLocked:$SkipPreventivelyLocked -SkipPreventivelyLockedExceptionTarget $SkipPreventivelyLockedExceptionTarget Write-Debug -Message ('$ComputerWorkload: ''{0}''' -f [string]$ComputerWorkload) Write-Debug -Message ('$HostLock: ''{0}''' -f $HostLock) Write-Debug -Message 'if ($HostLock)' if ($HostLock) { Write-Debug -Message ('Invoke-ComputerRestart -ComputerName ''{0}''' -f $ComputerName) Invoke-ComputerRestart -ComputerName $ComputerName Write-Debug -Message ('$ComputerWorkload: ''{0}''' -f [string]$ComputerWorkload) Write-Debug -Message ('$DestinationHostLock: ''{0}''' -f $DestinationHostLock) Write-Debug -Message ('Complete-ComputerMaintenance -ComputerName ''{0}'' -ComputerWorkload $ComputerWorkload -DestinationHostLock ([ref]$DestinationHostLock)' -f $ComputerName) Complete-ComputerMaintenance -ComputerName $ComputerName -ComputerWorkload $ComputerWorkload -DestinationHostLock ([ref]$DestinationHostLock) } } } } } Write-Debug -Message ('EXIT TRY {0}' -f $MyInvocation.MyCommand.Name) } catch { Write-Debug -Message ('ENTER CATCH {0}' -f $MyInvocation.MyCommand.Name) Write-Debug -Message ('$_.TargetObject: ''{0}''' -f $_.TargetObject) Write-Debug -Message ('if ($_.TargetObject -ne ''{0}'')' -f $SkipPreventivelyLockedExceptionTarget) if ($_.TargetObject -ne $SkipPreventivelyLockedExceptionTarget) { Write-Debug -Message ('{0}: $PSCmdlet.ThrowTerminatingError($_)' -f $MyInvocation.MyCommand.Name) $PSCmdlet.ThrowTerminatingError($_) } Write-Debug -Message ('EXIT CATCH {0}' -f $MyInvocation.MyCommand.Name) } finally { Write-Debug -Message ('ENTER FINALLY {0}' -f $MyInvocation.MyCommand.Name) Write-Debug -Message ('$FinallyVariables = Invoke-CustomScriptBlockCommand -Mode ''Finally'' -ComputerName ''{0}'' -Variables (Get-Variable | Where-Object -FilterScript {{$_ -is [System.Management.Automation.PSVariable]}})' -f $ComputerName) $FinallyVariables = Invoke-CustomScriptBlockCommand -ComputerName $ComputerName -Mode 'Finally' -Variables (Get-Variable | Where-Object -FilterScript { $_ -is [System.Management.Automation.PSVariable] }) Write-Debug -Message ('$FinallyVariables: ''{0}''' -f [string]$FinallyVariables) Write-Debug -Message 'if ($FinallyVariables)' if ($FinallyVariables) { foreach ($FinallyVariable in $FinallyVariables) { Write-Debug -Message ('$FinallyVariable: ''{0}''' -f [string]$FinallyVariable) Write-Debug -Message ('Set-Variable -Name ''{0}'' -Value ''{1}'' -Scope ''Script''' -f $FinallyVariable.Name, [string]$ErrorHandlingVariable.Value) Set-Variable -Name $FinallyVariable.Name -Value $FinallyVariable.Value -Scope 'Script' } } Write-Debug -Message ('$HostLock: ''{0}''' -f $HostLock) Write-Debug -Message 'if ($HostLock)' if ($HostLock) { Write-Debug -Message 'Unlock-Resource -LockObject $HostLock' Unlock-Resource -LockObject $HostLock } Write-Debug -Message ('$DestinationHostLock: ''{0}''' -f $DestinationHostLock) Write-Debug -Message 'if ($DestinationHostLock)' if ($DestinationHostLock) { Write-Debug -Message 'Unlock-Resource -LockObject $DestinationHostLock' Unlock-Resource -LockObject $DestinationHostLock } Write-Debug -Message ('EXIT FINALLY {0}' -f $MyInvocation.MyCommand.Name) } Write-Debug -Message ('EXIT {0}' -f $MyInvocation.MyCommand.Name) } |