#!/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([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()