Public/Register-cChocoExBootstrapTask.ps1
|
<#
.SYNOPSIS Creates a PowerShell Task that runs as SYSTEM to automatically runs cChocoEx bootstrap continuously .DESCRIPTION If no boostrap.ps1 script is present at "$env:ProgramData\cChocoEx\bootstrap.ps1" a miniminal one will be copied to use. .EXAMPLE PS C:\> Register-cChocoExBootStrapTask -LoopDelay 180 Creates a Powershell task that will execute "$env:ProgramData\cChocoEx\bootstrap.ps1" every 180 minutes and at startup. .INPUTS None .OUTPUTS None #> function Register-cChocoExBootStrapTask { [CmdletBinding()] param ( [Parameter(Mandatory = $true)] [int] $LoopDelay ) begin { #Gather Variables Set-GlobalVariables #Setup Folders Set-cChocoExFolders $TaskName = 'cChocoExBootstrapTask' $TaskPath = '\cChocoEx\' $UserID = "NT AUTHORITY\SYSTEM" $FilePath = "$env:ProgramData\cChocoEx\bootstrap.ps1" $Description = "This task is part of the loop functionality in cChocoEx. It runs $FilePath at startup and upon a defined interval" $ScheduledTaskSettingsSet = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries -MultipleInstances 'IgnoreNew' $ScheduledTaskPrincipal = New-ScheduledTaskPrincipal -UserID $UserID -LogonType ServiceAccount -RunLevel Highest $TaskTrigger01 = New-ScheduledTaskTrigger -AtStartup $TaskTrigger02 = New-ScheduledTaskTrigger -Once -At ((Get-Date).AddMinutes($LoopDelay)) -RepetitionInterval (New-TimeSpan -Minutes $LoopDelay) $ScheduledTaskAction = New-ScheduledTaskAction -Execute "PowerShell.exe" -Argument "-NoLogo -NonInteractive -WindowStyle Hidden -File `"$FilePath`"" #ScheduledTaskSplat $ScheduledTaskParams = @{ TaskName = $TaskName TaskPath = $TaskPath Settings = $ScheduledTaskSettingsSet Trigger = @($TaskTrigger01, $TaskTrigger02) Description = $Description Principal = $ScheduledTaskPrincipal Action = $ScheduledTaskAction Force = $true } } process { #Check for existing bootstrap.ps1 Write-Log -Severity 'Information' -Message "Checking for File - $FilePath" if (-Not(Test-Path -Path $FilePath)) { #Copy Minimal Bootstrap Write-Log -Severity 'Information' -Message "$FilePath Not Found - Copying minimal boostrap.ps1" try { $null = Copy-Item -Path (Join-Path -Path ($PSScriptRoot | Split-Path) -ChildPath 'scripts\bootstrap.min.ps1') -Destination $FilePath -Force Write-Log -Severity 'Information' -Message "Success" } catch { Write-Log -Severity 'Error' -Message "Failure" Write-Log -Severity 'Error' -Message "$($_.Exception.Message)" } } else { Write-Log -Severity 'Information' -Message "$FilePath Already Exists" } #Register Task Write-Log -Severity 'Information' -Message "Registering Scheduled Task $TaskName" try { $null = Register-ScheduledTask @ScheduledTaskParams Write-Log -Severity 'Information' -Message "Registering Scheduled Task $TaskName - Success" } catch { Write-Log -Severity 'Error' -Message "Registering Scheduled Task $TaskName - Failure" Write-Log -Severity 'Error' -Message "$($_.Exception.Message)" } } end { #Validate Task Exists try { $null = Get-ScheduledTask -TaskName $TaskName } catch { Write-Log -Severity 'Error' -Message "Required Scheduled Task $TaskName Not Found" Write-Log -Severity 'Error' -Message "$($_.Exception.Message)" } } } |