# Wizard Kit: Activate Windows using various methods

import os
import sys

# Init
os.chdir(os.path.dirname(os.path.realpath(__file__)))
sys.path.append(os.getcwd())
from functions.activation import *
from functions.common import *
init_global_vars()
os.system('title {}: Windows Activation Tool'.format(KIT_NAME_FULL))

def activate_with_bios():
    """Attempt to activate Windows with a key stored in the BIOS."""
    try_and_print(message='BIOS Activation:',
        function=activate_windows_with_bios, other_results=other_results)

if __name__ == '__main__':
    try:
        stay_awake()
        # Bail early if already activated
        if windows_is_activated():
            print_info('This system is already activated')
            sleep(5)
            exit_script()

        # Determine activation method
        activation_methods = [
            {'Name': 'Activate with BIOS key', 'Function': activate_with_bios},
            ]
        if global_vars['OS']['Version'] not in ['8', '10']:
            activation_methods[0]['Disabled'] = True
        actions = [
            {'Name': 'Quit', 'Letter': 'Q'},
            ]

        # Main loop
        while True:
            selection = menu_select(
                '{}: Windows Activation Menu'.format(KIT_NAME_FULL),
                activation_methods, actions)

            if (selection.isnumeric()):
                activation_methods[int(selection)-1]['Function']()
                break
            elif selection == 'Q':
                exit_script()

        # Done
        print_success('Done.')
        pause("Press Enter to exit...")
        exit_script()
    except SystemExit:
        pass
    except:
        major_exception()