# -*- coding: utf-8 -*-
from setuphelpers import *
try:
    import bs4 as BeautifulSoup
except:
    import BeautifulSoup
import platform
import requests

# Defining variables
bin_contains = 'virtualbox'
bin_ext_name_string = 'Oracle_VM_VirtualBox_Extension_Pack-%s.vbox-extpack'
bin_ext_name_joker = '*.vbox-extpack'

def install():
    bin_name = glob.glob('*%s*.deb' % bin_contains)[0]

    # Installing the package
    print('Installing: %s' % bin_name)
    install_deb(bin_name)
    

    #Installing the VirtualBox Extension Pack
    for ext in glob.glob(bin_ext_name_joker):
        print('Installing: %s' % ext)
        try:
            run('echo y | VBoxManage extpack install --replace "%s"' % ext)
        except:
            run('VBoxManage extpack cleanup ')
            run('echo y | VBoxManage extpack install --replace "%s"' % ext)

def uninstall():
    package_name = 'virtualbox'

    # Uninstalling the package 
    print('Uninstalling: %s' % package_name)
    uninstall_apt(package_name)


def update_package():
    # Getting proxy informations from WAPT settings
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()

    # Specific app values
    url = 'https://www.virtualbox.org/wiki/Linux_Downloads'

    # Getting latest version from official website
    
    for u in bs_find_all(url,'a','class','ext-link', proxies=proxies):
        if '.deb' in str(u['href']) and 'buster' in str(u['href']):
            url_dl = u['href']
            latest_bin = url_dl.split('/')[-1]
            version = url_dl.split('/')[-2]
            latest_bin_ext = bin_ext_name_string % version
            url_dl_ext = url_dl.replace(latest_bin,latest_bin_ext)
          

    print('Latest Version %s' % version)

    for f in glob.glob('*.deb'):
        if f != latest_bin:
            remove_file(f)

    for f in glob.glob(bin_ext_name_joker):
        if f != latest_bin_ext:
            remove_file(f)

    # Downloading latest binaries
    if not isfile(latest_bin):
        print('Downloading: ' + latest_bin)
        wget(url_dl,latest_bin,proxies=proxies)

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

    # Changing version of the package
    control.version = '%s-%s' % (version, control.version.split('-', 1)[-1])
    control.save_control_to_wapt()
    print("Changing package version to: %s in WAPT\\control" % control.version)


def bs_find_all(url, element, attribute=None, value=None, headers=None, proxies=None, features='html.parser', **kwargs):
    """"You may need to use a header for some websites. For example: headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0'}
    """
    page = requests.get(url, proxies=proxies, headers=headers, **kwargs).text
    try:
        import bs4 as BeautifulSoup
        soup = BeautifulSoup.BeautifulSoup(page, features=features)
    except:
        import BeautifulSoup
        soup = BeautifulSoup.BeautifulSoup(page)
    if value:
        return soup.findAll(element,{attribute:value})
    else:
        return soup.findAll(element)


def get_proxies_from_wapt_console():
    proxies = {}
    if platform.system() == 'Windows':
        waptconsole_ini_path = makepath(user_local_appdata(), 'waptconsole', 'waptconsole.ini')
    else:
        waptconsole_ini_path = makepath(user_home_directory(), '.config', 'waptconsole', 'waptconsole.ini')
    if isfile(waptconsole_ini_path):
        proxy_wapt = inifile_readstring(waptconsole_ini_path, 'global', 'http_proxy')
        if proxy_wapt:
            proxies = {'http': proxy_wapt, 'https': proxy_wapt}
    return proxies


def get_proxies():
    if platform.python_version_tuple()[0] == '3':
        from urllib.request import getproxies
    else:
        from urllib import getproxies
    return getproxies()