'''Wizard Kit: Functions - Cleanup'''
# pylint: disable=no-name-in-module,wildcard-import
# vim: sts=2 sw=2 ts=2

from functions.setup import *
from settings.cleanup import *

def cleanup_adwcleaner():
  """Move AdwCleaner folders into the ClientDir."""
  source_path = r'{SYSTEMDRIVE}\AdwCleaner'.format(**global_vars['Env'])
  source_quarantine = r'{}\Quarantine'.format(source_path)

  # Quarantine
  if os.path.exists(source_quarantine):
    os.makedirs(global_vars['QuarantineDir'], exist_ok=True)
    dest_name = r'{QuarantineDir}\AdwCleaner_{Date-Time}'.format(
      **global_vars)
    dest_name = non_clobber_rename(dest_name)
    shutil.move(source_quarantine, dest_name)

  # Delete source folder if empty
  delete_empty_folders(source_path)

  # Main folder
  if os.path.exists(source_path):
    os.makedirs(global_vars['LogDir'], exist_ok=True)
    dest_name = r'{LogDir}\Tools\AdwCleaner'.format(
      **global_vars)
    dest_name = non_clobber_rename(dest_name)
    shutil.move(source_path, dest_name)


def cleanup_cbs(dest_folder):
  """Safely cleanup a known CBS archive bug under Windows 7.

  If a CbsPersist file is larger than 2 Gb then the auto archive feature
  continually fails and will fill up the system drive with temp files.

  This function moves the temp files and CbsPersist file to a temp folder,
  compresses the CbsPersist files with 7-Zip, and then opens the temp folder
  for the user to manually save the backup files and delete the temp files.
  """
  backup_folder = r'{dest_folder}\CbsFix'.format(dest_folder=dest_folder)
  temp_folder = r'{backup_folder}\Temp'.format(backup_folder=backup_folder)
  os.makedirs(backup_folder, exist_ok=True)
  os.makedirs(temp_folder, exist_ok=True)

  # Move files into temp folder
  cbs_path = r'{SYSTEMROOT}\Logs\CBS'.format(**global_vars['Env'])
  for entry in os.scandir(cbs_path):
    # CbsPersist files
    if entry.name.lower().startswith('cbspersist'):
      dest_name = r'{}\{}'.format(temp_folder, entry.name)
      dest_name = non_clobber_rename(dest_name)
      shutil.move(entry.path, dest_name)
  temp_path = r'{SYSTEMROOT}\Temp'.format(**global_vars['Env'])
  for entry in os.scandir(temp_path):
    # cab_ files
    if entry.name.lower().startswith('cab_'):
      dest_name = r'{}\{}'.format(temp_folder, entry.name)
      dest_name = non_clobber_rename(dest_name)
      shutil.move(entry.path, dest_name)

  # Compress CbsPersist files with 7-Zip
  cmd = [
    global_vars['Tools']['SevenZip'],
    'a', '-t7z', '-mx=3', '-bso0', '-bse0',
    r'{}\CbsPersists.7z'.format(backup_folder),
    r'{}\CbsPersist*'.format(temp_folder)]
  run_program(cmd)


def cleanup_desktop():
  """Move known backup files and reports into the ClientDir."""
  dest_folder = r'{LogDir}\Tools'.format(**global_vars)
  os.makedirs(dest_folder, exist_ok=True)

  desktop_path = r'{USERPROFILE}\Desktop'.format(**global_vars['Env'])
  for entry in os.scandir(desktop_path):
    if DESKTOP_ITEMS.search(entry.name):
      dest_name = r'{}\{}'.format(dest_folder, entry.name)
      dest_name = non_clobber_rename(dest_name)
      shutil.move(entry.path, dest_name)

  # Remove dir if empty
  delete_empty_folders(dest_folder)


def delete_empty_folders(folder_path):
  """Delete all empty folders in path (depth first)."""
  if not os.path.exists(folder_path) or not os.path.isdir(folder_path):
    # Bail early (silently)
    return

  # Delete empty subfolders first
  for item in os.scandir(folder_path):
    if item.is_dir():
      delete_empty_folders(item.path)

  # Remove top folder
  try:
    os.rmdir(folder_path)
  except OSError:
    pass


def delete_registry_key(hive, key, recurse=False):
  """Delete a registry key and all it's subkeys."""
  access = winreg.KEY_ALL_ACCESS

  try:
    if recurse:
      # Delete all subkeys first
      with winreg.OpenKeyEx(hive, key, 0, access) as k:
        key_info = winreg.QueryInfoKey(k)
        for x in range(key_info[0]):
          subkey = r'{}\{}'.format(key, winreg.EnumKey(k, 0))
          delete_registry_key(hive, subkey)

    # Delete key
    winreg.DeleteKey(hive, key)
  except FileNotFoundError:
    # Ignore
    pass


def delete_registry_value(hive, key, value):
  """Delete a registry value."""
  access = winreg.KEY_ALL_ACCESS
  with winreg.OpenKeyEx(hive, key, 0, access) as k:
    winreg.DeleteValue(k, value)


def restore_default_uac():
  """Restores default UAC settings via the registry."""
  if global_vars['OS']['Version'] == '10':
    write_registry_settings(UAC_DEFAULTS_WIN10, all_users=True)
  else:
    # Haven't checked Win8 settings, only applying minimum set
    write_registry_settings(UAC_DEFAULTS_WIN7, all_users=True)


if __name__ == '__main__':
  print("This file is not meant to be called directly.")