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

r""" 
2023-02-13 16:17:58,558 CRITICAL Critical error during uninstall: CalledProcessErrorOutput: Command '"C:\\Program Files (x86)\\Microsoft\\EdgeWebView\\Application\\110.0.1587.41\\Installer\\setup.exe" --uninstall 
--msedgewebview --system-level --verbose-logging --force-uninstall' returned non-zero exit status 19.

"""
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(),
        accept_returncodes=[0, 3010, 2147747602],
    )

    # Avoiding the usage by WAPT of the app built-in Uninstallstring
    uninstallkey.remove(app_uninstallkey)

    # Uninstalling WebView2 user installs
    for user_dir in glob.glob(makepath(systemdrive, "Users", "**")):
        # "C:\Users\jpadmin\AppData\Local\Microsoft\EdgeWebView\Application\110.0.1587.41\Installer\setup.exe" --uninstall --msedgewebview --verbose-logging
        local_appdata_app_dir = makepath(user_dir, "AppData", "Local", "Microsoft", "EdgeWebView", "Application")
        for user_app_dir in glob.glob(makepath(local_appdata_app_dir, "**")):
            if not user_app_dir.endswith("SetupMetrics"):
                user_app_setup = makepath(user_app_dir, "Installer", "setup.exe")
                if isfile(user_app_setup):
                    try:
                        app_uninstall_cmd = user_app_setup + " --uninstall --msedgewebview --verbose-logging --force-uninstall"
                        print(app_uninstall_cmd)
                        run(app_uninstall_cmd, accept_returncodes=[0, 3010, 19])
                    except:
                        print("Unable to remove WebView2 for user: %s" % user_dir.split(os.sep)[-1])
                    # remove_tree(local_appdata_app_dir)


def audit():
    # Declaring local variables
    app_check = installed_softwares(uninstallkey=app_uninstallkey)
    if app_check:
        installed_version = app_check[0]["version"]
        control.name = app_check[0]["name"]
    else:
        installed_version = None
        control.name = control.package.split("-", 1)[-1].replace("-", " ").title()

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


def uninstall():
    # Uninstalling the software with different uninstall_string
    for to_uninstall in installed_softwares(uninstallkey=app_uninstallkey):
        print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
        killalltasks(control.get_impacted_process_list())
        app_uninstall_cmd = to_uninstall["uninstall_string"] + " --force-uninstall"
        run(app_uninstall_cmd, accept_returncodes=[0, 3010, 19])
        wait_uninstallkey_absent(to_uninstall["key"])

                                

# -*- 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") # security limitations
    chrome_options.add_argument("--lang=en-us")
    chrome_options.add_argument("--disable-notifications")
    chrome_options.add_argument(
        "--safebrowsing-disable-download-protection"
    )  # safebrowsing.enabled need to be enabled to work # no longer work on headless chrome 110+
    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": True,
        },
    )

    # 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/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()
    # driver.get("chrome://downloads/")
    # driver.find_element(By.XPATH, '/html/body/downloads-manager//div[2]/iron-list/downloads-item//div[2]/div[2]/div[5]/span[2]/cr-button').click()
    wait_download()
    driver.quit()

    # Changing version of the package
    if Version(version, 4) > Version(control.get_software_version(), 4):
        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)