# -*- 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 = "DontSleep "
editor_dir = makepath(programfiles, "DontSleep")
app_dir = makepath(editor_dir, "DontSleep")
app_path = makepath(app_dir,"DontSleep_x64.exe")
audit_version = False


def get_installed_version(app_path):
    return get_file_properties(app_path).get("FileVersion", "")


def install():
    # Declaring local variables
    zip_name = glob.glob(f"*.zip")[0]

    unzipped_dir = "DontSleep"


    # Installing software
    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir) and force:
        remove_tree(app_dir)
    mkdirs(app_dir)
    print("Extracting: %s to: %s" % (zip_name, unzipped_dir))
    unzip(zip_name, unzipped_dir)
    print('Copy DontSleep to %s' % app_dir)
    copytree2(unzipped_dir, app_dir, onreplace=default_overwrite)

    # Creating custom shortcuts
    create_desktop_shortcut(app_name, target=app_path)
    create_programs_menu_shortcut(app_name, target=app_path)


def audit():
    # Auditing software
    audit_status = "OK"
    installed_version = get_installed_version(app_path)
    if Version(installed_version) < Version(control.get_software_version()) and audit_version:
        print("%s is installed in version (%s) instead of (%s)" % (app_name, installed_version, control.get_software_version()))
        audit_status = "WARNING"
    elif isdir(app_dir) and not dir_is_empty(app_dir):
        print("%s (%s) is installed" % (app_name, installed_version))
        audit_status = "OK"
    else:
        print("%s is not installed" % app_name)
        audit_status = "ERROR"
    return audit_status


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

    # Removing shortcuts
    remove_desktop_shortcut(app_name)
    remove_programs_menu_shortcut(app_name)
                                

# -*- coding: utf-8 -*-
from setuphelpers import *
from setupdevhelpers import *
import zipfile
import os
import requests

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://www.softwareok.com/Download/DontSleep_x64.zip"
    }

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

    # Téléchargement avec User-Agent
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/115.0.0.0 Safari/537.36"
    }
    response = requests.get(download_url, headers=headers, proxies=proxies, allow_redirects=True)
    with open(latest_bin, 'wb') as f:
        f.write(response.content)

    # Vérifiez si le fichier est une page HTML
    with open(latest_bin, 'r', encoding='utf-8', errors='ignore') as f:
        content = f.read(1024)
        if "<html" in content.lower():
            print(f"Error: {latest_bin} appears to be an HTML file, not a ZIP archive.")
            print("Check the download URL or try accessing it manually.")
            return False

    # Vérification si c'est un ZIP valide
    if not zipfile.is_zipfile(latest_bin):
        print(f"Downloaded file {latest_bin} is not a valid ZIP file.")
        return False

    # Décompression
    print(f"Extracting: {latest_bin}")
    with zipfile.ZipFile(latest_bin, 'r') as zip_ref:
        zip_ref.extractall()

    # Recherche de l'exécutable .exe
    exe_file = None
    for file in os.listdir():
        if file.endswith(".exe"):
            exe_file = file
            print(f"Found executable: {exe_file}")
            break

    if not exe_file:
        print("Error: No executable found in the ZIP archive.")
        return False

    # Récupération de la version
    version = get_version_from_binary(exe_file)

    # Mise à jour de la version du 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)})")

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