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

uninstallkey = []

# Declaring specific app values (TO CHANGE)
bin_name_string = 'MobaXterm_installer_%s.msi'


def install():
    # Specific app values
    package_version = control.version.split('-',1)[0]

	# Getting the used storage on programfiles before installation (place it on the top)
    get_disk_free_space_before = get_disk_free_space(programfiles)

    # Installing the package
    install_msi_if_needed(bin_name_string % package_version
        ,min_version=package_version)

	# Return used storage of the program. (place it on the bottom)
    get_disk_free_space_after = get_disk_free_space(programfiles)
    free_space_after_diff = get_disk_free_space_before - get_disk_free_space_after
    print("Storage used: " + str(free_space_after_diff))


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

    from waptpackage import PackageEntry
    import requests

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


    # Update transition from BeautifulSoup 3 to 4
    try:
        import bs4 as BeautifulSoup
    except:
        import BeautifulSoup





    # Specific app values
    app_name = control.name
    url = control.sources

    # Getting latest version from official website
    page = requests.get(url,headers={'User-Agent':'Mozilla/5.0 (Windows NT 6.1; Win64; x64)'}).text
    bs = BeautifulSoup.BeautifulSoup(page)
    #bs = BeautifulSoup.BeautifulSoup(page,features="html.parser") # For bs4 only
    bs_raw_string = (bs.find('a',{'class':'btn_vert'}))
    url_dl = bs_raw_string['href']
    #https://download.mobatek.net/1242019111120613/MobaXterm_Installer_v12.4.zip

    latest_zip = url_dl.split('/')[-1]

    print('Download url is: ' + url_dl)

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

        # Deleting outdated binaries
        remove_file('MobaXterm_installer.dat')
        for bin_in_dir in glob.glob('*.exe') or glob.glob('*.msi'):
            print('Binary: ' + bin_in_dir + ' Deleted')
            remove_file(bin_in_dir)

        # Extracting
        print('Extracting ' + latest_zip)
        Sevenz_path = makepath(programfiles,'7-Zip','7z.exe')
        run('"%s" x %s' % (Sevenz_path,latest_zip))
        remove_file(latest_zip)

        # Getting version number
        for bin_in_dir in glob.glob('*.exe') or glob.glob('*.msi'):
            version = (bin_in_dir.split('_')[-1]).replace('.msi','')

        print('Latest ' + app_name + ' version is: ' + version)

		# Change version of the package
        print('Writing ' + version + '-0 in WAPT\\control')
        pe = PackageEntry()
        pe.load_control_from_wapt(os.getcwd())

        pe.version = version + '-0'
        pe.save_control_to_wapt(os.getcwd())
        print('Update package done. You can now build-upload your package')
    else:
        print('This package is already up-to-date')


if __name__ == '__main__':
    update_package()

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


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