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

# Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "QGIS-OSGeo4W-"


def install():
    # Declaring local variables
    package_version = control.get_software_version()
    bin_name = glob.glob("*%s*.msi" % bin_contains)[0]
    short_package_version = ".".join(package_version.split(".")[:2])
    app_uninstallkey = "QGIS %s" % short_package_version
    app_name = "QGIS"
    processes_to_kill = control.impacted_process.split(",")

    # Uninstalling all other versions of the software before installation
    for uninstall in installed_softwares(app_name):
        if uninstall["version"] != package_version:
            print("Removing: %s" % uninstall["name"])
            killalltasks(processes_to_kill)
            try:
                run(uninstall_cmd(uninstall["key"]), timeout=1200)
            except:
                print("Unable to uninstall %s" % uninstall["name"])
            folder_path = makepath("\\".join(uninstall["uninstall_string"].split("\\")[:3]))
            wait_uninstallkey_absent(uninstall["key"])
            time.sleep(60)
            if isdir(folder_path):
                remove_tree(folder_path)

    # Installing the software
    print("Installing: %s" % bin_name)
    install_msi_if_needed(bin_name, timeout=1200)

    # Removing desktop shortcuts
    shortcuts_folder = makepath(systemdrive, "Users", "Public", "Desktop", app_uninstallkey)
    if isdir(shortcuts_folder):
        remove_tree(shortcuts_folder)


def uninstall():
    # Specific app values
    package_version = control.version.split("-")[0]
    short_package_version = ".".join(package_version.split(".")[:2])
    app_uninstallkey = "QGIS %s" % short_package_version

    # Removing the remaining folder of the app in %programfiles%
    folder_name = app_uninstallkey
    appfolder_path = makepath(programfiles, folder_name)
    time.sleep(60)
    if isdir(appfolder_path):
        remove_tree(appfolder_path)


def session_setup():
    print("Disabling: auto update check")

    # AppData\Roaming\QGIS\QGIS3\profiles\default\QGIS\QGIS3.ini
    r"""[qgis]
    checkVersion=true
    [core]
    NewsFeed\httpsfeedqgisorg\disabled=false"""

    user_conf_folder = makepath(user_appdata, "QGIS", "QGIS3", "profiles", "default", "QGIS")
    user_conf_file = makepath(user_conf_folder, "QGIS3.ini")

    if not isdir(user_conf_folder):
        mkdirs(user_conf_folder)

    inifile_writestring(user_conf_file, "qgis", r"checkVersion", "false")
    inifile_writestring(user_conf_file, "core", r"NewsFeed\httpsfeedqgisorg\disabled", "true")

                                

# -*- coding: utf-8 -*-
from setuphelpers import *
import bs4 as BeautifulSoup

# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "QGIS-OSGeo4W-"


def update_package():
    # Declaring local variables
    result = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    app_name = control.name
    url = "https://www.qgis.org/en/site/forusers/download.html"
    html_source = wgets("https://www.qgis.org/en/site/forusers/download.html", proxies=proxies).splitlines()
    for line in html_source:
        if "The long-term repositories currently offer" in line:
            version = line.split(" ")[-2]
            print(version)
    for line in html_source:
        if ".msi" in line and version in line:
            download_url = line.split('"')[-2]
            latest_bin = download_url.split("/")[-1]

    print("Latest %s version is: %s" % (app_name, version))
    print("Download URL is: %s" % download_url)

    # Downloading latest binaries
    if not isfile(latest_bin):
        print("Downloading: %s" % latest_bin)
        wget(download_url, latest_bin, proxies=proxies)
        # Checking version from file
        version_from_file = get_version_from_binary(latest_bin)
        # if not version_from_file.startswith(version) and version_from_file != '':
        if Version(version_from_file) != Version(version) and version_from_file != "":
            print("Changing version to the version number of the binary")
            os.rename(latest_bin, bin_contains + version_from_file + ".msi")
            version = version_from_file
        else:
            print("Binary file version corresponds to online version")

        # Changing version of the package
    if Version(version) > Version(control.get_software_version()):
        print("Software version updated (from: %s to: %s)" % (control.get_software_version(), Version(version)))
        result = True
    else:
        print("Software version up-to-date (%s)" % Version(version))
    control.version = "%s-%s" % (Version(version), control.version.split("-", 1)[-1])
    # control.set_software_version(version)
    control.save_control_to_wapt()

    # Deleting outdated binaries
    remove_outdated_binaries(version)

    # Validating update-package-sources
    return result