# -*- coding: utf-8 -*-
from setuphelpers import *
import wmi
import platform

uninstallkey = []

# Defining variables
bin_name = 'iTunes64Setup.exe'
bin_msi_name = 'iTunes64.msi'
silent_args = '/qn /norestart'
bin_list = ['AppleApplicationSupport.msi','AppleApplicationSupport64.msi','AppleMobileDeviceSupport64.msi','AppleSoftwareUpdate.msi','Bonjour64.msi','iTunes64.msi','SetupAdmin.exe']
bin_dependencies_to_install = ['AppleApplicationSupport.msi','AppleApplicationSupport64.msi','AppleMobileDeviceSupport.msi','AppleMobileDeviceSupport64.msi']
instrusive_apps = ['Bonjour','Apple Software Update']
old_dependencies = ['Apple Application Support (32 bits)','Apple Application Support (64 bits)']


def install():
    # Initializing variables
    package_version = control.version.split('-',1)[0]
    is_already_installed = False

    # Uninstalling older iTunes versions
    for uninstall_itunes in installed_softwares(name='iTunes'):
        if 'Apple Inc.' in uninstall_itunes['publisher']:
            """ if uninstall_itunes['version'] < package_version:
                print('Removing %s' % uninstall_itunes['name'])
                run(uninstall_cmd(uninstall_itunes['key'])) """
            if uninstall_itunes['version'] == package_version:
                is_already_installed = True

    # Removing no longer required dependencies
    for old_dependency in old_dependencies:
        for uninstall in installed_softwares(old_dependency):
            print('Removing %s' % uninstall['name'])
            run(uninstall_cmd(uninstall['key']))

    # Installing the package
    for binary in bin_dependencies_to_install:
        if isfile(binary):
            print('Installing: %s' % binary)
            install_msi_if_needed(binary)

    if not is_already_installed:
        print('Installing: %s' % bin_msi_name)
        run('%s %s' % (bin_msi_name,silent_args))
    else:
        print("iTunes is already installed in the correct version.")

    print("Disabling: Update Check and Telemetry Apple")
    # Option used : (kParentalFlags_DisableFirstRunWelcomeWindow,kParentalFlags_DisableDiagnostics,kParentalFlags_DisableCheckForUpdates,kParentalFlags_Locked,kParentalFlags_DisableCheckForDeviceUpdates,kParentalFlags_DisableCheckForAppUpdates,kParentalFlags_DisableAutomaticDeviceSync)
    registry_set(HKEY_LOCAL_MACHINE,r'SOFTWARE\Apple Computer, Inc.\iTunes\Parental Controls\Default','AdminFlags',23202049,REG_DWORD)

    # Making sure that intrusive Apple's apps are not installed
    for intru_app in instrusive_apps:
        for uninstall_intru_app in installed_softwares(name=intru_app):
            if 'Apple Inc.' in uninstall_intru_app['publisher']:
                    print('Removing %s' % uninstall_intru_app['name'])
                    run(uninstall_cmd(uninstall_intru_app['key']))

    # Modifying default startup mode of Apple services
    """ if get_service_start_mode('Apple Mobile Device Service') != 'Manual':
        set_service_start_mode('Apple Mobile Device Service', 'Manual') """
    if get_service_start_mode('iPod Service') != 'Manual':
        set_service_start_mode('iPod Service', 'Manual')

    # Disable iTunesHelper at Startup
    registry_delete(HKEY_LOCAL_MACHINE,r'SOFTWARE\Microsoft\Windows\CurrentVersion\Run','iTunesHelper')


def uninstall():
    # Uninstalling iTunes
    for uninstall_itunes in installed_softwares(name='iTunes'):
    	if 'Apple Inc.' in uninstall_itunes['publisher']:
            print('Removing %s' % uninstall_itunes['name'])
            run(uninstall_cmd(uninstall_itunes['key']))


def update_package():
    print('Downloading/Updating package content from upstream binary sources')

    # Initializing variables
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()

    app_name = control.name
    url = control.sources
    arch = control.architecture
    if arch == 'x64':
        url_dl = url + 'win64'
    else:
        url_dl = url + 'win32'

    print("Download url is: %s" % url_dl)

    # Downloading latest binaries
    if not isfile(bin_name):
        print('Downloading ' + bin_name)
        wget(url_dl,bin_name,proxies=proxies)

        version = get_file_properties(bin_name)['ProductVersion']
        print("Latest %s version is: %s" % (app_name,version))

        # Extracting binaries from iTunesSetup
        for binary in bin_list:
            if isfile(binary):
                remove_file(binary)
        for bin_in_dir in glob.glob(bin_name):
            print('Extracting ' + bin_in_dir)
            Sevenz_path = makepath(programfiles,'7-Zip','7z.exe')
            run('"%s" x %s' % (Sevenz_path,bin_name))

        # Changing version of the package
        control.version = '%s-%s'%(version,int(control.version.split('-')[-1])+1)
        control.save_control_to_wapt()
        print('Changing version to: %s in WAPT\\control' % control.version)

    # Removing unwanted binaries
    for binary in glob.glob('*.exe') + glob.glob('*.msi'):
        if binary not in bin_dependencies_to_install or binary == bin_name:
            if binary != bin_msi_name:
                print('Removing %s' % binary)
                remove_file(binary)




def get_proxies():
    import platform
    if platform.python_version_tuple()[0] == '3':
        from urllib.request import getproxies
    else:
        from urllib import getproxies
    return getproxies()


def set_service_start_mode(service_name, start_mode):
    """Changing start mode of a service
    Usable values: 'Automatic', 'Manual', 'Disabled'
    Informations: https://msdn.microsoft.com/en-us/library/aa384896(v=vs.85).aspx
    """
    for svc in wmi.WMI().Win32_Service(Name=service_name):
        svc.ChangeStartMode(StartMode=start_mode)
        if start_mode == 'Disabled':
            service_stop(service_name)


def get_service_start_mode(service_name):
    """Getting actual start mode of a service
    Returned values: 'Auto', 'Manual', 'Disabled'
    """
    for svc in wmi.WMI().Win32_Service(Name=service_name):
        return svc.StartMode

def get_proxies_from_wapt_console():
    proxies = {}
    if platform.system() == 'Windows':
        waptconsole_ini_path = makepath(user_local_appdata(), 'waptconsole', 'waptconsole.ini')
    else:
        waptconsole_ini_path = makepath(user_home_directory(), '.config', 'waptconsole', 'waptconsole.ini')
    if isfile(waptconsole_ini_path):
        proxy_wapt = inifile_readstring(waptconsole_ini_path, 'global', 'http_proxy')
        if proxy_wapt:
            proxies = {'http': proxy_wapt, 'https': proxy_wapt}
    return proxies