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

r"""
Usable WAPT package functions: install(), uninstall(), session_setup(), audit(), update_package()

https://docs.microsoft.com/en-US/windows/security/identity-protection/access-control/security-identifiers#security-identifier-architecture

"""
# Declaring global variables - Warnings: 1) WAPT context is only available in package functions; 2) Global variables are not persistent between calls
bin_contains = "nomachine_"
silent_args = '/usbinstall="0" /VERYSILENT /SUPPRESSMSGBOXES /NORESTART /SP-'  # Silent # '/usbinstall="0" /silent' # Semi-silent
app_uninstallkey = "NoMachine_is1"
app_name = "NoMachine"
server_cfg_file = makepath(programfiles32, "NoMachine", "etc", "server.cfg")
node_cfg_file = makepath(programfiles32, "NoMachine", "etc", "node.cfg")


def install():
    # Declaring local variables
    package_version = control.get_software_version()
    bin_name = glob.glob("*%s*.exe" % bin_contains)[0]

    # Uninstalling older versions of the software
    for uninstall in installed_softwares(name=r"\bNoMachine\b"):
        if uninstall["version"] < control.get_software_version() or force:
            print("Removing: %s (%s)" % (uninstall["name"], uninstall["version"]))
            run(uninstall_cmd(uninstall["key"]))
            wait_uninstallkey_absent(uninstall["key"])
            time.sleep(60)

    # Installing the software
    print("Installing: %s" % bin_name)
    install_exe_if_needed(bin_name, silentflags=silent_args, key=app_uninstallkey, min_version=package_version)

    # Configuring server.cfg default preferences
    if isfile(server_cfg_file):
        file = open(server_cfg_file, "r")
        old_data = file.read()
        file.close()

        data = old_data
        data = data.replace("#UpdateFrequency 172800", "UpdateFrequency 0")
        data = data.replace("#StartNXDaemon Automatic", "StartNXDaemon 0")
        data = data.replace("StartNXDaemon Automatic", "StartNXDaemon 0")
        data = data.replace("#PhysicalDesktopSharing 1", "PhysicalDesktopSharing 0")
        data = data.replace("PhysicalDesktopSharing 1", "PhysicalDesktopSharing 0")
        data = data.replace("#PhysicalDesktopSharing 1", "PhysicalDesktopSharing 0")
        data = data.replace("#EnableNetworkBroadcast 1", "EnableNetworkBroadcast 0")
        data = data.replace("EnableNetworkBroadcast 1", "EnableNetworkBroadcast 0")

        new_data = data
        if not old_data == new_data:
            print("Writing configuration file with minimum instrusion parameters (%s)" % server_cfg_file)
            file = open(server_cfg_file, "w")
            file.write(data)
            file.close()

    # Configuring node.cfg default preferences
    if isfile(node_cfg_file):
        file = open(node_cfg_file, "r")
        old_data = file.read()
        file.close()

        data = old_data
        data = data.replace("#EnableDiskSharing", "EnableDiskSharing none")
        data = data.replace("#EnablePrinterSharing both", "EnablePrinterSharing none")
        data = data.replace("#EnableUSBSharing both", "EnableUSBSharing none")
        data = data.replace("#EnableNetworkSharing both", "EnableNetworkSharing none")
        data = data.replace("EnableSmartcardSharing 1", "EnableSmartcardSharing 0")
        data = data.replace("#AudioInterface disabled", "AudioInterface disabled")
        data = data.replace("#DisplayMonitorIcon 1", "DisplayMonitorIcon 0")

        new_data = data
        if not old_data == new_data:
            print("Writing configuration file with minimum instrusion parameters (%s)" % node_cfg_file)
            file = open(node_cfg_file, "w")
            file.write(data)
            file.close()

    # Changing default start mode of the application services
    set_service_start_mode("nxservice", "Disabled")

    # Removing nx user from local admins
    for adm_user in run_powershell('Get-LocalGroupMember -SID "S-1-5-32-544"'):
        if "nx" in adm_user:
            run_powershell('Remove-LocalGroupMember -SID "S-1-5-32-544" -Member nx')


def update_package():
    # Declaring local variables
    result = False
    proxies = get_proxies()
    if not proxies:
        proxies = get_proxies_from_wapt_console()
    app_name = control.name
    sub_bin_name = bin_contains + "%s.exe"
    url = "https://www.nomachine.com/download/download&id=8"

    # Getting latest version from official sources
    print("URL used is: %s" % url)
    for bs_search in bs_find_all(url, "a", "href", "javascript:void(0)", proxies=proxies):
        if bin_contains in bs_search["onclick"]:
            latest_bin = bs_search["onclick"].split("'")[1]
            version = latest_bin.split("_")[1]
            download_url = bs_search["onclick"].split("'")[-2]
            break

    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 Version(version) != Version(version_from_file) and version_from_file != "":
            print("Changing version to the version number of the binary (from: %s to: %s)" % (version, version_from_file))
            os.rename(latest_bin, sub_bin_name % version_from_file)
            version = version_from_file
        else:
            print("Binary file version correspond 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
    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 or not update-package-sources
    return result


def set_service_start_mode(service_name, start_mode, change_state=True):
    r"""Changing start mode of a service and changing his running status accordingly
    Usable values: 'Auto', 'Manual', 'Disabled'
    Informations: https://msdn.microsoft.com/en-us/library/aa384896(v=vs.85).aspx

    .. versionchanged:: 2.2
        Now interact on the running state

    """
    if not service_installed(service_name):
        return print("WARNING: %s service do not exist, skipped")
    to_start = False
    to_stop = False
    if start_mode == "Auto":
        start_mode = "Automatic"
        to_start = True
    elif start_mode == "Disabled" or start_mode == "Manual":
        to_stop = True
    elif start_mode == "Boot" or start_mode == "System":
        # Handle driver services but it do not seems to work
        pass
    else:
        raise Exception("Start mode value specified is incorrect")

    c = wmi.WMI()
    for svc in c.Win32_Service(Name=service_name):
        if svc.StartMode != start_mode:
            svc.ChangeStartMode(StartMode=start_mode)
            if to_start and change_state:
                if not service_is_running(service_name):
                    try:
                        service_start(service_name)
                    except:
                        print("Unable to start the service: %s" % service_name)
            elif to_stop and change_state:
                if service_is_running(service_name):
                    try:
                        service_stop(service_name)
                    except:
                        print("Unable to stop the service: %s" % service_name)


def get_service_start_mode(service_name):
    r"""Getting actual start mode of a service
    Returned values: 'Auto', 'Manual', 'Disabled'

    """
    c = wmi.WMI()
    for svc in c.Win32_Service(Name=service_name):
        return svc.StartMode