# 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.

from sos.report.plugins import Plugin, IndependentPlugin


class RabbitMQ(Plugin, IndependentPlugin):

    short_desc = 'RabbitMQ messaging service'
    plugin_name = 'rabbitmq'
    profiles = ('services',)
    var_puppet_gen = "/var/lib/config-data/puppet-generated/rabbitmq"
    files = (
        '/etc/rabbitmq/rabbitmq.conf',
        var_puppet_gen + '/etc/rabbitmq/rabbitmq.config'
    )
    packages = ('rabbitmq-server',)

    def setup(self):
        in_container = False
        container_names = []
        _containers = self.get_containers()
        for _con in _containers:
            if _con[1].startswith('rabbitmq'):
                in_container = True
                container_names.append(_con[1])

        if in_container:
            for container in container_names:
                self.add_container_logs(container)
                self.add_cmd_output(
                    'rabbitmqctl report',
                    container=container,
                    foreground=True,
                    tags="rabbitmq_report"
                )
                self.add_cmd_output(
                    "rabbitmqctl eval 'rabbit_diagnostics:maybe_stuck().'",
                    container=container,
                    foreground=True,
                    timeout=10
                )
        else:
            self.add_cmd_output("rabbitmqctl report")
            self.add_cmd_output(
                "rabbitmqctl eval 'rabbit_diagnostics:maybe_stuck().'",
                timeout=10)

        self.add_copy_spec([
            "/etc/rabbitmq/*",
            self.var_puppet_gen + "/etc/rabbitmq/*",
            self.var_puppet_gen + "/etc/security/limits.d/",
            self.var_puppet_gen + "/etc/systemd/"
        ])
        self.add_copy_spec([
            "/var/log/rabbitmq/*",
        ])

        self.add_file_tags({
            "/var/log/rabbitmq/rabbit@.*[^-sasl].log": "rabbitmq_logs",
            "/var/log/rabbitmq/startup_err": "rabbitmq_startup_err"
        })

        # Crash dump can be large in some situation but it is useful to
        # investigate why rabbitmq crashes. So capture the file without
        # sizelimit
        self.add_copy_spec([
            "/var/log/containers/rabbitmq/erl_crash.dump"
        ], sizelimit=0)

    def postproc(self):
        self.do_file_sub("/etc/rabbitmq/rabbitmq.conf",
                         r"(\s*default_pass\s*,\s*)\S+", r"\1<<***>>},")

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