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

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

Installation procedure: https://wiki.documentfoundation.org/Deployment_and_Migration

"""
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "LibreOffice_"
silent_args_dict = {
    "ALLUSERS": 1,
    "CREATEDESKTOPLINK": 0,
    "REBOOTYESNO": "No",
    "ISCHECKFORPRODUCTUPDATES": 0,
    "VC_REDIST": 0,
    "QUICKSTART": 0,
    "ADDLOCAL": "ALL",
    "REMOVE": "gm_o_Onlineupdate",
}
force_register_as_default_mso_app = False  # Make sure that LBO will BE the default application for any Office applications
force_no_register_as_default_mso_app = False  # Make sure that LBO will NOT BE the default application for any Office applications


def install():
    # Declaring local variables
    package_version = control.get_software_version()
    bin_name = glob.glob("*%s*.msi" % bin_contains)[0]
    major_version = package_version.split(".")[0]

    # Uninstalling OpenOffice if detected
    for uninstall in installed_softwares(name="OpenOffice"):
        print("Removing: %s (%s)" % (uninstall["name"], uninstall["version"]))
        run(uninstall_cmd(uninstall["key"]))
        wait_uninstallkey_absent(uninstall["key"])

    # Uninstalling LibreOffice if too old
    for uninstall in installed_softwares(name="LibreOffice"):
        if not uninstall["version"].startswith(major_version):
            if "help" in uninstall["name"].lower():
                continue
            print("Removing: %s (%s)" % (uninstall["name"], uninstall["version"]))
            killalltasks(control.impacted_process.split(","))
            run(uninstall_cmd(uninstall["key"]))
            wait_uninstallkey_absent(uninstall["key"])

    # Defining Office default applications
    """
    REGISTER_ALL_MSO_TYPES (default=0) - use LibreOffice as the default application for Microsoft Office file formats 
    REGISTER_NO_MSO_TYPES  (default=0) - don't use LibreOffice as default application for any Microsoft Office file formats 
    REGISTER_DOC           (default=0) - use LibreOffice as the default application for Microsoft Word file format .doc (You can use similar for .xls, .ppt etc.)

    """
    if not force_register_as_default_mso_app or not force_no_register_as_default_mso_app:
        # Checking if MSO file extensions refer to MSO apps and affect LBO if not
        if not get_file_assocation(".doc").startswith("Word") and not get_file_assocation(".docx").startswith("Word"):
            silent_args_dict["REGISTER_DOC"] = 1
            silent_args_dict["REGISTER_DOCX"] = 1
        if not get_file_assocation(".xls").startswith("Excel") and not get_file_assocation(".xlsx").startswith("Excel"):
            silent_args_dict["REGISTER_XLS"] = 1
            silent_args_dict["REGISTER_XLSX"] = 1
        if not get_file_assocation(".ppt").startswith("PowerPoint") and not get_file_assocation(".pptx").startswith("PowerPoint"):
            silent_args_dict["REGISTER_PPT"] = 1
            silent_args_dict["REGISTER_PPTX"] = 1
        if not get_file_assocation(".pub").startswith("Publisher"):
            silent_args_dict["REGISTER_PUB"] = 1

    if not force_register_as_default_mso_app or not force_no_register_as_default_mso_app:
        # Checking if a version of MSO is installed
        count_mso_apps = 0
        if installed_softwares(name="Microsoft Office "):
            count_mso_apps += 1
        if installed_softwares(name="Office 365"):
            count_mso_apps += 1
        if installed_softwares(name="Microsoft 365"):
            count_mso_apps += 1

    if force_register_as_default_mso_app:
        # Register LBO as default MSO apps
        silent_args_dict["REGISTER_ALL_MSO_TYPES"] = 1
    elif force_no_register_as_default_mso_app:
        # Do not register LBO as default MSO apps
        silent_args_dict["REGISTER_NO_MSO_TYPES"] = 1
    else:
        if count_mso_apps == 0:
            # Register LBO as default MSO apps if no MSO version was detected
            silent_args_dict["REGISTER_ALL_MSO_TYPES"] = 1

    # Installing the software
    print("Installing: %s" % bin_name)
    install_msi_if_needed(bin_name, properties=silent_args_dict, timeout=1200)


def get_file_assocation(ext):
    r"""Return the associated application for a file extension (example : '.doc')

    >>> get_file_assocation('.pptx')
    'LibreOffice.Pptx'

    """
    try:
        return winreg.QueryValue(winreg.HKEY_CLASSES_ROOT, ext)
    except:
        return ""

                                

# -*- 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.libreoffice.org/download/download"
    if control.architecture == "x64":
        dl_arch = "x86_64"
        ends_bin_name = "_Win_x64.msi"
    else:
        dl_arch = "x86"
        ends_bin_name = "_Win_x86.msi"
    bin_contains = "LibreOffice_"

    # Getting latest version from official website
    version = bs_find_all(url, "span", "class", "dl_version_number", proxies=proxies)[1].string  # [1] for Still, [0] for Fresh
    search_url = "https://download.documentfoundation.org/libreoffice/stable/%s/win/%s/" % (version, dl_arch)
    for bs_search in bs_find_all(search_url, "a", proxies=proxies):
        if bin_contains in bs_search["href"] and ends_bin_name in bs_search["href"]:
            latest_bin = bs_search.string
            download_url = search_url + latest_bin
            break

    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_version_from_binary(latest_bin)
        if not version_from_file.startswith(version) 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, bin_contains + version_from_file + ends_bin_name)
            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(version)

    # Validating update-package-sources
    return result