Fri, 17 Dec 2010 17:28:05 +0100
util.stanza: Iterate on childtags instead of all childs.
-- 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. -- local httpserver = require "net.httpserver"; local lfs = require "lfs"; local open = io.open; local t_concat = table.concat; local stat = lfs.attributes; local http_base = config.get("*", "core", "http_path") or "www_files"; local response_400 = { status = "400 Bad Request", body = "<h1>Bad Request</h1>Sorry, we didn't understand your request :(" }; local response_403 = { status = "403 Forbidden", body = "<h1>Forbidden</h1>You don't have permission to view the contents of this directory :(" }; local response_404 = { status = "404 Not Found", body = "<h1>Page Not Found</h1>Sorry, we couldn't find what you were looking for :(" }; -- 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"; }; local function preprocess_path(path) if path:sub(1,1) ~= "/" then path = "/"..path; end local level = 0; for component in path:gmatch("([^/]+)/") do if component == ".." then level = level - 1; elseif component ~= "." then level = level + 1; end if level < 0 then return nil; end end return path; end function serve_file(path) local full_path = http_base..path; if stat(full_path, "mode") == "directory" then if stat(full_path.."/index.html", "mode") == "file" then return serve_file(path.."/index.html"); end return response_403; end local f, err = open(full_path, "rb"); if not f then return response_404; end local data = f:read("*a"); f:close(); if not data then return response_403; end local ext = path:match("%.([^.]*)$"); local mime = mime_map[ext]; -- Content-Type should be nil when not known return { headers = { ["Content-Type"] = mime; }; body = data; }; end local function handle_file_request(method, body, request) local path = preprocess_path(request.url.path); if not path then return response_400; end path = path:gsub("^/[^/]+", ""); -- Strip /files/ return serve_file(path); end local function handle_default_request(method, body, request) local path = preprocess_path(request.url.path); if not path then return response_400; end return serve_file(path); end local function setup() local ports = config.get(module.host, "core", "http_ports") or { 5280 }; httpserver.set_default_handler(handle_default_request); httpserver.new_from_config(ports, handle_file_request, { base = "files" }); end if prosody.start_time then -- already started setup(); else prosody.events.add_handler("server-started", setup); end