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

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 = [
    ".001",
    ".7z",
    ".arj",
    ".bz2",
    ".bzip2",
    ".cab",
    ".cpio",
    ".deb",
    ".dmg",
    ".esd",
    ".fat",
    ".gz",
    ".gzip",
    ".hfs",
    ".lha",
    ".lzh",
    ".lzma",
    ".ntfs",
    ".rar",
    ".rpm",
    ".squashfs",
    ".swm",
    ".tar",
    ".taz",
    ".tbz",
    ".tbz2",
    ".tgz",
    ".tpz",
    ".txz",
    ".vhd",
    ".wim",
    ".xar",
    ".xz",
    ".z",
    ".zip",
    ".zipe",
]
#    ".iso",


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

    # Uninstalling other versions of the software
    for uninstall in installed_softwares("7-zip"):
        if Version(uninstall["version"], 2) < Version(package_version, 2) or force:
            print("Removing: %s (%s)" % (uninstall["name"], uninstall["version"]))
            killalltasks(control.impacted_process.split(","))
            try:
                run(uninstall_cmd(uninstall["key"]))
            except:
                if not "uninstall.exe" in str(uninstall_cmd(uninstall["key"])).lower():
                    raise
            if uninstall_key_exists(uninstall["key"]):
                time.sleep(10)
                if "uninstall.exe" in str(uninstall_cmd(uninstall["key"])).lower():
                    unregister_uninstall(uninstall["key"], win64app=iswin64())

    # Installing the software
    print("Installing: %s" % bin_name)
    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