from setuphelpers import *

"""
Installation procedures:
https://mozilla.github.io/policy-templates/
https://mozilla.github.io/policy-templates/#extensionsettings

"""
policies_file = "policies.json"


def get_firefox_policies_dirs():
    detected_apps = installed_softwares("Firefox")
    firefox_policies_dirs = []
    if get_os_name() == "Linux":
        firefox_policies_dirs.append("/etc/firefox/policies")
    else:
        if not detected_apps:
            error("Firefox is not installed on this machine.")
        for ffox in detected_apps:
            if get_os_name() == "Windows":
                firefox_policies_dirs.append(makepath(ffox["install_location"], "distribution"))
            if get_os_name() == "Darwin":
                firefox_policies_dirs.append(makepath(ffox["install_location"], "Contents", "Resources", "distribution"))

    return firefox_policies_dirs


def install():
    # Check if policies.json is valid
    try:
        json_load_file(policies_file)
    except:
        error(f"{policies_file} is malformed.")

    # Interact differently with Luti
    if params.get("running_as_luti", False):
        WAPT.install("tis-firefox")

    for ffox_dist_dir in get_firefox_policies_dirs():
        policies_path = makepath(ffox_dist_dir, policies_file)
        print(f"Writting Firefox configuration to: {policies_path}")
        killalltasks(ensure_list(control.impacted_process))
        if get_os_name() == "Darwin":
            run("sudo defaults write /Library/Preferences/org.mozilla.firefox EnterprisePoliciesEnabled -bool TRUE")
        mkdirs(ffox_dist_dir)
        filecopyto(policies_file, policies_path)

    if get_os_name() == "Windows":
        # Removing policies from registry since it bypass the policies.json file (source: https://support.mozilla.org/bm/questions/1236197)
        ffox_policies_reg_path = r"SOFTWARE\Policies\Mozilla\Firefox"
        if reg_key_exists(HKEY_LOCAL_MACHINE, ffox_policies_reg_path):
            registry_deletekey(HKEY_LOCAL_MACHINE, ffox_policies_reg_path, "", force=True)

    # Forget package instead of adding it as conflict
    WAPT.forget_packages(f"{control.package.split('-')[0]}-config-firefox")


def uninstall():
    for ffox_dist_dir in get_firefox_policies_dirs():
        policies_path = makepath(ffox_dist_dir, policies_file)
        remove_file(policies_path)

    if get_os_name() == "Windows":
        # Reapplying minimal configuration for Firefox
        if not isfile(policies_path):
            registry_set(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Mozilla\Firefox", "DisableAppUpdate", 1, REG_DWORD)
            registry_set(HKEY_LOCAL_MACHINE, r"SOFTWARE\Policies\Mozilla\Firefox", "DisableTelemetry", 1, REG_DWORD)


def audit():
    audit_status = "OK"
    for ffox_dist_dir in get_firefox_policies_dirs():
        policies_path = makepath(ffox_dist_dir, policies_file)
        if not isfile(policies_path):
            print("WARNING: Configuration is NOT applied for Firefox, re-applying...")
            install()
            audit_status = "WARNING"
        else:
            print("OK: Configuration is correctly applied for Firefox.")
            audit_status = "OK"

    return audit_status


# def merge_dicts(first, second):
#     merged = copy.deepcopy(first)
#     merged.update(second)
#     return merged


# def merge_json(first, second):
#     return merge_dicts(json.loads(first), json.loads(second))

                                

# -*- coding: utf-8 -*-
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