-- Copyright (C) 2009 Thilo Cestonaro

-- Copyright (C) 2009 Waqas Hussain

-- Copyright (C) 2009-2013 Matthew Wild

-- Copyright (C) 2013 Kim Alvefur

-- Copyright (C) 2013 Marco Cirillo

local hosts = prosody.hosts;

local tostring = tostring;

local split_jid = require "util.jid".split;

local datamanager = require"core.storagemanager".olddm;

local data_load, data_store = datamanager.load, datamanager.store;

local datastore = "muc_log";

local muc_form_config_option = "muc#roomconfig_enablelogging"

local log_by_default = module:get_option_boolean("muc_log_by_default", false);

local log_presences = module:get_option_boolean("muc_log_presences", true);

-- Module Definitions

local 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

local function logging_enabled(room)

	local enabled = room._data.logging;

	if enabled == nil then

		return log_by_default;

	end

	return enabled;

end

function log_if_needed(event)

	local stanza = event.stanza;

	if (stanza.name == "presence") or

		(stanza.name == "iq") or

		(stanza.name == "message" and tostring(stanza.attr.type) == "groupchat")

	then

		local node, host = split_jid(stanza.attr.to);

		if node and host then

			local bare = node .. "@" .. host;

			if get_room_from_jid(bare) then

				local room = get_room_from_jid(bare)

				local today = os.date("!%y%m%d");

				local now = os.date("!%H:%M:%S");

				local muc_to = nil

				local muc_from = nil;

				local already_joined = false;

				if room._data.hidden then -- do not log any data of private rooms

					return;

				end

				if not logging_enabled(room) then -- do not log where logging is not enabled

					return;

				end

				if stanza.name == "presence" and stanza.attr.type == nil then

					muc_from = stanza.attr.to;

					if room._occupants and room._occupants[stanza.attr.to] then

						already_joined = true;

						stanza:tag("alreadyJoined"):text("true");

					end

				elseif stanza.name == "iq" and stanza.attr.type == "set" then -- kick, to is the room, from is the admin, nick who is kicked is attr of iq->query->item

					if stanza.tags[1] and stanza.tags[1].name == "query" then

						local tmp = stanza.tags[1];

						if tmp.tags[1] ~= nil and tmp.tags[1].name == "item" and tmp.tags[1].attr.nick then

							tmp = tmp.tags[1];

							for jid, nick in pairs(room._jid_nick) do

								if nick == stanza.attr.to .. "/" .. tmp.attr.nick then

									muc_to = nick;

									break;

								end

							end

						end

					end

				else

					for jid, nick in pairs(room._jid_nick) do

						if jid == stanza.attr.from then

							muc_from = nick;

							break;

						end

					end

				end

				if (muc_from or muc_to) then

					local data = data_load(node, host, datastore .. "/" .. today);

					local realFrom = stanza.attr.from;

					local realTo = stanza.attr.to;

					if data == nil then

						data = {};

					end

					stanza.attr.from = muc_from;

					stanza.attr.to = muc_to;

					data[#data + 1] = "<stanza time=\"".. now .. "\">" .. tostring(stanza) .. "</stanza>\n";

					stanza.attr.from = realFrom;

					stanza.attr.to = realTo;

					if already_joined == true then

						if stanza[#stanza].name == "alreadyJoined" then  -- normaly the faked element should be the last, remove it when it is the last

							stanza[#stanza] = nil;

						else

							for i = 1, #stanza, 1 do

								if stanza[i].name == "alreadyJoined" then  -- remove the faked element

									stanza[i] = nil;

									break;

								end

							end

						end

					end

					datamanager.getpath(node, host, datastore, nil, true); -- create the datastore dir

					data_store(node, host, datastore .. "/" .. today, data);

				end

			end

		end

	end

end

module:hook("muc-config-form", function(event)

	local room, form = event.room, event.form;

	table.insert(form,

	{

		name = muc_form_config_option,

		type = "boolean",

		label = "Enable Logging?",

		value = logging_enabled(room),

	}

	);

end);

module:hook("muc-config-submitted", function(event)

	local room, fields, changed = event.room, event.fields, event.changed;

	local new = fields[muc_form_config_option];

	if new ~= room._data.logging then

		room._data.logging = new;

		if type(changed) == "table" then

			changed[muc_form_config_option] = true;

		else

			event.changed = true;

		end

	end

end);

module:hook("message/bare", log_if_needed, 1);

if log_presences then

	module:hook("iq/bare", log_if_needed, 1);

	module:hook("presence/full", log_if_needed, 1);

end

module:log("debug", "module mod_muc_log loaded!");