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

r"""
Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()

"""
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "7z"
ext_file_association = [
    ".7z",
    ".zip",
    ".rar",
    ".001",
    ".cab",
    # ".iso",
    ".xz",
    ".txz",
    ".lzma",
    ".tar",
    ".cpio",
    ".bz2",
    ".bzip2",
    ".tbz2",
    ".tbz",
    ".gz",
    ".gzip",
    ".tgz",
    ".tpz",
    ".z",
    ".taz",
    ".lzh",
    ".lha",
    ".rpm",
    ".deb",
    ".arj",
    # ".vhd",
    # ".vhdx",
    ".wim",
    ".swm",
    ".esd",
    ".fat",
    ".ntfs",
    ".dmg",
    ".hfs",
    ".xar",
    ".squashfs",
]


def install():
    # Initializing variables
    package_version = control.version.split("-", 1)[0]
    bin_name = glob.glob("*%s*.msi" % bin_contains)[0]
    to_force = False

    # Uninstalling other versions of the software
    if len(installed_softwares(name="7-Zip")) >= 2:
        to_force = True
    for to_uninstall in installed_softwares(name="7-Zip"):
        if Version(to_uninstall["version"], 2) < Version(package_version, 2) or to_uninstall["key"] == "7-Zip" or force or to_force:
            print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
            killalltasks(control.impacted_process.split(","))
            try:
                run(uninstall_cmd(to_uninstall["key"]))
            except:
                wait_uninstallkey_absent(to_uninstall["key"])
                unregister_uninstall(to_uninstall["key"], win64app=to_uninstall["win64"])

    # Installing the software
    print("Installing: %s" % bin_name)
    if to_force:
        install_msi_if_needed(bin_name, min_version=package_version, force=to_force)
    else:
        install_msi_if_needed(bin_name, min_version=package_version)

    # File association for 7-Zip
    for ext in ext_file_association:
        register_ext(
            "7-zip",
            ext,
            '"%s" "%%1"' % (makepath(programfiles, "7-zip", "7zFM.exe")),
            icon="%s,1" % (makepath(programfiles, "7-zip", "7z.dll")),
        )

                                

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


def update_package():
    # Declaring local variables
    result = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    app_name = control.name
    url = "https://www.7-zip.org/download.html"
    if control.architecture == "x64":
        sub_bin_name = "7z%s-x64.msi"
    else:
        sub_bin_name = "7z%s.msi"

    # Getting latest version from official website
    print("URL used is: " + url)
    version = bs_find_all(url, "b", proxies=proxies)[1].text.split("Zip ")[-1].split(" (")[0]
    splitted_version = version.replace(".", "")
    latest_bin = sub_bin_name % splitted_version
    download_url = "https://www.7-zip.org/a/%s" % latest_bin

    print("Latest %s version is: %s" % (app_name, version))
    print("Download URL is: %s" % download_url)

    # Downloading latest binaries
    if not isfile(latest_bin):
        print("Downloading: %s" % latest_bin)
        wget(download_url, latest_bin, proxies=proxies)

        # Checking version from file
        version_from_file = get_msi_properties(latest_bin)["ProductName"].split(" ")[1]
        if Version(version) != Version(version_from_file) and version_from_file != "":
            print("Changing version to the version number of the binary (from: %s to: %s)" % (version, version_from_file))
            os.rename(latest_bin, sub_bin_name % version_from_file)
            version = version_from_file
        else:
            print("Binary file version corresponds to online version")

    # 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)))
        result = True
    else:
        print("Software version up-to-date (%s)" % Version(version))
    control.version = "%s-%s" % (Version(version), control.version.split("-", 1)[-1])

    # control.set_software_version(version)
    control.save_control_to_wapt()

    # Deleting outdated binaries
    remove_outdated_binaries(splitted_version)

    # Validating update-package-sources
    return result