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