# -*- 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 = "Letsignit"
bin_name = glob.glob("*.exe")[0]
app_dir = makepath(programfiles32, app_name)
app_path = makepath(app_dir, bin_name)


def install():
    # Installing software
    killalltasks(ensure_list(control.impacted_process))
    if isdir(app_dir) and force:
        remove_tree(app_dir)
    mkdirs(app_dir)
    filecopyto(bin_name, app_path)

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

def audit():
    return "OK"

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 glob


def update_package():
    # Declaration of local variables
    package_updated = False
    proxies = get_proxies_from_wapt_console()
    if not proxies:
        proxies = get_proxies()
    url = "https://cloud.letsignit.com/agent/download"

    extensions_file = ['.dmg', '.exe', '.deb', '.rpm', '.AppImage', '.appimage', '.msi', '.zip']
    # Deleting outdated binaries
    for bin in extensions_file:
        for file in glob.glob(f"*{bin}"):
            remove_file(file)
    # Get filename
    last_bin = get_filename_from_url(url)
    # Get version
    version = last_bin.split(" ")[-1].split(".exe")[0]

    if Version(version) >= Version(control.get_software_version()):
        # Get download url
        download_url = get_download_url_from_url(url)
        wget(download_url, last_bin, proxies=proxies)
        print(
            "Software version updated (from: %s to: %s)"
            % (control.get_software_version(), Version(version))
        )
        package_updated = True
    else:
        print("Software version up-to-date (%s)" % Version(version))

    control.set_software_version(version)
    control.save_control_to_wapt()

    return package_updated


def get_filename_from_url(url):
    import requests
    from urllib.parse import unquote

    # Define header

    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:95.0) Gecko/20100101 Firefox/95.0"
    }

    # Send a HEAD request to get headers
    response = requests.head(url, headers=headers, allow_redirects=True)

    if response.status_code == 200:
        filename = unquote(response.url.split("agent-win/")[-1])
    else:
        print("Error:", response.status_code)

    return filename


def get_download_url_from_url(url):
    import requests
    from urllib.parse import unquote

    # Define header

    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:95.0) Gecko/20100101 Firefox/95.0"
    }

    # Send a HEAD request to get headers
    response = requests.head(url, headers=headers, allow_redirects=True)

    if response.status_code == 200:
        download_url = unquote(response.url)
    else:
        print("Error:", response.status_code)

    return download_url