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


def inventory_casa_fan(info):
    inventory = []
    for nr, _speed in info[0]:
        inventory.append((nr, None))
    return inventory


def check_casa_fan(item, no_params, info):
    for idx, (nr, speed) in enumerate(info[0]):
        if item == nr:
            fan_status = info[1][idx][1]
            if fan_status == "1":
                return (0, "%s RPM" % speed)
            elif fan_status == "3":
                return (1, "%s RPM, running over threshold (!)" % speed)
            elif fan_status == "2":
                return (1, "%s RPM, running under threshold (!)" % speed)
            elif fan_status == "0":
                return (3, "%s RPM, unknown fan status (!)" % speed)
            elif fan_status == "4":
                return (2, "FAN Failure (!!)")
    return (3, "Fan %s not found in snmp output" % item)


check_info["casa_fan"] = {
    "check_function": check_casa_fan,
    "inventory_function": inventory_casa_fan,
    "service_description": "Fan %s",
    "snmp_scan_function": lambda oid: oid(".1.3.6.1.2.1.1.2.0").startswith(".1.3.6.1.4.1.20858.2."),
    "snmp_info": [
        (".1.3.6.1.4.1.20858.10.31.1.1.1.2", [OID_END, ""]),  # FAN Speed
        (".1.3.6.1.4.1.20858.10.33.1.4.1.4", [OID_END, ""]),  # FAN State
    ],
}
