plugins/mod_http_files.lua
changeset 5262 4e58fde55594
parent 5261 b14f02671439
child 5263 736e7ec8cd2e
equal deleted inserted replaced
5261:b14f02671439 5262:4e58fde55594
    12 local os_date = os.date;
    12 local os_date = os.date;
    13 local open = io.open;
    13 local open = io.open;
    14 local stat = lfs.attributes;
    14 local stat = lfs.attributes;
    15 local build_path = require"socket.url".build_path;
    15 local build_path = require"socket.url".build_path;
    16 
    16 
    17 local base_path = module:get_option_string("http_files_dir", module:get_option_string("http_path", "www_files"));
    17 local base_path = module:get_option_string("http_files_dir", module:get_option_string("http_path"));
    18 local dir_indices = module:get_option("http_index_files", { "index.html", "index.htm" });
    18 local dir_indices = module:get_option("http_index_files", { "index.html", "index.htm" });
    19 local directory_index = module:get_option_boolean("http_dir_listing");
    19 local directory_index = module:get_option_boolean("http_dir_listing");
    20 
    20 
    21 local mime_map = module:shared("mime").types;
    21 local mime_map = module:shared("mime").types;
    22 if not mime_map then
    22 if not mime_map then
    47 	end
    47 	end
    48 end
    48 end
    49 
    49 
    50 local cache = setmetatable({}, { __mode = "kv" }); -- Let the garbage collector have it if it wants to.
    50 local cache = setmetatable({}, { __mode = "kv" }); -- Let the garbage collector have it if it wants to.
    51 
    51 
    52 function serve_file(event, path)
    52 function serve(opts)
    53 	local request, response = event.request, event.response;
    53 	local base_path = opts.path;
    54 	local orig_path = request.path;
    54 	local dir_indices = opts.index_files or dir_indices;
    55 	local full_path = base_path.."/"..path;
    55 	local directory_index = opts.directory_index;
    56 	local attr = stat(full_path);
    56 	local function serve_file(event, path)
    57 	if not attr then
    57 		local request, response = event.request, event.response;
    58 		return 404;
    58 		local orig_path = request.path;
       
    59 		local full_path = base_path .. (path and "/"..path or "");
       
    60 		local attr = stat(full_path);
       
    61 		if not attr then
       
    62 			return 404;
       
    63 		end
       
    64 
       
    65 		local request_headers, response_headers = request.headers, response.headers;
       
    66 
       
    67 		local last_modified = os_date('!%a, %d %b %Y %H:%M:%S GMT', attr.modification);
       
    68 		response_headers.last_modified = last_modified;
       
    69 
       
    70 		local etag = ("%02x-%x-%x-%x"):format(attr.dev or 0, attr.ino or 0, attr.size or 0, attr.modification or 0);
       
    71 		response_headers.etag = etag;
       
    72 
       
    73 		local if_none_match = request_headers.if_none_match
       
    74 		local if_modified_since = request_headers.if_modified_since;
       
    75 		if etag == if_none_match
       
    76 		or (not if_none_match and last_modified == if_modified_since) then
       
    77 			return 304;
       
    78 		end
       
    79 
       
    80 		local data = cache[path];
       
    81 		if data and data.etag == etag then
       
    82 			response_headers.content_type = data.content_type;
       
    83 			data = data.data;
       
    84 		elseif attr.mode == "directory" then
       
    85 			if full_path:sub(-1) ~= "/" then
       
    86 				local path = { is_absolute = true, is_directory = true };
       
    87 				for dir in orig_path:gmatch("[^/]+") do path[#path+1]=dir; end
       
    88 				response_headers.location = build_path(path);
       
    89 				return 301;
       
    90 			end
       
    91 			for i=1,#dir_indices do
       
    92 				if stat(full_path..dir_indices[i], "mode") == "file" then
       
    93 					return serve_file(event, path..dir_indices[i]);
       
    94 				end
       
    95 			end
       
    96 
       
    97 			if not directory_index then
       
    98 				return 403;
       
    99 			else
       
   100 				local html = require"util.stanza".stanza("html")
       
   101 					:tag("head"):tag("title"):text(path):up()
       
   102 						:tag("meta", { charset="utf-8" }):up()
       
   103 					:up()
       
   104 					:tag("body"):tag("h1"):text(path):up()
       
   105 						:tag("ul");
       
   106 				for file in lfs.dir(full_path) do
       
   107 					if file:sub(1,1) ~= "." then
       
   108 						local attr = stat(full_path..file) or {};
       
   109 						html:tag("li", { class = attr.mode })
       
   110 							:tag("a", { href = file }):text(file)
       
   111 						:up():up();
       
   112 					end
       
   113 				end
       
   114 				data = "<!DOCTYPE html>\n"..tostring(html);
       
   115 				cache[path] = { data = data, content_type = mime_map.html; etag = etag; };
       
   116 				response_headers.content_type = mime_map.html;
       
   117 			end
       
   118 
       
   119 		else
       
   120 			local f, err = open(full_path, "rb");
       
   121 			if f then
       
   122 				data, err = f:read("*a");
       
   123 				f:close();
       
   124 			end
       
   125 			if not data then
       
   126 				module:log("debug", "Could not open or read %s. Error was %s", full_path, err);
       
   127 				return 403;
       
   128 			end
       
   129 			local ext = path:match("%.([^./]+)$");
       
   130 			local content_type = ext and mime_map[ext];
       
   131 			cache[path] = { data = data; content_type = content_type; etag = etag };
       
   132 			response_headers.content_type = content_type;
       
   133 		end
       
   134 
       
   135 		return response:send(data);
    59 	end
   136 	end
    60 
   137 
    61 	local request_headers, response_headers = request.headers, response.headers;
   138 	return serve_file;
    62 
       
    63 	local last_modified = os_date('!%a, %d %b %Y %H:%M:%S GMT', attr.modification);
       
    64 	response_headers.last_modified = last_modified;
       
    65 
       
    66 	local etag = ("%02x-%x-%x-%x"):format(attr.dev or 0, attr.ino or 0, attr.size or 0, attr.modification or 0);
       
    67 	response_headers.etag = etag;
       
    68 
       
    69 	local if_none_match = request_headers.if_none_match
       
    70 	local if_modified_since = request_headers.if_modified_since;
       
    71 	if etag == if_none_match
       
    72 	or (not if_none_match and last_modified == if_modified_since) then
       
    73 		return 304;
       
    74 	end
       
    75 
       
    76 	local data = cache[path];
       
    77 	if data and data.etag == etag then
       
    78 		response_headers.content_type = data.content_type;
       
    79 		data = data.data;
       
    80 	elseif attr.mode == "directory" then
       
    81 		if full_path:sub(-1) ~= "/" then
       
    82 			local path = { is_absolute = true, is_directory = true };
       
    83 			for dir in orig_path:gmatch("[^/]+") do path[#path+1]=dir; end
       
    84 			response_headers.location = build_path(path);
       
    85 			return 301;
       
    86 		end
       
    87 		for i=1,#dir_indices do
       
    88 			if stat(full_path..dir_indices[i], "mode") == "file" then
       
    89 				return serve_file(event, path..dir_indices[i]);
       
    90 			end
       
    91 		end
       
    92 
       
    93 		if not directory_index then
       
    94 			return 403;
       
    95 		else
       
    96 			local html = require"util.stanza".stanza("html")
       
    97 				:tag("head"):tag("title"):text(path):up()
       
    98 					:tag("meta", { charset="utf-8" }):up()
       
    99 				:up()
       
   100 				:tag("body"):tag("h1"):text(path):up()
       
   101 					:tag("ul");
       
   102 			for file in lfs.dir(full_path) do
       
   103 				if file:sub(1,1) ~= "." then
       
   104 					local attr = stat(full_path..file) or {};
       
   105 					html:tag("li", { class = attr.mode })
       
   106 						:tag("a", { href = file }):text(file)
       
   107 					:up():up();
       
   108 				end
       
   109 			end
       
   110 			data = "<!DOCTYPE html>\n"..tostring(html);
       
   111 			cache[path] = { data = data, content_type = mime_map.html; etag = etag; };
       
   112 			response_headers.content_type = mime_map.html;
       
   113 		end
       
   114 
       
   115 	else
       
   116 		local f, err = open(full_path, "rb");
       
   117 		if f then
       
   118 			data, err = f:read("*a");
       
   119 			f:close();
       
   120 		end
       
   121 		if not data then
       
   122 			module:log("debug", "Could not open or read %s. Error was %s", full_path, err);
       
   123 			return 403;
       
   124 		end
       
   125 		local ext = path:match("%.([^./]+)$");
       
   126 		local content_type = ext and mime_map[ext];
       
   127 		cache[path] = { data = data; content_type = content_type; etag = etag };
       
   128 		response_headers.content_type = content_type;
       
   129 	end
       
   130 
       
   131 	return response:send(data);
       
   132 end
   139 end
   133 
   140 
   134 module:provides("http", {
       
   135 	route = {
       
   136 		["GET /*"] = serve_file;
       
   137 	};
       
   138 });
       
   139 
   141 
       
   142 if base_path then
       
   143 	module:provides("http", {
       
   144 		route = {
       
   145 			["GET /*"] = serve {
       
   146 				path = base_path;
       
   147 				directory_index = directory_index;
       
   148 			}
       
   149 		};
       
   150 	});
       
   151 else
       
   152 	module:log("debug", "http_files_dir not set, assuming use by some other module");
       
   153 end
       
   154