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

# TODO WATORule
factory_settings["informix_tabextents_default_levels"] = {
    "levels": (40, 70),
}


def parse_informix_tabextents(info):
    parsed = {}
    instance = None
    entry = None
    for line in info:
        if instance is not None and line == ["(constant)", "TABEXTENTS"]:
            entry = {}
            parsed.setdefault(instance, [])
            parsed[instance].append(entry)

        elif line[0].startswith("[[[") and line[0].endswith("]]]"):
            instance = line[0][3:-3]

        elif entry is not None:
            entry.setdefault(line[0], line[1])

    return parsed


def inventory_informix_tabextents(parsed):
    return [(instance, {}) for instance in parsed]


def check_informix_tabextents(item, params, parsed):
    if item in parsed:
        max_extents = -1
        long_output = []
        for entry in parsed[item]:
            extents = int(entry["extents"])
            if extents >= max_extents:
                max_extents = extents
            long_output.append(
                "[%s/%s] Extents: %s, Rows: %s"
                % (entry["db"], entry["tab"], entry["extents"], entry["nrows"])
            )

        warn, crit = params["levels"]
        state = 0
        infotext = "Maximal extents: %s" % max_extents
        if max_extents >= crit:
            state = 2
        elif max_extents >= warn:
            state = 1
        if state:
            infotext += " (warn/crit at %s/%s)" % (warn, crit)
        return state, "%s\n%s" % (infotext, "\n".join(long_output)), [("max_extents", max_extents)]


check_info["informix_tabextents"] = {
    "parse_function": parse_informix_tabextents,
    "inventory_function": inventory_informix_tabextents,
    "check_function": check_informix_tabextents,
    "has_perfdata": True,
    "service_description": "Informix Table Extents %s",
    "group": "informix_tabextents",
    "default_levels_variable": "informix_tabextents_default_levels",
}
