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

app_install_path = "/usr/lib/virtualbox/"
app_vbm = makepath(app_install_path, "VBoxManage")


def install():
    bin_name = glob.glob("virtualbox-*.deb")[0]

    
    
    # Avoid VMs to be shutted unexpectedly
    if isrunning("virtualbox") and not force:
        error("ERROR: VirtualBox is running")

    uninstall()
    
    update_apt()
    install_apt("linux-headers-generic")
    install_deb(bin_name)

    # 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")
    run('"%s" setextradata global GUI/UpdateDate never' % app_vbm)


def uninstall():
    for p in control.impacted_process.split(',') :
        killalltasks(p)
    uninstall_apt("'virtualbox*'")
    autoremove_apt()
    if not installed_softwares("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/Linux_Downloads"
    distri = control.target_os.split("-")[1].lower()

    # Downloading latest binaries
    print("URL used is: %s" % url)
    for bs_search in bs_find_all(url, "a", "class", "ext-link", proxies=proxies):
        if ".deb" in str(bs_search["href"]) and distri in str(bs_search["href"]).lower():
            url_dl = bs_search["href"]
            version = url_dl.split("/")[-2]
            latest_bin = url_dl.split("/")[-1]

    for bs_search in bs_find_all("https://www.virtualbox.org/wiki/Downloads", "a", "class", "ext-link", proxies=proxies):
        if bs_search["href"].split("/")[-1].endswith(".vbox-extpack"):
            url_dl_ext = bs_search["href"]
            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.replace("~", "_"), 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()

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

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

    # Validating or not update-package-sources
    return package_updated