# -*- coding: utf-8 -*-
from setuphelpers import *
import glob

r"""
Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()

"""

def install():
    # Déclaration des variables locales
    try:
        bin_name = glob.glob('Vivaldi.*.exe')[0]
    except IndexError:
        raise Exception("No Vivaldi installer found in the current directory")

    # Installation du logiciel
    install_exe_if_needed(bin_name,
                          silentflags='--vivaldi-silent --do-not-launch-chrome --system-level',
                          key='Vivaldi',
                          min_version=control.get_software_version(),
                          accept_returncodes=[0, 3010, 1641])
    
    # Nettoyer les clés de désinstallation pour éviter les audits sans code
    uninstallkey.clear()

def uninstall():
    for uninstall in installed_softwares('vivaldi'):
        uninstall_string = uninstall.get("uninstall_string", "")
        if uninstall_string:
            print(f"Uninstalling Vivaldi: {uninstall_string}")
            try:
                # Exécution de la commande de désinstallation avec gestion des erreurs
                run(f"{uninstall_string} --uninstall --force-uninstall --vivaldi --system-level", accept_returncodes=[0, 19])
                wait_uninstallkey_absent(uninstall['key'], max_loop=600)
            except CalledProcessErrorOutput as e:
                print(f"Error uninstalling Vivaldi: {e.output}")
                if 'Could not get application shortcuts location' in e.output:
                    print("Ignoring error related to application shortcuts location.")
                else:
                    raise
        else:
            print("No uninstall string found for Vivaldi")

def audit():
    if not uninstall_key_exists("Vivaldi"):
        print('uninstallkey Vivaldi not found')
        return "ERROR"
    else:
        return "OK"

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
from setupdevhelpers import *
import glob
import re



def update_package():
    # Declaring local variables
    package_updated = False
    
    proxies = get_proxies_from_wapt_console()
    if not proxies:
        proxies = get_proxies()

  
    url_base = "https://vivaldi.com/download/?platform=windows"
    response = requests.get(url_base,allow_redirects=True, proxies=proxies)
    # Extract the correct div using bs_find_all
    divs = bs_find_all(response.text, "div","class","download-menu", proxies=proxies)
    exe_file = None
    for div in divs:
        if exe_file:
            break
        links = div.find_all('a', href=True)
        for link in links:
            if link['href'].endswith('.exe'):
                href = link['href']
                exe_file = href
                download_url = exe_file
                latest_bin = exe_file.split('/')[-1]
                # Extraction de la version à partir du nom de fichier latest_bin
                pattern = r"Vivaldi\.(\d+\.\d+\.\d+\.\d+)\.x64\.exe"
                match = re.search(pattern, latest_bin)
                version = match.group(1)
                break


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


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

        
    version = get_version_from_binary(latest_bin)
    # Mettre à jour le package
    control.set_software_version(version)
    control.save_control_to_wapt()