From 589f77ef0e27e9df0d52558f67fdd2cb9791fce4 Mon Sep 17 00:00:00 2001
From: damencho <damencho@jitsi.org>
Date: Tue, 4 Apr 2017 13:25:37 -0500
Subject: [PATCH] Adds prosody plugin that query existing rooms for
 information.

Queries room for their size or room particiapnt's information. Depends on luarocks net-url module.
---
 resources/prosody-plugins/mod_muc_size.lua | 127 +++++++++++++++++++++
 1 file changed, 127 insertions(+)
 create mode 100644 resources/prosody-plugins/mod_muc_size.lua

diff --git a/resources/prosody-plugins/mod_muc_size.lua b/resources/prosody-plugins/mod_muc_size.lua
new file mode 100644
index 000000000..7a6265821
--- /dev/null
+++ b/resources/prosody-plugins/mod_muc_size.lua
@@ -0,0 +1,127 @@
+-- Prosody IM
+-- Copyright (C) 2017 Atlassian
+--
+-- This project is MIT/X11 licensed. Please see the
+-- COPYING file in the source package for more information.
+--
+-- This module requires net-url module
+-- Install it using #luarocks install net-url
+
+module:set_global(); -- Global module
+
+local split_jid = require "util.jid".split;
+local st = require "util.stanza";
+local it = require "util.iterators";
+local json = require "util.json";
+local iterators = require "util.iterators";
+local array = require"util.array";
+
+local tostring = tostring;
+local neturl = require "net.url";
+local parse = neturl.parseQuery;
+
+function get_room_from_jid(jid)
+	local node, host = split_jid(jid);
+	local component = hosts[host];
+	if component then
+		local muc = component.modules.muc
+		if muc and rawget(muc,"rooms") then
+			-- We're running 0.9.x or 0.10 (old MUC API)
+			return muc.rooms[jid];
+		elseif muc and rawget(muc,"get_room_from_jid") then
+			-- We're running >0.10 (new MUC API)
+			return muc.get_room_from_jid(jid);
+		else
+			return
+		end
+	end
+end
+
+function handle_get_room_size(event)
+	local params = parse(event.request.url.query);
+	local room_name = params["room"];
+	local room = get_room_from_jid(room_name);
+	local participant_count = 0;
+
+	log("debug", "Querying room %s", tostring(room_name));
+
+	if room then
+		local occupants = room._occupants;
+		if occupants then
+			participant_count = iterators.count(room:each_occupant());
+		end
+		log("debug", "there are %s occupants in room", tostring(participant_count));
+	else
+		log("debug", "no such room exists");
+	end
+
+	if participant_count > 1 then
+		participant_count = participant_count - 1;
+	end
+
+	local GET_response = {
+		headers = {
+			content_type = "application/json";
+		};
+		body = [[{"participants":]]..participant_count..[[}]];
+	};
+	return GET_response;
+end
+
+function handle_get_room (event)
+	local params = parse(event.request.url.query);
+	local room_name = params["room"];
+	local room = get_room_from_jid(room_name);
+	local participant_count = 0;
+	local occupants_json = array();
+
+	log("debug", "Querying room %s", tostring(room_name));
+
+	if room then
+		local occupants = room._occupants;
+		if occupants then
+			participant_count = iterators.count(room:each_occupant());
+			for _, occupant in room:each_occupant() do
+			    -- filter focus as we keep it as hidden participant
+			    if string.sub(occupant.nick,-string.len("/focus"))~="/focus" then
+				    for _, pr in occupant:each_session() do
+					local nick = pr:get_child_text("nick", "http://jabber.org/protocol/nick") or "";
+					local email = pr:get_child_text("email") or "";
+					occupants_json:push({
+					    jid = tostring(occupant.nick),
+					    email = tostring(email),
+					    display_name = tostring(nick)});
+				    end
+			    end
+			end
+		end
+		log("debug", "there are %s occupants in room", tostring(participant_count));
+	else
+		log("debug", "no such room exists");
+	end
+
+	if participant_count > 1 then
+		participant_count = participant_count - 1;
+	end
+
+	local GET_response = {
+		headers = {
+			content_type = "application/json";
+		};
+		body = json.encode(occupants_json);
+	};
+	return GET_response;
+end;
+
+function module.add_host(module)
+	module:depends("http");
+	module:provides("http", {
+		default_path = "/";
+		route = {
+			["GET room-size"] = handle_get_room_size;
+			["GET sessions"] = function () return tostring(it.count(it.keys(prosody.full_sessions))); end;
+			["GET room"] = handle_get_room;
+		};
+	});
+end
+