functions/ScheduledTasks.ps1

function Stop-BeamScheduledTask {
    param($taskPath, $taskName,$timeoutSec = 10, [switch][bool] $disable) 

    $timeout = $timeoutSec
    ipmo require
    req scheduledtasks

    $tn = $taskName
    $folder = $taskPath

    $folder = "\" + $folder.Trim("\","/") + "\" 

    write-host "looking for task '$($tn)' in folder '$folder'"
    try {
        $t = scheduledtasks\get-Scheduledtask -taskname $tn -taskpath $folder  -ErrorAction SilentlyContinue 
    } catch {
        # for some reason, get-scheduledtask throws if $folder is not found, even if erroraction is SilentlyContinue
    }
    
    if ($t -ne $null) {
        if ($disable) {
            write-host "$(get-date) disabling task '$folder$($t.Taskname)'"
            $t | scheduledtasks\disable-scheduledtask    
        }
        write-host "$(get-date) stopping task '$folder$($t.Taskname)'"
        $t | scheduledtasks\stop-scheduledtask
        $c = 0
        do {
            $c++
            Start-Sleep -Seconds 1
            $t = scheduledtasks\get-Scheduledtask -taskname $tn -taskpath $folder          
            write-host "$tn : $($t.State)"
        } while ($t.State -ne "Ready" -and $t.State -ne "Disabled" -and $c -lt $timeout) 



        $t | format-table | out-string | write-host
        if ($t.State -eq "Disabled") {
            write-warning "task '$fulltn' is Disabled!"
        }
        elseif ($t.State -ne "Ready") {
            throw "task '$fulltn' failed to stop. status = '$($t.State)' after $timeout s"
        }

        try {
        $processPath = ($t.Definition.Actions | select -first 1).Path
        if ($processpath -eq $null) {
            Write-Warning "failed to retrieve executable path from dask definition"
        }
        else {
            $processName = [System.IO.Path]::GetFileNameWithoutExtension($processPAth)
            Write-Verbose "looking for process path $processpath name $processname" -Verbose
            $hangingprocs = $true
            for($retries = 10; $hangingprocs -and $retries -gt 0; $retries--) {
                $procs = @(get-process $processName | ? { $_.Path -eq $processPath })
                if ($procs.Count -gt 0) {
                    write-warning "there are still processes running with task executable '$processPath'"
                foreach ($p in $procs) {
                        write-warning "killing process $($p.Id) (as user '$env:USERDOMAIN\$env:USERNAME')"
                        try {
                            $p.Kill()
                        } catch {
                            write-warning "failed to kill process $($p.Id): $($_.Message)"
                            if ($retries - 1 -eq 0) {
                                throw
                            }
                            #throw
                        }
                    }
                    Start-Sleep 1
            }
            else {
                    $hangingprocs = $false
                }
            } 
        }
        } catch {
            write-warning "failed to kill hanging '$tn' process: $($_.message)"
            write-warning "$($_.scriptstacktrace)"
            throw
        }

    }
}

function Start-BeamScheduledTask {
    param($taskPath, $taskName, $timeoutSec = 10) 

    $timeout = $timeoutSec
    ipmo require
    req scheduledtasks

    $folder = $taskPath
    $folder = "\" + $folder.Trim("\","/") + "\" 
    $tn = $taskName
    $fulltn = "$taskpath$taskname"

    

    write-host "$(get-date) starting task '$fulltn'" 
    $t = scheduledtasks\get-Scheduledtask -taskname $tn -taskpath $folder           

    if ($t -eq $null) { throw "task $tn not found in folder $folder" }
    # create-taskhere returns wrong task type (from taskscheduler module instead of scheduledtasks ?)
    # the task may have been disabled
    if (!$t.settings.enabled) {
        $t | scheduledtasks\Enable-ScheduledTask -ErrorAction stop
    }
    $t | scheduledtasks\Start-ScheduledTask -erroraction stop

    $c = 0
    do {
        $c++
        Start-Sleep -Seconds 1
        $t = scheduledtasks\get-Scheduledtask -taskname $tn -taskpath $folder           
        write-host "$tn : $($t.State)"
    } while ($t.State -ne "Running" -and $t.State -ne "Disabled" -and $c -lt $timeout) 

    $t | format-table | out-string | write-host
    if ($t.State -eq "Disabled") {
        write-warning "task '$fulltn' is Disabled! Assuming this is intentional, will not treat this as error."
    }
    elseif ($t.State -ne "Running") {
        throw "task '$fulltn' failed to start. status = '$($t.State)' after $timeout s"
    }
}

function Restart-BeamScheduledTask {
    param($taskPath, $taskName) 

    Stop-BeamScheduledTask -taskpath $taskpath -taskname $taskname
    Start-BeamScheduledTask -taskpath $taskpath -taskname $taskname
}

new-alias Restart-ScheduledTask Restart-BeamScheduledTask -force