Changeset

432:23a6289a00bd

mod_stanza_counter: initial draft.
author Marco Cirillo <maranda@lightwitch.org>
date Tue, 20 Sep 2011 21:15:29 +0000
parents 431:fb7898826026
children 433:967757965dbd
files mod_stanza_counter/mod_stanza_counter.lua
diffstat 1 files changed, 163 insertions(+), 0 deletions(-) [+]
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mod_stanza_counter/mod_stanza_counter.lua	Tue Sep 20 21:15:29 2011 +0000
@@ -0,0 +1,163 @@
+-- (C) 2011, Marco Cirillo (LW.Org)
+-- General Stanzas' Counter with web output.
+
+local jid_bare = require "util.jid".bare
+local httpserver = require "net.httpserver"
+
+module.host = "*" -- Needs to be global for stats web wise.
+
+local ports = module:get_option_array("stanza_counter_ports" or {{ port = 5280; base = "stanza-counter" }})
+
+-- http handlers
+
+local r_200 = "\n
+<html>\n
+<head>\n
+<title>Prosody's Stanza Counter</title>\n
+<meta name=\"robots\" content=\"noindex, nofollow\" />\n
+</head>\n
+\n
+<body>\n
+<h3>Incoming and Outgoing stanzas divided per type</h3>\n
+<p><strong>Incoming IQs</strong>: %d<br/>\n
+<strong>Outgoing IQs</strong>: %d<br/>\n
+<strong>Incoming Messages</strong>: %d<br/>\n
+<strong>Outgoing Messages</strong>: %d<br/>\n
+<strong>Incoming Presences</strong>: %d<br/>\n
+<strong>Outgoing Presences</strong>: %d<p>\n
+</body>\n
+\n
+</html>"
+
+local r_405 = "\n
+<html>\n
+<head>\n
+<title>Prosody's Stanza Counter - Error</title>\n
+<meta name=\"robots\" content=\"noindex, nofollow\" />\n
+</head>\n
+\n
+<body>\n
+<h3>Bad Method ... I only support GET.</h3>\n
+</body>\n
+\n"
+
+local function res(code, r, h)
+	local response = {
+		status = code;
+		body = r;
+		}
+	
+        if h then response.headers = h; end
+        return response
+end
+
+local function req(method, body, request)
+	if method = "GET" then
+		local forge_res = r_200:format(prosody.stanza_counter.iq["incoming"],
+					       prosody.stanza_counter.iq["outgoing"],
+					       prosody.stanza_counter.message["incoming"],
+					       prosody.stanza_counter.message["outgoing"],
+					       prosody.stanza_counter.presence["incoming"],
+					       prosody.stanza_counter.presence["outgoing"]);
+		return res(200, forge_res)
+	else
+		return res(405, r_405, {["Allow"] = "GET"})
+	end
+end
+
+-- Setup, Init functions.
+-- initialize function counter table on the global object on start
+local function init_counter()
+	prosody.stanza_counter = { 
+		iq = { incoming=0, outgoing=0 },
+		message = { incoming=0, outgoing=0 },
+		presence = { incoming=0, outgoing=0 }
+	}
+end
+
+-- init http interface
+local function init_web()
+	httpserver.new_from_config(ports, req)
+end
+
+-- Setup on server start
+local function setup()
+	init_counter(); init_web();
+end
+
+-- Basic Stanzas' Counters
+local function iq_callback(check)
+	return function(self)
+		local origin, stanza = self.origin, self.stanza
+		if not prosody.stanza_counter then init_counter() end
+		if check
+			if not stanza.attr.to or hosts[jid_bare(stanza.attr.to)] then return nil;
+			else
+				prosody.stanza_counter.iq["outgoing"] = prosody.stanza_counter.iq["outgoing"] + 1
+			end
+		else
+			prosody.stanza_counter.iq["incoming"] = prosody.stanza_counter.iq["incoming"] + 1
+		end
+	end
+end
+
+local function message_callback(check)
+	return function(self)
+		local origin, stanza = self.origin, self.stanza
+		if not prosody.stanza_counter then init_counter() end
+		if check
+			if not stanza.attr.to or hosts[jid_bare(stanza.attr.to)] then return nil;
+			else
+				prosody.stanza_counter.message["outgoing"] = prosody.stanza_counter.message["outgoing"] + 1
+			end
+		else
+			prosody.stanza_counter.message["incoming"] = prosody.stanza_counter.message["incoming"] + 1
+		end
+	end
+end
+
+local function presence_callback(check)
+	return function(self)
+		local origin, stanza = self.origin, self.stanza
+		if not prosody.stanza_counter then init_counter() end
+		if check
+			if not stanza.attr.to or hosts[jid_bare(stanza.attr.to)] then return nil;
+			else
+				prosody.stanza_counter.presence["outgoing"] = prosody.stanza_counter.presence["outgoing"] + 1
+			end
+		else
+			prosody.stanza_counter.presence["incoming"] = prosody.stanza_counter.presence["incoming"] + 1
+		end
+	end
+end
+
+
+
+-- Hook all pre-stanza events.
+module:hook("pre-iq/bare", iq_callback(true), 140)
+module:hook("pre-iq/full", iq_callback(true), 140)
+module:hook("pre-iq/host", iq_callback(true), 140)
+
+module:hook("pre-message/bare", mes_callback(true), 140)
+module:hook("pre-message/full", mes_callback(true), 140)
+module:hook("pre-message/host", mes_callback(true), 140)
+
+module:hook("pre-presence/bare", pre_callback(true), 140)
+module:hook("pre-presence/full", pre_callback(true), 140)
+module:hook("pre-presence/host", pre_callback(true), 140)
+
+-- Hook all stanza events.
+module:hook("iq/bare", iq_callback(false), 140)
+module:hook("iq/full", iq_callback(false), 140)
+module:hook("iq/host", iq_callback(false), 140)
+
+module:hook("message/bare", mes_callback(false), 140)
+module:hook("message/full", mes_callback(false), 140)
+module:hook("message/host", mes_callback(false), 140)
+
+module:hook("presence/bare", pre_callback(false), 140)
+module:hook("presence/full", pre_callback(false), 140)
+module:hook("presence/host", pre_callback(false), 140)
+
+-- Hook server start to initialize the counter.
+module:hook("server-started", setup)