-- Copyright (C) 2009 Thilo Cestonaro

--

-- This project is MIT/X11 licensed. Please see the

-- COPYING file in the source package for more information.

--

module:set_global();

local prosody = prosody;

local tabSort = table.sort;

local tonumber = _G.tonumber;

local tostring = _G.tostring;

local strchar = string.char;

local strformat = string.format;

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

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

local httpserver = require "net.httpserver";

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

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

local datamanager = require "util.datamanager";

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

local datastore = "muc_log";

local urlBase = "muc_log";

local config = nil;

local tostring = _G.tostring;

local tonumber = _G.tonumber;

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

local str_format = string.format;

local io_open = io.open;

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

local lom = require "lxp.lom";

--[[ LuaFileSystem

* URL: http://www.keplerproject.org/luafilesystem/index.html

* Install: luarocks install luafilesystem

* ]]

local lfs = require "lfs";

--[[

* Default templates for the html output.

]]--

local html = {};

local theme;

local function checkDatastorePathExists(node, host, today, create)

	create = create or false;

	local path = data_getpath(node, host, datastore, "dat", true);

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

	-- check existance

	local attributes, err = lfs.attributes(path);

	if attributes == nil or attributes.mode ~= "directory" then

		module:log("warn", "muc_log folder isn't a folder: %s", path);

		return false;

	end

	attributes, err = lfs.attributes(path .. "/" .. today);

	if attributes == nil then

		if create then

			return lfs.mkdir(path .. "/" .. today);

		else

			return false;

		end

	elseif attributes.mode == "directory" then

		return true;

	end

	return false;

end

local function htmlEscape(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 createDoc(body, title)

	if not body then

		return { status = "404 Not Found", body = "<h1>Page Not Found</h1>Sorry, we couldn't find what you were looking for :(" };

	end

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

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

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

end

function urlunescape (escapedUrl)

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

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

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

	return escapedUrl

end

local function generateComponentListSiteContent()

	local components = "";

	for component,host in pairs(hosts) do

		if host.modules.muc and host.modules.muc_log then

			components = components .. html.components.bit:gsub("###COMPONENT###", component);

		end

	end

	return (html.components.body:gsub("###COMPONENTS_STUFF###", components));

end

local function generateRoomListSiteContent(component)

	local rooms = "";

	if prosody.hosts[component] and prosody.hosts[component].muc ~= nil then

		for jid, room in pairs(prosody.hosts[component].muc.rooms) do

			local node = splitJid(jid);

			if not room._data.hidden and node then

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

			end

		end

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

	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 createMonth(month, year, dayCallback)

	local htmlStr = 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 weekDays = {"Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun"};

	local weekDay = 0;

	local weeks = 1;

	local logAvailableForMinimumOneDay = false;

	local weekDaysHtml = "";

	for _, tmp in ipairs(weekDays) do

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

	end

	htmlStr = htmlStr:gsub("###TITLE###", title):gsub("###WEEKDAYS###", weekDaysHtml);

	for i = 1, 31 do

		weekDay = weekDay + 1;

		if weekDay == 1 then htmlStr = htmlStr .. "<tr>\n"; end

		if i == 1 then

			for _, tmp in ipairs(weekDays) do

				if dow ~= tmp then

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

					weekDay = weekDay + 1;

				else

					break;

				end

			end

		end

		if i < days + 1 then

			local tmp = tostring(i);

			if dayCallback ~= nil and dayCallback.callback ~= nil then

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

			end

			if tmp == nil then

				tmp = tostring(i);

			else

				logAvailableForMinimumOneDay = true;

			end

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

		end

		if i >= days then

			break;

		end

		if weekDay == 7 then

			weekDay = 0;

			weeks = weeks + 1;

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

		end

	end

	if weekDay + 1 < 8 or weeks < 6 then

		weekDay = weekDay + 1;

		if weekDay > 7 then

			weekDay = 1;

		end

		if weekDay == 1 then

			weeks = weeks + 1;

		end

		for y = weeks, 6 do

			if weekDay == 1 then

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

			end

			for i = weekDay, 7 do

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

			end

			weekDay = 1

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

		end

	end

	htmlStr = htmlStr .. html.month.footer;

	if logAvailableForMinimumOneDay then

		return htmlStr;

	end

end

local function createYear(year, dayCallback)

	local year = year;

	local tmp;

	if tonumber(year) <= 99 then

		year = year + 2000;

	end

	local htmlStr = "";

	for i=1, 12 do

		tmp = createMonth(i, year, dayCallback);

		if tmp then

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

		end

	end

	if htmlStr ~= "" then

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

	end

	return "";

end

local function perDayCallback(path, day, month, year, room, webPath)

	local webPath = webPath or ""

	local year = year;

	if year > 2000 then

		year = year - 2000;

	end

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

	room = urlencode(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 .. bareDay);

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

		return s;

	end

	return;

end

local function generateDayListSiteContentByRoom(bareRoomJid)

	local days = "";

	local arrDays = {};

	local tmp;

	local node, host, resource = splitJid(bareRoomJid);

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

	local room = nil;

	local nextRoom = "";

	local previousRoom = "";

	local rooms = "";

	local attributes = nil;

	local since = "";

	local to = "";

	local topic = "";

	if not(prosody.hosts[host] and prosody.hosts[host].muc and prosody.hosts[host].muc.rooms[bareRoomJid]) then

		return;

	end

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

	attributes = lfs.attributes(path);

	do

		local found = 0;

		for jid, room in pairs(prosody.hosts[host].muc.rooms) do

			local node = splitJid(jid)

			if not room._data.hidden and node then

				if found == 0 then

					previousRoom = node

				elseif found == 1 then

					nextRoom = node

					found = -1

				end

				if jid == bareRoomJid then

					found = 1

				end

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

			end

		end

		room = prosody.hosts[host].muc.rooms[bareRoomJid];

		if room._data.hidden then

			room = nil

		end

	end

	if attributes ~= nil and room ~= nil then

		local alreadyDoneYears = {};

		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 alreadyDoneYears[year] then

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

					days = createYear(year, {callback=perDayCallback, path=path, room=node}) .. days;

					alreadyDoneYears[year] = true;

				end

			end

		end

	end

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

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

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

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

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

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

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

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

end

local function parseIqStanza(stanza, timeStuff, 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

		end

		if victim ~= nil then

			if text ~= nil then

				text = html.day.reason:gsub("###REASON###", htmlEscape(text));