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

r"""
Procedures: 
https://docs.microsoft.com/deployedge/microsoft-edge-update-policies
> This policy is available only on Windows instances that are joined to a Microsoft® Active Directory® domain.
https://admx.help/?Category=EdgeChromium&Policy=Microsoft.Policies.Update::Pol_DefaultUpdatePolicy

UninstallKey                           Software                                                              Version             Uninstallstring
-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
Microsoft Edge Update                  Microsoft Edge Update                                                 1.3.161.35
Microsoft EdgeWebView                  Microsoft Edge WebView2 Runtime                                       101.0.1210.53       "C:\Program Files (x86)\Microsoft\EdgeWebView\Application\101.0.1210.53\Installer\setup.exe" --uninstall --msedgewebview --system-level --verbose-logging
{235B0AE8-4971-3AF7-80AE-CEE8C161E846} Microsoft Edge                                                        101.0.1210.53       MsiExec.exe /X{235B0AE8-4971-3AF7-80AE-CEE8C161E846}

"""
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
app_schtasks_list = ["MicrosoftEdgeUpdateTaskMachineCore", "MicrosoftEdgeUpdateTaskMachineUA", "MicrosoftEdgeUpdateBrowserReplacementTask"]
app_services_list = ["MicrosoftEdgeElevationService", "edgeupdatem", "edgeupdate"]
try:
    app_dir_list = glob.glob("%s" % makepath(programfiles32, "Microsoft", "Edge"))
except:
    app_dir_list = []
# app_extra_key_list = ["Microsoft EdgeWebView", "Microsoft Edge Update"]


def install():
    # Declaring local variables
    package_version = control.get_software_version()
    bin_name = glob.glob("*MicrosoftEdgeEnterprise*.msi")[0]
    app_uninstallkey = get_msi_properties(bin_name)["ProductCode"]

    # Uninstalling Microsoft Edge when installed with an exe
    for to_uninstall in installed_softwares(uninstallkey="Microsoft Edge"):
        if not uninstall_key_exists(to_uninstall["key"]):
            continue
        print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
        uninstall_app_cmd = to_uninstall["uninstall_string"] + " --force-uninstall"
        run_notfatal(uninstall_app_cmd)
        wait_uninstallkey_absent(to_uninstall["key"])

    # # Uninstalling Microsoft EdgeWebView
    # for to_uninstall in installed_softwares(uninstallkey="Microsoft EdgeWebView"):
    #     if not uninstall_key_exists(to_uninstall["key"]):
    #         continue
    #     print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
    #     uninstall_app_cmd = to_uninstall["uninstall_string"] + " --force-uninstall"
    #     run_notfatal(uninstall_app_cmd)
    #     wait_uninstallkey_absent(to_uninstall["key"])

    registry_deletekey(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft", "EdgeUpdate")
    # Disabling auto-update and adding minimal preferences
    registry_setstring(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "AutoUpdateCheckPeriodMinutes", 0, type=REG_DWORD)
    # registry_setstring(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "CreateDesktopShortcutDefault", 0, type=REG_DWORD)
    registry_setstring(
        HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "UpdaterExperimentationAndConfigurationServiceControl", 0, type=REG_DWORD
    )

    # Installing the package
    print("Installing: %s" % bin_name)
    install_msi_if_needed(
        bin_name,
        min_version=package_version,
        timeout=900,
    )

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

    # # Deny updates and installs
    # registry_setstring(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "UpdateDefault", 0, type=REG_DWORD)
    # registry_setstring(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "InstallDefault", 0, type=REG_DWORD)
    # registry_setstring(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "Install%s" % app_uninstallkey, 0, type=REG_DWORD)
    # registry_setstring(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft\EdgeUpdate", "Update%s" % app_uninstallkey, 0, type=REG_DWORD)

    # Removing application desktop shortcut if allowed
    if params.get("remove_desktop_shortcuts"):
        remove_desktop_shortcut(control.name)

    # Disabling application scheduled tasks
    for task in app_schtasks_list:
        if task_exists(task):
            try:
                disable_task(task)
            except:
                print("Unable to disable the task: %s" % task)

    # Changing default start mode of the application services
    for service_name in app_services_list:
        set_service_start_mode(service_name, "Manual")


def audit():
    # Declaring local variables
    package_version = control.version.split("-", 1)[0]
    app_name = control.name
    installed_version = None

    # Getting installed software version
    for installed_app in installed_softwares(name=r"Microsoft Edge$"):
        installed_version = installed_app["version"]
        break

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


def removing_extra_edge_apps():
    for to_uninstall in installed_softwares(name="Microsoft Edge "):
        if not to_uninstall["uninstall_string"]:
            unregister_uninstall(uninstallkey=to_uninstall["key"])
            continue
        print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
        uninstall_app_cmd = to_uninstall["uninstall_string"] + " --force-uninstall"
        run_notfatal(uninstall_app_cmd)
        wait_uninstallkey_absent(to_uninstall["key"])


def uninstall():
    # Avoiding unexpected behavoir by rolling-back install changes
    registry_deletekey(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Microsoft", "EdgeUpdate")

    # # Uninstalling all other Microsoft Edge apps
    # removing_extra_edge_apps()

    # Force uninstalling Microsoft Edge
    for to_uninstall in installed_softwares(name=r"Microsoft Edge\b"):
        print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
        killalltasks(control.impacted_process.split(","))
        try:
            run(uninstall_cmd(to_uninstall["key"]))
        except:
            wait_uninstallkey_absent(to_uninstall["key"])
            unregister_uninstall(uninstallkey=to_uninstall["key"])

    # Removing Microsoft Edge remaining files
    for app_folder in app_dir_list:
        if isdir(app_folder):
            killalltasks(control.get_impacted_process_list())
            try:
                remove_tree(app_folder)
            except:
                print("Unable to remove: %s" % app_folder)

                                

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


def update_package():
    # Declaring local variables
    result = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    if control.architecture == "x64":
        bin_name_sub = "MicrosoftEdgeEnterprise_%s_X64.msi"
        bin_temp = "MicrosoftEdgeEnterpriseX64.msi"
        url_dl = "http://go.microsoft.com/fwlink/?LinkID=2093437"
    else:
        bin_name_sub = "MicrosoftEdgeEnterprise_%s_X86.msi"
        bin_temp = "MicrosoftEdgeEnterpriseX86.msi"
        url_dl = "http://go.microsoft.com/fwlink/?LinkID=2093505"

    print("Download url is: %s" % url_dl)

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

    # Downloading latest binaries
    if not isfile(bin_temp):
        print("Downloading: %s" % bin_temp)
        wget(url_dl, bin_temp, proxies=proxies)

        # Getting version from binary and renaming binary
        version = get_version_from_binary(bin_temp)
        latest_bin = bin_name_sub % version
        if not isfile(latest_bin):
            print("Renaming binary with the version number of the binary (%s)" % latest_bin)
            os.rename(bin_temp, latest_bin)

    # 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)))
        result = True
    control.version = "%s-%s" % (Version(version), control.version.split("-", 1)[-1])
    # control.set_software_version(Version(version))
    control.save_control_to_wapt()

    # Validating update-package-sources
    return result