-- Prosody IM

-- Copyright (C) 2012 Kim Alvefur

--

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

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

--

module:set_global();

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

local lfs = require "lfs";

local stat = lfs.attributes;

local build_path = require"socket.url".build_path;

local base64_encode = require"util.encodings".base64.encode;

local st = require"util.stanza";

local render = require"util.interpolation".new("%b{}", st.xml_escape);

local mime = module:shared("/*/http_files/mime");

local function get_resource(resource)

	local fh = assert(module:load_resource(resource));

	local data = fh:read"*a";

	fh:close();

	return data;

end

local dir_index_template = get_resource("resources/template.html");

local style = get_resource("resources/style.css"):gsub("url%((.-)%)", function(url)

	--module:log("debug", "Inlineing %s", url);

	return "url(data:image/png;base64,"..base64_encode(get_resource("resources/"..url))..")";

end);

local function generate_directory_index(path, full_path)

	local filelist = {};

	if path ~= "/" then

		table.insert(filelist, { class = "parent directory", href = "..", rel = "up", text = "Parent Directory" });

	end

	local mime_map = mime.types;

	for file in lfs.dir(full_path) do

		if file:sub(1,1) ~= "." then

			local attr = stat(full_path..file) or {};

			local path = { file };

			local file_ext = file:match"%.([^.]+)$";

			local type = attr.mode == "file" and file_ext and mime_map and mime_map[file_ext] or nil;

			local class = table.concat({ attr.mode or "unknown", file_ext, type and type:match"^[^/]+" }, " ");

			path.is_directory = attr.mode == "directory";

			table.insert(filelist, { class = class, href = build_path(path), type = type, text = file });

		end

	end

	table.sort(filelist, function (a, b)

		if a.href == ".." then return true end

		if b.href == ".." then return false end

		if a.class:match"directory" and not b.class:match"directory" then return true end

		if not a.class:match"directory" and b.class:match"directory" then return false end

		return a.text < b.text;

	end);

	return render(dir_index_template, {

		path = path,

		style = style,

		filelist = filelist,

		footer = "Prosody "..prosody.version,

	});

end

module:hook_object_event(server, "directory-index", function (event)

	local ok, data = pcall(generate_directory_index, event.path, event.full_path);

	if ok then return data end

	module:log("warn", data);

end);