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

properties = {"ADDLOCAL": "OpenVPN.GUI,OpenVPN.Service,OpenVPN,Drivers,Drivers.Wintun,Drivers.TAPWindows6"}  # ,OpenVPN.GUI.OnLogon


def install():
    # declaring local variables
    bin_name = glob.glob("*OpenVPN*.msi")[0]

    # Uninstalling specific versions of the software
    for to_uninstall in installed_softwares(name="OpenVPN "):
        if Version(to_uninstall["version"]) > Version("2.6") and Version(to_uninstall["version"]) <= Version("2.6.010"):
            print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
            killalltasks(control.get_impacted_process_list())
            run(uninstall_cmd(to_uninstall["key"]))
            wait_uninstallkey_absent(to_uninstall["key"])

    # adding "OpenVPN Inc."" and "WireGuard LLC" trusted driver certificates
    try:
        run('"%s" -addstore TrustedPublisher "%s"' % (makepath(system32, "certutil.exe"), makepath(basedir, "openvpn.cer")), timeout=60)
        run('"%s" -addstore TrustedPublisher "%s"' % (makepath(system32, "certutil.exe"), makepath(basedir, "wireguard.cer")), timeout=60)
    except:
        print("WARNING: Required Publisher Driver certificates have not been applied, you may need to apply them manually or with GPO")

    # installing the package
    install_msi_if_needed(
        bin_name,
        properties=properties,
        min_version=get_msi_properties(bin_name)["ProductVersion"],
    )

    # Removing application's desktop shortcut if allowed
    if params.get("remove_desktop_shortcut"):
        remove_desktop_shortcut("OpenVPN GUI")


def uninstall():
    # removing "OpenVPN Inc." and "WireGuard LLC" trusted driver certificates
    try:
        run('"%s" -delstore TrustedPublisher "OpenVPN Inc."' % (makepath(system32, "certutil.exe")), timeout=60)
        run('"%s" -delstore TrustedPublisher "WireGuard LLC"' % (makepath(system32, "certutil.exe")), timeout=60)
    except:
        print("WARNING: Driver certificates have not been removed")

                                

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


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    url = "https://openvpn.net/community-downloads/"
    arch_dict = {"x64": "amd64", "x86": "x86"}

    # Getting latest version from official sources
    print("URL used is: %s" % url)
    for bs_search in bs_find_all(url, "a", "class", "ov-button w-100", proxies=proxies):
        if arch_dict[control.architecture] in bs_search.text:
            latest_bin = bs_search.text
            version = latest_bin.split("-")[1]
            download_url = bs_search["href"]
            break

    # Downloading latest binaries
    print("Latest %s version is: %s" % (control.name, version))
    print("Download URL is: %s" % download_url)
    if not isfile(latest_bin):
        print("Downloading: %s" % latest_bin)
        wget(download_url, latest_bin, proxies=proxies)
    else:
        print("Binary is present: %s" % latest_bin)

    # Checking version from file
    version_from_file = get_version_from_binary(latest_bin)
    # if not version_from_file.startswith(version) and version_from_file != "":
    if Version(version_from_file) != Version(version) and version_from_file != "":
        print("Changing version to the version number of the binary (from: %s to: %s)" % (version, version_from_file))
        # os.rename(latest_bin, latest_bin.replace(version, version_from_file))
        version = version_from_file
    else:
        print("Binary file version corresponds to online version")

    # Changing name of the package
    # OpenVPN 2.6.0-I004 amd64 (2.6.009)
    control.name = get_msi_properties(latest_bin)["ProductName"] + " (%s)" % get_msi_properties(latest_bin)["ProductVersion"]

    # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        print("Software version updated (from: %s to: %s)" % (control.get_software_version(), Version(version)))
        package_updated = True
    else:
        print("Software version up-to-date (%s)" % Version(version))
    control.set_software_version(version)
    control.save_control_to_wapt()

    # Deleting outdated binaries
    remove_outdated_binaries(latest_bin.split("-")[1], filename_contains=latest_bin.split("-")[3])

    # Validating or not update-package-sources
    return package_updated