+
−module:set_global();
+
−
+
−local stats = module:require("mod_statistics/stats");
+
−local filters = require "util.filters";
+
−local serialize = require "util.serialization".serialize;
+
−
+
−local cached_values = {};
+
−
+
−local sessions = {};
+
−
+
−local function push_stat(conn, name, value)
+
−	local value_str = serialize(value);
+
−	return conn:write((("STAT %q (%s)\n"):format(name, value_str):gsub("\\\n", "\\n")));
+
−end
+
−
+
−local function push_stat_to_all(name, value)
+
−	for conn in pairs(sessions) do
+
−		push_stat(conn, name, value);
+
−	end
+
−end
+
−
+
−local session_stats_tpl = ([[{
+
−	message_in = %d, message_out = %d;
+
−	presence_in = %d, presence_out = %d;
+
−	iq_in = %d, iq_out = %d;
+
−	bytes_in = %d, bytes_out = %d;
+
−}]]):gsub("%s", "");
+
−
+
−
+
−local jid_fields = {
+
−	c2s = "full_jid";
+
−	s2sin = "from_host";
+
−	s2sout = "to_host";
+
−	component = "host";
+
−};
+
−
+
−local function push_session_to_all(session, stats)
+
−	local id = tostring(session):match("[a-f0-9]+$"); -- FIXME: Better id? :/
+
−	local stanzas_in, stanzas_out = stats.stanzas_in, stats.stanzas_out;
+
−	local s = (session_stats_tpl):format(
+
−		stanzas_in.message, stanzas_out.message,
+
−		stanzas_in.presence, stanzas_out.presence,
+
−		stanzas_in.iq, stanzas_out.iq,
+
−		stats.bytes_in, stats.bytes_out);
+
−	local jid = session[jid_fields[session.type]] or "";
+
−	for conn in pairs(sessions) do
+
−		conn:write(("SESS %q %q %s\n"):format(id, jid, s));
+
−	end
+
−end
+
−
+
−local available_stats = stats.stats;
+
−local active_sessions = stats.active_sessions;
+
−
+
−-- Handle statistics provided by other modules
+
−local function item_handlers(host)
+
−	host = host and (host.."/") or "";
+
−
+
−	return function (event) -- Added
+
−		local stats = event.item.statistics;
+
−		local group = host..(stats.name and (stats.name.."::") or "");
+
−		for name, stat in pairs(stats) do
+
−			available_stats[group..name] = stat;
+
−		end
+
−	end, function (event) -- Removed
+
−		local stats = event.item.statistics;
+
−		local group = host..(stats.name and (stats.name.."::") or "");
+
−		for name, stat in pairs(stats) do
+
−			available_stats[group..name] = nil;
+
−		end
+
−	end;
+
−end
+
−
+
−module:handle_items("statistics-provider", item_handlers());
+
−function module.add_host(module)
+
−	module:handle_items("statistics-provider", item_handlers(module.host));
+
−end
+
−
+
−-- Network listener
+
−local listener = {};
+
−
+
−function listener.onconnect(conn)
+
−	sessions[conn] = {};
+
−	push_stat(conn, "version", prosody.version);
+
−	for name, value in pairs(cached_values) do
+
−		push_stat(conn, name, value);
+
−	end
+
−	conn:write("\n"); -- Signal end of first batch (for non-streaming clients)
+
−end
+
−
+
−function listener.onincoming(conn, data)
+
−end
+
−
+
−function listener.ondisconnect(conn)
+
−	sessions[conn] = nil;
+
−end
+
−
+
−function listener.onreadtimeout()
+
−	return true;
+
−end
+
−
+
−function module.load()
+
−	if not(prosody and prosody.arg) then
+
−		return;
+
−	end
+
−	filters.add_filter_hook(stats.filter_hook);
+
−
+
−	module:add_timer(1, function ()
+
−		for stat_name, stat in pairs(available_stats) do
+
−			if stat.get then
+
−				local cached = cached_values[stat_name];
+
−				local new_value = stat.get();
+
−				if new_value ~= cached then
+
−					push_stat_to_all(stat_name, new_value);
+
−					cached_values[stat_name] = new_value;
+
−				end
+
−			end
+
−		end
+
−		for session, session_stats in pairs(active_sessions) do
+
−			active_sessions[session] = nil;
+
−			push_session_to_all(session, session_stats);
+
−		end
+
−		return 1;
+
−	end);
+
−
+
−end
+
−function module.unload()
+
−	filters.remove_filter_hook(stats.filter_hook);
+
−end
+
−function module.command( args )
+
−	local command = args[1];
+
−	if command == "top" then
+
−		local dir = module:get_directory();
+
−		package.path = dir.."/?.lua;"..dir.."/?.lib.lua;"..package.path;
+
−		local prosodytop = module:require "prosodytop";
+
−		prosodytop.run();
+
−	end
+
−end
+
−
+
−if prosody and prosody.arg then
+
−	module:provides("net", {
+
−		default_port = 5782;
+
−		listener = listener;
+
−		private = true;
+
−	});
+
−end