diff options
Diffstat (limited to 'examples/web_server/src/directory_handler.erl')
| -rw-r--r-- | examples/web_server/src/directory_handler.erl | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/examples/web_server/src/directory_handler.erl b/examples/web_server/src/directory_handler.erl new file mode 100644 index 0000000..ed342b5 --- /dev/null +++ b/examples/web_server/src/directory_handler.erl @@ -0,0 +1,51 @@ +%% Feel free to use, reuse and abuse the code in this file. + +%% @doc Directory handler. +-module(directory_handler). + +%% REST Callbacks +-export([init/3]). +-export([rest_init/2]). +-export([allowed_methods/2]). +-export([resource_exists/2]). +-export([content_types_provided/2]). + +%% Callback Callbacks +-export([list_json/2]). +-export([list_html/2]). + +init(_Transport, _Req, _Paths) -> + {upgrade, protocol, cowboy_rest}. + +rest_init(Req, Paths) -> + {ok, Req, Paths}. + +allowed_methods(Req, State) -> + {[<<"GET">>], Req, State}. + +resource_exists(Req, {ReqPath, FilePath}) -> + case file:list_dir(FilePath) of + {ok, Fs} -> {true, Req, {ReqPath, lists:sort(Fs)}}; + _Err -> {false, Req, {ReqPath, FilePath}} + end. + +content_types_provided(Req, State) -> + {[ + {{<<"application">>, <<"json">>, []}, list_json}, + {{<<"text">>, <<"html">>, []}, list_html} + ], Req, State}. + +list_json(Req, {Path, Fs}) -> + Files = [[ <<(list_to_binary(F))/binary>> || F <- Fs ]], + {jsx:encode(Files), Req, Path}. + +list_html(Req, {Path, Fs}) -> + Body = [[ links(Path, F) || F <- [".."|Fs] ]], + HTML = [<<"<!DOCTYPE html><html><head><title>Index</title></head>", + "<body>">>, Body, <<"</body></html>\n">>], + {HTML, Req, Path}. + +links(<<>>, File) -> + ["<a href='/", File, "'>", File, "</a><br>\n"]; +links(Prefix, File) -> + ["<a href='/", Prefix, $/, File, "'>", File, "</a><br>\n"]. |