#!/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_blade_bx_powermod(info):
    for line in info:
        yield (line[0], None)


def check_blade_bx_powermod(item, _no_param, info):
    power_status = {
        "1": ("unknown", 3),
        "2": ("ok", 0),
        "3": ("not-present", 2),
        "4": ("error", 2),
        "5": ("critical", 2),
        "6": ("off", 2),
        "7": ("dummy", 2),
        "8": ("fanmodule", 0),
    }
    for line in info:
        index, status, product_name = line
        if not index == item:
            continue
        state_readable, state = power_status[status]
    return state, "%s Status is %s" % (product_name, state_readable)


check_info["blade_bx_powermod"] = {
    "inventory_function": inventory_blade_bx_powermod,
    "check_function": check_blade_bx_powermod,
    "service_description": "Power Module %s",
    "snmp_info": (
        ".1.3.6.1.4.1.7244.1.1.1.3.2.4.1",
        [
            1,  # s31SysPowerSupplyUnitId
            2,  # s31SysPowerSupplyUnitStatus
            4,  # s31SysPowerSupplyUnitProductName
        ],
    ),
    "snmp_scan_function": lambda oid: "BX600" in oid(".1.3.6.1.2.1.1.1.0")
    or oid(".1.3.6.1.2.1.1.2.0") == ".1.3.6.1.4.1.7244.1.1.1",
}
