# Copyright (C) 2018 Amit Ghadge <amitg.b14@gmail.com>
#
# This file is part of the sos project: https://github.com/sosreport/sos
#
# This copyrighted material is made available to anyone wishing to use,
# modify, copy, or redistribute it subject to the terms and conditions of
# version 2 of the GNU General Public License.
#
# See the LICENSE file in the source distribution for further information.

import re
from sos.report.plugins import Plugin, IndependentPlugin


class Elastic(Plugin, IndependentPlugin):

    short_desc = 'ElasticSearch service'
    plugin_name = 'elastic'
    profiles = ('services', )

    packages = ('elasticsearch',)

    def get_hostname_port(self, els_config_file):
        """ Get hostname and port number """
        hostname = "localhost"
        port = "9200"
        try:
            with open(els_config_file, encoding='UTF-8') as fread:
                for line in fread:
                    network_host = re.search(r'(^network.host):(.*)', line)
                    network_port = re.search(r'(^http.port):(.*)', line)
                    if network_host and len(network_host.groups()) == 2:
                        hostname = network_host.groups()[-1].strip()
                        hostname = re.sub(r'"|\'', '', hostname)
                        continue
                    if network_port and len(network_port.groups()) == 2:
                        port = network_port.groups()[-1].strip()
        except Exception as err:  # pylint: disable=broad-except
            self._log_info(f"Failed to parse {els_config_file}: {err}")
        return hostname, port

    def setup(self):
        els_config_file = self.path_join(
            "/etc/elasticsearch/elasticsearch.yml"
        )
        self.add_copy_spec(els_config_file)

        if self.get_option("all_logs"):
            self.add_copy_spec("/var/log/elasticsearch/*")
        else:
            self.add_copy_spec("/var/log/elasticsearch/*.log")

        host, port = self.get_hostname_port(els_config_file)
        endpoint = host + ":" + port
        self.add_cmd_output([
                f"curl -X GET '{endpoint}/_cluster/settings?pretty'",
                f"curl -X GET '{endpoint}/_cluster/health?pretty'",
                f"curl -X GET '{endpoint}/_cluster/stats?pretty'",
                f"curl -X GET '{endpoint}/_cat/nodes?v'",
                f"curl -X GET '{endpoint}/_cat/indices'",
                f"curl -X GET '{endpoint}/_cat/shards'",
                f"curl -X GET '{endpoint}/_cat/aliases'",
        ])

# vim: set et ts=4 sw=4 :