#!/usr/bin/env python
# -*- coding: iso-8859-1 -*-

import os
import sys
import urllib2
from xml.dom import minidom


# Hostname of Icecast server
# Just canonical name, no http:// nor ending /
host = "foo.bar.com"
username = "admin"
# Password for admin access to Icecast2 server to fetch statistics
password = ""
realm = "Icecast2 Server"

# Bitrates the MP3 stream is served with
mp3bitrates = [56, 128]
# Bitrates the Ogg Stream is served with
oggbitrates = [56, 128, 172]

# This plugin shows the statistics of a specific subset of sources connected to an Icecast2 server.
# Place the file in /usr/share/munin/plugins and then symlink to it from
# /etc/munin/plugins
# The name icecast2_total will show total number of listeners on server, as
# well the total number of listeners for any configured stream.
# For each stream with multiple bitrates, create one
# icecast2_streamname
# If the name contains a "-" exchange it with a "_", and the script will change it back for you.
# This is to satisfy internal requirements of Munin.
# For each streamname, the plugin will check for the configured bitrates
# Expecting the mountpoints to be on the form of
# /streamname_<bitrate> for mp3
# /streamname_<bitrate>.ogg for Ogg/Vorbis


def hent_XML():
    auth_handler = urllib2.HTTPBasicAuthHandler()
    auth_handler.add_password(realm, host, username, password)
    opener = urllib2.build_opener(auth_handler)
    urllib2.install_opener(opener)

    xmlweb = urllib2.urlopen("http://%s/admin/stats" % host)
    xml = xmlweb.read()
    xmlweb.close()

    # Parser oversikt

    xmldoc = minidom.parseString(xml)
    xmldoc = xmldoc.firstChild

    # Totalt antall lyttere
    total_lyttere = xmldoc.getElementsByTagName("clients")[0].firstChild.nodeValue
    # Totalt antall kilder
    total_kilder = xmldoc.getElementsByTagName("sources")[0].firstChild.nodeValue
    # Status for enkelt strøm
    sources = xmldoc.getElementsByTagName("source")
    sourcelist = {}
    for source in sources:
        mount = source.getAttribute("mount")
        listeners = source.getElementsByTagName("listeners")[0].firstChild.nodeValue
        name = source.getElementsByTagName("server_name")[0].firstChild.nodeValue
        mount = mount.replace("-", "_")
        sourcelist[mount[1:]] = (listeners, name)

    sourcename = sys.argv[0].split("/")[-1][len("icecast2_"):]
    if len(sys.argv) == 1:
        sys.argv.append("")
    if sys.argv[1] == "autoconf":
        print "yes"
    elif sys.argv[1] == "config":
        if sourcename == "total":
            print "graph_title Totalt antall lyttere"
            print "graph_vlabel lyttere"
            print "graph_category streaming"
            print "totallyttere.label Totalt antall lyttere"
            print "totalkilder.label Totalt antall kilder"
            chanlist = {}
            for a, b, filelist in os.walk("/etc/munin/plugins"):
                for file in filelist:
                    if file.find("icecast2_") != -1:
                        channelname = file[len("icecast2_"):]
                        if channelname != "total" and channelname not in chanlist:
                            chanlist[channelname] = 0
            chanlist = chanlist.keys()
            chanlist.sort()
            for chan in chanlist:
                graphtitle = ""
                for key in sourcelist.keys():
                    if key.find(chan) != -1:
                        l, graphtitle = sourcelist[key]
                        break
                if graphtitle == "":
                    graphtitle = chan
                print "%s.label %s" % (chan, graphtitle)

        else:
            sumstring = ""
            graphtitle = ""
            for key in sourcelist.keys():
                if key.find(sourcename) != -1:
                    l, graphtitle = sourcelist[key]
                    break
            if graphtitle == "":
                graphtitle = sourcename
            print "graph_title %s" % graphtitle
            print "graph_vlabel lyttere"
            print "graph_category streaming"
            for bitrate in mp3bitrates:
                print "%s_%s.label %s-%s" % (sourcename, bitrate,
                                             "/" + sourcename.replace("_", "-"), bitrate)
                sumstring += "%s_%s " % (sourcename, bitrate)
                print "%s_%s.critical -0.5:" % (sourcename, bitrate)
            for bitrate in oggbitrates:
                print "%s_%s_ogg.label %s-%s.ogg" % (sourcename, bitrate,
                                                     "/" + sourcename.replace("_", "-"), bitrate)
                print "%s_%s_ogg.critical -0.5:" % (sourcename, bitrate)
                sumstring += "%s_%s_ogg " % (sourcename, bitrate)
            print "%slyttere.label Totalt antall lyttere" % sourcename
            print "%slyttere.sum %s" % (sourcename, sumstring)
    elif sys.argv[1] != "config":
        if sourcename == "total":
            print "totallyttere.value %s" % total_lyttere
            print "totalkilder.value %s" % total_kilder
            statslist = {}
            for a, b, filelist in os.walk("/etc/munin/plugins"):
                for file in filelist:
                    if file.find("icecast2_") != -1:
                        channelname = file[len("icecast2_"):]
                        if channelname != "total" and channelname not in statslist:
                            statslist[channelname] = 0

            for source in sourcelist:
                listeners, name = sourcelist[source]
                if source[:source.rfind("_")] not in statslist:
                    statslist[source[:source.rfind("_")]] = 0
                statslist[source[:source.rfind("_")]] += int(listeners)
            for stat in statslist:
                print "%s.value %s" % (stat, statslist[stat])
        else:
            for bitrate in mp3bitrates:
                if ("%s_%s" % (sourcename, bitrate)) in sourcelist:
                    listeners = sourcelist["%s_%s" % (sourcename, bitrate)][0]
                    print listeners
                else:
                    listeners = -1
                print "%s_%s.value %s" % (sourcename, bitrate, listeners)
            for bitrate in oggbitrates:
                if ("%s_%s.ogg" % (sourcename, bitrate)) in sourcelist:
                    listeners = sourcelist["%s_%s.ogg" % (sourcename, bitrate)][0]
                else:
                    listeners = -1
                print "%s_%s_ogg.value %s" % (sourcename, bitrate, listeners)
    else:
        print sys.argv[1]


if __name__ == "__main__":
    hent_XML()