diff options
author | Loïc Hoguin <[email protected]> | 2016-06-13 13:23:22 +0200 |
---|---|---|
committer | Loïc Hoguin <[email protected]> | 2016-06-13 13:26:24 +0200 |
commit | 88227898edd26a823d0942fc7226adb61a20cb5d (patch) | |
tree | af1aeb41e7b92145b3be017cfdf18bf66f4a8e6e /examples/file_server/src/directory_handler.erl | |
parent | 4293a40d9e922bc00f25d6c7e328f86e102275a0 (diff) | |
download | cowboy-88227898edd26a823d0942fc7226adb61a20cb5d.tar.gz cowboy-88227898edd26a823d0942fc7226adb61a20cb5d.tar.bz2 cowboy-88227898edd26a823d0942fc7226adb61a20cb5d.zip |
Merge static_world and web_server examples
The new example is called file_server and it's basically
the same as web_server was. The name is clearer than the
original, all examples being "Web servers".
The new example is also tested and the test suite has
been refactored a little.
Diffstat (limited to 'examples/file_server/src/directory_handler.erl')
-rw-r--r-- | examples/file_server/src/directory_handler.erl | 47 |
1 files changed, 47 insertions, 0 deletions
diff --git a/examples/file_server/src/directory_handler.erl b/examples/file_server/src/directory_handler.erl new file mode 100644 index 0000000..d16e014 --- /dev/null +++ b/examples/file_server/src/directory_handler.erl @@ -0,0 +1,47 @@ +%% Feel free to use, reuse and abuse the code in this file. + +%% @doc Directory handler. +-module(directory_handler). + +%% REST Callbacks +-export([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(Req, Paths) -> + {cowboy_rest, 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) -> + {[ + {{<<"text">>, <<"html">>, []}, list_html}, + {{<<"application">>, <<"json">>, []}, list_json} + ], 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"]. |