import agent_util


class ElasticSearchPlugin(agent_util.Plugin):
    textkey = "elasticsearch"
    label = "Elastic Search"

    @classmethod
    def get_metadata(self, config):
        status = agent_util.SUPPORTED
        msg = None

        if not config:
            self.log.info(
                "The [elasticsearch] config block not found in the config file"
            )
            return {}

        if not "hostname" in config or not "port" in config:
            self.log.info(
                "The [elasticsearch] config block does not contain variables for hostname and/or port"
            )
            return {}

        if not agent_util.which("curl", exc=False):
            self.log.info("curl not found!")
            status = agent_util.UNSUPPORTED
            msg = "Curl is not installed - please install"
            return {}

        options = []

        data = {
            "number_of_nodes": {
                "label": "Total number of nodes in cluster",
                "options": None,
                "status": status,
                "error_message": msg,
            },
            "number_of_data_nodes": {
                "label": "Number of data nodes in cluster",
                "options": None,
                "status": status,
                "error_message": msg,
            },
            "active_primary_shards": {
                "label": "Number of active primary shards",
                "options": None,
                "status": status,
                "error_message": msg,
            },
            "active_shards": {
                "label": "Total number of shards",
                "options": None,
                "status": status,
                "error_message": msg,
            },
            "relocating_shards": {
                "label": "Number of shards getting relocated",
                "options": None,
                "status": status,
                "error_message": msg,
            },
            "initializing_shards": {
                "label": "Number of initializing shards",
                "options": None,
                "status": status,
                "error_message": msg,
            },
            "unassigned_shards": {
                "label": "Number of unassigned shards",
                "options": None,
                "status": status,
                "error_message": msg,
            },
        }
        return data

    def check(self, textkey, data, config):
        user_string = ""
        if config.get("username") and config.get("password"):
            user_string = "--user %s:%s" % (
                config.get("username"),
                config.get("password"),
            )

        cmd = "curl %s %s:%s/_cluster/health" % (
            user_string,
            config["hostname"],
            config["port"],
        )
        ret, output = agent_util.execute_command(cmd)
        reply = agent_util.json_loads(output)
        return int(reply[textkey])