#!/usr/bin/python
# -*- encoding: utf-8; py-indent-offset: 4 -*-
# +------------------------------------------------------------------+
# |             ____ _               _        __  __ _  __           |
# |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
# |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
# |           | |___| | | |  __/ (__|   <    | |  | | . \            |
# |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
# |                                                                  |
# | Copyright Mathias Kettner 2016             mk@mathias-kettner.de |
# +------------------------------------------------------------------+
#
# This file is part of Check_MK.
# The official homepage is at http://mathias-kettner.de/check_mk.
#
# check_mk is free software;  you can redistribute it and/or modify it
# under the  terms of the  GNU General Public License  as published by
# the Free Software Foundation in version 2.  check_mk is  distributed
# in the hope that it will be useful, but WITHOUT ANY WARRANTY;  with-
# out even the implied warranty of  MERCHANTABILITY  or  FITNESS FOR A
# PARTICULAR PURPOSE. See the  GNU General Public License for more de-
# ails.  You should have  received  a copy of the  GNU  General Public
# License along with GNU Make; see the file  COPYING.  If  not,  write
# to the Free Software Foundation, Inc., 51 Franklin St,  Fifth Floor,
# Boston, MA 02110-1301 USA.

# .1.3.6.1.2.1.47.1.1.1.1.4.10 1
# .1.3.6.1.2.1.47.1.1.1.1.2.10 Chassis Temperature Sensor
# .1.3.6.1.2.1.47.1.1.1.1.5.10 8
# .1.3.6.1.2.1.47.1.1.1.1.7.10 Chassis Temperature Sensor
# .1.3.6.1.2.1.47.1.1.1.1.11.10
# .1.3.6.1.2.1.47.1.1.1.1.13.10

# Note: Some bogus devices such as Cisco 2500 Series WLC or
# Cisco Wide Area Application Services do not have exactly
# one entry == "0" which means device itself or do not provide
# any information about physical class.
# In the first case {{Hardware > System}} will be omitted.
# In the second case we inventorize these components below physical
# components table {{Unknown entities}}.
# In all cases ports are ignored because we have an extra view
# for them.


def inv_snmp_extended_info(info):
    map_type = {
        "1": ("Other", "others"),
        "2": ("Unknown", "unknowns"),
        "3": ("Chassis", "chassis"),
        "4": ("Backplane", "backplanes"),
        "5": ("Container", "containers"),
        "6": ("PSU", "psus"),
        "7": ("Fan", "fans"),
        "8": ("Sensor", "sensors"),
        "9": ("Module", "modules"),
        "10": ("Port", "ports"),
        "11": ("Stack", "stacks"),
    }

    parsed = {}
    count_parents = 0
    for child, description, parent, child_type, name, software, \
        serial, manufacturer, model in info:
        if parent == "0":
            count_parents += 1
        if child_type in map_type:
            parsed.setdefault(
                child,
                (parent, description, child_type, name, software, serial, manufacturer, model))
        else:
            parsed.setdefault(
                child, (parent, description, "2", name, software, serial, manufacturer, model))

    node = None
    for index, (parent, description, entity_type, name, software, \
        serial, manufacturer, model) in parsed.items():
        if count_parents == 1 and parent == "0":
            node = inv_tree("hardware.system.")
            if serial:
                node["serial"] = serial
            if model:
                node["model"] = model

        elif entity_type != "10":
            if parsed.get(parent):
                location_info = (map_type[parsed[parent][2]][0], parent)
            elif parent == "0":
                location_info = ("Device", "0")
            else:
                location_info = ("Missing in ENTITY table", parent)
            node = inv_tree_list("hardware.components.%s:" % (map_type[entity_type][1]))
            node.append({
                "index": index,
                "name": name,
                "description": description,
                "software": software,
                "serial": serial,
                "manufacturer": manufacturer,
                "model": model,
                "location": "%s (%s)" % location_info,
            })

    return node


inv_info['snmp_extended_info'] = {
    'inv_function': inv_snmp_extended_info,
    'snmp_info': (
        '.1.3.6.1.2.1.47.1.1.1.1',
        [
            OID_END,
            "2",  # entPhysicalDescr
            "4",  # entPhysicalContainedIn
            "5",  # entPhysicalClass
            "7",  # entPhysicalName
            "10",  # entPhysicalSoftwareRev (NEW)
            "11",  # entPhysicalSerialNum
            "12",  # entPhysicalMfgName (NEW)
            "13",  # entPhysicalModelName
        ]),
    'snmp_scan_function': lambda oid: oid(".1.3.6.1.2.1.47.1.1.1.1.*"),
}
