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

uninstallkey = []
bin_name = 'FastCopy_%s_installer.exe'
path = makepath(programfiles,'FastCopy')
silent_args = '/SILENT /DIR="%s"' %path
app_name = 'FastCopy'
label = "FastCopy"

def install():
    # Declare variables
    package_version = control.version.split('-')[0]
    # Installing binary
    print('installing FastCopy')
    install_exe_if_needed(bin_name % package_version,silentflags=silent_args,key='',min_version=package_version)


def uninstall():
    # Variables
    setup_path = makepath(programfiles,'FastCopy','setup.exe')

    print('closing FastCopy')
    killalltasks('FastCopy.exe')
    print('uninstalling FastCopy')
    remove_user_desktop_shortcut(label)

     # Uninstalling
    if isfile (setup_path):
        run(r'"%s" /silent /r' % setup_path)
        time.sleep(30)
        if isdir(path):
            remove_tree(path)
    else:
        print('file does not exist or the software is not installed')



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

    # Initializing variables
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    bin_name_temp = 'FastCopy_installer.exe'
    version = '1.0'
    package_version = control.version.split('-')[0]

    url = 'https://fastcopy.jp/'

    # Getting latest version from official website
    print('URL used is: %s' % url)

    # Finding version from file
##    for bs_search in bs_find_all(url,'a',proxies=proxies):
##        if bs_search['href'].startswith('/archive'):
##            version_from_file = bs_search.text.split('Fastcopy')[-1].split('_')[0]
##            break

    download_html = wgets(url,proxies=proxies).splitlines()
    for line in download_html:
        if 'FastCopy ver' in line:
            version_from_file = line.split(' ')[-1]


    # Finding download url
    for bs_read in bs_find_all(url,'a',proxies=proxies):
        if bs_read['href'].endswith('_installer.exe'):
            dl_url = url + bs_read['href']
            break

    # download binary
    if isfile(bin_name % package_version):
        print("removing old binary !")
        remove_file(bin_name % package_version)

    print("downloading binary")
    wget(dl_url,bin_name_temp,proxies=proxies)


    # Checking version from file
    if version != version_from_file:
        os.rename(bin_name_temp,bin_name % version_from_file)
        version = version_from_file

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


def bs_find_all(url, element, attribute=None, value=None, headers=None, proxies=None, features='html.parser', **kwargs):
    """"You may need to use a header for some websites. For example: headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0'}
    """
    import requests
    page = requests.get(url, proxies=proxies, headers=headers, **kwargs).text
    try:
        import bs4 as BeautifulSoup
        soup = BeautifulSoup.BeautifulSoup(page, features=features)
    except:
        import BeautifulSoup
        soup = BeautifulSoup.BeautifulSoup(page)
    if value:
        return soup.findAll(element,{attribute:value})
    else:
        return soup.findAll(element)


def bs_find(url, element, attribute=None, value=None, headers=None, proxies=None, features='html.parser', **kwargs):
    """"You may need to use a header for some websites. For example: headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0'}
    """
    import requests
    page = requests.get(url, proxies=proxies, headers=headers, **kwargs).text
    try:
        import bs4 as BeautifulSoup
        soup = BeautifulSoup.BeautifulSoup(page, features=features)
    except:
        import BeautifulSoup
        soup = BeautifulSoup.BeautifulSoup(page)
    if value:
        return soup.find(element,{attribute:value})
    else:
        return soup.find(element)


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