#!/usr/bin/python3

import json
import SNMPUtil
import argparse

### Monitoring iDRAC Servers - Performance
### 
### This plugin can be used to get the overall status of the hardwares of iDRAC Servers
### It uses snmpwalk command to get the hadrware data from the iDRAC Servers.
### SNMPUtil.py is used to get the snmp raw data and parsed to get the output json
### Download and install the latest version of Site24x7 Linux Agent. The agent will execute the plugin and push the data to the Site24x7 server
### 
### Author: Anita, Zoho Corp
### Language : Python
### Tested in Ubuntu
### Tested for snmp version 1 and 2c

### iDRAC Server Configuration Details

OIDREPLACE = "1.3.6.1.4.1"
SMISTR="SNMPv2-SMI::enterprises"
ISOSTR="iso.3.6.1.4.1" 


OIDS = {
        'battery'   : ['1.3.6.1.4.1.674.10892.5.4.600.50.1'], 
        'cpu'       : ['1.3.6.1.4.1.674.10892.5.4.1100.30.1'],  
        'fan'       : ['1.3.6.1.4.1.674.10892.5.4.700.12.1'], 
        'memory'    : ['1.3.6.1.4.1.674.10892.5.4.1100.50.1'],
        'pdisk'     : ['1.3.6.1.4.1.674.10892.5.5.1.20.130.4.1'], 
        'PS'        : ['1.3.6.1.4.1.674.10892.5.4.600.12.1'],
        'PU'        : ['1.3.6.1.4.1.674.10892.5.4.600.10.1'],
        'sensor'    : ['1.3.6.1.4.1.674.10892.5.4.700.20.1'],  
        'vdisk'     : ['1.3.6.1.4.1.674.10892.5.5.1.20.140.1.1']
}

hardware = {
        'battery'   : ['1.3.6.1.4.1.674.10892.5.4.600.50.1.5'],
        'cpu'       : ['1.3.6.1.4.1.674.10892.5.4.1100.30.1.17'],
        'fan'       : ['1.3.6.1.4.1.674.10892.5.4.700.12.1.6'],
        'memory'    : ['1.3.6.1.4.1.674.10892.5.4.1100.50.1.14'],
        'pdisk'     : ['1.3.6.1.4.1.674.10892.5.5.1.20.130.4.1.11'],
        'PS'        : ['1.3.6.1.4.1.674.10892.5.4.600.12.1.5'],
        'PU'        : ['1.3.6.1.4.1.674.10892.5.4.600.10.1.8'],
        'sensor'    : ['1.3.6.1.4.1.674.10892.5.4.700.20.1.6'],
        'vdisk'     : ['1.3.6.1.4.1.674.10892.5.5.1.20.140.1.1.23']
}


names = {
        'battery'   : ['status'],
        'cpu'       : ['cores'],
        'fan'       : [{'rotations':'rpm'}],
        'memory'    : [{'size': 'KB'}],
        'pdisk'     : [{'size': 'MB'}],
        'PS'        :  ['status'], 
        'PU'        : ['status'],
        'sensor'    : ['reading'],
        'vdisk'     : ['badblocks']
        }



class HardwareParser:
    def __init__(self, hostname, snmp_version, snmp_community_str, mib_location):
        self.hostname = hostname
        self.snmp_version = snmp_version
        self.snmp_community_str = snmp_community_str
        self.mib_location = mib_location
        
        self.hardware = ''
        self.oids = ''
        self.pattern = ''
        
    def getData(self):
        output_data = {}
        output_data['data'] = {}
        output_data['units'] = {}
        for _ in OIDS:
            self.hardware = _
            self.oids = OIDS[self.hardware]
            
            for _ in self.oids:
                try:
                    snmpdata = SNMPUtil.SNMPPARSER('snmpwalk',self.hostname, self.snmp_version, self.snmp_community_str,_, self.mib_location, hardware[self.hardware])
                    self.snmp_data = snmpdata.getRawData()
                    output_data = self.parseSNMPData(output_data)
                except ValueError as e:
                    raise ValueError(e)
        return output_data
    
    def parseSNMPData(self,output_data):
        jsondata = output_data['data'] 
        unitdata = output_data['units']
        try:
        
            for _ in self.snmp_data:
                if ( not _.startswith(OIDREPLACE) and _.startswith(SMISTR) ):
                    _ = _.replace(SMISTR, OIDREPLACE)
                elif ( not _.startswith(OIDREPLACE) and _.startswith(ISOSTR) ):
                    _ = _.replace(ISOSTR, OIDREPLACE)
                #print(_)
            
                for index, __ in enumerate(hardware[self.hardware]) :
                    if __ in _:        
                        _ = _.replace('\n','').replace('\r','')
                        name = _.split(' ')[0]
                        elementname = name[len(name)-1]
                        l = _.split(' ')
                        l.pop(0)
                        value = ' '.join(l)
                    
                        #print(name, elementname, value)
                    
                        if ':' in value:
                            val = value.split(':')[1:] 
                            value = val[len(val)-1]
                        
                        elem = names[self.hardware][index]
                        attribute = ''
                        unit = ''
                    
                        if type(elem) is str:
                            attribute = elem
                        elif type(elem) is dict:
                            attribute = list(elem.keys())[0]
                            unit = elem[list(elem.keys())[0]]
                        
                        key = (self.hardware +'_'+  attribute +'_'+elementname).replace(' ','')
                        jsondata[key] = value
                     
                        if unit!='':
                            unitdata[key] = unit 
                            
        except Exception as e:
            jsondata['status']=0
            jsondata['msg']=str(e)
                 
        output_data['data'] = jsondata
        output_data['units'] = unitdata

        return output_data



if __name__ == '__main__':
    
    result = {}
    
    parser = argparse.ArgumentParser()
    parser.add_argument('--hostname', help='hostname', nargs='?', default='localhost')
    parser.add_argument('--snmp_version', help='snmp version', type=str, nargs='?', default="2c")
    parser.add_argument('--snmp_community_str', help='snmp community version',  nargs='?', default='public')
    parser.add_argument('--idrac_mib_file_locn', help='idrac mib file location',  nargs='?', default='')
    parser.add_argument('--plugin_version', help='plugin template version',  nargs='?', default='1')
    parser.add_argument('--heartbeat_required', help='Enable heartbeat for monitoring',  nargs='?', default="true")
    args = parser.parse_args()
    
    try:
        parser = HardwareParser(args.hostname, args.snmp_version, args.snmp_community_str, args.idrac_mib_file_locn)
        output = parser.getData()
        result = output['data']
        result['units'] = output['units']
    except Exception as e:
        result['msg'] = str(e)
    
    result['plugin_version'] = args.plugin_version
    result['heartbeat_required'] = args.heartbeat_required
    print(json.dumps(result))