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

app_name = "geek"
app_dir = makepath(programfiles, app_name)
exe_path = makepath(app_dir, "geek.exe")


def install():
    bin_name = glob.glob('geek.zip')[0]

    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir) and force:
        remove_tree(app_dir)

    mkdirs(app_dir)
    unzip(bin_name, app_dir)

    # Creating shortcuts
    create_desktop_shortcut(app_name, target=exe_path)
    create_programs_menu_shortcut(app_name, target=exe_path)


def uninstall():
    # Uninstalling software
    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir):
        remove_tree(app_dir)

    # Removing shortcuts
    remove_desktop_shortcut(app_name)
    remove_programs_menu_shortcut(app_name)
                                

# -*- 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 *
import waptlicences
import zipfile  
import os      


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

    download_dict = {
        "windows": "https://geekuninstaller.com/geek.zip"
    }

    download_url = download_dict[control.target_os]
    latest_bin = download_url.rsplit('/', 1)[-1]
    latest_bin_extension = latest_bin.rsplit('.', 1)[-1]

    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}")


    # Décompression du fichier ZIP
    if latest_bin_extension == "zip":
        print(f"Extracting: {latest_bin}")
        with zipfile.ZipFile(latest_bin, 'r') as zip_ref:
            zip_ref.extractall()  # Extraction dans le répertoire courant

        # Recherche de l'exécutable .exe dans les fichiers extraits
        exe_file = None
        for file in os.listdir():  # Parcours uniquement dans le répertoire courant
            if file.endswith(".exe"):
                exe_file = file
                print(f"Found executable: {exe_file}")
                break
    else:
        print("Downloaded file is not a ZIP archive.")
    version = get_version_from_binary(exe_file)

    # 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