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

uninstallkey = []

# Declaring specific app values (TO CHANGE)
bin_name_string = 'ccsetup%s_slim.exe'
silent_inst_arg = '/S'
uninstall_key = 'CCleaner'
app_path = makepath(programfiles,'CCleaner')
conf_file_path = makepath(programfiles,'CCleaner','ccleaner.ini')


def install():
    # Specific app values
    package_version = control.version.split('-',1)[0]
    package_version_split = package_version.split('.')
    short_package_version = '%s.%s' % (package_version_split[0],package_version_split[1])
    version = short_package_version.replace('.','')
    app_name = control.name
    app_publisher = control.editor

    # Uninstalling older versions
    check_installed = installed_softwares(app_name)
    #print(check_installed)
    for uninstall in check_installed:
    	if app_publisher in uninstall['publisher']:
    		if uninstall['version'] < package_version:
    			print('Removing %s' % uninstall['name'])
    			cmd = uninstall_cmd(uninstall['key'])
    			run(cmd)

	# Getting the used storage on programfiles before installation (place it on the top)
    get_disk_free_space_before = get_disk_free_space(programfiles)

    # Installing the package
    install_exe_if_needed(bin_name_string % version
        ,silentflags=silent_inst_arg
        ,key=uninstall_key
        ,min_version=short_package_version)

    # Adding configuration file
    if isfile(conf_file_path):
        print('CCleaner configuration file detected and kept')
    else:
        if isdir(app_path):
            print('CCleaner configuration file not detected')
            filecopyto('ccleaner.ini',app_path)
            print('CCleaner configuration file with minimal intrusion added')

	# Return used storage of the installation. (place it on the bottom)
    get_disk_free_space_after = get_disk_free_space(programfiles)
    free_space_after_diff = get_disk_free_space_before - get_disk_free_space_after
    print ("Storage used: " + str(free_space_after_diff))


def session_setup():
    print('Disabling: update check, telemetry, monitoring, publicity')

    registry_set(HKEY_CURRENT_USER,'software\\Piriform\\CCleaner', 'Monitoring','0')
    registry_set(HKEY_CURRENT_USER,'software\\Piriform\\CCleaner', 'SystemMonitoring','0')
    registry_set(HKEY_CURRENT_USER,'software\\Piriform\\CCleaner', 'UpdateCheck','0')
    registry_set(HKEY_CURRENT_USER,'software\\Piriform\\CCleaner', 'RunICS','0')
    registry_set(HKEY_CURRENT_USER,'software\\Piriform\\CCleaner', 'AutoICS','0')
    registry_set(HKEY_CURRENT_USER,'software\\Piriform\\CCleaner', 'CheckTrialOffer','0')


def update_package():
    print('Update package content from upstream binary sources')

    import requests,re
    from waptpackage import PackageEntry

    # Update transition from BeautifulSoup 3 to 4
    try:
        import BeautifulSoup
    except:
        import bs4 as BeautifulSoup

    # Specific app values
    app_name = control.name
    url = control.sources

    # Getting proxy informations from WAPT settings
    proxy = {}
    if isfile(makepath(application_data(),'waptconsole','waptconsole.ini')):
        proxywapt = inifile_readstring(makepath(user_local_appdata(),'waptconsole','waptconsole.ini'),'global','http_proxy')
        if proxywapt :
            proxy = {'http':proxywapt,'https':proxywapt}

    # Get latest version number from official website
    page = requests.get(url,headers={'User-Agent':'Mozilla/5.0 (Windows NT 6.1; Win64; x64)'}).text
    bs = BeautifulSoup.BeautifulSoup(page)
    url_dl = bs.find('p', attrs={"class": u"fs-13"}).a["href"]
    latest_bin = url_dl.rsplit('/',1)[1]

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

		# Change version of the package
        version = get_file_properties(latest_bin)['ProductVersion']
        print('Writing ' + version + '-0 in WAPT\\control')
        pe = PackageEntry()
        pe.load_control_from_wapt(os.getcwd())

        pe.version = version + '-0'
        pe.save_control_to_wapt(os.getcwd())
        print('Update package done. You can now build-upload your package')
    else:
        print('This package is already up-to-date')

    # Deleting outdated binaries
    for bin_in_dir in glob.glob('*.exe') or glob.glob('*.msi'):
        if bin_in_dir != latest_bin :
            print('Outdated binary: ' + bin_in_dir + ' Deleted')
            remove_file(bin_in_dir)


if __name__ == '__main__':
    update_package()