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

# Author: Lars Michelsen <lm@mathias-kettner.de>
# Modified for Dell Sensors By: Chris Bowlby <cbowlby@tenthpowertech.com>

# Tested with Dell PowerConnect 5448 and 5424 models.
# Relevant SNMP OIDs:
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.1.67109249 = INTEGER: 67109249
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.1.67109250 = INTEGER: 67109250
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.1.67109251 = INTEGER: 67109251
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.2.67109249 = STRING: "fan1_unit1"
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.2.67109250 = STRING: "fan2_unit1"
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.2.67109251 = STRING: "fan3_unit1"
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.3.67109249 = INTEGER: 1
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.3.67109250 = INTEGER: 1
# .1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1.3.67109251 = INTEGER: 1

# Status codes:
# 1 => normal,
# 2 => warning,
# 3 => critical,
# 4 => shutdown,
# 5 => notPresent,
# 6 => notFunctioning

# GENERAL MAPS:
dell_powerconnect_fans_status_map = {
    "1": "normal",
    "2": "warning",
    "3": "critical",
    "4": "shutdown",
    "5": "notPresent",
    "6": "notFunctioning",
}
dell_powerconnect_fans_status2nagios_map = {
    "normal": 0,
    "warning": 1,
    "critical": 2,
    "shutdown": 3,
    "notPresent": 1,
    "notFunctioning": 2,
}
dell_powerconnect_fans_state_performance_map = {"1": 1, "2": 0, "3": 0, "4": 0, "5": 0, "6": 0}


# Inventory of all fan related elements
def inventory_dell_powerconnect_fans(info):
    inventory = []
    for _device_id, name, state in info:
        if dell_powerconnect_fans_status_map[state] != "notPresent":
            inventory.append((name, None))
    return inventory


# The check for the states and details of each fan
def check_dell_powerconnect_fans(item, _not_used, info):
    for _device_id, name, state in info:
        if name == item:
            dell_powerconnect_status = dell_powerconnect_fans_status_map[state]
            status = dell_powerconnect_fans_status2nagios_map[dell_powerconnect_status]

            return (status, 'Condition of FAN "%s" is %s' % (name, dell_powerconnect_status))

    return (3, "item not found in snmp data")


# Auto-detection of fan related details.

check_info["dell_powerconnect_fans"] = {
    "check_function": check_dell_powerconnect_fans,
    "inventory_function": inventory_dell_powerconnect_fans,
    "service_description": "Sensor %s",
    "has_perfdata": True,
    "snmp_info": (".1.3.6.1.4.1.674.10895.3000.1.2.110.7.1.1", ["1", "2", "3"]),
    "snmp_scan_function": lambda oid: ".1.3.6.1.4.1.674.10895" in oid(".1.3.6.1.2.1.1.2.0")
    or ".1.3.6.1.4.1.6027.1.3.22" in oid(".1.3.6.1.2.1.1.2.0"),
}
