Software /
code /
prosody
File
plugins/mod_http_files.lua @ 5254:df3552822054
mod_http_files: Make sure file extensions are not nil or empty string
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Fri, 21 Dec 2012 08:19:58 +0100 |
parent | 5253:ad45612199e0 |
child | 5255:bf34f1ee08ee |
line wrap: on
line source
-- 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 os_date = os.date; 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" }); local show_file_list = module:get_option_boolean("http_files_show_list"); local mime_map = module:shared("mime").types; if not mime_map then mime_map = { html = "text/html", htm = "text/html", xml = "application/xml", txt = "text/plain", css = "text/css", js = "application/javascript", png = "image/png", gif = "image/gif", jpeg = "image/jpeg", jpg = "image/jpeg", svg = "image/svg+xml", }; module:shared("mime").types = mime_map; local mime_types, err = open(module:get_option_string("mime_types_file", "/etc/mime.types"),"r"); if mime_types then local mime_data = mime_types:read("*a"); mime_types:close(); setmetatable(mime_map, { __index = function(t, ext) local typ = mime_data:match("\n(%S+)[^\n]*%s"..(ext:lower()).."%s") or "application/octet-stream"; t[ext] = typ; return typ; end }); end end local cache = setmetatable({}, { __mode = "kv" }); -- Let the garbage collector have it if it wants to. 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 request_headers, response_headers = request.headers, response.headers; local last_modified = os_date('!%a, %d %b %Y %H:%M:%S GMT', attr.modification); response_headers.last_modified = last_modified; local etag = ("%02x-%x-%x-%x"):format(attr.dev or 0, attr.ino or 0, attr.size or 0, attr.modification or 0); response_headers.etag = etag; local if_none_match = request_headers.if_none_match local if_modified_since = request_headers.if_modified_since; if etag == if_none_match or last_modified == if_modified_since then return 304; end local data = cache[path]; if data then response_headers.content_type = data.content_type; data = data.data; elseif 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 if not show_file_list then return 403; else local html = require"util.stanza".stanza("html") :tag("head"):tag("title"):text(path):up() :tag("meta", { charset="utf-8" }):up() :up() :tag("body"):tag("h1"):text(path):up() :tag("ul"); for file in lfs.dir(full_path) do if file:sub(1,1) ~= "." then local attr = stat(full_path..file) or {}; html:tag("li", { class = attr.mode }) :tag("a", { href = file }):text(file) :up():up(); end end data = "<!DOCTYPE html>\n"..tostring(html); cache[path] = { data = data, content_type = mime_map.html; hits = 0 }; response_headers.content_type = mime_map.html; end else local f, err = open(full_path, "rb"); if f then data = f:read("*a"); f:close(); end if not data then return 403; end local ext = path:match("%.([^./]+)$"); local content_type = ext and mime_map[ext]; cache[path] = { data = data; content_type = content_type; }; response_headers.content_type = content_type; end return response:send(data); end module:provides("http", { route = { ["GET /*"] = serve_file; }; });