# Wizard Kit: User Checklist

import os
import sys

# Init
os.chdir(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(os.getcwd())
from functions.browsers import *
from functions.cleanup import *
from functions.setup import *
init_global_vars()
os.system('title {}: User Checklist Tool'.format(KIT_NAME_FULL))
global_vars['LogFile'] = r'{LogDir}\User Checklist ({USERNAME}).log'.format(
    **global_vars, **global_vars['Env'])

if __name__ == '__main__':
    try:
        stay_awake()
        os.system('cls')
        other_results = {
            'Warning': {
                'NotInstalledError': 'Not installed',
                'NoProfilesError': 'No profiles found',
            }}
        answer_config_browsers = ask('Install adblock?')
        if answer_config_browsers:
            answer_reset_browsers = ask(
                'Reset browsers to safe defaults first?')
        if global_vars['OS']['Version'] == '10':
            answer_config_classicshell = ask('Configure ClassicShell?')
            answer_config_explorer_user = ask('Configure Explorer?')
        
        # Cleanup
        print_info('Cleanup')
        try_and_print(message='Desktop...',
            function=cleanup_desktop, cs='Done')
        
        # Scan for supported browsers
        print_info('Scanning for browsers')
        scan_for_browsers()
        
        # Homepages
        print_info('Current homepages')
        list_homepages()
        
        # Backup
        print_info('Backing up browsers')
        backup_browsers()
        
        # Reset
        if answer_config_browsers and answer_reset_browsers:
            print_info('Resetting browsers')
            reset_browsers()
        
        # Configure
        print_info('Configuring programs')
        if answer_config_browsers:
            install_adblock()
        if global_vars['OS']['Version'] == '10':
            if answer_config_classicshell:
                try_and_print(message='ClassicStart...',
                    function=config_classicstart, cs='Done')
            if answer_config_explorer_user:
                try_and_print(message='Explorer...',
                    function=config_explorer_user, cs='Done')
            if (not answer_config_browsers
                and not answer_config_classicshell
                and not answer_config_explorer_user):
                print_warning('        Skipped')
        else:
            if not answer_config_browsers:
                print_warning('        Skipped')
        
        # Done
        print_standard('\nDone.')
        pause('Press Enter to exit...')
        exit_script()
    except SystemExit:
        pass
    except:
        major_exception()