# -*- coding: utf-8 -*-
##################################################
# This file is part of WAPT Enterprise
# All right reserved, (c) Tranquil IT Systems 2024
# For more information please refer to
# https://wapt.tranquil.it/store/licences.html
##################################################
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


def install():
    # Declaring local variables
    bin_name = glob.glob('MEGA*.exe')[0]
    # Installing the software
    
    install_exe_if_needed(bin_name,
        silentflags='/VERYSILENT /SUPPRESSMSGBOXES /NORESTART',
        key='{D3D535FA-0A6E-4A2E-97D8-2BEF54485683}_is1',
        min_version=control.get_software_version()
    )


                                

# -*- coding: utf-8 -*-
##################################################
# This file is part of WAPT Enterprise
# All right reserved, (c) Tranquil IT Systems 2024
# For more information please refer to
# https://wapt.tranquil.it/store/licences.html
##################################################
from setuphelpers import *
from setupdevhelpers import *


def update_package():
    # Initialisation
    package_updated = False
    proxies = get_proxies_from_wapt_console()
    if not proxies:
        proxies = get_proxies()
    app_name = control.name

    base_url = "https://www.megasoftware.net"
    url_release = f"{base_url}/releases"
    url_history = f"{base_url}/history"

    # With this part we avoid getting beta release
    latest_major_release_version = max({
        v.text.split(' ')[1]
        for v in bs_find(url_release, "select", "id", "home-version-select").find_all('option')
        if not 'Older Version' in v.text
    })

    valid_versions = [
        v.text.split("version ")[1].split(" ")[0]
        for v in bs_find_all(url_history, "div", "class", "qi-q")
        if f"MEGA {latest_major_release_version}" in v.text
    ]
    
    version = sorted(valid_versions, key=Version, reverse=True)[0]
    latest_bin = f"MEGA_{version}_win64_setup.exe"
    latest_bin_extension = latest_bin.rsplit('.', 1)[-1]

    download_url = "https://www.megasoftware.net/releases/" + latest_bin

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

    # 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