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

app_install_path = makepath(programfiles, "Oracle", "VirtualBox")
app_vbm = makepath(app_install_path, "VBoxManage.exe")


def install():
    # Avoid VMs to be shutted unexpectedly
    if isrunning("virtualbox") and not force:
        error("ERROR: VirtualBox is running")
    else:
        # Uninstalling the software if install forced
        for to_uninstall in installed_softwares("VirtualBox"):
            print("Removing: %s" % to_uninstall["name"])
            killalltasks(control.get_impacted_process_list())
            run(uninstall_cmd(to_uninstall["key"]))
            wait_uninstallkey_absent(to_uninstall["key"])

    # Adding certs
    try:
        for crt in glob.glob("oracle*.cer"):
            run(r"C:\Windows\System32\certutil.exe -addstore TrustedPublisher %s" % crt)
    except:
        for crt in glob.glob("oracle*.cer"):
            run(r"C:\Windows\System32\certutil.exe -addstore -f TrustedPublisher %s" % crt)
    # Installing the package
    bin_name = glob.glob("VirtualBox-*.exe")[0]
    install_exe_if_needed(
        bin_name,
        silentflags="--silent --msiparams REBOOT=ReallySuppress ALLUSERS=2 VBOX_INSTALLDESKTOPSHORTCUT=0 VBOX_START=0 VBOX_INSTALLQUICKLAUNCHSHORTCUT=0",
        name="Oracle VirtualBox " + control.get_software_version(),
        min_version=control.get_software_version(),
    )

    # Remove all file extpack
    for file in glob.glob(makepath(app_install_path, "Oracle_VM_VirtualBox_Extension_Pack-*.vbox-extpack")):
        print("Removing: %s" % file)
        remove_file(file)
    # Copy new extpack
    for ext in glob.glob("Oracle_VM_VirtualBox_Extension_Pack-*.vbox-extpack"):
        print("Copying: %s to: %s" % (ext, app_install_path))
        filecopyto(ext, app_install_path)
        # Install new extpack
        try:
            print(
                run(
                    'echo y | "%s" extpack install --replace "%s" --accept-license=33d7284dc4a0ece381196fda3cfe2ed0e1e8e7ed7f27b9a9ebc4ee22e24bd23c'
                    % (app_vbm, ext)
                )
            )
        except:
            print(run('"%s" extpack cleanup' % (app_vbm)))
            print(
                run(
                    'echo y | "%s" extpack install --replace "%s" --accept-license=33d7284dc4a0ece381196fda3cfe2ed0e1e8e7ed7f27b9a9ebc4ee22e24bd23c'
                    % (app_vbm, ext)
                )
            )


def session_setup():
    print("Disabling: Auto update checks")
    try:
        run('"%s" setextradata global GUI/UpdateDate never' % app_vbm)
    except:
        pass


def uninstall():
    if not installed_softwares("Oracle VM VirtualBox"):
        if isdir(app_install_path):
            remove_tree(app_install_path)

                                

# -*- 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()
    app_name = control.name
    url = "https://www.virtualbox.org/wiki/Downloads"

    # Downloading latest binaries
    print("URL used is: %s" % url)
    for bs_search in bs_find_all(url, "a", "class", "ext-link", proxies=proxies):
        if bs_search["href"].split("/")[-1].endswith("-Win.exe"):
            url_dl = bs_search["href"]
            version = url_dl.split("/")[-2]
            latest_bin = url_dl.split("/")[-1]
    url_dl_ext = "https://download.virtualbox.org/virtualbox/%s/Oracle_VirtualBox_Extension_Pack-%s.vbox-extpack" % (version,version)
    latest_bin_ext = url_dl_ext.split('/')[-1]

    print("Latest %s version is: %s" % (app_name, version))
    print("Download URL is: %s" % url_dl)
    print("Download Extension Pack URL is: %s" % url_dl_ext)

    # Downloading latest binaries
    if not isfile(latest_bin):
        print("Downloading: %s" % latest_bin)
        wget(url_dl, latest_bin, proxies=proxies)

        # Checking version from file
        version_from_file = get_version_from_binary(latest_bin)
        if not version_from_file.startswith(version) and version_from_file != "":
            print("Changing version to the version number of the binary")
            os.rename(latest_bin, latest_bin.replace(version, version_from_file))
            version = version_from_file
        else:
            print("Binary file version corresponds to online version")

    # 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()

    # Downloading extension
    if not isfile(latest_bin_ext):
        print("Downloading: %s" % latest_bin_ext)
        wget(url_dl_ext, latest_bin_ext, proxies=proxies)

    # Deleting outdated binaries
    remove_outdated_binaries(version, list_extensions=["exe", "vbox-extpack"])

    # Validating or not update-package-sources
    return package_updated