From 50fb12d8edf4569f7ae0598153ab9ed39fca7463 Mon Sep 17 00:00:00 2001
From: Nicole Rappe <nicole.rappe@bunny-lab.io>
Date: Fri, 5 Sep 2025 15:14:20 -0600
Subject: [PATCH] Increased Auto-Agent Recovery Options

---
 Data/Agent/Scripts/register_agent_tasks.ps1 | 4 ++--
 Data/Agent/agent_deployment.py              | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/Data/Agent/Scripts/register_agent_tasks.ps1 b/Data/Agent/Scripts/register_agent_tasks.ps1
index b1a31d9..ce422fa 100644
--- a/Data/Agent/Scripts/register_agent_tasks.ps1
+++ b/Data/Agent/Scripts/register_agent_tasks.ps1
@@ -35,7 +35,7 @@ try {
     $wdArg = ('-NoProfile -ExecutionPolicy Bypass -File "{0}" -SupervisorTaskName "{1}"' -f $wdDest, $SupName)
     $wdAction = New-ScheduledTaskAction -Execute 'powershell.exe' -Argument $wdArg
     $wdTrigger = New-ScheduledTaskTrigger -Once -At ([datetime]::Now.AddMinutes(1)) -RepetitionInterval (New-TimeSpan -Minutes 5) -RepetitionDuration (New-TimeSpan -Days 365)
-    $wdSettings = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries -Hidden -ExecutionTimeLimit ([TimeSpan]::Zero)
+    $wdSettings = New-ScheduledTaskSettingsSet -AllowStartIfOnBatteries -DontStopIfGoingOnBatteries -Hidden -RestartCount 3 -RestartInterval (New-TimeSpan -Minutes 1) -ExecutionTimeLimit ([TimeSpan]::Zero)
     Register-ScheduledTask -TaskName $WdName -Action $wdAction -Trigger $wdTrigger -Settings $wdSettings -Principal $principal -Force | Out-Null
 
     # Ensure supervisor is running
@@ -52,7 +52,7 @@ try {
             $usrArg   = ('-W ignore::SyntaxWarning "{0}"' -f $UserScript)
             $usrAction = New-ScheduledTaskAction -Execute $UserExe -Argument $usrArg
             $usrTrig   = New-ScheduledTaskTrigger -AtLogOn
-            $usrSet    = New-ScheduledTaskSettingsSet -Hidden -ExecutionTimeLimit ([TimeSpan]::Zero)
+            $usrSet    = New-ScheduledTaskSettingsSet -Hidden -RestartCount 3 -RestartInterval (New-TimeSpan -Minutes 1) -ExecutionTimeLimit ([TimeSpan]::Zero)
             $usrPrin   = New-ScheduledTaskPrincipal -UserId $targetUser -LogonType Interactive -RunLevel Limited
             Register-ScheduledTask -TaskName $UserTaskName -Action $usrAction -Trigger $usrTrig -Settings $usrSet -Principal $usrPrin -Force | Out-Null
             Start-ScheduledTask -TaskName $UserTaskName | Out-Null
diff --git a/Data/Agent/agent_deployment.py b/Data/Agent/agent_deployment.py
index 824fc4e..1bd9b2c 100644
--- a/Data/Agent/agent_deployment.py
+++ b/Data/Agent/agent_deployment.py
@@ -126,7 +126,7 @@ $arg  = "-W ignore::SyntaxWarning {paths['agent_script']}"
 try {{ Unregister-ScheduledTask -TaskName $task -Confirm:$false -ErrorAction SilentlyContinue }} catch {{}}
 $action = New-ScheduledTaskAction -Execute $py -Argument $arg
 $trigger= New-ScheduledTaskTrigger -AtLogOn
-$settings = New-ScheduledTaskSettingsSet -Hidden -ExecutionTimeLimit ([TimeSpan]::Zero)
+$settings = New-ScheduledTaskSettingsSet -Hidden -RestartCount 3 -RestartInterval (New-TimeSpan -Minutes 1) -ExecutionTimeLimit ([TimeSpan]::Zero)
 Register-ScheduledTask -TaskName $task -Action $action -Trigger $trigger -Settings $settings -Force | Out-Null
 Start-ScheduledTask -TaskName $task | Out-Null
 """