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

uninstallkey = []

# Declaring specific app values (TO CHANGE)
bin_name = 'rufus.exe'
app_name = 'Rufus'
rufus_dir = makepath(programfiles32,app_name)
rufus_conf_file = makepath(rufus_dir,'rufus.ini')
app_processes = bin_name


def install():
    # Initializing variables
    package_version = control.version.split('-')[0]
    rufus_path = makepath(rufus_dir,bin_name)

    # Installing the package
    killalltasks(app_processes)
    if not isdir(rufus_dir):
        mkdirs(rufus_dir)
    else:
        for bin_in_dir in glob.glob(makepath(rufus_dir,'*.exe')):
            if bin_in_dir.endswith('.exe'):
                remove_file(bin_in_dir)
    filecopyto(bin_name,rufus_dir)

    #create_desktop_shortcut(app_name,rufus_path)
    create_programs_menu_shortcut(app_name,rufus_path)


def session_setup():
    print("Disabling: update-check at start-up")
    registry_set(HKEY_CURRENT_USER,r'SOFTWARE\Akeo Consulting\Rufus','UpdateCheckInterval',0xffffffff,type=REG_DWORD)


def uninstall():
    # Uninstalling the package
    killalltasks(app_processes)
    remove_tree(rufus_dir)
    #remove_desktop_shortcut(app_name)
    remove_programs_menu_shortcut(app_name)


def update_package():
    print('Download/Update package content from upstream binary sources')

    # Getting proxy informations from WAPT settings
    proxy = {}
    if platform.system()=='Windows' and isfile(makepath(user_local_appdata(),'waptconsole','waptconsole.ini')):
        proxywapt = inifile_readstring(makepath(user_local_appdata(),'waptconsole','waptconsole.ini'),'global','http_proxy')
        if proxywapt :
            proxy = {'http':proxywapt,'https':proxywapt}

    # Initializing variables
    git_repo = 'pbatard/rufus'
    url_api = 'https://api.github.com/repos/%s/releases/latest' % git_repo
    bin_end = bin_name.split('.')[-1]
    avoid_bin_end = 'p.exe'


    # Getting latest version from official website
    print('API used is: ' + url_api)
    json_load = json.loads(wgets(url_api,proxies=proxy))

    for download in json_load['assets']:
        if download['name'].endswith(bin_end) and download['name'] != avoid_bin_end:
            url_dl = download['browser_download_url']
            break

    version = json_load['tag_name'].replace('v','')
    latest_bin = bin_name

    print("Latest %s version is: %s" % (app_name,version))
    print("Download url is: %s" % url_dl)

    # Downloading latest binaries
    if isfile(bin_name):
        remove_file(bin_name)
    if not isfile(latest_bin):
        print('Downloading: %s' % latest_bin)
        wget(url_dl,latest_bin,proxies=proxy)

        # Checking version from file
        version = get_file_properties(latest_bin)['ProductVersion']
        print('Verified latest ' + app_name + ' version is: ' + version)

        # 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)

    # Deleting outdated binaries
    for bin_in_dir in glob.glob('*.exe') or glob.glob('*.msi') or glob.glob('*.zip'):
        if bin_in_dir != latest_bin :
            print('Outdated binary: %s Deleted' % bin_in_dir)
            remove_file(bin_in_dir)