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


appx_package = "Microsoft.Services.Store.Engagement"
appx_dir = makepath(programfiles, "WindowsAppsInstallers")
windows_min_os_version = WindowsVersions.Windows10v1607
allow_remove = False


def install():
    # Declaring local variables
    bin_name = glob.glob("%s*%s*" % (appx_package, control.get_software_version()))[0]
    bin_path = makepath(appx_dir, bin_name)

    # Checking minimum Windows version for this UWP application
    if windows_version() >= windows_min_os_version:
        if not isdir(appx_dir):
            mkdirs(appx_dir)
        else:
            for appx_file in glob.glob(makepath(appx_dir, "%s*%s*" % (appx_package, control.get_software_version()))):
                if not control.get_software_version() in appx_file:
                    remove_file(appx_file)
        if not isfile(bin_path):
            print("Copying: %s to: %s" % (bin_name, appx_dir))
            filecopyto(bin_name, appx_dir)
    else:
        error(
            "This Windows version (%s) need to be upgraded to minimum version (%s) to use this package" % (windows_version(), windows_min_os_version)
        )


def session_setup():
    # Declaring local variables
    bin_path = glob.glob(makepath(appx_dir, "%s*%s*" % (appx_package, control.get_software_version())))[0]

    # Installing the software in user env
    appx_version = get_powershell_str("Get-AppxPackage -Name %s" % appx_package, "Version")
    if not appx_version or appx_version < control.get_software_version() or force:
        print("Installing: %s" % bin_path.split("\\")[-1])
        run_powershell('Add-AppxPackage -Path "%s"' % bin_path)
    else:
        print("%s is installed in correct version (%s)" % (appx_package, appx_version))


def uninstall():
    if allow_remove:
        remove_appx(appx_package)
    for appx_file in glob.glob(makepath(appx_dir, "%s*%s*" % (appx_package, control.get_software_version()))):
        remove_file(appx_file)
    if isdir(appx_dir) and dir_is_empty(appx_dir):
        print("Removing: %s since he is empty" % (appx_dir))
        remove_tree(appx_dir)

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
import requests
from bs4 import BeautifulSoup
import json
import waptguihelper
import re


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    store_url = control.sources
    if not store_url:
        store_id = waptguihelper.input_dialog(
            "Choice of app",
            "Enter the windows store app id who depend on the desired dependency (foundable in package url: https://apps.microsoft.com/store/apps)",
            store_url,
        )
    else:
        store_id = store_url.split("/")[-1]
    url_ms = "https://apps.microsoft.com/store/detail/%s" % store_id
    control.sources = url_ms
    package_prefix = control.package.split("-")[0]

    # Getting info from adguard api
    res = requests.post(
        "https://store.rg-adguard.net/api/GetFiles",
        "type=ProductId&url=%s&ring=RP&lang=fr-FR" % store_id,
        headers={"content-type": "application/x-www-form-urlencoded"},
        proxies=proxies,
    )
    all_files = []
    all_files_dict = {}
    page = BeautifulSoup(res.content, features="html.parser")
    for bs_search in page.find_all("a"):
        if not "BlockMap" in bs_search.text and not "eappxbundle" in bs_search.text and not "emsixbundle" in bs_search.text:
            version = bs_search.text.split("_")[1]
            bin_name = bs_search.text
            download_url = bs_search["href"]
            pkg_splitted = re.split(r".\d+.", bin_name)[0]
            package_name = package_prefix + "-" + pkg_splitted.split("_")[0].replace(".", "-").lower()
            all_files.append({"version": version, "bin_name": bin_name, "package_name": package_name})
            all_files_dict[bin_name] = {"version": version, "bin_name": bin_name, "download_url": download_url, "package_name": package_name}
    if "tis-template-microsoft-store" in control.package:
        selected = waptguihelper.grid_dialog(
            "Please select the proper file",
            json.dumps(all_files),
            waptguihelper.GRT_SELECTED,
            '{"columns":[{"propertyname":"version","datatype":"String","required":false,"readonly":false,"width":130},{"propertyname":"bin_name","datatype":"String","required":false,"readonly":false,"width":500},{"propertyname":"package_name","datatype":"String","required":false,"readonly":false,"width":250}]}',
        )
    else:
        selected = [a for a in all_files if control.architecture in a["bin_name"] and control.package == a["package_name"]]
        if not selected:
            higer_version = "0"
            for a in all_files:
                if Version(higer_version) < Version(a["version"]) and control.package == a["package_name"]:
                    higer_version = a["version"]
            selected = [a for a in all_files if higer_version == a["version"] and control.package == a["package_name"]]

    for bin_selected in selected:
        latest_bin = bin_selected["bin_name"]
        version = all_files_dict[latest_bin]["version"]
        download_url = all_files_dict[latest_bin]["download_url"]
        package_name = all_files_dict[latest_bin]["package_name"]
        if download_url.split("/")[2].endswith("microsoft.com"):
            if not isfile(latest_bin):
                print("Downloading: %s" % latest_bin)
                wget(download_url, latest_bin, proxies=proxies)
            else:
                print("Binary file version corresponds to online version")
        else:
            print("ERROR: The retrieved url will not download from microsoft's servers")

    # Specifying arch
    if "arm" in latest_bin:
        control.architecture = "arm"
    elif "arm64" in latest_bin:
        control.architecture = "arm64"
    elif "x64" in latest_bin:
        control.architecture = "x64"
    elif "x86" in latest_bin:
        control.architecture = "x86"
    else:
        control.architecture = "all"

    # Updating pkg infos
    if "tis-template-microsoft-store" in control.package:
        control.package = package_name
        # Removing template files
        remove_file("WAPT\\changelog.txt")
        control.description = ""
        control.categories = "Dependency"

    control.name = re.split(r" \d", latest_bin.split("_")[0].replace(".", " "))[0]
    control.save_control_to_wapt()

    # Changing setup.py appx_package variable
    new_lines = []
    with open("setup.py", "r") as f:
        for line in f.readlines():
            if line.startswith("appx_package"):
                line = 'appx_package = "%s"\n' % latest_bin.split("_")[0]
            new_lines.append(line)
    with open("setup.py", "w") as f:
        f.writelines(new_lines)

    # 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)))
        package_updated = True
    else:
        print("Software version up-to-date (%s)" % Version(version))
    control.set_software_version(version)
    control.save_control_to_wapt()

    # Deleting outdated binaries
    remove_outdated_binaries(version, ["appxbundle", "msixbundle", "appx", "msix"])

    # Validating or not update-package-sources
    return package_updated