Add 'Backup Settings' group functions

scripts/auto_repairs.py

@@ -52,39 +52,40 @@ def placeholder_reboot(group, name):
 BASE_MENUS = {
   'Groups': {
     'Backup Settings': (
-      MenuEntry('Enable RegBack', placeholder_function),
+      MenuEntry('Enable RegBack',           'auto_enable_regback'),
-      MenuEntry('Enable System Restore', placeholder_function),
+      MenuEntry('Enable System Restore',    'auto_system_restore_enable'),
-      MenuEntry('Create System Restore', placeholder_function),
+      MenuEntry('Set System Restore Size',  'auto_system_restore_set_size'),
-      MenuEntry('Backup Browsers', placeholder_function),
+      MenuEntry('Create System Restore',    'auto_system_restore_create'),
-      MenuEntry('Backup Power Plans', placeholder_function),
+      MenuEntry('Backup Browsers',          placeholder_function),
-      MenuEntry('Backup Registry', placeholder_function),
+      MenuEntry('Backup Power Plans',       'auto_backup_power_plans'),
+      MenuEntry('Backup Registry',          'auto_backup_registry'),
       ),
     'Windows Repairs': (
-      MenuEntry('Disable Windows Updates', placeholder_function),
+      MenuEntry('Disable Windows Updates',  placeholder_function),
-      MenuEntry('Reset Windows Updates', placeholder_function),
+      MenuEntry('Reset Windows Updates',    placeholder_function),
-      MenuEntry('Reboot', placeholder_reboot),
+      MenuEntry('Reboot',                   placeholder_reboot),
-      MenuEntry('CHKDSK', placeholder_function),
+      MenuEntry('CHKDSK',                   placeholder_function),
-      MenuEntry('DISM RestoreHealth', 'auto_dism'),
+      MenuEntry('DISM RestoreHealth',       'auto_dism'),
-      MenuEntry('SFC Scan', placeholder_function),
+      MenuEntry('SFC Scan',                 placeholder_function),
-      MenuEntry('Fix File Associations', placeholder_function),
+      MenuEntry('Fix File Associations',    placeholder_function),
-      MenuEntry('Clear Proxy Settings', placeholder_function),
+      MenuEntry('Clear Proxy Settings',     placeholder_function),
-      MenuEntry('Disable Pending Renames', placeholder_function),
+      MenuEntry('Disable Pending Renames',  placeholder_function),
-      MenuEntry('Registry Repairs', placeholder_function),
+      MenuEntry('Registry Repairs',         placeholder_function),
-      MenuEntry('Repair Safe Mode', placeholder_function),
+      MenuEntry('Repair Safe Mode',         placeholder_function),
-      MenuEntry('Reset UAC', placeholder_function),
+      MenuEntry('Reset UAC',                placeholder_function),
-      MenuEntry('Reset Windows Policies', placeholder_function),
+      MenuEntry('Reset Windows Policies',   placeholder_function),
       ),
     'Malware Cleanup': (
-      MenuEntry('BleachBit', placeholder_function),
+      MenuEntry('BleachBit',                placeholder_function),
-      MenuEntry('HitmanPro', placeholder_function),
+      MenuEntry('HitmanPro',                placeholder_function),
-      MenuEntry('KVRT', placeholder_function),
+      MenuEntry('KVRT',                     placeholder_function),
-      MenuEntry('Windows Defender', placeholder_function),
+      MenuEntry('Windows Defender',         placeholder_function),
-      MenuEntry('Reboot', placeholder_reboot),
+      MenuEntry('Reboot',                   placeholder_reboot),
       ),
     'Manual Steps': (
-      MenuEntry('AdwCleaner', placeholder_function),
+      MenuEntry('AdwCleaner',               placeholder_function),
-      MenuEntry('IO Bit Uninstaller', placeholder_function),
+      MenuEntry('IO Bit Uninstaller',       placeholder_function),
-      MenuEntry('Enable Windows Updates', placeholder_function),
+      MenuEntry('Enable Windows Updates',   placeholder_function),
       ),
     },
   'Options': (

scripts/wk/repairs/win.py

@@ -5,9 +5,11 @@ import atexit
 import logging
 import os
 import platform
+import re
 import sys
+import time
 
-from subprocess import CalledProcessError
+from subprocess import CalledProcessError, DEVNULL
 
 from wk.cfg.main  import KIT_NAME_FULL
 from wk.exe       import get_procs, run_program, popen_program, wait_for_procs
@@ -47,11 +49,12 @@ AUTO_REPAIR_KEY = fr'Software\{KIT_NAME_FULL}\Auto Repairs'
 CONEMU = 'ConEmuPID' in os.environ
 OS_VERSION = float(platform.win32_ver()[0])
 WIDTH = 50
+SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE')
 TRY_PRINT = TryAndPrint()
 TRY_PRINT.width = WIDTH
 TRY_PRINT.verbose = True
-#for error in ('subprocess.CalledProcessError', 'FileNotFoundError'):
+for error in ('CalledProcessError', 'FileNotFoundError'):
-#  TRY_PRINT.add_error(error)
+  TRY_PRINT.add_error(error)
 
 
 # Auto Repairs
@@ -442,8 +445,20 @@ def update_main_menu(menus):
 
 
 # Auto Repairs: Wrapper Functions
+def auto_backup_power_plans(group, name):
+  """Backup power plans."""
+  result = TRY_PRINT.run('Backup Power Plans...', export_power_plans)
+  save_settings(group, name, result=result)
+
+
+def auto_backup_registry(group, name):
+  """Backup registry."""
+  result = TRY_PRINT.run('Backup Registry...', backup_registry)
+  save_settings(group, name, result=result)
+
+
 def auto_dism(group, name):
-  """Auto DISM repairs, returns bool."""
+  """Run DISM repairs."""
   needs_reboot = False
   result = TRY_PRINT.run('DISM (RestoreHealth)...', run_dism)
 
@@ -467,7 +482,73 @@ def auto_dism(group, name):
     reboot()
 
 
+def auto_enable_regback(group, name):
+  """Enable RegBack."""
+  result = TRY_PRINT.run(
+    'Enable RegBack...', reg_set_value, 'HKLM',
+    r'System\CurrentControlSet\Control\Session Manager\Configuration Manager',
+    'EnablePeriodicBackup', 1, 'DWORD',
+    )
+  save_settings(group, name, result=result)
+
+
+def auto_system_restore_create(group, name):
+  """Create a System Restore point."""
+  result = TRY_PRINT.run(
+    'Create System Restore...', create_system_restore_point,
+    )
+  save_settings(group, name, result=result)
+
+
+def auto_system_restore_enable(group, name):
+  """Enable System Restore."""
+  cmd = [
+    'powershell', '-Command', 'Enable-ComputerRestore',
+    '-Drive', SYSTEMDRIVE,
+    ]
+  result = TRY_PRINT.run('Enable System Restore...', run_program, cmd=cmd)
+  save_settings(group, name, result=result)
+
+
+def auto_system_restore_set_size(group, name):
+  """Set System Restore size."""
+  result = TRY_PRINT.run('Set System Restore Size...', set_system_restore_size)
+  save_settings(group, name, result=result)
+
+
+# Misc Functions
+def set_backup_path(name, date=False):
+  """Set backup path, returns pathlib.Path."""
+  backup_path = format_log_path(log_name=f'../Backups/{name}').with_suffix('')
+  if date:
+    backup_path = backup_path.joinpath(time.strftime('%Y-%m-%d'))
+  return backup_path.resolve()
+
+
+# Tool Functions
+def backup_registry():
+  """Backup Registry."""
+  backup_path = set_backup_path('Registry', date=True)
+  backup_path.parent.mkdir(parents=True, exist_ok=True)
+  run_tool('Erunt', 'ERUNT', backup_path, 'sysreg', 'curuser', 'otherusers')
+
+
 # OS Built-in Functions
+def create_system_restore_point():
+  """Create System Restore point."""
+  cmd = [
+    'powershell', '-Command', 'Checkpoint-Computer',
+    '-Description', f'{KIT_NAME_FULL}-AutoRepairs',
+    ]
+  too_recent = (
+    'WARNING: A new system restore point cannot be created'
+    'because one has already been created within the past'
+    )
+  proc = run_program(cmd)
+  if too_recent in proc.stdout:
+    raise GenericWarning('Skipped, a restore point was created too recently')
+
+
 def disable_windows_updates():
   """Disable and stop Windows Updates."""
   stop_service('wuauserv')
@@ -479,6 +560,31 @@ def enable_windows_updates():
   enable_service('wuauserv', 'demand')
 
 
+def export_power_plans():
+  """Export existing power plans."""
+  backup_path = set_backup_path('Power Plans', date=True)
+  backup_path.mkdir(parents=True, exist_ok=True)
+  cmd = ['powercfg', '/L']
+  proc = run_program(cmd)
+  plans = {}
+
+  # Get plans
+  for line in proc.stdout.splitlines():
+    line = line.strip()
+    match = re.match(r'^Power Scheme GUID: (.{36})\s+\((.*)\)\s*(\*?)', line)
+    if match:
+      name = match.group(2)
+      if match.group(3):
+        name += ' (Default)'
+      plans[name] = match.group(1)
+
+  # Backup plans to disk
+  for name, guid in plans.items():
+    out_path = backup_path.joinpath(f'{name}.pow')
+    cmd = ['powercfg', '-export', out_path, guid]
+    run_program(cmd)
+
+
 def kill_explorer():
   """Kill all Explorer processes."""
   cmd = ['taskkill', '/im', 'explorer.exe', '/f']
@@ -608,6 +714,15 @@ def run_sfc_scan():
     raise OSError
 
 
+def set_system_restore_size(size=8):
+  """Set System Restore size."""
+  cmd = [
+    'vssadmin', 'Resize', 'ShadowStorage',
+    f'/On={SYSTEMDRIVE}', f'/For={SYSTEMDRIVE}', f'/MaxSize={size}%',
+    ]
+  run_program(cmd, pipe=False, stderr=DEVNULL, stdout=DEVNULL)
+
+
 def start_explorer():
   """Start Explorer."""
   popen_program(['explorer.exe'])