From 487b80b9b06d6f75a5142e8d04b1ec8ea487fe3e Mon Sep 17 00:00:00 2001 From: Badlop Date: Tue, 16 Jun 2009 13:52:07 +0000 Subject: [PATCH] Initial commit of mod_http_fileserver (thanks to Massimiliano Mirra) SVN Revision: 2183 --- src/web/mod_http_fileserver.erl | 112 ++++++++++++++++++++++++++++++++ 1 file changed, 112 insertions(+) create mode 100644 src/web/mod_http_fileserver.erl diff --git a/src/web/mod_http_fileserver.erl b/src/web/mod_http_fileserver.erl new file mode 100644 index 000000000..248870c8e --- /dev/null +++ b/src/web/mod_http_fileserver.erl @@ -0,0 +1,112 @@ +%%%---------------------------------------------------------------------- +%%% File : mod_http_fileserver.erl +%%% Author : Massimiliano Mirra +%%% Purpose : Simple file server plugin for embedded ejabberd web server +%%% Created : +%%% Id : +%%%---------------------------------------------------------------------- + +-module(mod_http_fileserver). +-author('mmirra@process-one.net'). +-vsn(''). +-define(ejabberd_debug, true). +-compile([export_all]). + +-behaviour(gen_mod). + +-export([ + start/2, + stop/1, + process/2 + ]). + +-include("ejabberd.hrl"). +-include("jlib.hrl"). +-include("ejabberd_http.hrl"). + +%%%---------------------------------------------------------------------- +%%% REQUEST HANDLERS +%%%---------------------------------------------------------------------- + +%%----------------------------------------------------------------------- +%% FUNCTION +%% +%% process/2 +%% +%% PURPOSE +%% +%% Handle an HTTP request. +%% +%% RETURNS +%% +%% Page to be sent back to the client and/or HTTP status code. +%% +%% ARGUMENTS +%% +%% - LocalPath: part of the requested URL path that is "local to the +%% module". +%% +%%----------------------------------------------------------------------- + +process(LocalPath, _Request) -> + ?DEBUG("Requested ~p", [LocalPath]), + [{docroot, DocRoot}] = ets:lookup(mod_http_fileserver, docroot), + FileName = filename:join(filename:split(DocRoot) ++ LocalPath), + case file:read_file(FileName) of + {ok, FileContents} -> + ?DEBUG("Delivering content.", []), + {200, + [{"Server", "ejabberd"}, + {"Content-type", content_type(FileName)}], + FileContents}; + {error, Error} -> + ?DEBUG("Delivering error: ~p", [Error]), + case Error of + eacces -> {403, [], "Forbidden"}; + enoent -> {404, [], "Not found"}; + _Else -> {404, [], atom_to_list(Error)} + end + end. + + +%%%---------------------------------------------------------------------- +%%% UTILITIES +%%%---------------------------------------------------------------------- + +content_type(Filename) -> + case httpd_util:to_lower(filename:extension(Filename)) of + ".jpg" -> "image/jpeg"; + ".jpeg" -> "image/jpeg"; + ".gif" -> "image/gif"; + ".png" -> "image/png"; + ".html" -> "text/html"; + ".css" -> "text/css"; + ".txt" -> "text/plain"; + ".xul" -> "application/vnd.mozilla.xul+xml"; + ".jar" -> "application/java-archive"; + ".xpi" -> "application/x-xpinstall"; + _Else -> "application/octet-stream" + end. + + +%%%---------------------------------------------------------------------- +%%% BEHAVIOUR CALLBACKS +%%%---------------------------------------------------------------------- + +start(_Host, Opts) -> + case gen_mod:get_opt(docroot, Opts, undefined) of + undefined -> + {'EXIT', {missing_document_root, ?MODULE}}; + DocRoot -> + case filelib:is_dir(DocRoot) of + true -> + ets:new(mod_http_fileserver, [named_table, bag]), + ets:insert(mod_http_fileserver, [{docroot, DocRoot}]), + ok; + _Else -> + {'EXIT', {unaccessible_document_root, ?MODULE}} + end + end. + +stop(_Host) -> + ok.