Munin contributed plugins

#!/usr/bin/env python

# -*- coding: utf-8 -*-

# Copyright (c) 2010 Christoph Heer (Christoph.Heer@googlemail.com)

#

# Permission is hereby granted, free of charge, to any person obtaining a

# copy of this software and associated documentation files (the \"Software\"),

# to deal in the Software without restriction, including without limitation

# the rights to use, copy, modify, merge, publish, distribute, sublicense,

# and/or sell copies of the Software, and to permit persons to whom the

# Software is furnished to do so, subject to the following conditions:

#

# The above copyright notice and this permission notice shall be included in

# all copies or substantial portions of the Software.

#

# THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL

# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER

# DEALINGS IN THE SOFTWARE.

import sys

import os

import telnetlib

import re

def main(): 

    try:

        mode = sys.argv[1]

    except IndexError:

        mode = ""

    wildcard = sys.argv[0].split("prosody_")[1].split("_")[0]

    host = os.environ.get('host', 'localhost')

    port = int(os.environ.get('port', 5582))

    if mode == "suggest":

        print "c2s"

        print "s2s"

        print "presence"

        print "uptime"

        print "users"

        sys.exit(0)

    if wildcard == "c2s":

        if mode == "config":

            print "graph_title Prosody C2S Connections"

            print "graph_vlabel users"

            print "graph_category Prosody"

            print "all_client_connections.label client connections"

            print "secure_client_connections.label secure client connections"

            print "insecure_client_connections.label insecure client " \

                  "connections"

            sys.exit(0)

        else:

            connection_count_re = re.compile(r"Total:\s(\d+)\s")

            telnet = telnetlib.Telnet(host, port)

            telnet.write("c2s:show_secure()\n")

            telnet_response = telnet.read_until("secure client connections",

                                                5)

            parsed_info = connection_count_re.findall(telnet_response)

            secure_client_connections = int(parsed_info[0])

            print "secure_client_connections.value %s" % \

                  (secure_client_connections)

            telnet.write("c2s:show_insecure()\n")

            telnet_response = telnet.read_until("insecure client connections",

                                                5)

            parsed_info = connection_count_re.findall(telnet_response)

            insecure_client_connections = int(parsed_info[0])

            print "insecure_client_connections.value %s" % \

                  (insecure_client_connections)

            all_client_connections = secure_client_connections + \

                                     insecure_client_connections

            print "all_client_connections.value %s" % (all_client_connections)

            telnet.write("quit")

    elif wildcard == "s2s":

        if mode == "config":

            print "graph_title Prosody S2S Connections"

            print "graph_vlabel servers"

            print "graph_category Prosody"

            print "outgoing_connections.label outgoing connections"

            print "incoming_connections.label incoming connections"

            sys.exit(0)

        else:   

            server_connections_re = re.compile(r"(\d+) outgoing, (\d+)")

            telnet = telnetlib.Telnet(host, port)

            telnet.write("s2s:show()\n")

            telnet_response = telnet.read_until("connections", 5)

            parsed_info = server_connections_re.findall(telnet_response)

            print "outgoing_connections.value %s" % (parsed_info[0][0])

            print "incoming_connections.value %s" % (parsed_info[0][1])

            telnet.write("quit")

    elif wildcard == "presence":

        if mode == "config":

            print "graph_title Prosody Client Presence"

            print "graph_vlabel clients"

            print "graph_category Prosody"

            print "available.label Avaible Clients"

            print "chat.label Ready for Chat Clients"

            print "away.label Away Clients"

            print "xa.label Extended Away Clients"

            print "dnd.label Do Not Disturb Clients"

            sys.exit(0)

        else:   

            client_presence_re = re.compile(r"- (.*?)\(\d+\)")

            telnet = telnetlib.Telnet(host, port)

            telnet.write("c2s:show()\n")

            telnet_response = telnet.read_until("clients", 5)

            parsed_info = client_presence_re.findall(telnet_response)

            print "available.value %s" % (parsed_info.count("available"))

            print "chat.value %s" % (parsed_info.count("chat"))

            print "away.value %s" % (parsed_info.count("away"))

            print "xa.value %s" % (parsed_info.count("xa"))

            print "dnd.value %s" % (parsed_info.count("dnd"))

            telnet.write("quit")

    elif wildcard == "uptime":

        if mode == "config":

            print "graph_title Prosody Uptime"

            print "graph_args --base 1000 -l 0"

            print "graph_scale no"

            print "graph_vlabel uptime in days"

            print "graph_category Prosody"

            print "graph_order uptime"

            print "uptime.draw AREA"

            print "uptime.min U"

            print "uptime.max U"

            print "uptime.label uptime"

            print "uptime.type GAUGE"

            sys.exit(0)

        else:

            uptime_re = re.compile(r"\d+")

            telnet = telnetlib.Telnet(host, port)

            telnet.write("server:uptime()\n")

            telnet_response = telnet.read_until("minutes (", 5)

            parsed_info = uptime_re.findall(telnet_response)

            uptime_value = float(parsed_info[0]) + float(parsed_info[1])/24 +\

                           float(parsed_info[2])/60/24

            print "uptime.value %s" % (uptime_value)

            telnet.write("quit")

    elif wildcard == "users":

        if mode == "config":

            print "graph_title Prosody Registered Users"

            print "graph_vlabel users"

            print "graph_category Prosody"

        base_dir = os.environ.get('internal_storage_path', "/var/lib/prosody")

        if os.path.isdir(base_dir):

            vhosts = listdirs(base_dir)

            for vhost in vhosts:

                account_dir = os.path.join(base_dir, vhost, "accounts")

                if os.path.isdir(account_dir):

                    vhost = vhost.replace("%2e",".")

                    munin_var = vhost.replace(".","_")

                    if mode == "config":

                        print "%s.label %s" % (munin_var, vhost)

                    else:

                        accounts = len(list(listfiles(account_dir)))

                        print "%s.value %s" % (munin_var, accounts)

def listdirs(folder):

    for x in os.listdir(folder):

        if os.path.isdir(os.path.join(folder, x)):

            yield x

def listfiles(folder):

    for x in os.listdir(folder):

        if os.path.isfile(os.path.join(folder, x)):

            yield x

if __name__ == '__main__':

    main()