From e55cc41e2b479326f65f66e020372db761bf1cee Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Wed, 28 Sep 2022 00:00:26 -0700
Subject: [PATCH] Update the AutoRepairs scheduled task every run

This is useful if the kit is being run from a different port and the
letter changed between runs.
---
 scripts/wk/repairs/win.py | 27 +++++++++++++++------------
 1 file changed, 15 insertions(+), 12 deletions(-)

diff --git a/scripts/wk/repairs/win.py b/scripts/wk/repairs/win.py
index a1e8fc3f..6cdc0bd5 100644
--- a/scripts/wk/repairs/win.py
+++ b/scripts/wk/repairs/win.py
@@ -162,6 +162,20 @@ def build_menus(base_menus, title):
   return menus
 
 
+def update_scheduled_task():
+  """Create (or update) scheduled task to start repairs."""
+  cmd = [
+    'schtasks', '/create', '/f',
+    '/sc', 'ONLOGON',
+    '/tn', f'{KIT_NAME_FULL}-AutoRepairs',
+    '/rl', 'HIGHEST',
+    '/tr', fr'C:\Windows\System32\cmd.exe "/C {sys.executable} {sys.argv[0]}"',
+    ]
+  if IN_CONEMU:
+    cmd[-1] = f'{CONEMU_EXE} -run {sys.executable} {sys.argv[0]}'
+  run_program(cmd)
+
+
 def end_session():
   """End Auto Repairs session."""
   # Remove logon task
@@ -254,6 +268,7 @@ def init(menus):
 
 def init_run(options):
   """Initialize Auto Repairs Run."""
+  update_scheduled_task()
   if options['Kill Explorer']['Selected']:
     atexit.register(start_explorer)
     TRY_PRINT.run('Killing Explorer...', kill_explorer, msg_good='DONE')
@@ -292,18 +307,6 @@ def init_session(options):
   if zone != WINDOWS_TIME_ZONE and ask(msg):
     set_timezone(WINDOWS_TIME_ZONE)
 
-  # Create logon task for Auto Repairs
-  cmd = [
-    'schtasks', '/create', '/f',
-    '/sc', 'ONLOGON',
-    '/tn', f'{KIT_NAME_FULL}-AutoRepairs',
-    '/rl', 'HIGHEST',
-    '/tr', fr'C:\Windows\System32\cmd.exe "/C {sys.executable} {sys.argv[0]}"',
-    ]
-  if IN_CONEMU:
-    cmd[-1] = f'{CONEMU_EXE} -run {sys.executable} {sys.argv[0]}'
-  run_program(cmd)
-
   # One-time tasks
   if options['Run TDSSKiller (once)']['Selected']:
     TRY_PRINT.run('Running TDSSKiller...', run_tdsskiller, msg_good='DONE')