aboutsummaryrefslogtreecommitdiffstats
path: root/examples/file_server/src/directory_handler.erl
diff options
context:
space:
mode:
authorLoïc Hoguin <[email protected]>2016-06-13 13:23:22 +0200
committerLoïc Hoguin <[email protected]>2016-06-13 13:26:24 +0200
commit88227898edd26a823d0942fc7226adb61a20cb5d (patch)
treeaf1aeb41e7b92145b3be017cfdf18bf66f4a8e6e /examples/file_server/src/directory_handler.erl
parent4293a40d9e922bc00f25d6c7e328f86e102275a0 (diff)
downloadcowboy-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.erl47
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"].