#!/usr/bin/python
# -*- coding:utf-8 -*-

# Author: Pablo Saavedra Rodinho
# Contact: pablo.saavedra@interoud.com

"""
Save a group of streams from http or udp and generates a db with related info

"""

from optparse import OptionParser

import os
import sys
import subprocess

reload(sys)
sys.setdefaultencoding('utf-8')

import ConfigParser
import simplejson as json


import time
from datetime import datetime
import struct
import socket

import urllib2
import httplib2

DEV_NULL=open('/dev/null', "w")

conffile = "./chic.cfg"

logfile = "/dev/stdout"
loglevel = 20
workdir = "./chic/"
template_name = "Template_Sources"
check_name = "stream"

_template = '''<?xml version="1.0" encoding="UTF-8"?>
<zabbix_export>
    <version>2.0</version>
    <date>2013-07-25T22:04:07Z</date>
    <groups>
        <group>
            <name>Templates</name>
        </group>
    </groups>
    <templates>
        <template>
            <template>%(template_name)s</template>
            <name>%(template_name)s</name>
            <groups>
                <group>
                    <name>Templates</name>
                </group>
            </groups>
            <applications/>
            <items>
            %(items)s
            </items>
            <discovery_rules/>
            <macros/>
            <templates/>
            <screens/>
        </template>
    </templates>
    <triggers>
    %(triggers)s
    </triggers>
</zabbix_export>

'''

_i = '''
                <item>
                    <name>%(check_name)s $1</name>
                    <type>0</type>
                    <snmp_community/>
                    <multiplier>0</multiplier>
                    <snmp_oid/>
                    <key>%(check_name)s[%(key)s]</key>
                    <delay>180</delay>
                    <history>90</history>
                    <trends>365</trends>
                    <status>0</status>
                    <value_type>4</value_type>
                    <allowed_hosts/>
                    <units/>
                    <delta>0</delta>
                    <snmpv3_securityname/>
                    <snmpv3_securitylevel>0</snmpv3_securitylevel>
                    <snmpv3_authpassphrase/>
                    <snmpv3_privpassphrase/>
                    <formula>1</formula>
                    <delay_flex/>
                    <params/>
                    <ipmi_sensor/>
                    <data_type>0</data_type>
                    <authtype>0</authtype>
                    <username/>
                    <password/>
                    <publickey/>
                    <privatekey/>
                    <port/>
                    <description/>
                    <inventory_link>0</inventory_link>
                    <applications/>
                    <valuemap/>
                </item>
'''

_t = '''
        <trigger>
            <expression>{%(template_name)s:%(check_name)s[%(key)s].diff(0)}=1 | {%(template_name)s:%(check_name)s[%(key)s].str([Ok])}=0</expression>
            <name>STREAM - %(key)s down of %(template_name)s on {HOST.NAME}</name>
            <url/>
            <status>0</status>
            <priority>%(priority)s</priority>
            <description/>
            <type>0</type>
            <dependencies/>
        </trigger>
'''

_t2 = '''
        <trigger>
            <expression>{%(template_name)s:%(check_name)s[%(key)s].diff(0)}=1</expression>
            <name>STREAM - Some source down of %(template_name)s on {HOST.NAME}</name>
            <url/>
            <status>0</status>
            <priority>%(priority)s</priority>
            <description/>
            <type>0</type>
            <dependencies/>
        </trigger>
'''



# command line options parser ##################################################

parser = OptionParser()

parser.add_option("-f", "--file", dest="conffile",
        help="The configuration file (default: %s)" % conffile,
        default=conffile)
parser.add_option("-l", "--logfile",
        dest="logfile", help="Log file (default: %s)" % logfile,
        default=logfile)
parser.add_option("--loglevel",
        dest="loglevel", help="Log level (default: %s)" % loglevel,
        default=loglevel)
parser.add_option("-t", "--template", dest="template_name",
        help="Template name (default: %s)" % template_name,
        default=template_name)
parser.add_option("-c", "--check", dest="check_name",
        help="Zabbix custmon check name (default: %s)" % check_name,
        default=check_name)


(options, args) = parser.parse_args()

conffile = options.conffile

logfile = options.logfile
loglevel = options.loglevel
template_name = options.template_name
check_name = options.check_name

# logging ######################################################################
import logging
hdlr = logging.FileHandler(logfile)
hdlr.setFormatter(logging.Formatter('%(levelname)s %(asctime)s %(message)s'))
logger = logging.getLogger('chic')
logger.addHandler(hdlr)
logger.setLevel(int(loglevel))


# setting up ###################################################################

logger.info("Default encoding: %s" % sys.getdefaultencoding())

################################################################################


def main():
    global conffile
    global template_name

    res = {}

    try:
      # RawConfigParser not interpolate attribute values
      cfg = ConfigParser.RawConfigParser()
      cfg.readfp(file(conffile))

    except Exception, e:
        print ("Conffile %s error: %s" % (conffile,e))

    items = ""
    triggers = ""

    items += _i % {
        "key": "all",
        "template_name":template_name,
        "check_name":check_name,
    }
    triggers += _t2 % {
        "key": "all",
        "template_name":template_name,
        "check_name":check_name,
        "priority":4,
    }
    for s in cfg.sections():
        key = s
        try:
          trigger = True
          if s == 'global':
              for o in cfg.options(s):
                  try:
                      vars()[o] = cfg.get(s,o)
                  except Exception, e:
                      print "Error parsing %s - %s: %s" % (s,o,e)
              continue

          for o in cfg.options(s):
              trigger = eval (cfg.get(s,o)) if o == 'trigger' else trigger

          if trigger:
              items += _i % {
                          "key": key,
                          "template_name":template_name,
                          "check_name":check_name,
                      }
              triggers += _t % {
                          "key": key,
                          "template_name":template_name,
                          "check_name":check_name,
                          "priority":2,
                      }
        except Exception, e:
          logger.error ("Problem on %s key: %s" % (key,e))

    print _template % {
            "items":items,
            "triggers":triggers,
            "template_name":template_name,
            "check_name":check_name,
            }

# main #########################################################################

if __name__ == '__main__':
    main()

