local it = require "util.iterators";

local log = require "util.logger".init("stats");

local last_cpu_wall, last_cpu_clock;

local get_time = require "socket".gettime;

local active_sessions, active_jids = {}, {};

local s2s_sessions = module:shared"/*/s2s/sessions";

local c2s_sessions = module:shared"/*/c2s/sessions";

local stats = {

	total_users = {

		get = function () return it.count(it.keys(bare_sessions)); end

	};

	total_c2s = {

		get = function () return it.count(it.keys(full_sessions)); end

	};

	total_s2sin = {

		get = function () local i = 0; for conn,sess in next,s2s_sessions do if sess.direction == "incoming" then i = i + 1 end end return i end

	};

	total_s2sout = {

		get = function () local i = 0; for conn,sess in next,s2s_sessions do if sess.direction == "outgoing" then i = i + 1 end end return i end

	};

	total_s2s = {

		get = function () return it.count(it.keys(s2s_sessions)); end

	};

	total_component = {

		get = function ()

			local count = 0;

			for host, host_session in pairs(hosts) do

				if host_session.type == "component" then

					local c = host_session.modules.component;

					if c and c.connected then -- 0.9 only

						count = count + 1;

					end

				end

			end

			return count;

		end

	};

	up_since = {

		get = function () return prosody.start_time; end;

		tostring = function (up_since)

			return tostring(os.time()-up_since).."s";

		end;

	};

	};

	time = {

		tostring = function () return os.date("%T"); end;

	};

	cpu = {

		get = function ()

			local new_wall, new_clock = get_time(), os.clock();

			local pc = 0;

			if last_cpu_wall then

				pc = 100/((new_wall-last_cpu_wall)/(new_clock-last_cpu_clock));

			end

			last_cpu_wall, last_cpu_clock = new_wall, new_clock;

			return math.ceil(pc);

		end;

		tostring = "%s%%";

	};

};

local add_statistics_filter; -- forward decl

if prosody and prosody.arg then -- ensures we aren't in prosodyctl

	setmetatable(active_sessions, {

		__index = function ( t, k )

			local v = {

				bytes_in = 0, bytes_out = 0;

				stanzas_in = {

					message = 0, presence = 0, iq = 0;

				};

				stanzas_out = {

					message = 0, presence = 0, iq = 0;

				};

			}

			rawset(t, k, v);

			return v;

		end

	});

	local filters = require "util.filters";

	local function handle_stanza_in(stanza, session)

		local s = active_sessions[session].stanzas_in;

		local n = s[stanza.name];

		if n then

			s[stanza.name] = n + 1;

		end

		return stanza;

	end

	local function handle_stanza_out(stanza, session)

		local s = active_sessions[session].stanzas_out;

		local n = s[stanza.name];

		if n then

			s[stanza.name] = n + 1;

		end

		return stanza;

	end

	local function handle_bytes_in(bytes, session)

		local s = active_sessions[session];

		s.bytes_in = s.bytes_in + #bytes;

		return bytes;

	end

	local function handle_bytes_out(bytes, session)

		local s = active_sessions[session];

		s.bytes_out = s.bytes_out + #bytes;

		return bytes;

	end

	function add_statistics_filter(session)

		filters.add_filter(session, "stanzas/in", handle_stanza_in);

		filters.add_filter(session, "stanzas/out", handle_stanza_out);

		filters.add_filter(session, "bytes/in", handle_bytes_in);

		filters.add_filter(session, "bytes/out", handle_bytes_out);

	end

end

return {

	stats = stats;

	active_sessions = active_sessions;

	filter_hook = add_statistics_filter;

};