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


def install():
    # Define the installation file name
    bin_name = glob.glob('TheBrain*Installer.exe')[0]
    
    # Install the software
    install_exe_if_needed(bin_name,
                          silentflags='-q -overwrite',
                          key='',
                          min_version=control.get_software_version())


def uninstall():
    # Retrieve installed software with "TheBrain" in the name
    installed_thebrain_versions = [soft for soft in installed_softwares() if "TheBrain" in soft['name']]
    
    # Loop through installed versions to initiate uninstallation
    for soft in installed_thebrain_versions:
        key = soft["key"]
        uninstall_string = soft["uninstall_string"]
        
        if uninstall_string:
            print(f"Uninstalling TheBrain with key {key}")
            run(uninstall_string + " /quiet /norestart")
        else:
            print(f"Key {key} has no uninstall string. Nothing to uninstall for this key.")
                                

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




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

  
    url_base = "https://www.thebrain.com/download"
    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","is-win", proxies=proxies)
    zip_file = None
    for div in divs:
        if zip_file:
            break
        links = div.find_all('a', href=True)
        for link in links:
            if 'Download for Windows' in link.text :
                href = link['href']
                zip_file = href
                download_url = requests.get(zip_file,allow_redirects=True, proxies=proxies).url
                latest_bin = download_url.split('/')[-1].replace("%"," ")
                version = latest_bin.split(" ")[-2]
                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('*.exe'):
        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()