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

bin_name_string = "Ubuntu_%s.Appx"
package_name = "CanonicalGroupLimited.Ubuntu"


def set_variables():
    # Declaring variables
    global package_version
    package_version = control.version.split("-")[0]
    global bin_name
    bin_name = bin_name_string % package_version
    global app_name
    app_name = control.name
    global wsl_dir
    wsl_dir = makepath(programfiles, "WSL")
    global wsl_ubuntu
    wsl_ubuntu = makepath(wsl_dir, bin_name)


def install():
    # Initializing variables
    set_variables()

    # Installing the package
    if not isdir(wsl_dir):
        mkdirs(wsl_dir)
    for bin_in_dir in glob.glob(makepath(wsl_dir, "*ubuntu*")):
        if bin_in_dir != wsl_ubuntu:
            print("Removing %s" % bin_in_dir)
            remove_file(bin_in_dir)

    if not isfile(wsl_ubuntu):
        print("Copying %s in %s " % (bin_name, wsl_dir))
        filecopyto(bin_name, wsl_dir)


def uninstall():
    # Initializing variables
    set_variables()

    # Uninstalling the package
    print("Removing %s for all users" % package_name)
    run_powershell("Get-AppxPackage -Name %s -AllUsers | Remove-AppxPackage" % package_name)


def session_setup():
    # Initializing variables
    set_variables()

    # Install the Appx in user environment
    try:
        installed_wsl_ubuntu_version = run_powershell('Get-AppxPackage -Name "%s" | Select-Object version' % package_name)["Version"]
    except:
        installed_wsl_ubuntu_version = "0.0.0"

    if package_version > installed_wsl_ubuntu_version:
        print("Installing %s for the current user" % app_name)
        run_powershell('Add-AppxPackage -Path "%s"' % wsl_ubuntu)

    # Upgrading WSL Ubuntu to WSL 2
    # Upgrade procedure: https://docs.microsoft.com/en-us/windows/wsl/install-win10#set-your-distribution-version-to-wsl-1-or-wsl-2
    if windows_version() >= "10.0.19041":
        try:
            print("Upgrading Ubuntu to WSL 2")
            run_powershell("wsl --set-version Ubuntu-18.04 2")
        except:
            print("Ubuntu is already in WSL 2, or the upgrade do not complete")
    else:
        print("You need Windows 10 version 2004 minimum for upgrading Ubuntu to WSL 2")

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
import platform
import xml.etree.ElementTree as ET

bin_name_string = "Ubuntu_%s.Appx"


def set_variables():
    # Declaring variables
    global package_version
    package_version = control.version.split("-")[0]
    global bin_name
    bin_name = bin_name_string % package_version
    global app_name
    app_name = control.name
    global wsl_dir
    wsl_dir = makepath(programfiles, "WSL")
    global wsl_debian
    wsl_debian = makepath(wsl_dir, bin_name)


def update_package():
    print("Download/Update package content from upstream binary sources")

    # Getting proxy informations from WAPT settings
    proxy = {}
    if platform.system() == "Windows" and isfile(makepath(user_local_appdata(), "waptconsole", "waptconsole.ini")):
        proxywapt = inifile_readstring(makepath(user_local_appdata(), "waptconsole", "waptconsole.ini"), "global", "http_proxy")
        if proxywapt:
            proxy = {"http": proxywapt, "https": proxywapt}

    # Initializing variables
    set_variables()
    latest_bin = bin_name_string.split("_")[0] + ".Appxbundle"
    url_dl = "https://aka.ms/wslubuntu2204"

    print("Download url is: " + url_dl)

    # Downloading latest binaries
    if isfile(latest_bin):
        remove_file(latest_bin)
    if not isfile(latest_bin):
        print("Downloading: " + latest_bin)
        wget(url_dl, latest_bin, proxies=proxy)

        # Unzip to get file version
        unzip(latest_bin, target="ubuntu_wsl_temp")

        # Checking version from file
        xml_to_read = ET.parse(r"ubuntu_wsl_temp\AppxMetadata\AppxBundleManifest.xml")
        xml_content = xml_to_read.getroot()

        for vers in list(xml_content):
            if vers.get("Name") == "CanonicalGroupLimited.Ubuntu":
                version_from_file = vers.get("Version")

        version = version_from_file
        old_latest_bin = latest_bin
        latest_bin = bin_name_string % version
        if isfile(latest_bin):
            remove_file(latest_bin)
        os.rename(old_latest_bin, latest_bin)
        print("Verified latest " + app_name + " version is: " + version)

        # Removing temp folder
        remove_tree("ubuntu_wsl_temp")

        # Changing version of the package
        control.version = "%s-%s" % (version, int(control.version.split("-")[-1]) + 1)
        control.save_control_to_wapt()
        print("Changing version to " + control.version + " in WAPT\\control")

    # Deleting outdated binaries
    for bin_in_dir in glob.glob("*.Appx"):
        if bin_in_dir != latest_bin:
            print("Outdated binary: " + bin_in_dir + " Deleted")
            remove_file(bin_in_dir)