# Wizard Kit: Install uBlock Origin for Firefox

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'])
D7_MODE = 'd7mode' in sys.argv

if __name__ == '__main__':
    try:
        stay_awake()
        clear_screen()
        print_info('{}: Firefox Fix for d7')
        other_results = {
            'Warning': {
                'NotInstalledError': 'Not installed',
                'NoProfilesError': 'No profiles found',
            }}

        # Scan for Firefox browsers
        print_info('Scanning for Firefox browsers')
        scan_for_browsers(just_firefox=True)

        # Install uBlock Origin
        print_info('Installing uBlock Origin')
        install_adblock(just_firefox=True)

        # Done
        print_standard('\nDone.')
        pause('Press Enter to exit...')
        exit_script()
    except SystemExit:
        pass
    except:
        major_exception()