Diff

mod_http_user_count/mod_http_user_count.lua @ 1830:a45f2f79e99b

mod_http_user_count: Support for host and MUC session counts
author Matthew Wild <mwild1@gmail.com>
date Sat, 05 Sep 2015 23:08:40 +0100
parent 1263:fe29627a5ed8
line wrap: on
line diff
--- a/mod_http_user_count/mod_http_user_count.lua	Fri Sep 04 00:41:29 2015 +0200
+++ b/mod_http_user_count/mod_http_user_count.lua	Sat Sep 05 23:08:40 2015 +0100
@@ -1,10 +1,33 @@
 local it = require "util.iterators";
+local jid_split = require "util.jid".prepped_split;
 
 module:depends("http");
 
+local function check_muc(jid)
+	local room_name, host = jid_split(jid);
+	if not hosts[host] then
+		return nil, "No such host: "..host;
+	elseif not hosts[host].modules.muc then
+		return nil, "Host '"..host.."' is not a MUC service";
+	end
+	return room_name, host;
+end
+
 module:provides("http", {
     route = {
         ["GET /sessions"] = function () return tostring(it.count(it.keys(prosody.full_sessions))); end;
         ["GET /users"] = function () return tostring(it.count(it.keys(prosody.bare_sessions))); end;
+        ["GET /host"] = function () return tostring(it.count(it.keys(prosody.hosts[module.host].sessions))); end;
+        ["GET /room/*"] = function (request, room_jid)
+        	local name, host = check_muc(room_jid);
+        	if not name then
+        		return "0";
+        	end
+       		local room = prosody.hosts[host].modules.muc.rooms[name.."@"..host];
+       		if not room then
+       			return "0";
+       		end
+        	return tostring(it.count(it.keys(room._occupants)));
+        end;
     };
 });