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

# Agent / MIB output
# SysChassisPowerSupplyEntry ::=
#        SEQUENCE {
#                sysChassisPowerSupplyIndex                   INTEGER,
#                sysChassisPowerSupplyStatus                  INTEGER
#        }

# sysChassisPowerSupplyStatus
#   bad(0),
#   good(1),
#   notpresent(2)


def inventory_f5_bigip_psu(info):
    inventory = []
    for line in info:
        psu = line[0]
        state = line[1]
        # inventorize the PSU unless it's in state 2 (notpresent)
        if state != "2":
            inventory.append((psu, None))
    return inventory


def check_f5_bigip_psu(item, _no_params, info):
    for line in info:
        psu = line[0]
        state = int(line[1])
        if psu == item:
            if state == 1:
                return (0, "PSU state: good")
            elif state == 0:
                return (2, "PSU state: bad!!")
            elif state == 2:
                return (1, "PSU state: notpresent!")
            return (3, "PSU state is unknown")

    return (3, "item not found in SNMP output")


check_info["f5_bigip_psu"] = {
    "check_function": check_f5_bigip_psu,
    "inventory_function": inventory_f5_bigip_psu,
    "service_description": "PSU %s",
    # Get ID and status from the SysChassisPowerSupplyTable
    "snmp_info": (".1.3.6.1.4.1.3375.2.1.3.2.2.2.1", [1, 2]),
    "snmp_scan_function": scan_f5_bigip,  # Check using the vendor mib Id and verify it's a loadbalancer
}
