#!/usr/bin/python
# -*- encoding: utf-8; py-indent-offset: 4 -*-
# +------------------------------------------------------------------+
# |             ____ _               _        __  __ _  __           |
# |            / ___| |__   ___  ___| | __   |  \/  | |/ /           |
# |           | |   | '_ \ / _ \/ __| |/ /   | |\/| | ' /            |
# |           | |___| | | |  __/ (__|   <    | |  | | . \            |
# |            \____|_| |_|\___|\___|_|\_\___|_|  |_|_|\_\           |
# |                                                                  |
# | Copyright Mathias Kettner 2010             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.

# Author: Lars Michelsen <lm@mathias-kettner.de>

hp_proliant_mem_type_map = {
                1: 'other',
                2: 'board',
                3: 'cpqSingleWidthModule',
                4: 'cpqDoubleWidthModule',
                5: 'simm',
                6: 'pcmcia',
                7: 'compaq-specific',
                8: 'dimm',
                9: 'smallOutlineDimm',
                10: 'rimm',
                11: 'srimm',
                12: 'fb-dimm',
}
hp_proliant_mem_status_map = {
                1: "other",
                2: "notPresent",
                3: "present",
                4: "good",
                5: "add",
                6: "upgrade",
                7: "missing",
                8: "doesNotMatch",
                9: "notSupported",
                10: "badConfig",
                11: "degraded",
}
hp_proliant_mem_status2nagios_map = {
                'n/a': 3,
                'other': 3,
                'notPresent': 3,
                'present': 1,
                'good': 0,
                'add': 1,
                'upgrade': 1,
                'missing': 2,
                'doesNotMatch': 2,
                'notSupported': 2,
                'badConfig': 2,
                'degraded': 2,
}

hp_proliant_mem_condition_status2nagios_map = { 'other': 3, 'ok': 0, 'degraded': 2, 'failed': 2 }
hp_proliant_mem_condition_map = {
                0: 'n/a',
                1: 'other',
                2: 'ok',
                3: 'degraded',
}

def inventory_hp_proliant_mem(info):
    if len(info) > 0:
        return [ (line[1], None) for line in info if int(line[2]) > 0 and int(line[4]) != 2 ]

def check_hp_proliant_mem(item, params, info):
    for line in info:
        if line[1] == item:
            board_index, module_index, module_size, \
            module_type, module_status, module_condition = line

            module_size_mb = int(module_size) / 1024

            type = 'n/a'
            if int(module_type) in hp_proliant_mem_type_map:
                type = hp_proliant_mem_type_map[int(module_type)]

            snmp_status = 'n/a'
            if int(module_status) in hp_proliant_mem_status_map:
                snmp_status = hp_proliant_mem_status_map[int(module_status)]

            detail_output = ', Status: %s ' % snmp_status
            status = hp_proliant_mem_status2nagios_map[snmp_status]
            if status > 0:
                detail_output += '(%s) ' % nagios_state_names[status]

            condition = 'n/a'
            if int(module_condition) in hp_proliant_mem_condition_map:
                condition = hp_proliant_mem_condition_map[int(module_condition)]
            condition_status = hp_proliant_mem_condition_status2nagios_map[condition]

            detail_output += ', Condition: %s ' % condition
            if condition_status > 0:
                detail_output += '(%s) ' % nagios_state_names[condition_status]
            if condition_status > status:
                status = condition_status

            return (status, '%s - Board: %s, Num: %s, Type: %s, Size: %s MB%s' %
                            (nagios_state_names[status], board_index, module_index,
                             type, module_size_mb, detail_output))
    return (3, "UNKNOWN - item not found in snmp data")


check_info['hp_proliant_mem'] = (check_hp_proliant_mem, "HW Mem %s", 0,  inventory_hp_proliant_mem)
snmp_info['hp_proliant_mem']  = ( ".1.3.6.1.4.1.232.6.2.14.13.1", [  "2", # cpqHeResMem2BoardNum
                                                                     "1", # cpqHeResMem2Module
                                                                     "6", # cpqHeResMem2ModuleSize
                                                                     "7", # cpqHeResMem2ModuleType
                                                                     "19", # cpqHeResMem2ModuleStatus
                                                                     "20", # cpqHeResMem2ModuleCondition
                                                                ] )

snmp_scan_functions['hp_proliant_mem'] = \
    lambda oid: "proliant" in oid(".1.3.6.1.4.1.232.2.2.4.2.0").lower()
