"""
Interact with Consul

https://www.consul.io

"""

import base64
import http.client
import logging
import urllib

import salt.utils.http
import salt.utils.json
from salt.exceptions import SaltInvocationError

log = logging.getLogger(__name__)


# Don't shadow built-ins.
__func_alias__ = {"list_": "list"}

__virtualname__ = "consul"


def _get_config():
    """
    Retrieve Consul configuration
    """
    return __salt__["config.get"]("consul.url") or __salt__["config.get"]("consul:url")


def _get_token():
    """
    Retrieve Consul configuration
    """
    return __salt__["config.get"]("consul.token") or __salt__["config.get"](
        "consul:token"
    )


def _query(
    function,
    consul_url,
    token=None,
    method="GET",
    api_version="v1",
    data=None,
    query_params=None,
    decode=True,
    text=False,
):
    """
    Consul object method function to construct and execute on the API URL.

    :param api_url:     The Consul api url.
    :param api_version  The Consul api version
    :param function:    The Consul api function to perform.
    :param method:      The HTTP method, e.g. GET or POST.
    :param data:        The data to be sent for POST method. This param is ignored for GET requests.
    :return:            The json response from the API call or False.
    """

    if not query_params:
        query_params = {}

    ret = {"data": "", "res": True}

    if not token:
        token = _get_token()

    headers = {"X-Consul-Token": token, "Content-Type": "application/json"}
    base_url = urllib.parse.urljoin(consul_url, f"{api_version}/")
    url = urllib.parse.urljoin(base_url, function, False)

    if method == "GET":
        data = None
    else:
        if data is not None:
            if type(data) != str:
                data = salt.utils.json.dumps(data)
        else:
            data = salt.utils.json.dumps({})

    result = salt.utils.http.query(
        url,
        method=method,
        params=query_params,
        data=data,
        decode=decode,
        text=text,
        status=True,
        header_dict=headers,
        opts=__opts__,
    )

    if result.get("status", None) == http.client.OK:
        ret["data"] = result.get("dict", result)
        ret["res"] = True
    elif result.get("status", None) == http.client.NO_CONTENT:
        ret["data"] = "No content available."
        ret["res"] = False
    elif result.get("status", None) == http.client.NOT_FOUND:
        ret["data"] = "Key not found."
        ret["res"] = False
    elif result.get("error", None):
        ret["data"] = "An error occurred."
        ret["error"] = result["error"]
        ret["res"] = False
    else:
        if result:
            ret["data"] = result
            ret["res"] = True
        else:
            ret["res"] = False
    return ret


def list_(consul_url=None, token=None, key=None, **kwargs):
    """
    List keys in Consul

    :param consul_url: The Consul server URL.
    :param key: The key to use as the starting point for the list.
    :return: The list of keys.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.list
        salt '*' consul.list key='web'

    """
    ret = {}

    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    query_params = {}

    if "recurse" in kwargs:
        query_params["recurse"] = "True"

    # No key so recurse and show all values
    if not key:
        query_params["recurse"] = "True"
        function = "kv/"
    else:
        function = f"kv/{key}"

    query_params["keys"] = "True"
    query_params["separator"] = "/"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def get(consul_url=None, key=None, token=None, recurse=False, decode=False, raw=False):
    """
    Get key from Consul

    :param consul_url: The Consul server URL.
    :param key: The key to use as the starting point for the list.
    :param recurse: Return values recursively beginning at the value of key.
    :param decode: By default values are stored as Base64 encoded values,
                   decode will return the whole key with the value decoded.
    :param raw: Simply return the decoded value of the key.
    :return: The keys in Consul.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.get key='web/key1'
        salt '*' consul.get key='web' recurse=True
        salt '*' consul.get key='web' recurse=True decode=True

    By default values stored in Consul are base64 encoded, passing the
    decode option will show them as the decoded values.

    .. code-block:: bash

        salt '*' consul.get key='web' recurse=True decode=True raw=True

    By default Consult will return other information about the key, the raw
    option will return only the raw value.

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not key:
        raise SaltInvocationError('Required argument "key" is missing.')

    query_params = {}
    function = f"kv/{key}"
    if recurse:
        query_params["recurse"] = "True"
    if raw:
        query_params["raw"] = True
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )

    if ret["res"]:
        if decode:
            for item in ret["data"]:
                if item["Value"] is not None:
                    item["Value"] = base64.b64decode(item["Value"])
                else:
                    item["Value"] = ""
    return ret


def put(consul_url=None, token=None, key=None, value=None, **kwargs):
    """
    Put values into Consul

    :param consul_url: The Consul server URL.
    :param key: The key to use as the starting point for the list.
    :param value: The value to set the key to.
    :param flags: This can be used to specify an unsigned value
                  between 0 and 2^64-1. Clients can choose to use
                  this however makes sense for their application.
    :param cas: This flag is used to turn the PUT into a
                Check-And-Set operation.
    :param acquire: This flag is used to turn the PUT into a
                    lock acquisition operation.
    :param release: This flag is used to turn the PUT into a
                    lock release operation.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.put key='web/key1' value="Hello there"

        salt '*' consul.put key='web/key1' value="Hello there" acquire='d5d371f4-c380-5280-12fd-8810be175592'

        salt '*' consul.put key='web/key1' value="Hello there" release='d5d371f4-c380-5280-12fd-8810be175592'

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not key:
        raise SaltInvocationError('Required argument "key" is missing.')

    # Invalid to specified these together
    conflicting_args = ["cas", "release", "acquire"]
    for _l1 in conflicting_args:
        for _l2 in conflicting_args:
            if _l1 in kwargs and _l2 in kwargs and _l1 != _l2:
                raise SaltInvocationError(
                    "Using arguments `{}` and `{}` together is invalid.".format(
                        _l1, _l2
                    )
                )

    query_params = {}

    available_sessions = session_list(consul_url=consul_url, return_list=True)
    _current = get(consul_url=consul_url, token=token, key=key)

    if "flags" in kwargs:
        if kwargs["flags"] >= 0 and kwargs["flags"] <= 2**64:
            query_params["flags"] = kwargs["flags"]

    if "cas" in kwargs:
        if _current["res"]:
            if kwargs["cas"] == 0:
                ret["message"] = f"Key {key} exists, index must be non-zero."
                ret["res"] = False
                return ret

            if kwargs["cas"] != _current["data"]["ModifyIndex"]:
                ret["message"] = f"Key {key} exists, but indexes do not match."
                ret["res"] = False
                return ret
            query_params["cas"] = kwargs["cas"]
        else:
            ret["message"] = f"Key {key} does not exists, CAS argument can not be used."
            ret["res"] = False
            return ret

    if "acquire" in kwargs:
        if kwargs["acquire"] not in available_sessions:
            ret["message"] = "{} is not a valid session.".format(kwargs["acquire"])
            ret["res"] = False
            return ret

        query_params["acquire"] = kwargs["acquire"]

    if "release" in kwargs:
        if _current["res"]:
            if "Session" in _current["data"]:
                if _current["data"]["Session"] == kwargs["release"]:
                    query_params["release"] = kwargs["release"]
                else:
                    ret["message"] = f"{key} locked by another session."
                    ret["res"] = False
                    return ret

            else:
                ret["message"] = "{} is not a valid session.".format(kwargs["acquire"])
                ret["res"] = False
        else:
            log.error("Key {0} does not exist. Skipping release.")

    data = value
    function = f"kv/{key}"
    method = "PUT"
    res = _query(
        consul_url=consul_url,
        token=token,
        function=function,
        method=method,
        data=data,
        query_params=query_params,
    )

    if res["res"]:
        ret["res"] = True
        ret["data"] = f"Added key {key} with value {value}."
    else:
        ret["res"] = False
        ret["data"] = f"Unable to add key {key} with value {value}."
        if "error" in res:
            ret["error"] = res["error"]
    return ret


def delete(consul_url=None, token=None, key=None, **kwargs):
    """
    Delete values from Consul

    :param consul_url: The Consul server URL.
    :param key: The key to use as the starting point for the list.
    :param recurse: Delete values recursively beginning at the value of key.
    :param cas: This flag is used to turn the DELETE into
                a Check-And-Set operation.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.delete key='web'
        salt '*' consul.delete key='web' recurse='True'

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not key:
        raise SaltInvocationError('Required argument "key" is missing.')

    query_params = {}

    if "recurse" in kwargs:
        query_params["recurse"] = True

    if "cas" in kwargs:
        if kwargs["cas"] > 0:
            query_params["cas"] = kwargs["cas"]
        else:
            ret["message"] = (
                "Check and Set Operation ",
                "value must be greater than 0.",
            )
            ret["res"] = False
            return ret

    function = f"kv/{key}"
    res = _query(
        consul_url=consul_url,
        token=token,
        function=function,
        method="DELETE",
        query_params=query_params,
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Deleted key {key}."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to delete key {key}."
        if "error" in res:
            ret["error"] = res["error"]
    return ret


def agent_checks(consul_url=None, token=None):
    """
    Returns the checks the local agent is managing

    :param consul_url: The Consul server URL.
    :return: Returns the checks the local agent is managing

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_checks

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "agent/checks"
    ret = _query(consul_url=consul_url, function=function, token=token, method="GET")
    return ret


def agent_services(consul_url=None, token=None):
    """
    Returns the services the local agent is managing

    :param consul_url: The Consul server URL.
    :return: Returns the services the local agent is managing

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_services

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "agent/services"
    ret = _query(consul_url=consul_url, function=function, token=token, method="GET")
    return ret


def agent_members(consul_url=None, token=None, **kwargs):
    """
    Returns the members as seen by the local serf agent

    :param consul_url: The Consul server URL.
    :return: Returns the members as seen by the local serf agent

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_members

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "wan" in kwargs:
        query_params["wan"] = kwargs["wan"]

    function = "agent/members"
    ret = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="GET",
        query_params=query_params,
    )
    return ret


def agent_self(consul_url=None, token=None):
    """
    Returns the local node configuration

    :param consul_url: The Consul server URL.
    :return: Returns the local node configuration

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_self

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "agent/self"
    ret = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="GET",
        query_params=query_params,
    )
    return ret


def agent_maintenance(consul_url=None, token=None, **kwargs):
    """
    Manages node maintenance mode

    :param consul_url: The Consul server URL.
    :param enable: The enable flag is required.
                   Acceptable values are either true
                   (to enter maintenance mode) or
                   false (to resume normal operation).
    :param reason: If provided, its value should be a
                   text string explaining the reason for
                   placing the node into maintenance mode.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_maintenance enable='False' reason='Upgrade in progress'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "enable" in kwargs:
        query_params["enable"] = kwargs["enable"]
    else:
        ret["message"] = 'Required parameter "enable" is missing.'
        ret["res"] = False
        return ret

    if "reason" in kwargs:
        query_params["reason"] = kwargs["reason"]

    function = "agent/maintenance"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="PUT",
        query_params=query_params,
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = "Agent maintenance mode {}ed.".format(kwargs["enable"])
    else:
        ret["res"] = True
        ret["message"] = "Unable to change maintenance mode for agent."
    return ret


def agent_join(consul_url=None, token=None, address=None, **kwargs):
    """
    Triggers the local agent to join a node

    :param consul_url: The Consul server URL.
    :param address: The address for the agent to connect to.
    :param wan: Causes the agent to attempt to join using the WAN pool.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_join address='192.168.1.1'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not address:
        raise SaltInvocationError('Required argument "address" is missing.')

    if "wan" in kwargs:
        query_params["wan"] = kwargs["wan"]

    function = f"agent/join/{address}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="GET",
        query_params=query_params,
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = "Agent joined the cluster"
    else:
        ret["res"] = False
        ret["message"] = "Unable to join the cluster."
    return ret


def agent_leave(consul_url=None, token=None, node=None):
    """
    Used to instruct the agent to force a node into the left state.

    :param consul_url: The Consul server URL.
    :param node: The node the agent will force into left state
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_leave node='web1.example.com'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not node:
        raise SaltInvocationError('Required argument "node" is missing.')

    function = f"agent/force-leave/{node}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="GET",
        query_params=query_params,
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Node {node} put in leave state."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to change state for {node}."
    return ret


def agent_check_register(consul_url=None, token=None, **kwargs):
    """
    The register endpoint is used to add a new check to the local agent.

    :param consul_url: The Consul server URL.
    :param name: The description of what the check is for.
    :param id: The unique name to use for the check, if not
               provided 'name' is used.
    :param notes: Human readable description of the check.
    :param script: If script is provided, the check type is
                   a script, and Consul will evaluate that script
                   based on the interval parameter.
    :param http: Check will perform an HTTP GET request against
                 the value of HTTP (expected to be a URL) based
                 on the interval parameter.
    :param ttl: If a TTL type is used, then the TTL update endpoint
                must be used periodically to update the state of the check.
    :param interval: Interval at which the check should run.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_check_register name='Memory Utilization' script='/usr/local/bin/check_mem.py' interval='15s'

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "name" in kwargs:
        data["Name"] = kwargs["name"]
    else:
        raise SaltInvocationError('Required argument "name" is missing.')

    if True not in [True for item in ("script", "http", "ttl") if item in kwargs]:
        ret["message"] = 'Required parameter "script" or "http" is missing.'
        ret["res"] = False
        return ret

    if "id" in kwargs:
        data["ID"] = kwargs["id"]

    if "notes" in kwargs:
        data["Notes"] = kwargs["notes"]

    if "script" in kwargs:
        if "interval" not in kwargs:
            ret["message"] = 'Required parameter "interval" is missing.'
            ret["res"] = False
            return ret
        data["Script"] = kwargs["script"]
        data["Interval"] = kwargs["interval"]

    if "http" in kwargs:
        if "interval" not in kwargs:
            ret["message"] = 'Required parameter "interval" is missing.'
            ret["res"] = False
            return ret
        data["HTTP"] = kwargs["http"]
        data["Interval"] = kwargs["interval"]

    if "ttl" in kwargs:
        data["TTL"] = kwargs["ttl"]

    function = "agent/check/register"
    res = _query(
        consul_url=consul_url, function=function, token=token, method="PUT", data=data
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = "Check {} added to agent.".format(kwargs["name"])
    else:
        ret["res"] = False
        ret["message"] = "Unable to add check to agent."
    return ret


def agent_check_deregister(consul_url=None, token=None, checkid=None):
    """
    The agent will take care of deregistering the check from the Catalog.

    :param consul_url: The Consul server URL.
    :param checkid: The ID of the check to deregister from Consul.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_check_deregister checkid='Memory Utilization'

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not checkid:
        raise SaltInvocationError('Required argument "checkid" is missing.')

    function = f"agent/check/deregister/{checkid}"
    res = _query(consul_url=consul_url, function=function, token=token, method="GET")
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Check {checkid} removed from agent."
    else:
        ret["res"] = False
        ret["message"] = "Unable to remove check from agent."
    return ret


def agent_check_pass(consul_url=None, token=None, checkid=None, **kwargs):
    """
    This endpoint is used with a check that is of the TTL type. When this
    is called, the status of the check is set to passing and the TTL
    clock is reset.

    :param consul_url: The Consul server URL.
    :param checkid: The ID of the check to mark as passing.
    :param note: A human-readable message with the status of the check.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_check_pass checkid='redis_check1' note='Forcing check into passing state.'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not checkid:
        raise SaltInvocationError('Required argument "checkid" is missing.')

    if "note" in kwargs:
        query_params["note"] = kwargs["note"]

    function = f"agent/check/pass/{checkid}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        query_params=query_params,
        method="GET",
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Check {checkid} marked as passing."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to update check {checkid}."
    return ret


def agent_check_warn(consul_url=None, token=None, checkid=None, **kwargs):
    """
    This endpoint is used with a check that is of the TTL type. When this
    is called, the status of the check is set to warning and the TTL
    clock is reset.

    :param consul_url: The Consul server URL.
    :param checkid: The ID of the check to deregister from Consul.
    :param note: A human-readable message with the status of the check.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_check_warn checkid='redis_check1' note='Forcing check into warning state.'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not checkid:
        raise SaltInvocationError('Required argument "checkid" is missing.')

    if "note" in kwargs:
        query_params["note"] = kwargs["note"]

    function = f"agent/check/warn/{checkid}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        query_params=query_params,
        method="GET",
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Check {checkid} marked as warning."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to update check {checkid}."
    return ret


def agent_check_fail(consul_url=None, token=None, checkid=None, **kwargs):
    """
    This endpoint is used with a check that is of the TTL type. When this
    is called, the status of the check is set to critical and the
    TTL clock is reset.

    :param consul_url: The Consul server URL.
    :param checkid: The ID of the check to deregister from Consul.
    :param note: A human-readable message with the status of the check.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_check_fail checkid='redis_check1' note='Forcing check into critical state.'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not checkid:
        raise SaltInvocationError('Required argument "checkid" is missing.')

    if "note" in kwargs:
        query_params["note"] = kwargs["note"]

    function = f"agent/check/fail/{checkid}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        query_params=query_params,
        method="GET",
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Check {checkid} marked as critical."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to update check {checkid}."
    return ret


def agent_service_register(
    consul_url=None, token=None, decode=False, text=True, **kwargs
):
    """
    The used to add a new service, with an optional
    health check, to the local agent.

    :param consul_url: The Consul server URL.
    :param name: A name describing the service.
    :param address: The address used by the service, defaults
                    to the address of the agent.
    :param port: The port used by the service.
    :param id: Unique ID to identify the service, if not
               provided the value of the name parameter is used.
    :param tags: Identifying tags for service, string or list.
    :param script: If script is provided, the check type is
                   a script, and Consul will evaluate that script
                   based on the interval parameter.
    :param http: Check will perform an HTTP GET request against
                 the value of HTTP (expected to be a URL) based
                 on the interval parameter.
    :param check_ttl: If a TTL type is used, then the TTL update
                      endpoint must be used periodically to update
                      the state of the check.
    :param check_interval: Interval at which the check should run.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_service_register name='redis' tags='["master", "v1"]' address="127.0.0.1" port="8080" check_script="/usr/local/bin/check_redis.py" interval="10s"

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    lc_kwargs = dict()
    for k, v in kwargs.items():
        lc_kwargs[k.lower()] = v

    if "name" in lc_kwargs:
        data["Name"] = lc_kwargs["name"]
    else:
        raise SaltInvocationError('Required argument "name" is missing.')

    if "address" in lc_kwargs:
        data["Address"] = lc_kwargs["address"]

    if "port" in lc_kwargs:
        data["Port"] = lc_kwargs["port"]

    if "id" in lc_kwargs:
        data["ID"] = lc_kwargs["id"]

    if "tags" in lc_kwargs:
        _tags = lc_kwargs["tags"]
        if not isinstance(_tags, list):
            _tags = [_tags]
        data["Tags"] = _tags

    if "enabletagoverride" in lc_kwargs:
        data["EnableTagOverride"] = lc_kwargs["enabletagoverride"]

    if "check" in lc_kwargs:
        dd = dict()
        for k, v in lc_kwargs["check"].items():
            dd[k.lower()] = v
        interval_required = False
        check_dd = dict()

        if "script" in dd:
            interval_required = True
            check_dd["Script"] = dd["script"]
        if "http" in dd:
            interval_required = True
            check_dd["HTTP"] = dd["http"]
        if "ttl" in dd:
            check_dd["TTL"] = dd["ttl"]
        if "interval" in dd:
            check_dd["Interval"] = dd["interval"]

        if interval_required:
            if "Interval" not in check_dd:
                ret["message"] = 'Required parameter "interval" is missing.'
                ret["res"] = False
                return ret
        else:
            if "Interval" in check_dd:
                del check_dd["Interval"]  # not required, so ignore it

        if len(check_dd.keys()) > 0:
            data["Check"] = check_dd  # if empty, ignore it

    function = "agent/service/register"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="PUT",
        data=data,
        decode=decode,
        text=text,
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = "Service {} registered on agent.".format(kwargs["name"])
    else:
        ret["res"] = False
        ret["message"] = "Unable to register service {}.".format(kwargs["name"])
    return ret


def agent_service_deregister(
    consul_url=None, token=None, serviceid=None, decode=False, text=True
):
    """
    Used to remove a service.

    :param consul_url: The Consul server URL.
    :param serviceid: A serviceid describing the service.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_service_deregister serviceid='redis'

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not serviceid:
        raise SaltInvocationError('Required argument "serviceid" is missing.')

    function = f"agent/service/deregister/{serviceid}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="PUT",
        data=data,
        decode=decode,
        text=text,
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Service {serviceid} removed from agent."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to remove service {serviceid}."
    return ret


def agent_service_maintenance(consul_url=None, token=None, serviceid=None, **kwargs):
    """
    Used to place a service into maintenance mode.

    :param consul_url: The Consul server URL.
    :param serviceid: A name of the service.
    :param enable: Whether the service should be enabled or disabled.
    :param reason: A human readable message of why the service was
                   enabled or disabled.
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.agent_service_deregister serviceid='redis' enable='True' reason='Down for upgrade'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not serviceid:
        raise SaltInvocationError('Required argument "serviceid" is missing.')

    if "enable" in kwargs:
        query_params["enable"] = kwargs["enable"]
    else:
        ret["message"] = 'Required parameter "enable" is missing.'
        ret["res"] = False
        return ret

    if "reason" in kwargs:
        query_params["reason"] = kwargs["reason"]

    function = f"agent/service/maintenance/{serviceid}"
    res = _query(
        consul_url=consul_url, token=token, function=function, query_params=query_params
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Service {serviceid} set in maintenance mode."
    else:
        ret["res"] = False
        ret["message"] = "Unable to set service {} to maintenance mode.".format(
            serviceid
        )
    return ret


def session_create(consul_url=None, token=None, **kwargs):
    """
    Used to create a session.

    :param consul_url: The Consul server URL.
    :param lockdelay: Duration string using a "s" suffix for seconds.
                      The default is 15s.
    :param node: Must refer to a node that is already registered,
                 if specified. By default, the agent's own node
                 name is used.
    :param name: A human-readable name for the session
    :param checks: A list of associated health checks. It is highly
                   recommended that, if you override this list, you
                   include the default "serfHealth".
    :param behavior: Can be set to either release or delete. This controls
                     the behavior when a session is invalidated. By default,
                     this is release, causing any locks that are held to be
                     released. Changing this to delete causes any locks that
                     are held to be deleted. delete is useful for creating
                     ephemeral key/value entries.
    :param ttl: Session is invalidated if it is not renewed before
                the TTL expires
    :return: Boolean and message indicating success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.session_create node='node1' name='my-session' behavior='delete' ttl='3600s'

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret
    data = {}

    if "lockdelay" in kwargs:
        data["LockDelay"] = kwargs["lockdelay"]

    if "node" in kwargs:
        data["Node"] = kwargs["node"]

    if "name" in kwargs:
        data["Name"] = kwargs["name"]
    else:
        raise SaltInvocationError('Required argument "name" is missing.')

    if "checks" in kwargs:
        data["Touch"] = kwargs["touch"]

    if "behavior" in kwargs:
        if not kwargs["behavior"] in ("delete", "release"):
            ret["message"] = ("Behavior must be ", "either delete or release.")
            ret["res"] = False
            return ret
        data["Behavior"] = kwargs["behavior"]

    if "ttl" in kwargs:
        _ttl = kwargs["ttl"]
        if str(_ttl).endswith("s"):
            _ttl = _ttl[:-1]

        if int(_ttl) < 0 or int(_ttl) > 3600:
            ret["message"] = ("TTL must be ", "between 0 and 3600.")
            ret["res"] = False
            return ret
        data["TTL"] = f"{_ttl}s"

    function = "session/create"
    res = _query(
        consul_url=consul_url, function=function, token=token, method="PUT", data=data
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = "Created session {}.".format(kwargs["name"])
    else:
        ret["res"] = False
        ret["message"] = "Unable to create session {}.".format(kwargs["name"])
    return ret


def session_list(consul_url=None, token=None, return_list=False, **kwargs):
    """
    Used to list sessions.

    :param consul_url: The Consul server URL.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :param return_list: By default, all information about the sessions is
                        returned, using the return_list parameter will return
                        a list of session IDs.
    :return: A list of all available sessions.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.session_list

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    query_params = {}

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = "session/list"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )

    if return_list:
        _list = []
        for item in ret["data"]:
            _list.append(item["ID"])
        return _list
    return ret


def session_destroy(consul_url=None, token=None, session=None, **kwargs):
    """
    Destroy session

    :param consul_url: The Consul server URL.
    :param session: The ID of the session to destroy.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.session_destroy session='c1c4d223-91cb-3d1f-1ee8-f2af9e7b6716'

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not session:
        raise SaltInvocationError('Required argument "session" is missing.')

    query_params = {}

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = f"session/destroy/{session}"
    res = _query(
        consul_url=consul_url,
        function=function,
        token=token,
        method="PUT",
        query_params=query_params,
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Destroyed Session {session}."
    else:
        ret["res"] = False
        ret["message"] = f"Unable to destroy session {session}."
    return ret


def session_info(consul_url=None, token=None, session=None, **kwargs):
    """
    Information about a session

    :param consul_url: The Consul server URL.
    :param session: The ID of the session to return information about.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.session_info session='c1c4d223-91cb-3d1f-1ee8-f2af9e7b6716'

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not session:
        raise SaltInvocationError('Required argument "session" is missing.')

    query_params = {}

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = f"session/info/{session}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def catalog_register(consul_url=None, token=None, **kwargs):
    """
    Registers a new node, service, or check

    :param consul_url: The Consul server URL.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :param node: The node to register.
    :param address: The address of the node.
    :param service: The service that will be registered.
    :param service_address: The address that the service listens on.
    :param service_port: The port for the service.
    :param service_id: A unique identifier for the service, if this is not
                       provided "name" will be used.
    :param service_tags: Any tags associated with the service.
    :param check: The name of the health check to register
    :param check_status: The initial status of the check,
                         must be one of unknown, passing, warning, or critical.
    :param check_service: The service that the check is performed against.
    :param check_id: Unique identifier for the service.
    :param check_notes: An opaque field that is meant to hold human-readable text.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_register node='node1' address='192.168.1.1' service='redis' service_address='127.0.0.1' service_port='8080' service_id='redis_server1'

    """
    ret = {}
    data = {}
    data["NodeMeta"] = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "datacenter" in kwargs:
        data["Datacenter"] = kwargs["datacenter"]

    if "node" in kwargs:
        data["Node"] = kwargs["node"]
    else:
        ret["message"] = "Required argument node argument is missing."
        ret["res"] = False
        return ret

    if "address" in kwargs:
        if isinstance(kwargs["address"], list):
            _address = kwargs["address"][0]
        else:
            _address = kwargs["address"]
        data["Address"] = _address
    else:
        ret["message"] = "Required argument address argument is missing."
        ret["res"] = False
        return ret

    if "ip_interfaces" in kwargs:
        data["TaggedAddresses"] = {}
        for k in kwargs["ip_interfaces"]:
            if kwargs["ip_interfaces"].get(k):
                data["TaggedAddresses"][k] = kwargs["ip_interfaces"][k][0]

    if "service" in kwargs:
        data["Service"] = {}
        data["Service"]["Service"] = kwargs["service"]

        if "service_address" in kwargs:
            data["Service"]["Address"] = kwargs["service_address"]

        if "service_port" in kwargs:
            data["Service"]["Port"] = kwargs["service_port"]

        if "service_id" in kwargs:
            data["Service"]["ID"] = kwargs["service_id"]

        if "service_tags" in kwargs:
            _tags = kwargs["service_tags"]
            if not isinstance(_tags, list):
                _tags = [_tags]
            data["Service"]["Tags"] = _tags

    if "cpu" in kwargs:
        data["NodeMeta"]["Cpu"] = kwargs["cpu"]

    if "num_cpus" in kwargs:
        data["NodeMeta"]["Cpu_num"] = kwargs["num_cpus"]

    if "mem" in kwargs:
        data["NodeMeta"]["Memory"] = kwargs["mem"]

    if "oscode" in kwargs:
        data["NodeMeta"]["Os"] = kwargs["oscode"]

    if "osarch" in kwargs:
        data["NodeMeta"]["Osarch"] = kwargs["osarch"]

    if "kernel" in kwargs:
        data["NodeMeta"]["Kernel"] = kwargs["kernel"]

    if "kernelrelease" in kwargs:
        data["NodeMeta"]["Kernelrelease"] = kwargs["kernelrelease"]

    if "localhost" in kwargs:
        data["NodeMeta"]["localhost"] = kwargs["localhost"]

    if "nodename" in kwargs:
        data["NodeMeta"]["nodename"] = kwargs["nodename"]

    if "os_family" in kwargs:
        data["NodeMeta"]["os_family"] = kwargs["os_family"]

    if "lsb_distrib_description" in kwargs:
        data["NodeMeta"]["lsb_distrib_description"] = kwargs["lsb_distrib_description"]

    if "master" in kwargs:
        data["NodeMeta"]["master"] = kwargs["master"]

    if "check" in kwargs:
        data["Check"] = {}
        data["Check"]["Name"] = kwargs["check"]

        if "check_status" in kwargs:
            if kwargs["check_status"] not in (
                "unknown",
                "passing",
                "warning",
                "critical",
            ):
                ret[
                    "message"
                ] = "Check status must be unknown, passing, warning, or critical."
                ret["res"] = False
                return ret
            data["Check"]["Status"] = kwargs["check_status"]

        if "check_service" in kwargs:
            data["Check"]["ServiceID"] = kwargs["check_service"]

        if "check_id" in kwargs:
            data["Check"]["CheckID"] = kwargs["check_id"]

        if "check_notes" in kwargs:
            data["Check"]["Notes"] = kwargs["check_notes"]

    function = "catalog/register"
    res = _query(
        consul_url=consul_url, function=function, token=token, method="PUT", data=data
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = "Catalog registration for {} successful.".format(
            kwargs["node"]
        )
    else:
        ret["res"] = False
        ret["message"] = "Catalog registration for {} failed.".format(kwargs["node"])
    ret["data"] = data
    return ret


def catalog_deregister(consul_url=None, token=None, **kwargs):
    """
    Deregisters a node, service, or check

    :param consul_url: The Consul server URL.
    :param node: The node to deregister.
    :param datacenter: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :param checkid: The ID of the health check to deregister.
    :param serviceid: The ID of the service to deregister.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_register node='node1' serviceid='redis_server1' checkid='redis_check1'

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "datacenter" in kwargs:
        data["Datacenter"] = kwargs["datacenter"]

    if "node" in kwargs:
        data["Node"] = kwargs["node"]
    else:
        ret["message"] = "Node argument required."
        ret["res"] = False
        return ret

    if "checkid" in kwargs:
        data["CheckID"] = kwargs["checkid"]

    if "serviceid" in kwargs:
        data["ServiceID"] = kwargs["serviceid"]

    function = "catalog/deregister"
    res = _query(
        consul_url=consul_url, function=function, token=token, method="PUT", data=data
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = "Catalog item {} removed.".format(kwargs["node"])
    else:
        ret["res"] = False
        ret["message"] = "Removing Catalog item {} failed.".format(kwargs["node"])
    return ret


def catalog_datacenters(consul_url=None, token=None):
    """
    Return list of available datacenters from catalog.

    :param consul_url: The Consul server URL.
    :return: The list of available datacenters.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_datacenters

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "catalog/datacenters"
    ret = _query(consul_url=consul_url, function=function, token=token)
    return ret


def catalog_nodes(consul_url=None, token=None, **kwargs):
    """
    Return list of available nodes from catalog.

    :param consul_url: The Consul server URL.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: The list of available nodes.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_nodes

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = "catalog/nodes"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def catalog_services(consul_url=None, token=None, **kwargs):
    """
    Return list of available services rom catalog.

    :param consul_url: The Consul server URL.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: The list of available services.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_services

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = "catalog/services"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def catalog_service(consul_url=None, token=None, service=None, **kwargs):
    """
    Information about the registered service.

    :param consul_url: The Consul server URL.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :param tag: Filter returned services with tag parameter.
    :return: Information about the requested service.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_service service='redis'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not service:
        raise SaltInvocationError('Required argument "service" is missing.')

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    if "tag" in kwargs:
        query_params["tag"] = kwargs["tag"]

    function = f"catalog/service/{service}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def catalog_node(consul_url=None, token=None, node=None, **kwargs):
    """
    Information about the registered node.

    :param consul_url: The Consul server URL.
    :param node: The node to request information about.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: Information about the requested node.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.catalog_service service='redis'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not node:
        raise SaltInvocationError('Required argument "node" is missing.')

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = f"catalog/node/{node}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def health_node(consul_url=None, token=None, node=None, **kwargs):
    """
    Health information about the registered node.

    :param consul_url: The Consul server URL.
    :param node: The node to request health information about.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: Health information about the requested node.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.health_node node='node1'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not node:
        raise SaltInvocationError('Required argument "node" is missing.')

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = f"health/node/{node}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def health_checks(consul_url=None, token=None, service=None, **kwargs):
    """
    Health information about the registered service.

    :param consul_url: The Consul server URL.
    :param service: The service to request health information about.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: Health information about the requested node.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.health_checks service='redis1'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not service:
        raise SaltInvocationError('Required argument "service" is missing.')

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    function = f"health/checks/{service}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def health_service(consul_url=None, token=None, service=None, **kwargs):
    """
    Health information about the registered service.

    :param consul_url: The Consul server URL.
    :param service: The service to request health information about.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :param tag: Filter returned services with tag parameter.
    :param passing: Filter results to only nodes with all
                    checks in the passing state.
    :return: Health information about the requested node.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.health_service service='redis1'

        salt '*' consul.health_service service='redis1' passing='True'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not service:
        raise SaltInvocationError('Required argument "service" is missing.')

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    if "tag" in kwargs:
        query_params["tag"] = kwargs["tag"]

    if "passing" in kwargs:
        query_params["passing"] = kwargs["passing"]

    function = f"health/service/{service}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def health_state(consul_url=None, token=None, state=None, **kwargs):
    """
    Returns the checks in the state provided on the path.

    :param consul_url: The Consul server URL.
    :param state: The state to show checks for. The supported states
                  are any, unknown, passing, warning, or critical.
                  The any state is a wildcard that can be used to
                  return all checks.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :return: The checks in the provided state.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.health_state state='redis1'

        salt '*' consul.health_state service='redis1' passing='True'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not state:
        raise SaltInvocationError('Required argument "state" is missing.')

    if "dc" in kwargs:
        query_params["dc"] = kwargs["dc"]

    if state not in ("any", "unknown", "passing", "warning", "critical"):
        ret["message"] = "State must be any, unknown, passing, warning, or critical."
        ret["res"] = False
        return ret

    function = f"health/state/{state}"
    ret = _query(
        consul_url=consul_url, function=function, token=token, query_params=query_params
    )
    return ret


def status_leader(consul_url=None, token=None):
    """
    Returns the current Raft leader

    :param consul_url: The Consul server URL.
    :return: The address of the Raft leader.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.status_leader

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "status/leader"
    ret = _query(consul_url=consul_url, function=function, token=token)
    return ret


def status_peers(consul_url, token=None):
    """
    Returns the current Raft peer set

    :param consul_url: The Consul server URL.
    :return: Retrieves the Raft peers for the
             datacenter in which the agent is running.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.status_peers

    """
    ret = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "status/peers"
    ret = _query(consul_url=consul_url, function=function, token=token)
    return ret


def acl_create(consul_url=None, token=None, **kwargs):
    """
    Create a new ACL token.

    :param consul_url: The Consul server URL.
    :param name: Meaningful indicator of the ACL's purpose.
    :param token: Authentication token
    :param type: Type is either client or management. A management
                 token is comparable to a root user and has the
                 ability to perform any action including creating,
                 modifying, and deleting ACLs.
    :param rules: The Consul server URL.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.acl_create

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "name" in kwargs:
        data["Name"] = kwargs["name"]
    else:
        raise SaltInvocationError('Required argument "name" is missing.')

    if "type" in kwargs:
        data["Type"] = kwargs["type"]

    if "rules" in kwargs:
        rules_str = ""
        rules = kwargs["rules"]
        for item in rules:
            for k, v in item.items():
                if k != "policy":
                    rules_str += f'{k} "{v}" {{\n'
                else:
                    rules_str += f'  {k} = "{v}"\n}}\n'

        data["Rules"] = rules_str

    function = "acl/create"
    res = _query(
        consul_url=consul_url, token=token, data=data, method="PUT", function=function
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = "ACL {} created.".format(kwargs["name"])
    else:
        ret["res"] = False
        ret["message"] = "Removing Catalog item {} failed.".format(kwargs["name"])
    return ret


def acl_update(consul_url=None, token=None, **kwargs):
    """
    Update an ACL token.

    :param consul_url: The Consul server URL.
    :param name: Meaningful indicator of the ACL's purpose.
    :param id: Unique identifier for the ACL to update.
    :param type: Type is either client or management. A management
                 token is comparable to a root user and has the
                 ability to perform any action including creating,
                 modifying, and deleting ACLs.
    :param rules: The Consul server URL.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.acl_update

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "id" in kwargs:
        data["ID"] = kwargs["id"]
    else:
        ret["message"] = 'Required parameter "id" is missing.'
        ret["res"] = False
        return ret

    if "name" in kwargs:
        data["Name"] = kwargs["name"]
    else:
        raise SaltInvocationError('Required argument "name" is missing.')

    if "type" in kwargs:
        data["Type"] = kwargs["type"]

    if "rules" in kwargs:
        data["Rules"] = kwargs["rules"]

    function = "acl/update"
    res = _query(
        consul_url=consul_url, token=token, data=data, method="PUT", function=function
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = "ACL {} created.".format(kwargs["name"])
    else:
        ret["res"] = False
        ret["message"] = "Updating ACL {} failed.".format(kwargs["name"])

    return ret


def acl_delete(consul_url=None, token=None, **kwargs):
    """
    Delete an ACL token.

    :param consul_url: The Consul server URL.
    :param id: Unique identifier for the ACL to update.
    :return: Boolean & message of success or failure.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.acl_delete id='c1c4d223-91cb-3d1f-1ee8-f2af9e7b6716'

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "id" not in kwargs:
        ret["message"] = 'Required parameter "id" is missing.'
        ret["res"] = False
        return ret

    function = "acl/destroy/{}".format(kwargs["id"])
    res = _query(
        consul_url=consul_url, token=token, data=data, method="PUT", function=function
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = "ACL {} deleted.".format(kwargs["id"])
    else:
        ret["res"] = False
        ret["message"] = "Removing ACL {} failed.".format(kwargs["id"])
        ret["changes"] = res

    return ret


def acl_info(consul_url=None, **kwargs):
    """
    Information about an ACL token.

    :param consul_url: The Consul server URL.
    :param id: Unique identifier for the ACL to update.
    :return: Information about the ACL requested.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.acl_info id='c1c4d223-91cb-3d1f-1ee8-f2af9e7b6716'

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "id" not in kwargs:
        ret["message"] = 'Required parameter "id" is missing.'
        ret["res"] = False
        return ret

    function = "acl/info/{}".format(kwargs["id"])
    ret = _query(consul_url=consul_url, data=data, method="GET", function=function)
    return ret


def acl_clone(consul_url=None, token=None, **kwargs):
    """
    Information about an ACL token.

    :param consul_url: The Consul server URL.
    :param id: Unique identifier for the ACL to update.
    :return: Boolean, message of success or
             failure, and new ID of cloned ACL.

    CLI Example:

    .. code-block:: bash

        salt '*' consul.acl_info id='c1c4d223-91cb-3d1f-1ee8-f2af9e7b6716'

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "id" not in kwargs:
        ret["message"] = 'Required parameter "id" is missing.'
        ret["res"] = False
        return ret

    function = "acl/clone/{}".format(kwargs["id"])
    res = _query(
        consul_url=consul_url, token=token, data=data, method="PUT", function=function
    )
    if res["res"]:
        ret["res"] = True
        ret["message"] = "ACL {} cloned.".format(kwargs["name"])
        ret["ID"] = res["data"]
    else:
        ret["res"] = False
        ret["message"] = "Cloning ACL item {} failed.".format(kwargs["name"])
    return ret


def acl_list(consul_url=None, token=None, **kwargs):
    """
    List the ACL tokens.

    :param consul_url: The Consul server URL.
    :return: List of ACLs

    CLI Example:

    .. code-block:: bash

        salt '*' consul.acl_list

    """
    ret = {}
    data = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    function = "acl/list"
    ret = _query(
        consul_url=consul_url, token=token, data=data, method="GET", function=function
    )
    return ret


def event_fire(consul_url=None, token=None, name=None, **kwargs):
    """
    List the ACL tokens.

    :param consul_url: The Consul server URL.
    :param name: The name of the event to fire.
    :param dc: By default, the datacenter of the agent is queried;
               however, the dc can be provided using the "dc" parameter.
    :param node: Filter by node name.
    :param service: Filter by service name.
    :param tag: Filter by tag name.
    :return: List of ACLs

    CLI Example:

    .. code-block:: bash

        salt '*' consul.event_fire name='deploy'

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if not name:
        raise SaltInvocationError('Required argument "name" is missing.')

    if "dc" in kwargs:
        query_params = kwargs["dc"]

    if "node" in kwargs:
        query_params = kwargs["node"]

    if "service" in kwargs:
        query_params = kwargs["service"]

    if "tag" in kwargs:
        query_params = kwargs["tag"]

    function = f"event/fire/{name}"
    res = _query(
        consul_url=consul_url,
        token=token,
        query_params=query_params,
        method="PUT",
        function=function,
    )

    if res["res"]:
        ret["res"] = True
        ret["message"] = f"Event {name} fired."
        ret["data"] = res["data"]
    else:
        ret["res"] = False
        ret["message"] = "Cloning ACL item {} failed.".format(kwargs["name"])
    return ret


def event_list(consul_url=None, token=None, **kwargs):
    """
    List the recent events.

    :param consul_url: The Consul server URL.
    :param name: The name of the event to fire.
    :return: List of ACLs

    CLI Example:

    .. code-block:: bash

        salt '*' consul.event_list

    """
    ret = {}
    query_params = {}
    if not consul_url:
        consul_url = _get_config()
        if not consul_url:
            log.error("No Consul URL found.")
            ret["message"] = "No Consul URL found."
            ret["res"] = False
            return ret

    if "name" in kwargs:
        query_params = kwargs["name"]
    else:
        raise SaltInvocationError('Required argument "name" is missing.')

    function = "event/list/"
    ret = _query(
        consul_url=consul_url, token=token, query_params=query_params, function=function
    )
    return ret
