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


def install():
    # Declaring local variables
    app_uninstallstring = makepath(programfiles32, "Bitvise SSH Client", "uninst.exe")
    app_uninstallkey = "BvSshClient"

    # Installing the software
    print("Installing: BvSshClient-Inst.exe")
    install_exe_if_needed('BvSshClient-Inst.exe',
        silentflags='/acceptEULA',
        key='BvSshClient',
        min_version=control.get_software_version(),
    )
    #Adding QuietUninstallString
    quiet_uninstall_string = rf'"{app_uninstallstring}" -unat BvSshClient'
    register_uninstall(app_uninstallkey, quiet_uninstall_string=quiet_uninstall_string)

    uninstallkey.clear()


def uninstall():
    # Force uninstalling the software
    for to_uninstall in installed_softwares(name="Bitvise SSH Client"):
        print("Removing: %s (%s)" % (to_uninstall["name"], to_uninstall["version"]))
        killalltasks(ensure_list(control.impacted_process))

        try:
            # Adding silent flags to the uninstall command
            uninstall_command = uninstall_cmd(to_uninstall["key"])
            print(uninstall_command)
            run(uninstall_command)
            wait_uninstallkey_absent(to_uninstall["key"], max_loop=60)
        except Exception as e:
            print(f"Error during uninstallation: {e}")
            # Handling the case where Bitvise SSH Clien Director might already be partially uninstalled
            print("Bitvise SSH Clien Director might have already been uninstalled.")
            print("Removing Bitvise SSH Clien Director from the list of installed programs.")

        if uninstall_key_exists(to_uninstall["key"]):
            unregister_uninstall(to_uninstall["key"], win64app=to_uninstall["win64"])

        # Clean up any remaining files
        if isdir(to_uninstall["install_location"]):
            print(f"Removing remaining files in {to_uninstall['install_location']}")
            remove_tree(to_uninstall["install_location"])


                                

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


def update_package():
    # Declaring local variables
    package_updated = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    url = control.sources
    app_name = control.name
    download_url = "https://dl.bitvise.com/BvSshClient-Inst.exe"
    latest_bin = download_url.split("/")[-1]

    # Getting latest version from official sources
    print("URL used is: %s" % url)
    for bs_search in bs_find_all(
        url,
        "p",
        user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:91.0) Gecko/20100101 Firefox/91.0",
        proxies=proxies,
    ):
        if "Current version:" in bs_search.text:
            version = bs_search.text.split(": ")[1].split(",")[0]
            break

    # Downloading latest binaries
    print("Latest %s version is: %s" % (app_name, version))
    print("Download URL is: %s" % download_url)
    if Version(version) > Version(get_version_from_binary(latest_bin)):
        print("Downloading: %s" % latest_bin)
        wget(download_url, latest_bin, proxies=proxies)
    else:
        print("Binary is present: %s" % latest_bin)

    # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        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()

    # Validating or not update-package-sources
    return package_updated