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

app_uninstallkey = "Microsoft EdgeWebView"


def install():
    bin_name = glob.glob("MicrosoftEdgeWebView2RuntimeInstaller*.exe")[0]
    install_exe_if_needed(
        bin_name,
        silentflags="/silent /install",
        key=app_uninstallkey,
        min_version=control.get_software_version(),
    )
    uninstallkey.clear()


def uninstall():
    for to_uninstall in installed_softwares(uninstallkey=app_uninstallkey):
        print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
        killalltasks(control.impacted_process.split(","))
        app_uninstall_cmd = uninstall_cmd(to_uninstall["key"])
        app_uninstall_cmd[5] = "--force-uninstall"
        app_uninstall_cmd.remove("--verbose-logging")
        run(app_uninstall_cmd, accept_returncodes=[0, 300, 19])
        wait_uninstallkey_absent(to_uninstall["key"])


def audit():
    # Getting installed software version
    if installed_softwares(uninstallkey=app_uninstallkey):
        installed_version = installed_softwares(uninstallkey=app_uninstallkey)[0]["version"]
    else:
        installed_version = None

    # Auditing software
    print("Auditing: %s" % control.package)
    if installed_version is None:
        print("%s is not installed" % control.name)
        return "ERROR"
    # elif Version(installed_version) != Version(control.get_software_version()):
    #     print("%s is installed in version (%s) instead of (%s)" % (control.name, installed_version, control.get_software_version()))
    #     return "WARNING"
    else:
        print("%s is installed in correct version (%s)" % (control.name, installed_version))
        return "OK"

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
import sys
import time
import os
import glob

"""     
    "update_package_dependencies": [
        "tis-wapt-selenium",
        "tis-chrome"
    ]
"""


def update_package():
    # Importing Selenium with tis-wapt-selenium package
    sys.path.insert(0, makepath(WAPT.wapt_base_dir, "selenium"))
    from selenium import webdriver
    from selenium.webdriver.chrome.options import Options
    from selenium.webdriver.common.by import By

    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    arch_dict = {
        "x64": "MicrosoftEdgeWebView2RuntimeInstallerX64.exe",
        "x86": "MicrosoftEdgeWebView2RuntimeInstallerX86.exe",
        "arm64": "MicrosoftEdgeWebView2RuntimeInstallerARM64.exe",
    }
    latest_bin = arch_dict[control.architecture]

    # Deleting binaries
    for f in glob.glob("*.exe") + glob.glob("*.msi"):
        print("Removing: %s" % f)
        remove_file(f)

    version = control.get_software_version()
    current_path = os.getcwd()
    chrome_options = Options()
    chrome_options.add_argument("--headless")
    chrome_options.add_argument("--disable-notifications")
    chrome_options.add_experimental_option(
        "prefs",
        {
            "download.default_directory": f"{current_path}",
            "download.prompt_for_download": False,
            "download.directory_upgrade": True,
            "safebrowsing_for_trusted_sources_enabled": False,
            "safebrowsing.enabled": False,
        },
    )

    # download with selenium
    driver = webdriver.Chrome(options=chrome_options, executable_path=makepath(WAPT.wapt_base_dir, "selenium", "chromedriver.exe"))
    driver.get("https://developer.microsoft.com/fr-fr/microsoft-edge/webview2/#download-section")

    # get latest version version
    version = driver.find_element(By.XPATH, "/html/body/div[1]/div[2]/article/section[2]/section[2]/div/div[2]/div[3]/div/select[1]/option[1]").text
    driver.find_element(By.ID, control.architecture).click()
    driver.find_element(By.ID, "accept-evergreen").click()
    wait_download()
    driver.quit()

    # 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()

    # Validating or not update-package-sources
    return package_updated


def wait_download(delay=360):
    loop = 0
    while not glob.glob("*.exe"):
        loop += 3
        if loop > delay:
            break
        time.sleep(3)