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

bin_name_string = "Win10Pcap-v%s.msi"


def install():
    # Specific app values
    package_version = control.get_software_version()
    bin_name = bin_name_string % package_version

    # Adding "SoftEther Corporation" as trusted driver signer
    with disable_file_system_redirection():
        run('"%s" -addstore TrustedPublisher "%s"' % (makepath(system32, "certutil.exe"), makepath(basedir, "softether-drivers.cer")))

    # Installing the package
    install_msi_if_needed(
        bin_name,
        min_version=package_version,
    )

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
import platform
import requests
import bs4 as BeautifulSoup

bin_name_string = "Win10Pcap-v%s.msi"


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}

    # Specific app values
    app_name = control.name
    url = "http://www.win10pcap.org/download/"

    # Getting latest version from official website
    page = requests.get(url, proxies=proxy).text
    bs = BeautifulSoup.BeautifulSoup(page)
    bs_raw_string = bs.findAll("a")
    for download in bs_raw_string:
        if download["href"].endswith(".msi"):
            latest_bin = download["href"]
            version = latest_bin.split("-")[1].replace("v", "")
            url_dl = url + latest_bin

    print("Latest " + app_name + " version is: " + version)
    print("Download url is: " + url_dl)

    # Downloading latest binaries
    if not isfile(latest_bin):
        print("Downloading: " + latest_bin)
        wget(url_dl, latest_bin, proxies=proxy)

        # Get version from file
        version_from_file = get_msi_properties(latest_bin)["ProductVersion"]
        if version != version_from_file and version_from_file != "":
            version = version_from_file
            old_latest_bin = latest_bin
            latest_bin = bin_name_string % version
            if isfile(latest_bin):
                remove_file(latest_bin)
            os.rename(old_latest_bin, latest_bin)

        # Change version of the package
        pe = PackageEntry().load_control_from_wapt(".")
        pe.version = "%s-%s" % (version, int(pe.version.split("-", 1)[1]) + 1)
        pe.save_control_to_wapt(".")
        print("Changing version to " + pe.version + " in WAPT\\control")
        print("Update package done. You can now build-upload your package")
    else:
        print("This package is already up-to-date")

    # 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: " + bin_in_dir + " Deleted")
            remove_file(bin_in_dir)