Diff

mod_stanza_counter/mod_stanza_counter.lua @ 434:ac4a6cc5776c

mod_stanza_counter: cleanup, corrections. (prosody starts, web stats work)
author Marco Cirillo <maranda@lightwitch.org>
date Tue, 20 Sep 2011 22:05:14 +0000
parent 433:967757965dbd
child 435:b6abe463b4fc
line wrap: on
line diff
--- a/mod_stanza_counter/mod_stanza_counter.lua	Tue Sep 20 21:21:06 2011 +0000
+++ b/mod_stanza_counter/mod_stanza_counter.lua	Tue Sep 20 22:05:14 2011 +0000
@@ -6,41 +6,13 @@
 
 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" }})
+local ports = module:get_option_array("stanza_counter_ports" or {{ port = 5280 }})
 
 -- 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>\n"
+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>\n"
 
-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
-</html>\n"
+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</html>\n"
 
 local function res(code, r, h)
 	local response = {
@@ -53,7 +25,7 @@
 end
 
 local function req(method, body, request)
-	if method = "GET" then
+	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"],
@@ -78,7 +50,7 @@
 
 -- init http interface
 local function init_web()
-	httpserver.new_from_config(ports, req)
+	httpserver.new_from_config(ports, req, { base = "stanza-counter" })
 end
 
 -- Setup on server start
@@ -91,7 +63,7 @@
 	return function(self)
 		local origin, stanza = self.origin, self.stanza
 		if not prosody.stanza_counter then init_counter() end
-		if check
+		if check then
 			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
@@ -102,11 +74,11 @@
 	end
 end
 
-local function message_callback(check)
+local function mes_callback(check)
 	return function(self)
 		local origin, stanza = self.origin, self.stanza
 		if not prosody.stanza_counter then init_counter() end
-		if check
+		if check then
 			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
@@ -117,11 +89,11 @@
 	end
 end
 
-local function presence_callback(check)
+local function pre_callback(check)
 	return function(self)
 		local origin, stanza = self.origin, self.stanza
 		if not prosody.stanza_counter then init_counter() end
-		if check
+		if check then
 			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