# -*- 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 *

app_name = "BEAST"
app_dir = makepath(programfiles, app_name)


def install():
    # Declaring local variables
    zip_name = glob.glob("BEAST.*.zip")[0]
    unzipped_dir = app_name
    version = control.get_software_version()
    exe_path = makepath(app_dir, f"{app_name} v{version}", f"{app_name} v{version}.exe")

    # Installing software
    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir):
        remove_tree(app_dir)
    print("Extracting: %s to: %s" % (zip_name, app_dir))
    unzip(zip_name, app_dir)

    create_programs_menu_shortcut("BEAST", target = exe_path)

def uninstall() : 
    version = control.get_software_version()
    app_install_dir = makepath(programfiles, app_name)

    # Uninstalling software
    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir):
        remove_tree(app_dir)
    
    remove_programs_menu_shortcut("BEAST")

                                

# -*- 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 *
import re


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    update_dict = {"windows": ".zip", "macos": ".dmg"}
    api_url = "https://api.github.com/repos/beast-dev/beast-mcmc/releases/latest"

    # Get data from API
    releases_dict = json.loads(wgets(api_url, proxies=proxies))
    zip_found = False  # Flag pour indiquer la découverte d'un fichier .zip
    for release in releases_dict:
        if zip_found:
            break
        for asset in releases_dict["assets"]:
            if asset["browser_download_url"].endswith(".zip") and update_dict[control.target_os] in asset["browser_download_url"]:
                url_download = asset["browser_download_url"]
                latest_bin = url_download.split("/")[-1]
                version = releases_dict["name"].split(' ')[1].replace("v","")
                zip_found = True  # Mettre à jour le flag pour indiquer qu'un zip a été trouvé
                break

    # Deleting binaries
    for f in glob.glob("*.zip"):
        if f != latest_bin:
            remove_file(f)

    # Downloading latest binaries
    print("Download URL is: %s" % url_download)
    if not isfile(latest_bin):
        print("Downloading: %s" % latest_bin)
        wget(url_download, latest_bin, proxies=proxies)
    else:
        print("Binary is present: %s" % latest_bin)

    control.set_software_version(version)
    control.save_control_to_wapt()