#!/usr/bin/env python
"""
Plugin to monitor Wowza streaming servers.

Author: Srijan Choudhary <srijan4@gmail.com>
Version: 2011031507

This program 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, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

Usage:
        - Link or copy to /etc/munin/plugins
        - To enable extra graphs, also link to one or more of the
        possible links (given below)
        - Then restart munin-node

Links possible:
        <default>                       total number of listeners in wowza
        wowza_duration                  duration of listeners (avg and mdn)

        wowza_vhost_listeners           number of listeners per vhost
        wowza_vhost_duration            duration of listeners per vhost
        wowza_vhost_uptime              uptime of vhosts

        wowza_app_listeners             number of listeners per application
        wowza_app_duration              duration of listeners per application
        wowza_app_uptime                uptime of applications

Configuration:
        - Enter your server, username, and password below
        (The plugin does not need to run on the same host
        as the Wowza media server)
        - Optionally provide clients to exclude
        - Optionally provide apps to exclude
        - Optionally provide vhosts to exclude

Possible TODOs:
        - use autoconf
        - use munin's configuration system

"""
from __future__ import print_function
from sys import argv, exit, stderr
import urllib2
from xml.etree.ElementTree import ElementTree
from os.path import basename

# CONFIGURATION

server = "127.0.0.1:8086"
user = "admin"
pw = "password"

# Exclude these in all calculations
client_exclude = ("127.0.0.1")
app_exclude = ("testapp")
vhost_exclude = ("testhost")

# /CONFIGURATION

url = "http://%s/serverinfo" %server

passman = urllib2.HTTPPasswordMgrWithDefaultRealm()
passman.add_password(None, url, user, pw)
authhandler = urllib2.HTTPDigestAuthHandler(passman)

opener = urllib2.build_opener(authhandler)
urllib2.install_opener(opener)
f = urllib2.urlopen(url)

tree = ElementTree()
tree.parse(f)
f.close()

vhosts = []
for vh in tree.findall("VHost"):
        if vh.find("Name").text not in vhost_exclude:
                applications = []
                for app in vh.findall("Application"):
                        if app.find("Name").text not in app_exclude:
                                if app.find("Status").text == "loaded":
                                        clients = []
                                        for client in app.findall("Client"):
                                                if client.find("IpAddress").text not in client_exclude:
                                                        clients.append({"ClientId": client.find("ClientId").text,
                                                                        "FlashVersion": client.find("FlashVersion").text,
                                                                        "IpAddress": client.find("IpAddress").text,
                                                                        "TimeRunning": float(client.find("TimeRunning").text),
                                                                        "DateStarted": client.find("DateStarted").text,
                                                                        "URI": client.find("URI").text,
                                                                        "Protocol": client.find("Protocol").text,
                                                                        "IsSSL": client.find("IsSSL").text,
                                                                        "IsEncrypted": client.find("IsEncrypted").text,
                                                                        "Port": client.find("Port").text
                                                                        })
                                        applications.append({"Name": app.find("Name").text,
                                                             "Status": app.find("Status").text,
                                                             "TimeRunning": float(app.find("TimeRunning").text),
                                                             "ConnectionsCurrent" : app.find("ConnectionsCurrent").text,
                                                             "ConnectionsTotal": app.find("ConnectionsTotal").text,
                                                             "ConnectionsTotalAccepted" : app.find("ConnectionsTotalAccepted").text,
                                                             "ConnectionsTotalRejected" : app.find("ConnectionsTotalRejected").text,
                                                             "Clients": clients})
                                else:
                                        applications.append({"Name": app.find("Name").text,
                                                             "Status": app.find("Status").text,
                                                             "TimeRunning": float(0),
                                                             "ConnectionsCurrent" : 0,
                                                             "ConnectionsTotal": 0,
                                                             "ConnectionsTotalAccepted" : 0,
                                                             "ConnectionsTotalRejected" : 0,
                                                             "Clients": []})
                vhosts.append({"Name": vh.find("Name").text,
			       "TimeRunning": float(vh.find("TimeRunning").text),
			       "ConnectionsLimit": vh.find("ConnectionsLimit").text,
			       "ConnectionsCurrent" : vh.find("ConnectionsCurrent").text,
			       "ConnectionsTotal": vh.find("ConnectionsTotal").text,
			       "ConnectionsTotalAccepted" : vh.find("ConnectionsTotalAccepted").text,
			       "ConnectionsTotalRejected" : vh.find("ConnectionsTotalRejected").text,
                               "Applications": applications})

plugin_name = basename(argv[0])

try:
	if argv[1] == "config":
		if plugin_name == "wowza_duration":
			print ("graph_title Wowza clients listening duration")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel minutes")
			print ("avg.label average listening duration")
			print ("mdn.label median listening duration")

		elif plugin_name == "wowza_vhost_listeners":
			print ("graph_title Wowza listeners count by vhosts")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel listeners")
			is_first = True
			for vh in vhosts:
                                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                                print (vname,".label vhost: ",vh["Name"],sep='')
				if is_first:
					print (vname,".draw AREA",sep='')
					is_first = False
				else:
					print (vname,".draw STACK",sep='')

		elif plugin_name == "wowza_vhost_duration":
			print ("graph_title Wowza clients listening duration by vhosts")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel minutes")
			for vh in vhosts:
                                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                                print (vname,"_avg.label average listening duration for ",vh["Name"],sep='')
                                print (vname,"_mdn.label median listening duration for ",vh["Name"],sep='')

		elif plugin_name == "wowza_vhost_uptime":
			print ("graph_title Wowza vhosts uptime")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel hours")
			for vh in vhosts:
                                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
				print (vname,".label vhost: ",vh["Name"],sep='')

		elif plugin_name == "wowza_app_listeners":
			print ("graph_title Wowza listeners count by apps")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel listeners")
			is_first = True
			for vh in vhosts:
                                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                                for app in vh["Applications"]:
                                        aname = app["Name"].strip("/").replace(".","_").replace("-","_")
                                        print (vname,"_",aname,".label vhost.app: ",vh["Name"],".",app["Name"],sep='')
                                        if is_first:
                                                print (vname,"_",aname,".draw AREA",sep='')
                                                is_first = False
                                        else:
                                                print (vname,"_",aname,".draw STACK",sep='')

		elif plugin_name == "wowza_app_duration":
			print ("graph_title Wowza clients listening duration by apps")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel minutes")
			for vh in vhosts:
                                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                                for app in vh["Applications"]:
                                        aname = app["Name"].strip("/").replace(".","_").replace("-","_")
                                        print (vname,"_",aname,"_avg.label average listening duration for ",vh["Name"],".",app["Name"],sep='')
                                        print (vname,"_",aname,"_mdn.label median listening duration for ",vh["Name"],".",app["Name"],sep='')

		elif plugin_name == "wowza_app_uptime":
			print ("graph_title Wowza apps uptime")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel hours")
			for vh in vhosts:
                                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                                for app in vh["Applications"]:
                                        aname = app["Name"].strip("/").replace(".","_").replace("-","_")
                                        print (vname,"_",aname,".label vhost.app: ",vh["Name"],".",app["Name"],sep='')

		else:                   # wowza_listeners
                        print ("graph_title Wowza listeners count")
			print ("graph_args --base 1000 -l 0")
			print ("graph_scale no")
			print ("graph_category streaming")
			print ("graph_vlabel listeners")
			print ("wowza_listeners.label Total Listeners")
			print ("wowza_listeners.draw AREA")
                        pass
	exit(0)

except IndexError:
	pass

if plugin_name == "wowza_duration":
        alldurations = []
        for vh in vhosts:
                for app in vh["Applications"]:
                        for client in app["Clients"]:
                                alldurations.append(client["TimeRunning"])
        alldurations.sort()
        if len(alldurations) == 0:
                average = 0
        else:
                average = (sum(alldurations) / float(len(alldurations)) / 60.)
        print ("avg.value ",average,sep='')
        if len(alldurations) % 2:
                median = alldurations[len(alldurations) / 2] / 60.
        elif len(alldurations):
                median = (alldurations[len(alldurations) / 2 - 1] + alldurations[len(alldurations) / 2]) / 2. / 60.
        else:
                median = 0
        print ("mdn.value ",median,sep='')

elif plugin_name == "wowza_vhost_listeners":
        for vh in vhosts:
                print (vh["Name"].strip("/").replace(".","_").replace("-","_"),".value ",vh["ConnectionsCurrent"],sep='')

elif plugin_name == "wowza_vhost_duration":
        alldurations = {}
        for vh in vhosts:
                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                alldurations[vh["Name"]] = []
                for app in vh["Applications"]:
                        for client in app["Clients"]:
                                alldurations[vh["Name"]].append(client["TimeRunning"])
                alldurations[vh["Name"]].sort()
                if len(alldurations[vh["Name"]]) == 0:
                       average = 0
                else:
                       average = (sum(alldurations[vh["Name"]]) / float(len(alldurations[vh["Name"]])) / 60.)
                print (vname,"_avg.value ",average,sep='')
                if len(alldurations[vh["Name"]]) % 2:
                        median = alldurations[vh["Name"]][len(alldurations[vh["Name"]]) / 2] / 60.
                elif len(alldurations):
                        median = (alldurations[vh["Name"]][len(alldurations[vh["Name"]]) / 2 - 1] + alldurations[vh["Name"]][len(alldurations[vh["Name"]]) / 2]) / 2. / 60.
                else:
                        median = 0
                print (vname,"_mdn.value ",median,sep='')

elif plugin_name == "wowza_vhost_uptime":
        for vh in vhosts:
            print (vh["Name"].strip("/").replace(".","_").replace("-","_"),".value ",vh["TimeRunning"]/3600.,sep='')

elif plugin_name == "wowza_app_listeners":
        for vh in vhosts:
                for app in vh["Applications"]:
                        print (vh["Name"].strip("/").replace(".","_").replace("-","_"),"_",app["Name"].strip("/").replace(".","_").replace("-","_"),".value ",app["ConnectionsCurrent"],sep='')

elif plugin_name == "wowza_app_duration":
        alldurations = {}
        for vh in vhosts:
                vname = vh["Name"].strip("/").replace(".","_").replace("-","_")
                for app in vh["Applications"]:
                        aname = app["Name"].strip("/").replace(".","_").replace("-","_")
                        name = ''.join([vname,'_',aname])
                        alldurations[name] = []
                        for client in app["Clients"]:
                                alldurations[name].append(client["TimeRunning"])
                        alldurations[name].sort()
                        if len(alldurations[name]) == 0:
                                average = 0
                        else:
                                average = (sum(alldurations[name]) / float(len(alldurations[name])) / 60.)
                        print (name,"_avg.value ",average,sep='')
                        if len(alldurations[name]) % 2:
                                median = alldurations[name][len(alldurations[name]) / 2] / 60.
                        elif len(alldurations):
                                median = (alldurations[name][len(alldurations[name]) / 2 - 1] + alldurations[name][len(alldurations[name]) / 2]) / 2. / 60.
                        else:
                                median = 0
                        print (name,"_mdn.value ",median,sep='')

elif plugin_name == "wowza_app_uptime":
	for vh in vhosts:
                for app in vh["Applications"]:
                        print (vh["Name"].strip("/").replace(".","_").replace("-","_"),"_",app["Name"].strip("/").replace(".","_").replace("-","_"),".value ",app["TimeRunning"]/3600.,sep='')

else:                   # wowza_listeners
        listeners = 0
        for vh in vhosts:
                listeners += int(vh["ConnectionsCurrent"])
        print ("wowza_listeners.value ",listeners,sep='')