-- Prosody IM

-- Copyright (C) 2008-2010 Matthew Wild

-- Copyright (C) 2008-2010 Waqas Hussain

-- 

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

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

--

module:depends("http");

local lfs = require "lfs";

local open = io.open;

local stat = lfs.attributes;

local http_base = module:get_option_string("http_files_dir", module:get_option_string("http_path", "www_files"));

local dir_indices = module:get_option("http_files_index", { "index.html", "index.htm" });

-- TODO: Should we read this from /etc/mime.types if it exists? (startup time...?)

local mime_map = {

	html = "text/html";

	htm = "text/html";

	xml = "text/xml";

	xsl = "text/xml";

	txt = "text/plain; charset=utf-8";

	js = "text/javascript";

	css = "text/css";

};

function serve_file(event, path)

	local request, response = event.request, event.response;

	local orig_path = request.path;

	local full_path = http_base.."/"..path;

	local attr = stat(full_path);

	if not attr then

		return 404;

	end

	local tag = ("%02x-%x-%x-%x"):format(attr.dev or 0, attr.ino or 0, attr.size or 0, attr.modification or 0);

	response.headers.etag = tag;

	if tag == request.headers.if_none_match then

		return 304;

	end

	if attr.mode == "directory" then

		if full_path:sub(-1) ~= "/" then

			response.headers.location = orig_path.."/";

			return 301;

		end

		for i=1,#dir_indices do

			if stat(full_path..dir_indices[i], "mode") == "file" then

				return serve_file(event, path..dir_indices[i]);

			end

		end

		-- TODO File listing

		return 403;

	end

	local f, err = open(full_path, "rb");

	if not f then

		module:log("warn", "Failed to open file: %s", err);

		return 404;

	end

	local data = f:read("*a");

	f:close();

	if not data then

		return 403;

	end

	local ext = path:match("%.([^.]*)$");

	response.headers.content_type = mime_map[ext]; -- Content-Type should be nil when not known

	return response:send(data);

end

module:provides("http", {

	route = {

		["GET /*"] = serve_file;

	};

});