#!/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.temperature import *  # pylint: disable=wildcard-import,unused-wildcard-import

# Example:
# .1.3.6.1.4.1.89.53.15.1.9.1 = INTEGER: 42
# .1.3.6.1.4.1.89.53.15.1.10.1 = INTEGER: ok(1)

# Temperature is in Celcius by default.
# Tested with Dell PowerConnect 5448 and 5424 models.

factory_settings["dell_powerconnect_temp_default_values"] = {
    "levels": (35, 40),
}


def parse_dell_powerconnect_temp(info):
    try:
        temp_str, dev_status = info[0]
    except (IndexError, ValueError):
        return None
    try:
        temp = float(temp_str)
    except ValueError:
        temp = None
    return (
        temp,
        {
            "1": "OK",
            "2": "unavailable",
            "3": "non operational",
        }.get(dev_status, "unknown[%s]" % dev_status),
    )


def inventory_dell_powerconnect_temp(parsed):
    if parsed:
        return [("Ambient", {})]


def check_dell_powerconnect_temp(_no_item, params, parsed):
    if parsed is None:
        return

    temp, dev_state_readable = parsed
    if dev_state_readable == "OK":
        state = 0
    elif dev_state_readable == "unavailable":
        state = 1
    elif dev_state_readable == "non operational":
        state = 2
    else:
        state = 3

    if temp is None:
        return state, "Status: %s" % dev_state_readable
    return check_temperature(
        temp, params, "dell_powerconnect", dev_status=state, dev_status_name=dev_state_readable
    )


check_info["dell_powerconnect_temp"] = {
    "parse_function": parse_dell_powerconnect_temp,
    "check_function": check_dell_powerconnect_temp,
    "inventory_function": inventory_dell_powerconnect_temp,
    "service_description": "Temperature %s",
    "default_levels_variable": "dell_powerconnect_temp_default_values",
    "has_perfdata": True,
    "snmp_info": (
        ".1.3.6.1.4.1.89.53.15.1",
        [
            "9",  # RADLAN-Physicaldescription-MIB::rlPhdUnitEnvParamTempSensorValue
            "10",  # RADLAN-Physicaldescription-MIB::rlPhdUnitEnvParamTempSensorStatus
        ],
    ),
    "snmp_scan_function": lambda oid: oid(".1.3.6.1.2.1.1.2.0").startswith(
        ".1.3.6.1.4.1.674.10895"
    ),
    "group": "temperature",
}
