#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (C) 2019 tribe29 GmbH - License: GNU General Public License v2
# This file is part of Checkmk (https://checkmk.com). It is subject to the terms and
# conditions defined in the file COPYING, which is part of this source code package.

# NOTE: Careful when replacing the *-import below with a more specific import. This can cause
# problems because it might remove variables from the check-context which are necessary for
# resolving legacy discovery results such as [("SUMMARY", "diskstat_default_levels")]. Furthermore,
# it might also remove variables needed for accessing discovery rulesets.
from cmk.base.check_legacy_includes.couchbase import *  # pylint: disable=wildcard-import,unused-wildcard-import


def discover_couchbase_nodes_services(parsed):
    for key, data in parsed.items():
        yield key, {"discovered_services": data.get("services", [])}


@get_parsed_item_data
def check_couchbase_nodes_services(_item, params, data):
    services_present = set(data.get("services", []))
    services_discovered = set(params["discovered_services"])

    services_appeared = services_present - services_discovered
    services_vanished = services_discovered - services_present
    services_unchanged = services_discovered & services_present

    if services_vanished:
        srt = sorted(services_vanished)
        yield 2, "%d services vanished: %s" % (len(srt), ", ".join(srt))
    if services_appeared:
        srt = sorted(services_appeared)
        yield 2, "%d services appeared: %s" % (len(srt), ", ".join(srt))

    srt = sorted(services_unchanged)
    yield 0, "%d services unchanged: %s" % (len(srt), ", ".join(srt))


check_info["couchbase_nodes_services"] = {
    "parse_function": parse_couchbase_lines,
    "inventory_function": discover_couchbase_nodes_services,
    "check_function": check_couchbase_nodes_services,
    "service_description": "Couchbase %s Services",
    "has_perfdata": False,
}
