# -*- coding: utf-8 -*-
from setuphelpers import *
import requests
from bs4 import BeautifulSoup
import time

# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "SpotifyAB"
appx_package = "SpotifyAB.SpotifyMusic"
appx_dir = makepath(programfiles, "WindowsAppsInstallers")
file_extension = "appx"


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

    # Checking minimum Windows version for this UWP application
    if windows_version() >= WindowsVersions.Windows10v1607:
        # Copying the software
        if not isdir(appx_dir):
            mkdirs(appx_dir)
        else:
            for appx_file in glob.glob(makepath(appx_dir, "*%s*.%s" % (bin_contains, file_extension))):
                if not package_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(), WindowsVersions.Windows10v1607)
        )


def uninstall():
    # Uninstalling the package
    remove_appx(appx_package)
    for appx_file in glob.glob(makepath(appx_dir, "*%s*.%s" % (bin_contains, file_extension))):
        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)


def session_setup():
    # Declaring local variables
    package_version = control.get_software_version()
    app_name = control.name
    bin_path = glob.glob(makepath(appx_dir, "*%s*.%s" % (bin_contains, file_extension)))[0]
    bin_name = bin_path.split("\\")[-1]
    user_app_dir = makepath(user_appdata, "Spotify")
    user_app_path = makepath(user_app_dir, "Spotify.exe")

    # Uninstalling Spotify Desktop from user environment
    try:
        if isfile(user_app_path):
            print("Uninstalling: Spotify Desktop from user environment")
            killalltasks(control.get_impacted_process_list())
            registry_deletekey(HKEY_CURRENT_USER, r"SOFTWARE\WAPT", "Spotify")
            run('"%s" %s' % (user_app_path, "/uninstall /silent"))
            time.sleep(60)
    except:
        print("Force uninstalling: Spotify Desktop from user environment")
        killalltasks(control.get_impacted_process_list())
        # HKEY_CURRENT_USER\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall\Spotify
        registry_deletekey(HKEY_CURRENT_USER, r"SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall", "Spotify")
        app_user_local_appdata = makepath(user_local_appdata, "Spotify")
        app_application_data = makepath(application_data, "Spotify")
        if isdir(app_user_local_appdata):
            remove_tree(app_user_local_appdata)
        if isdir(app_application_data):
            remove_tree(app_application_data)
        remove_user_desktop_shortcut("Spotify")
        remove_user_programs_menu_folder("Spotify")

    # Installing Spotify UWP in user env
    appx_version = get_powershell_str("Get-AppxPackage -Name %s" % appx_package, "Version")
    if not appx_version or appx_version < package_version or force:
        print("Installing: %s" % bin_name)
        run_powershell('Add-AppxPackage -Path "%s"' % bin_path)
    else:
        print("%s is installed in correct version (%s)" % (app_name, appx_version))


def update_package():
    # Declaring local variables
    result = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    app_name = control.name
    store_id = "9ncbcszsjrsb"

    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"},
    )
    page = BeautifulSoup(res.content, features="html.parser")
    for bs_search in page.find_all("a"):
        if bin_contains in bs_search.text and ".%s" % file_extension in bs_search.text:
            version = bs_search.text.split("_")[1]
            latest_bin = bs_search.text
            download_url = bs_search["href"]
            break
    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")

        if Version(version) > Version(control.get_software_version()):
            print("Software version updated (from: %s to: %s)" % (control.get_software_version(), Version(version)))
            result = True
        else:
            print("Software version up-to-date (%s)" % Version(version))
        control.version = "%s-%s" % (Version(version), control.version.split("-", 1)[-1])
    else:
        print("ERROR: The retrieved url will not download from microsoft's servers")
    # control.set_software_version(version)
    control.save_control_to_wapt()

    # Deleting outdated binaries
    remove_outdated_binaries(version)

    # Validating or not update-package-sources
    return result


def remove_appx(package, default_user=True):
    r"""Uninstall and remove Windows AppX package from the computer in machine and user environnement
    Args:
        package (str): package name, asterisk character (*) can be used as joker
        default_user (bool): Remove from image that will be installed for each new user
    .. versionadded:: 2.2
    """
    if running_as_admin() or running_as_system():
        run_powershell('Get-AppxPackage "%s" -AllUsers | Remove-AppxPackage -AllUsers' % package)
        if default_user:
            run_powershell(
                'Get-AppXProvisionedPackage -Online | Where-Object DisplayName -Like "%s" | Remove-AppxProvisionedPackage -Online -AllUsers' % package
            )
    else:
        run_powershell('Get-AppxPackage "%s" | Remove-AppxPackage' % package)


def get_powershell_str(powershell_cmd, value=None):
    r"""Get a value as string from the output of a powershell command
    WARNING: Adding -AllUsers parameter may be necessary to make sure WAPT is enable to get the request value
    Args:
        powershell_cmd (str): Powershell command
        value (str)         : Value to get as string
    .. versionadded:: 2.2
    """
    if value:
        return str(run_powershell("%s | Select-Object -ExpandProperty %s | Write-Host -NoNewline" % (powershell_cmd, value), output_format="text"))
    else:
        return str(run_powershell("%s | Write-Host -NoNewline" % powershell_cmd, output_format="text"))