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


r"""
Sources:
https://docs.microsoft.com/windows-hardware/manufacture/desktop/windows-setup-command-line-options
https://www.prajwaldesai.com/windows-10-upgrade-setup-failed-with-exit-code-0xc1900200/
https://www.tomshardware.com/how-to/bypass-windows-11-tpm-requirement
https://docs.microsoft.com/windows-hardware/manufacture/desktop/windows-setup-command-line-options?view=windows-11#compat
https://docs.microsoft.com/en-us/windows-server/administration/windows-commands/cleanmgr

Infos:
Forced install will automatically reboot the PC
This package can be updated graphically

"""
AllowUpgradesWithUnsupportedTPMOrCPU = False
DynamicUpdate = False
upgrade_path = makepath(systemdrive, "WindowsUpgrade")
upgrade_logs_path = makepath(systemdrive, "WindowsUpgradeLogs")
windows_old_path = makepath(systemdrive, "Windows.old")
cleanmgr_timeout = 660


def install():
    # Declaring local variables
    package_version = control.get_software_version()
    iso_path = makepath(basedir, "iso")
    cleanmgr_keys = [
        "Active Setup Temp Folders",
        "BranchCache",
        "Content Indexer Cleaner",
        "D3D Shader Cache",
        "Delivery Optimization Files",
        "Device Driver Packages",
        "Diagnostic Data Viewer database files",
        "Downloaded Program Files",
        "DownloadsFolder",
        "Internet Cache Files",
        "Language Pack",
        "Offline Pages Files",
        "Old ChkDsk Files",
        "Previous Installations",
        "Recycle Bin",
        "RetailDemo Offline Content",
        "Setup Log Files",
        "System error memory dump files",
        "System error minidump files",
        "Temporary Files",
        "Temporary Setup Files",
        "Temporary Sync Files",
        "Thumbnail Cache",
        "Update Cleanup",
        "Upgrade Discarded Files",
        "User file versions",
        "Windows Defender",
        "Windows Error Reporting Files",
        "Windows ESD installation files",
        "Windows Upgrade Log Files",
    ]

    # Checking if Windows must be upgraded
    if Version(windows_version(), 3) < Version(package_version, 3):
        # Trying to remove previous version of Windows if needed
        if isdir(windows_old_path) or isdir(r"C:\$Windows.~WS"):
            key_path = r"SOFTWARE\Microsoft\Windows\CurrentVersion\Explorer\VolumeCaches"
            key_value = r"StateFlags0042"
            for key in cleanmgr_keys:
                if "Upgrade Discarded Files" in key or "Previous Installations" in key:
                    registry_set(HKEY_LOCAL_MACHINE, makepath(key_path, key), key_value, 2)
                else:
                    registry_set(HKEY_LOCAL_MACHINE, makepath(key_path, key), key_value, 0)

            print("Removing previous version of Windows")
            cleanmgr_cmd = "cleanmgr /verylowdisk /sagerun:42"
            run_notfatal(cleanmgr_cmd, timeout=cleanmgr_timeout)  # seems to timeout on missing user graphical interface
            # run_powershell(r"Start-Process -FilePath CleanMgr.exe -ArgumentList '/sagerun:42 /verylowdisk' -WindowStyle Hidden -Wait")

            print("Waiting for custom CleanMgr")
            max_loop = cleanmgr_timeout
            while isrunning("cleanmgr"):
                loop = 0
                loop += 10
                if loop > max_loop:
                    print("CleanMgr timed out after %s seconds, trying to continue" % max_loop)
                    print('If the upgrade fail, please run: "%s" manually as asministrator on the PC' % cleanmgr_cmd)
                time.sleep(10)
                print("CleanMgr is running since %s seconds" % loop)
            print("CleanMgr finished")

        # Move iso
        if isdir(upgrade_path):
            remove_tree(upgrade_path)
        os.rename(iso_path, upgrade_path)

        if not isdir(upgrade_logs_path):
            mkdirs(upgrade_logs_path)

        # Upgrading Windows if needed
        if isdir(upgrade_path):
            print("%s folder exists, continuing..." % upgrade_path)
            print("Free disk space: %d bytes" % get_disk_free_space(upgrade_path))
            if get_disk_free_space(upgrade_path) < 25000000000:
                error("Not enough free space on disk for upgrade, please cleanup")
            else:
                # Changing cmd options
                additionnal_cmd = ""
                if AllowUpgradesWithUnsupportedTPMOrCPU:
                    registry_set(HKEY_LOCAL_MACHINE, r"SYSTEM\Setup\MoSetup", "AllowUpgradesWithUnsupportedTPMOrCPU", 1)
                    additionnal_cmd = additionnal_cmd + " /compat ignorewarning"
                if DynamicUpdate:
                    additionnal_cmd = additionnal_cmd + " /dynamicupdate enable"
                else:
                    additionnal_cmd = additionnal_cmd + " /dynamicupdate disable"

                if force:
                    try:
                        print("Force Upgrading Windows from version: %s to Windows version: %s" % (windows_version(), package_version))
                        run(
                            rf'"{makepath(upgrade_path, "setup.exe")}" /auto upgrade /quiet /eula accept /telemetry disable /showoobe none /copylogs "{makepath(upgrade_logs_path, "win11_upgrade_logs")}" {additionnal_cmd}',
                            timeout=5000,
                        )
                    except Exception as error_force_upgrade:
                        error(error_force_upgrade)
                else:
                    additionnal_cmd = additionnal_cmd + " /noreboot"
                    try:
                        print("Upgrading Windows from version: %s to Windows version: %s" % (windows_version(), package_version))
                        run(
                            rf'"{makepath(upgrade_path, "setup.exe")}" /auto upgrade /quiet /eula accept /telemetry disable /showoobe none /copylogs "{makepath(upgrade_logs_path, "win11_upgrade_logs")}" {additionnal_cmd}',
                            timeout=5000,
                        )
                    except Exception as error_upgrade:
                        error(error_upgrade)
        else:
            error("%s do not exist, please reinstall this package" % upgrade_path)
    else:
        print("This PC is already up-to-date on Windows version: %s" % windows_version())


def uninstall():
    if AllowUpgradesWithUnsupportedTPMOrCPU:
        registry_delete(HKEY_LOCAL_MACHINE, r"SYSTEM\Setup\MoSetup", "AllowUpgradesWithUnsupportedTPMOrCPU")
    if isdir(upgrade_path):
        print("Removing: %s" % upgrade_path)
        remove_tree(upgrade_path)
    if isdir(upgrade_logs_path):
        print("Removing: %s" % upgrade_logs_path)
        remove_tree(upgrade_logs_path)


def audit():
    if Version(windows_version()) < Version(control.get_software_version()):
        if isrunning("SetupHost"):
            print("WARNING: Windows upgrade is running...")
            return "WARNING"
        elif is_pending_reboot():
            print("WARNING: Please reboot this PC to finalize the upgrade")
            return "WARNING"
        else:
            print("ERROR: Windows upgrade failed")
            return "ERROR"
    else:
        print("OK: This PC is now up-to-date on Windows version: %s" % windows_version())
        print("Cleaning up since the upgrade was successful...")
        if isdir(upgrade_path):
            print("Removing: %s" % upgrade_path)
            remove_tree(upgrade_path)
        if isdir(upgrade_logs_path):
            print("Removing: %s" % upgrade_logs_path)
            remove_tree(upgrade_logs_path)
        return "OK"

                                

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


def update_package():
    # Declaring local variables
    package_updated = False
    iso_path = makepath(basedir, "iso")
    setup_path = makepath(iso_path, "setup.exe")
    iso_present = False
    iso_extracted = False
    iso_copy_path = None
    iso_to_download = False
    keep_extracted_iso = False
    keep_current_iso = False

    lang_dict = {
        "fr": "French",
        "en": "English",
        "pl": "Polish",
        "de": "German",
        "es": "Spanish",
        "pt": "Portuguese",
        "it": "Italian",
        "nl": "Dutch",
        "ru": "Russian",
        "all": "Others",
    }
    lang_list = []
    for entry in lang_dict:
        lang_list.append({"short name": entry, "long name": lang_dict[entry]})

    # Detect ISO if placed
    for iso_in_dir in glob.glob("*.iso"):
        if isfile(iso_in_dir):
            iso_present = True

    # Detect ISO if already extracted
    if isfile(setup_path):
        iso_extracted = True

    if iso_present:
        keep_iso_present_question = waptguihelper.message_dialog("ISO found", f"Do you want to keep: {iso_in_dir}?", waptguihelper.ID_YES)
        if keep_iso_present_question == waptguihelper.ID_YES:
            keep_current_iso = True

    if iso_extracted:
        keep_iso_extracted_question = waptguihelper.message_dialog("An ISO is already extracted", "Do you want to keep it?", waptguihelper.ID_YES)
        if keep_iso_extracted_question == waptguihelper.ID_YES:
            keep_extracted_iso = True

    if keep_current_iso and keep_extracted_iso:
        keep_only_one_question = waptguihelper.grid_dialog(
            "Please select only one", [{"keep :": iso_in_dir}, {"keep :": "iso already extracted"}], waptguihelper.GRT_SELECTED
        )
        if "extracted" in keep_only_one_question[0]["keep :"]:
            keep_current_iso = False
        else:
            keep_extracted_iso = False

    if not keep_extracted_iso:
        if not iso_present:
            iso_absent_question = waptguihelper.message_dialog(
                "No ISO detected", "Do you want to download Windows ISO from Microsoft's servers?\nClick No to use your own ISO", waptguihelper.ID_YES
            )
            if iso_absent_question == waptguihelper.ID_YES:
                iso_to_download = True

    if iso_to_download:
        # Download from browser
        try:
            if isdir(makepath(programfiles32, "Google", "Chrome", "Application")):
                chrome_app_path = makepath(programfiles32, "Google", "Chrome", "Application")
            else:
                chrome_app_path = makepath(programfiles, "Google", "Chrome", "Application")
            chrome_bin_path = makepath(chrome_app_path, "chrome.exe")
            cmd_browser = rf'"{chrome_bin_path}" --user-agent="Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko)Chrome/61.0.3163.100 Safari/537.36" --user-data-dir=%tmp%\chrome_tmp "{control.sources}"'
            waptguihelper.message_dialog(
                "Chrome will open with a custom user-agent",
                'Select "Download Windows 11 Disk Image (ISO)" => "Windows 11 (multi-edition ISO) => then your language. When the download is finished, please close the opened Chrome',
            )
            subprocess.Popen(cmd_browser, shell=True, start_new_session=True)
            waptguihelper.message_dialog("Waiting for download", "Click OK when the download is complete", waptguihelper.ID_OK)
        except:
            error(cmd_browser)

    # Copy to pkg
    if not keep_current_iso and not iso_present and not keep_extracted_iso:
        # Cleanup ISO files
        for iso_in_dir in glob.glob("*.iso"):
            remove_file(iso_in_dir)

        iso_copy_path = waptguihelper.filename_dialog("Please provide the Windows ISO to copy in this package", "", "", "ISO Files|*.iso")
        print("Copying: " + iso_copy_path)
        filecopyto(iso_copy_path, basedir)

    # Extracting ISO
    if not keep_extracted_iso:
        # Cleanup ISO dir
        if isdir(iso_path):
            remove_tree(iso_path)
        iso_in_dir = glob.glob("*.iso")[0]
        print("Extracting: " + iso_in_dir)
        unzip_with_7zip(iso_in_dir, iso_path)
        remove_file(iso_in_dir)

    # Getting version from the ISO setup.exe
    version = ".".join(get_file_properties(setup_path)["ProductVersion"].split(".")[:3])

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

    # Changing locale of the package
    lang_short = waptguihelper.grid_dialog("Please select a language", lang_list, waptguihelper.GRT_SELECTED)[0]["short name"]

    # Updating pkg infos
    if "-template" in control.package:
        package_name = waptguihelper.input_dialog(control.package, "You can redefine the package name", control.package.split("-template")[0])
        control.package = package_name
        if " Template" in control.name:
            control.name = control.name.split(" Template")[0]
        print("Changing package name to: %s" % package_name)
    control.locale = lang_short
    control.save_control_to_wapt()
    print("Changing locale to: %s in WAPT\\control" % control.locale)

    # Validating update-package-sources
    return package_updated