# -*- coding: utf-8 -*-
from setuphelpers import *
import glob
import requests
import shutil

try:
    import bs4 as BeautifulSoup
except:
    import BeautifulSoup

uninstallkey = []

def install():
    versionpaquet=get_file_properties("TeamViewer_Setup.exe")['ProductVersion'].strip('.0')
    print('Installing Teamviewer {}'.format(versionpaquet))
    install_exe_if_needed("TeamViewer_Setup.exe", '/S',key='TeamViewer', min_version=versionpaquet)

def update_package():
    proxies = {}
    if isfile(makepath(application_data(),'waptconsole','waptconsole.ini')):
        proxywapt = inifile_readstring(makepath(user_local_appdata(),'waptconsole','waptconsole.ini'),'global','http_proxy')
        if proxywapt :
            proxies = {'http':proxywapt,'https':proxywapt}

    os.chdir(os.path.dirname(__file__))

    for fteam in glob.glob("*.exe"):
        print('remove {}'.format(fteam))
        remove_file(fteam)

    page = wgets('https://www.teamviewer.com/en-us/download/windows/',proxies=proxies)
    for line in page.splitlines() :
        if 'TeamViewer_Setup.exe' in line :
            exe = line[line.find('https'):line.find('.exe')]+'.exe'
            print("Downloading {}".format(exe))
            wget(exe,'TeamViewer_Setup.exe',proxies=proxies)
            break

    for elem in bs_find_all('https://www.teamviewer.com/en-us/download/windows/','div','class','wpb_wrapper',proxies=proxies):
        if 'Changelog' in elem.text:
            version = elem.text.split('\n')[1]

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


def bs_find_all(url, element, attribute=None, value=None, headers={'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0'}, 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
    soup = BeautifulSoup.BeautifulSoup(page, features=features)
    if value:
        return soup.find_all(element, {attribute:value})
    else:
        return soup.find_all(element)