# -*- coding: utf-8 -*-
##################################################
# This file is part of WAPT Enterprise
# All right reserved, (c) Tranquil IT Systems 2024
# For more information please refer to
# https://wapt.tranquil.it/store/licences.html
##################################################
from setuphelpers import *
import winreg

"""
https://docs.fortinet.com/document/forticlient/7.2.3/xml-reference-guide/10217

"""
remove_other_configurations = False
conf_name = "VPN_CONF"
conf_file = "company.conf"
unencrypted_password = "************"


def install():
    vpn_configs = []
    if remove_other_configurations:
        if reg_key_exists(HKEY_LOCAL_MACHINE, r"SOFTWARE\Fortinet\FortiClient\Sslvpn\Tunnels"):
            vpn_configs = unm_key(HKEY_LOCAL_MACHINE, r"SOFTWARE\Fortinet\FortiClient\Sslvpn\Tunnels")

        print("old VPN configs:\n%s" % vpn_configs)
        for vpn_config in vpn_configs:
            if vpn_config != conf_name or force:
                print("deleting: %s" % vpn_config)
                registry_deletekey(HKEY_LOCAL_MACHINE, r"SOFTWARE\Fortinet\FortiClient\Sslvpn\Tunnels", vpn_config)

    if not conf_name in vpn_configs or force:
        print(f"Applying: {conf_file}")
        app_dir = installed_softwares(name="FortiClient VPN")[0]["install_location"]

        # Interact differently with Luti
        if params.get("running_as_luti"):
            return "OK"

        # "C:\Program Files\Fortinet\FortiClient\FCConfig.exe"
        config_exe = makepath(app_dir, "FCConfig.exe")

        run(f'"{config_exe}" -m all -f "{conf_file}" -o import -i 1 -p "{unencrypted_password}"')
    else:
        print(f"{conf_file} is already applied.")


def unm_key(ruche, keylist):
    result = []
    key = reg_openkey_noredir(ruche, keylist)
    i = 0
    while True:
        try:
            value = winreg.EnumKey(key, i)
            result.append(value)

            i += 1
        except WindowsError as e:
            # WindowsError: [Errno 259] No more data is available
            if e.winerror == 259:
                break
            else:
                raise
    return result

                                

# -*- coding: utf-8 -*-
##################################################
# This file is part of WAPT Enterprise
# All right reserved, (c) Tranquil IT Systems 2024
# For more information please refer to
# https://wapt.tranquil.it/store/licences.html
##################################################
from setuphelpers import *
from setupdevhelpers import *


def update_package():
    # Declaring local variables
    package_updated = False

    # Interact differently with Luti
    if params.get("running_as_luti", False):
        return "OK"

    # Checking wapt version
    up_min_wapt_version = "2.5"
    if Version(get_version_from_binary("wapt-get.exe")) < Version(up_min_wapt_version):
        error(f"Your WAPT Console must be {up_min_wapt_version} minimum to continue")

    # Renaming package
    if "template" in control.package:
        complete_control_package(control)
        complete_control_name(control)
        if "template" in control.package:
            error("Please rename the package.")
        else:
            last_version_digit = str(int(control.get_software_version().split(".")[-1]) + 1)
            version = ".".join(control.get_software_version().split(".")[:-1] + [last_version_digit])
        ask_message(
            control.package,
            "Please note that you will still need to edit this package to apply your own configuration that is compliant with your company's internal policies. You are sole responsible for the use of this package on your network.",
            48,
        )
    else:
        last_version_digit = str(int(control.get_software_version().split(".")[-1]) + 1)
        version = ".".join(control.get_software_version().split(".")[:-1] + [last_version_digit])

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

    # Validating or not update-package-sources
    return package_updated