# -*- coding: utf-8 -*-
from setuphelpers import *
import platform
import xml.etree.ElementTree as ET
import bs4 as BeautifulSoup
import requests

uninstallkey = []

# Defining variables
bin_name_string = 'TreeSizeFreeSetup-%s.exe'
silent_args = '/VERYSILENT /SUPPRESSMSGBOXES /NORESTART'
app_uninstallkey = 'TreeSize Free_is1'


def install():
    # Initializing variables
    package_version = control.version.split('-')[0]
    bin_name = bin_name_string % package_version

    # Installing the package
    install_exe_if_needed(bin_name
        ,silentflags=silent_args
        ,key=app_uninstallkey
        ,min_version=package_version)


def session_setup():
    print('Disabling: update check, context menu')

    # Initializing variables
    #C:\Users\username\AppData\Roaming\JAM Software\TreeSize Free\GlobalOptions.xml
    user_conf_dir = makepath(application_data,'JAM Software','TreeSize Free')
    user_conf_file = makepath(user_conf_dir,'GlobalOptions.xml')

    user_conf_content = """<?xml version="1.0" encoding="UTF-8"?>
<GlobalOptions Version="3" AppVersion="4.41" Type="TGlobalOptions">
	<ContextMenuEntries>4</ContextMenuEntries>
	<CheckForUpdateOptions Type="TCheckForUpdateOptions">
		<LatestOnlineVersion>-1</LatestOnlineVersion>
		<LastAutomaticCheckForUpdate>-1</LastAutomaticCheckForUpdate>
		<CheckForUpdatesAutomatically>False</CheckForUpdatesAutomatically>
	</CheckForUpdateOptions>
</GlobalOptions>
"""

    if not isdir(user_conf_dir):
        mkdirs(user_conf_dir)
    if isfile(user_conf_file):
        user_conf_tree = ET.parse(user_conf_file)
        user_conf_content = user_conf_tree.getroot()

        for conf in user_conf_content.iter('CheckForUpdateOptions'):
            if conf.get('Type') == 'TCheckForUpdateOptions':
                for children in conf.getchildren():
                    if children.tag == 'CheckForUpdatesAutomatically':
                        children.text='False'
                        user_conf_tree.write(user_conf_file,encoding='utf-8')

        for conf in user_conf_content.iter('ContextMenuEntries'):
            conf.text='4'
            user_conf_tree.write(user_conf_file,encoding='utf-8')
    else:
        file_open = open(user_conf_file,'w')
        file_open.write(user_conf_content)
        file_open.close()


def update_package():
    # Initializing variables
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    app_name = control.name
    url = 'https://www.jam-software.com/treesize_free/changes.shtml'
    url_dl = 'https://downloads.jam-software.de/treesize_free/TreeSizeFreeSetup.exe'

    # Getting latest version from official sources
    print('URL used is: %s' % url)
    version = bs_find(url, 'h3', 'class', 'collapsed-item__ttl', proxies=proxies).text.replace('Version ','')
    latest_bin = bin_name_string % version

    print("Latest %s version is: %s" % (app_name, version))
    print("Download url is: %s" % url_dl)

    # 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 version != version_from_file:
            os.rename(latest_bin,bin_name_string % version_from_file)
            version = version_from_file

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

    # Deleting outdated binaries
    remove_outdated_binaries(version)




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


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_version_from_binary(filename):
    if filename.endswith('.msi'):
        return get_msi_properties(filename)['ProductVersion']
    else:
        return get_file_properties(filename)['ProductVersion']


def remove_outdated_binaries(version, list_extensions=['exe','msi','deb','rpm','dmg','pkg'], list_filename_contain=None):
    if type(list_extensions) != list:
        list_extensions = [list_extensions]
    if list_filename_contain:
        if type(list_filename_contain) != list:
            list_filename_contain = [list_filename_contain]
    list_extensions = ['.' + ext for ext in list_extensions if ext[0] != '.']
    for file_ext in list_extensions:
        for bin_in_dir in glob.glob('*%s' % file_ext):
            if not version in bin_in_dir:
                remove_file(bin_in_dir)
            if list_filename_contain:
                for filename_contain in list_filename_contain:
                    if not filename_contain in bin_in_dir:
                        remove_file(bin_in_dir)


def bs_find(url, element, attribute=None, value=None, user_agent=None, proxies=None, features='html.parser', **kwargs):
    """"You may need to use a user agent for some websites.
    Example: user_agent='Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:78.0) Gecko/20100101 Firefox/78.0')
    """
    if user_agent:
        page = requests.get(url, proxies=proxies, headers={'User-Agent':'%s' % user_agent}, **kwargs).text
    else:
        page = requests.get(url, proxies=proxies, **kwargs).text
    soup = BeautifulSoup.BeautifulSoup(page, features=features)
    if value:
        return soup.find(element, {attribute: value})
    else:
        return soup.find(element)