# -*- 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(),
        accept_returncodes=[0, 3010, 2147747602],
        timeout=900,
    )

    # Adding QuietUninstallString in registry for software but no longer uninstallable
    quiet_uninstall_string = installed_softwares(uninstallkey=app_uninstallkey)[0]["uninstall_string"] + " --force-uninstall"
    register_uninstall(app_uninstallkey, quiet_uninstall_string=quiet_uninstall_string)

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

    # Uninstalling WebView2 user installs
    uninstall_users_msedgewebview()


def uninstall_users_msedgewebview():
    for user_dir in glob.glob(makepath(systemdrive, "Users", "**")):
        # "C:\Users\username\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)

                                

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

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


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    update_dict = {
        "latest_bin": {
            "windows-x64": "MicrosoftEdgeWebView2RuntimeInstallerX64.exe",
            "windows-x86": "MicrosoftEdgeWebView2RuntimeInstallerX86.exe",
            "windows-arm64": "MicrosoftEdgeWebView2RuntimeInstallerARM64.exe",
        },
        "download_url": {
            "windows-x64": "https://go.microsoft.com/fwlink/?linkid=2124701",
            "windows-x86": "https://go.microsoft.com/fwlink/?linkid=2099617",
            "windows-arm64": "https://go.microsoft.com/fwlink/?linkid=2099616",
        },
    }
    url = "https://developer.microsoft.com/en-us/microsoft-edge/webview2?form=MA13LH#download-section"
    latest_bin = update_dict["latest_bin"][f'{ensure_list(control.target_os)[0].split("(")[0]}-{ensure_list(control.architecture)[0]}']
    download_url = update_dict["download_url"][f'{ensure_list(control.target_os)[0].split("(")[0]}-{ensure_list(control.architecture)[0]}']

    # Deleting binaries
    remove_outdated_binaries("*")

    # Importing Selenium using package tis-wapt-selenium
    wapt_selenium_dir = os.sep.join([sys.executable.rsplit(os.path.sep, 1)[0].replace("/bin", ""), "selenium"])
    driver_path = makepath(wapt_selenium_dir, "chromedriver.exe")
    sys.path.insert(0, wapt_selenium_dir)
    from selenium import webdriver
    from selenium.webdriver.chrome.options import Options
    from selenium.webdriver.chrome.service import Service
    from selenium.webdriver.common.by import By

    # Selenium WebDriver initialization
    driver_options = Options()
    # driver_options.add_argument("--headless") # security limitations with headless flag
    driver_options.add_argument("--lang=en-us")
    driver_options.add_argument("--disable-notifications")
    driver_options.add_argument("--safebrowsing-disable-download-protection")
    driver_options.add_experimental_option(
        "prefs",
        {
            "download.default_directory": f"{basedir}",
            "download.prompt_for_download": False,
            "download.directory_upgrade": True,
            "safebrowsing_for_trusted_sources_enabled": False,
            "safebrowsing.enabled": True,
        },
    )
    driver = webdriver.Chrome(options=driver_options, service=Service(driver_path))

    # Downloading using Selenium
    driver.get(url)
    version = driver.find_element(By.ID, "version").text
    # if control.architecture == "x86":
    #     driver.find_element(
    #         By.XPATH, "/html/body/div[1]/div/div/main/div[2]/div[1]/section[3]/div[2]/div[2]/div/div/div[2]/div/div[2]/div[2]/button[1]"
    #     ).click()
    # if control.architecture == "arm64":
    #     driver.find_element(
    #         By.XPATH, "/html/body/div[1]/div/div/main/div[2]/div[1]/section[3]/div[2]/div[2]/div/div/div[2]/div/div[2]/div[2]/button[3]"
    #     ).click()
    # else:
    #     driver.find_element(
    #         By.XPATH, "/html/body/div[1]/div/div/main/div[2]/div[1]/section[3]/div[2]/div[2]/div/div/div[2]/div/div[2]/div[2]/button[3]"
    #     ).click()
    # time.sleep(10)
    # driver.find_element(By.XPATH, "/html/body/div[5]/div/div[2]/div/div/div[4]").click()
    # wait_download()
    driver.quit()

    # # Checking version from file
    # if get_os_name() == "Windows" and "windows" in control.target_os.lower():
    #     version_from_file = get_version_from_binary(latest_bin)
    #     if Version(version_from_file, 4) == Version(version, 4):
    #         print(f"INFO: Binary file version ({version_from_file}) corresponds to online version ({version})")
    #     else:
    #         error(f"ERROR: Binary file version ({version_from_file}) do NOT corresponds to online version ({version})")

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

    # 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

    # # Changing version of the package and validating update-package-sources
    # return complete_control_version(control, version)


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