-- Copyright (C) 2009 Thilo Cestonaro

-- 

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

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

--

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 datamanager = require "util.datamanager";

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

local datastore = "muc_log";

local urlBase = "muc_log";

local muc_hosts = {};

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 = (CFG_PLUGINDIR 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 = "default";

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

function createDoc(body)

	if body then

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

	end

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 htmlEscape(t)

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

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

		h = urlunescape(h)

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

	end);

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

	return t;

end

function splitUrl(url)

	local tmp = url:sub(string.len("/muc_log/") + 1);

	local day = nil;

	local room = nil;

	local component = nil;

	local at = nil;

	local slash = nil;

	local slash2 = nil;

	slash = tmp:find("/");

	if slash then

	 	component = tmp:sub(1, slash - 1);

		if tmp:len() > slash then

			room = tmp:sub(slash + 1);

			slash = room:find("/");

			if slash then

				tmp = room;

				room = tmp:sub(1, slash - 1);

				if tmp:len() > slash then

					day = tmp:sub(slash + 1);

					slash = day:find("/");

					if slash then

						day = day:sub(1, slash - 1);

					end

				end

			end

		end

	end

	return room, component, day;

end

local function generateComponentListSiteContent()

	local components = "";

	for component,muc_host in pairs(muc_hosts or {}) do

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

	end

	if components ~= "" then

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

	end

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

		if rooms ~= "" then

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

		end

	end

end

-- Calendar stuff

local function getDaysForMonth(month, year)

    local daysCount = 30;

    local leapyear = false;

    if year%4 == 0 and year%100 == 0 then

        if year%400 == 0 then

            leapyear = true;

        else

            leapyear = false; -- turn of the century but not a leapyear

        end

    elseif year%4 == 0 then

        leapyear = true;

    end

    if month == 2 and leapyear then

        daysCount = 29;

    elseif month == 2 and not leapyear then

        daysCount = 28;

    elseif  month < 8 and month%2 == 1 or

            month >= 8 and month%2 == 0

    then

        daysCount = 31;

    end

    return daysCount;

end

local function createMonth(month, year, dayCallback)

    local htmlStr = html.month.header;

    local days = getDaysForMonth(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("<span style='color:#DDDDDD'>"..tostring(i).."</span>");

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

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

            end

			if tmp == nil then

            	tmp = tostring("<span style='color:#DDDDDD'>"..tostring(i).."</span>");

			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 = "<div name='yearDiv' style='padding: 40px; text-align: center;'>" .. html.year.title:gsub("###YEAR###", tostring(year));

    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

	return htmlStr .. "</div><br style='clear:both;'/> \n";

end

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

	local year = year;

	if year > 2000 then

		year = year - 2000;

	end

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

	local attributes, err = lfs.attributes(path.."/"..bareDay)

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

		local s = html.days.bit;

		s = s:gsub("###BARE_DAY###", 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 attributes = nil;

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

	attributes = lfs.attributes(path);

	if muc_hosts ~= nil and muc_hosts[host] and prosody.hosts[host] ~= nil and prosody.hosts[host].muc ~= nil and prosody.hosts[host].muc.rooms[bareRoomJid] ~= nil then

		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 = {};

		for file in lfs.dir(path) do

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

			if year ~= nil and alreadyDoneYears[year] == nil then

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

				alreadyDoneYears[year] = true;

			end

		end

	end

	if days ~= "" then

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

		return tmp:gsub("###JID###", bareRoomJid);

	end

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 tostring(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));

			else

				text = "";

			end	

			return html.day.kick:gsub("###TIME_STUFF###", timeStuff):gsub("###VICTIM###", victim):gsub("###REASON_STUFF###", text);

		end

	end

	return;

end

local function parsePresenceStanza(stanza, timeStuff, nick)

	local ret = "";

	local showJoin = "block"

	if config and not config.showJoin then

		showJoin = "none";

	end

	if stanza.attr.type == nil then

		local showStatus = "block"

		if config and not config.showStatus then

			showStatus = "none";

		end

		local show, status = nil, "";

		local alreadyJoined = false;

		for _, tag in ipairs(stanza) do