# -*- coding: utf-8 -*-
from setuphelpers import *
from setupdevhelpers import *
from xml.etree.ElementTree import parse


def install():
    bin_name = glob.glob("TreeSizeFreeSetup*.exe")[0]
    app_uninstallkey = "TreeSize Free_is1"

    install_exe_if_needed(
        bin_name,
        silentflags="/VERYSILENT /SUPPRESSMSGBOXES /NORESTART /SP-",
        key=app_uninstallkey,
        min_version=control.get_software_version(),
    )

    # Adding QuietUninstallString
    quiet_uninstall_string = installed_softwares(uninstallkey=app_uninstallkey)[0]["uninstall_string"] + " /VERYSILENT /SUPPRESSMSGBOXES /NORESTART"
    register_uninstall(app_uninstallkey, quiet_uninstall_string=quiet_uninstall_string)


def session_cleanup():
    print("Removing the TreeSize shortcut on user appdata")
    remove_user_programs_menu_shortcut('TreeSizeFree.exe')


def session_setup():
    print("Disabling: update check, pop-ups and context menu once")
    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="0" Type="TGlobalOptions">
	<ContextMenuEntries>4</ContextMenuEntries>
	<AskForRestartAsAdministrator>False</AskForRestartAsAdministrator>
	<DoNotAskAgainList Type="TSerializableStringIntDictionary"></DoNotAskAgainList>
	<CheckForUpdateOptions Type="TCheckForUpdateOptions">
		<LatestOnlineVersion>0</LatestOnlineVersion>
		<LastAutomaticCheckForUpdate>0</LastAutomaticCheckForUpdate>
		<CheckForUpdatesAutomatically>False</CheckForUpdatesAutomatically>
	</CheckForUpdateOptions>
</GlobalOptions>
"""
    if not isdir(user_conf_dir):
        mkdirs(user_conf_dir)

    if isfile(user_conf_file) and os.path.getsize(user_conf_file) != 0:
        tree = parse(user_conf_file)
        root = tree.getroot()

        # Modify ContextMenuEntries
        context_menu_entries = root.find('ContextMenuEntries')
        if context_menu_entries is not None:
            context_menu_entries.text = '4'

        # Modify AskForRestartAsAdministrator
        ask_restart = root.find('AskForRestartAsAdministrator')
        if ask_restart is not None:
            ask_restart.text = 'False'

        # Modify CheckForUpdateOptions
        update_options = root.find('CheckForUpdateOptions')
        if update_options is not None:
            latest_version = update_options.find('LatestOnlineVersion')
            latest_version.text = '0'

            last_check = update_options.find('LastAutomaticCheckForUpdate')
            last_check.text = '0'

            auto_check = update_options.find('CheckForUpdatesAutomatically')
            auto_check.text = 'False'

        tree.write(user_conf_file, encoding='utf-8', xml_declaration=True)
    else:
        with open(user_conf_file, 'w') as xml_file:
            xml_file.write(user_conf_content)
                                

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


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

    download_url = "https://downloads.jam-software.de/treesize_free/TreeSizeFreeSetup.exe"
    latest_bin = download_url.rsplit("/", 1)[-1]
    latest_bin_extension = latest_bin.rsplit(".", 1)[-1]

    print(f"Download URL is: {download_url}")
    if not isfile(latest_bin):
        print(f"Downloading: {latest_bin}")
        wget(download_url, latest_bin, proxies=proxies)
    else:
        print(f"Binary is present: {latest_bin}")

    version = get_version_from_binary(latest_bin)
    print(f"Latest {app_name} version is: {version}")

    expected_issuer = "JAM Software GmbH"
    sign_name = waptlicences.check_msi_signature(latest_bin)[0]
    if sign_name != expected_issuer:
        error(f'Bad issuer {sign_name} != {expected_issuer} ')

    # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        print(f"Software version updated (from: {control.get_software_version()} to: {Version(version)})")
        package_updated = True
    else:
        print(f"Software version up-to-date ({Version(version)})")

    for f in glob.glob(f'*.{latest_bin_extension}'):
        if f != latest_bin:
            remove_file(f)

    control.set_software_version(version)
    control.save_control_to_wapt()

    return package_updated