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


appx_package_name = "TheBrowserCompany.Arc"
appx_dir = makepath(programfiles, "WindowsAppsInstallers")


def install():
    # Declare local variables
    bin_path = glob.glob(f"Arc*.msix")[0]
    add_appx_cmd = f'Add-AppxProvisionedPackage -Online -PackagePath "{bin_path}" -SkipLicense'

    # Installing the UWP application if needed
    appxprovisionedpackage = run_powershell(f'Get-AppXProvisionedPackage -Online | Where-Object DisplayName -Like "{appx_package_name}"')
    if appxprovisionedpackage is None:
        remove_appx(appx_package_name, False)
        appxprovisionedpackage = {"Version": "0"}
    elif force:
        uninstall()

    if Version(appxprovisionedpackage["Version"], 4) < Version(control.get_software_version(), 4):
        print(f"Installing: {bin_path.split(os.sep)[-1]} ({control.get_software_version()})")
        killalltasks(ensure_list(control.impacted_process))
        run_powershell(add_appx_cmd, output_format="text")
    else:
        print(f'{appxprovisionedpackage["PackageName"]} is already installed and up-to-date.')


def uninstall():
    print(f"Removing AppX: {appx_package_name}")
    remove_appx(appx_package_name)


def audit():
    # Declaring local variables
    audit_result = "OK"
    audit_version = True
    appxprovisionedpackage = run_powershell(f'Get-AppXProvisionedPackage -Online | Where-Object DisplayName -Like "{appx_package_name}"')

    # Auditing software
    if appxprovisionedpackage is None:
        print(f"{appx_package_name} is not installed.")
        audit_result = "ERROR"
    elif audit_version:
        if Version(appxprovisionedpackage.get("Version", "0"), 4) < Version(control.get_software_version(), 4):
            print(f'{appxprovisionedpackage["PackageName"]} is installed in version: {appxprovisionedpackage["Version"]} instead of: {control.get_software_version()}.')
            audit_result = "WARNING"
        else:
            print(f'{appxprovisionedpackage["PackageName"]} is installed and up-to-date.')
    else:
        print(f'{appxprovisionedpackage["PackageName"]} is installed.')

    return audit_result
                                

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


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()

    xml_latest_version = "https://releases.arc.net/windows/prod/Arc.appinstaller"

    namespaces = {'ai': 'http://schemas.microsoft.com/appx/appinstaller/2018'}
    root = ET.fromstring(wgets(xml_latest_version, proxies=proxies))

    main_package = root.find('ai:MainPackage', namespaces=namespaces)

    msix_url = main_package.get('Uri')
    latest_bin = msix_url.rsplit('/', 1)[-1]
    latest_bin_extension = latest_bin.rsplit('.', 1)[-1]
    version = main_package.get('Version')

    # dependencies = root.find('ai:Dependencies', namespaces=namespaces)
    # packages = dependencies.findall('ai:Package', namespaces=namespaces)
    # dependencies_uri = [ pkg.get('Uri') for pkg in packages ]

    # Downloading latest binaries
    print(f"MSIX Download URL is: {msix_url}")
    if not isfile(latest_bin):
        print(f"Downloading: {latest_bin}")
        wget(msix_url, latest_bin, proxies=proxies)
    else:
        print(f"Binary is present: {latest_bin}")

    # Cleanup dependencies
    # for dep in glob.glob('*.appx'):
    #     remove_file(dep)

    # Downloading dependencies
    # for dep in dependencies_uri:
    #     dep_bin_name = dep.rsplit('/', 1)[-1]
    #     print(f"Download URL is: {dep}")
    #     if not isfile(dep_bin_name):
    #         print(f"Downloading: {dep_bin_name}")
    #         wget(dep, dep_bin_name, proxies=proxies)
    #     else:
    #         print(f"Binary is present: {dep}")

    # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        print(f"Software version updated (from: {control.get_software_version()} to: {Version(version)})")
        package_updated = True
    else:
        print(f"Software version up-to-date ({Version(version)})")

    for f in glob.glob(f'*.{latest_bin_extension}'):
        if f != latest_bin:
            remove_file(f)

    control.set_software_version(version)
    control.save_control_to_wapt()

    return package_updated