module:depends("http");

local prosody = prosody;

local hosts = prosody.hosts;

local my_host = module:get_host();

local strchar = string.char;

local strformat = string.format;

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

local config_get = require "core.configmanager".get;

local urldecode = require "net.http".urldecode;

local http_event = require "net.http.server".fire_event;

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

local data_load, data_getpath = datamanager.load, datamanager.getpath;

local datastore = "muc_log";

local url_base = "muc_log";

local config = nil;

local table, tostring, tonumber = table, tostring, tonumber;

local os_date, os_time = os.date, os.time;

local str_format = string.format;

local io_open = io.open;

local themes_parent = (module.path and module.path:gsub("[/\\][^/\\]*$", "")  or (prosody.paths.plugins or "./plugins") .. "/muc_log_http") .. "/themes";

local lom = require "lxp.lom";

local lfs = require "lfs";

local html = {};

local theme;

-- Helper Functions

local p_encode = datamanager.path_encode;

local function store_exists(node, host, today)

	if lfs.attributes(data_getpath(node, host, datastore .. "/" .. today), "mode") then return true; else return false; end

end

-- Module Definitions

local function html_escape(t)

	if t then

		t = t:gsub("<", "&lt;");

		t = t:gsub(">", ">");

		t = t:gsub("(http://[%a%d@%.:/&%?=%-_#%%~]+)", function(h)

			h = urlunescape(h)

			return "<a href='" .. h .. "'>" .. h .. "</a>";

		end);

		t = t:gsub("\n", "<br />");

		t = t:gsub("%%", "%%%%");

	else

		t = "";

	end

	return t;

end

function create_doc(body, title)

	if not body then return "" end

	body = body:gsub("%%", "%%%%");

	return html.doc:gsub("###BODY_STUFF###", body)

		:gsub("<title>muc_log</title>", "<title>"..(title and html_escape(title) or "Chatroom logs").."</title>");

end

function urlunescape (url)

	url = url:gsub("+", " ")

	url = url:gsub("%%(%x%x)", function(h) return strchar(tonumber(h,16)) end)

	url = url:gsub("\r\n", "\n")

	return url

end

local function urlencode(s)

	return s and (s:gsub("[^a-zA-Z0-9.~_-]", function (c) return ("%%%02x"):format(c:byte()); end));

end

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 get_room_list(host)

	local component = hosts[host];

	local list = {};

	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)

			for _, room in pairs(muc.rooms) do

				list[room.jid] = room;

			end

			return list;

		elseif muc and rawget(muc,"each_room") then

			-- We're running >0.10 (new MUC API)

			for room, _ in muc.each_room() do

				list[room.jid] = room;

			end

			return list;

		end

	end

end

local function generate_room_list(host)

		local rooms;

		for jid, room in pairs(get_room_list(host)) do

			local node = split_jid(jid);

			if not room._data.hidden and room._data.logging and node then

				rooms = (rooms or "") .. html.rooms.bit:gsub("###ROOM###", urlencode(node)):gsub("###COMPONENT###", host);

			end

		end

		if rooms then

			return html.rooms.body:gsub("###ROOMS_STUFF###", rooms):gsub("###COMPONENT###", host), "Chatroom logs for "..host;

		end

end

-- Calendar stuff

local function get_days_for_month(month, year)

	if month == 2 then

		local is_leap_year = (year % 4 == 0 and year % 100 ~= 0) or year % 400 == 0;

		return is_leap_year and 29 or 28;

	elseif (month < 8 and month%2 == 1) or (month >= 8 and month%2 == 0) then

		return 31;

	end

	return 30;

end

local function create_month(month, year, callback)

	local html_str = html.month.header;

	local days = get_days_for_month(month, year);

	local time = os_time{year=year, month=month, day=1};

	local dow = tostring(os_date("%a", time))

	local title = tostring(os_date("%B", time));

	local week_days = {"Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"};

	local week_day = 0;

	local weeks = 1;

	local _available_for_one_day = false;

	local week_days_html = "";

	for _, tmp in ipairs(week_days) do

		week_days_html = week_days_html .. html.month.weekDay:gsub("###DAY###", tmp) .. "\n";

	end

	html_str = html_str:gsub("###TITLE###", title):gsub("###WEEKDAYS###", week_days_html);

	for i = 1, 31 do

		week_day = week_day + 1;

		if week_day == 1 then html_str = html_str .. "<tr>\n"; end

		if i == 1 then

			for _, tmp in ipairs(week_days) do

				if dow ~= tmp then

					html_str = html_str .. html.month.emptyDay .. "\n";

					week_day = week_day + 1;

				else

					break;

				end

			end

		end

		if i < days + 1 then

			local tmp = tostring(i);

			if callback and callback.callback then

				tmp = callback.callback(callback.path, i, month, year, callback.room, callback.webpath);

			end

			if tmp == nil then

				tmp = tostring(i);

			else

				_available_for_one_day = true;

			end

			html_str = html_str .. html.month.day:gsub("###DAY###", tmp) .. "\n";

		end

		if i >= days then

			break;

		end

		if week_day == 7 then

			week_day = 0;

			weeks = weeks + 1;

			html_str = html_str .. "</tr>\n";

		end

	end

	if week_day + 1 < 8 or weeks < 6 then

		week_day = week_day + 1;

		if week_day > 7 then

			week_day = 1;

		end

		if week_day == 1 then

			weeks = weeks + 1;

		end

		for y = weeks, 6 do

			if week_day == 1 then

				html_str = html_str .. "<tr>\n";

			end

			for i = week_day, 7 do

				html_str = html_str .. html.month.emptyDay .. "\n";

			end

			week_day = 1

			html_str = html_str .. "</tr>\n";

		end

	end

	html_str = html_str .. html.month.footer;

	if _available_for_one_day then

		return html_str;

	end

end

local function create_year(year, callback)

	local year = year;

	local tmp;

	if tonumber(year) <= 99 then

		year = year + 2000;

	end

	local html_str = "";

	for i=1, 12 do

		tmp = create_month(i, year, callback);

		if tmp then

			html_str = html_str .. "<div style='float: left; padding: 5px;'>\n" .. tmp .. "</div>\n";

		end

	end

	if html_str ~= "" then

		return "<div name='yearDiv' style='padding: 40px; text-align: center;'>" .. html.year.title:gsub("###YEAR###", tostring(year)) .. html_str .. "</div><br style='clear:both;'/> \n";

	end

	return "";

end

local function day_callback(path, day, month, year, room, webpath)

	local webpath = webpath or ""

	local year = year;

	if year > 2000 then

		year = year - 2000;

	end

	local bare_day = str_format("20%.02d-%.02d-%.02d", year, month, day);

	room = p_encode(room);

	local attributes, err = lfs.attributes(path.."/"..str_format("%.02d%.02d%.02d", year, month, day).."/"..room..".dat");

	if attributes ~= nil and attributes.mode == "file" then

		local s = html.days.bit;

		s = s:gsub("###BARE_DAY###", webpath .. bare_day);

		s = s:gsub("###DAY###", day);

		return s;

	end

	return;

end

local function generate_day_room_content(bare_room_jid)

	local days = "";

	local days_array = {};

	local tmp;

	local node, host = split_jid(bare_room_jid);

	local path = data_getpath(node, host, datastore);

	local room = nil;

	local next_room = "";

	local previous_room = "";

	local rooms = "";

	local attributes = nil;

	local since = "";

	local to = "";

	local topic = "";

	local component = hosts[host];

	if not(get_room_from_jid(bare_room_jid)) then

		return;

	end

	path = path:gsub("/[^/]*$", "");

	attributes = lfs.attributes(path);

	do

		local found = 0;

		module:log("debug", generate_room_list(host));

		for jid, room in pairs(get_room_list(host)) do

			local node = split_jid(jid)

			if not room._data.hidden and room._data.logging and node then

				if found == 0 then

					previous_room = node

				elseif found == 1 then

					next_room = node

					found = -1

				end

				if jid == bare_room_jid then

					found = 1

				end

				rooms = rooms .. html.days.rooms.bit:gsub("###ROOM###", urlencode(node));

			end

		end

		room = get_room_from_jid(bare_room_jid);

		if room._data.hidden or not room._data.logging then

			room = nil;

		end

	end

	if attributes and room then

		local already_done_years = {};

		topic = room._data.subject or "(no subject)"

		if topic:len() > 135 then

			topic = topic:sub(1, topic:find(" ", 120)) .. " ..."

		end

		local folders = {};

		for folder in lfs.dir(path) do table.insert(folders, folder); end

		table.sort(folders);

		for _, folder in ipairs(folders) do

			local year, month, day = folder:match("^(%d%d)(%d%d)(%d%d)");

			if year then

				to = tostring(os_date("%B %Y", os_time({ day=tonumber(day), month=tonumber(month), year=2000+tonumber(year) })));

				if since == "" then since = to; end

				if not already_done_years[year] then

					module:log("debug", "creating overview for: %s", to);

					days = create_year(year, {callback=day_callback, path=path, room=node}) .. days;

					already_done_years[year] = true;

				end

			end

		end

	end

	tmp = html.days.body:gsub("###DAYS_STUFF###", days);

	tmp = tmp:gsub("###PREVIOUS_ROOM###", previous_room == "" and node or previous_room);

	tmp = tmp:gsub("###NEXT_ROOM###", next_room == "" and node or next_room);

	tmp = tmp:gsub("###ROOMS###", rooms);

	tmp = tmp:gsub("###ROOMTOPIC###", topic);

	tmp = tmp:gsub("###SINCE###", since);

	tmp = tmp:gsub("###TO###", to);

	return tmp:gsub("###JID###", bare_room_jid), "Chatroom logs for "..bare_room_jid;

end

local function parse_iq(stanza, time, nick)

	local text = nil;

	local victim = nil;

	if(stanza.attr.type == "set") then

		for _,tag in ipairs(stanza) do

			if tag.tag == "query" then

				for _,item in ipairs(tag) do

					if item.tag == "item" and item.attr.nick ~= nil and item.attr.role == 'none' then

						victim = item.attr.nick;

						for _,reason in ipairs(item) do

							if reason.tag == "reason" then

								text = reason[1];

								break;

							end

						end

						break;

					end

				end

				break;

			end