diff options
| author | Adam Cammack <[email protected]> | 2013-02-16 02:21:54 -0600 |
|---|---|---|
| committer | Adam Cammack <[email protected]> | 2013-02-16 02:21:54 -0600 |
| commit | f112cdf643c520bb6554a2d8efa882e8e904b984 (patch) | |
| tree | 640ef3b6570c1359345cc8f98854936ca3a67540 /examples/web_server/src/directory_handler.erl | |
| parent | e3daf439da42283cf65faa3311ff73bb7ffe413b (diff) | |
| download | cowboy-f112cdf643c520bb6554a2d8efa882e8e904b984.tar.gz cowboy-f112cdf643c520bb6554a2d8efa882e8e904b984.tar.bz2 cowboy-f112cdf643c520bb6554a2d8efa882e8e904b984.zip | |
Add a web server example
Explore re-routing in middleware.
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"]. |