mirror of
https://github.com/ninenines/cowboy.git
synced 2025-07-14 20:30:23 +00:00
Remove cowboy_dispatcher
Types and code are moved to cowboy_router. The match/3 export from cowboy_dispatcher isn't available anymore as it is called internally.
This commit is contained in:
parent
bcb1222df9
commit
68da864aeb
3 changed files with 277 additions and 301 deletions
|
@ -1,291 +0,0 @@
|
||||||
%% Copyright (c) 2011-2013, Loïc Hoguin <essen@ninenines.eu>
|
|
||||||
%% Copyright (c) 2011, Anthony Ramine <nox@dev-extend.eu>
|
|
||||||
%%
|
|
||||||
%% Permission to use, copy, modify, and/or distribute this software for any
|
|
||||||
%% purpose with or without fee is hereby granted, provided that the above
|
|
||||||
%% copyright notice and this permission notice appear in all copies.
|
|
||||||
%%
|
|
||||||
%% THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
||||||
%% WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
||||||
%% MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
||||||
%% ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
||||||
%% WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
||||||
%% ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
||||||
%% OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
||||||
|
|
||||||
%% @doc Dispatch requests according to a hostname and path.
|
|
||||||
-module(cowboy_dispatcher).
|
|
||||||
|
|
||||||
%% API.
|
|
||||||
-export([match/3]).
|
|
||||||
|
|
||||||
-type bindings() :: [{atom(), binary()}].
|
|
||||||
-type tokens() :: [binary()].
|
|
||||||
-type match_rule() :: '_' | <<_:8>> | [binary() | '_' | '...' | atom()].
|
|
||||||
-type dispatch_path() :: [{match_rule(), module(), any()}].
|
|
||||||
-type dispatch_rule() :: {Host::match_rule(), Path::dispatch_path()}.
|
|
||||||
-type dispatch_rules() :: [dispatch_rule()].
|
|
||||||
|
|
||||||
-export_type([bindings/0]).
|
|
||||||
-export_type([tokens/0]).
|
|
||||||
-export_type([dispatch_rules/0]).
|
|
||||||
|
|
||||||
-ifdef(TEST).
|
|
||||||
-include_lib("eunit/include/eunit.hrl").
|
|
||||||
-endif.
|
|
||||||
|
|
||||||
%% API.
|
|
||||||
|
|
||||||
%% @doc Match hostname tokens and path tokens against dispatch rules.
|
|
||||||
%%
|
|
||||||
%% It is typically used for matching tokens for the hostname and path of
|
|
||||||
%% the request against a global dispatch rule for your listener.
|
|
||||||
%%
|
|
||||||
%% Dispatch rules are a list of <em>{Hostname, PathRules}</em> tuples, with
|
|
||||||
%% <em>PathRules</em> being a list of <em>{Path, HandlerMod, HandlerOpts}</em>.
|
|
||||||
%%
|
|
||||||
%% <em>Hostname</em> and <em>Path</em> are match rules and can be either the
|
|
||||||
%% atom <em>'_'</em>, which matches everything, `<<"*">>', which match the
|
|
||||||
%% wildcard path, or a list of tokens.
|
|
||||||
%%
|
|
||||||
%% Each token can be either a binary, the atom <em>'_'</em>,
|
|
||||||
%% the atom '...' or a named atom. A binary token must match exactly,
|
|
||||||
%% <em>'_'</em> matches everything for a single token, <em>'...'</em> matches
|
|
||||||
%% everything for the rest of the tokens and a named atom will bind the
|
|
||||||
%% corresponding token value and return it.
|
|
||||||
%%
|
|
||||||
%% The list of hostname tokens is reversed before matching. For example, if
|
|
||||||
%% we were to match "www.ninenines.eu", we would first match "eu", then
|
|
||||||
%% "ninenines", then "www". This means that in the context of hostnames,
|
|
||||||
%% the <em>'...'</em> atom matches properly the lower levels of the domain
|
|
||||||
%% as would be expected.
|
|
||||||
%%
|
|
||||||
%% When a result is found, this function will return the handler module and
|
|
||||||
%% options found in the dispatch list, a key-value list of bindings and
|
|
||||||
%% the tokens that were matched by the <em>'...'</em> atom for both the
|
|
||||||
%% hostname and path.
|
|
||||||
-spec match(dispatch_rules(), Host::binary() | tokens(), Path::binary())
|
|
||||||
-> {ok, module(), any(), bindings(),
|
|
||||||
HostInfo::undefined | tokens(),
|
|
||||||
PathInfo::undefined | tokens()}
|
|
||||||
| {error, notfound, host} | {error, notfound, path}
|
|
||||||
| {error, badrequest, path}.
|
|
||||||
match([], _, _) ->
|
|
||||||
{error, notfound, host};
|
|
||||||
match([{'_', PathMatchs}|_Tail], _, Path) ->
|
|
||||||
match_path(PathMatchs, undefined, Path, []);
|
|
||||||
match([{HostMatch, PathMatchs}|Tail], Tokens, Path)
|
|
||||||
when is_list(Tokens) ->
|
|
||||||
case list_match(Tokens, lists:reverse(HostMatch), []) of
|
|
||||||
false ->
|
|
||||||
match(Tail, Tokens, Path);
|
|
||||||
{true, Bindings, undefined} ->
|
|
||||||
match_path(PathMatchs, undefined, Path, Bindings);
|
|
||||||
{true, Bindings, HostInfo} ->
|
|
||||||
match_path(PathMatchs, lists:reverse(HostInfo),
|
|
||||||
Path, Bindings)
|
|
||||||
end;
|
|
||||||
match(Dispatch, Host, Path) ->
|
|
||||||
match(Dispatch, split_host(Host), Path).
|
|
||||||
|
|
||||||
-spec match_path(dispatch_path(),
|
|
||||||
HostInfo::undefined | tokens(), binary() | tokens(), bindings())
|
|
||||||
-> {ok, module(), any(), bindings(),
|
|
||||||
HostInfo::undefined | tokens(),
|
|
||||||
PathInfo::undefined | tokens()}
|
|
||||||
| {error, notfound, path} | {error, badrequest, path}.
|
|
||||||
match_path([], _, _, _) ->
|
|
||||||
{error, notfound, path};
|
|
||||||
match_path([{'_', Handler, Opts}|_Tail], HostInfo, _, Bindings) ->
|
|
||||||
{ok, Handler, Opts, Bindings, HostInfo, undefined};
|
|
||||||
match_path([{<<"*">>, Handler, Opts}|_Tail], HostInfo, <<"*">>, Bindings) ->
|
|
||||||
{ok, Handler, Opts, Bindings, HostInfo, undefined};
|
|
||||||
match_path([{PathMatch, Handler, Opts}|Tail], HostInfo, Tokens,
|
|
||||||
Bindings) when is_list(Tokens) ->
|
|
||||||
case list_match(Tokens, PathMatch, []) of
|
|
||||||
false ->
|
|
||||||
match_path(Tail, HostInfo, Tokens, Bindings);
|
|
||||||
{true, PathBinds, PathInfo} ->
|
|
||||||
{ok, Handler, Opts, Bindings ++ PathBinds, HostInfo, PathInfo}
|
|
||||||
end;
|
|
||||||
match_path(_Dispatch, _HostInfo, badrequest, _Bindings) ->
|
|
||||||
{error, badrequest, path};
|
|
||||||
match_path(Dispatch, HostInfo, Path, Bindings) ->
|
|
||||||
match_path(Dispatch, HostInfo, split_path(Path), Bindings).
|
|
||||||
|
|
||||||
%% Internal.
|
|
||||||
|
|
||||||
%% @doc Split a hostname into a list of tokens.
|
|
||||||
-spec split_host(binary()) -> tokens().
|
|
||||||
split_host(Host) ->
|
|
||||||
split_host(Host, []).
|
|
||||||
|
|
||||||
split_host(Host, Acc) ->
|
|
||||||
case binary:match(Host, <<".">>) of
|
|
||||||
nomatch when Host =:= <<>> ->
|
|
||||||
Acc;
|
|
||||||
nomatch ->
|
|
||||||
[Host|Acc];
|
|
||||||
{Pos, _} ->
|
|
||||||
<< Segment:Pos/binary, _:8, Rest/bits >> = Host,
|
|
||||||
false = byte_size(Segment) == 0,
|
|
||||||
split_host(Rest, [Segment|Acc])
|
|
||||||
end.
|
|
||||||
|
|
||||||
%% @doc Split a path into a list of path segments.
|
|
||||||
%%
|
|
||||||
%% Following RFC2396, this function may return path segments containing any
|
|
||||||
%% character, including <em>/</em> if, and only if, a <em>/</em> was escaped
|
|
||||||
%% and part of a path segment.
|
|
||||||
-spec split_path(binary()) -> tokens().
|
|
||||||
split_path(<< $/, Path/bits >>) ->
|
|
||||||
split_path(Path, []);
|
|
||||||
split_path(_) ->
|
|
||||||
badrequest.
|
|
||||||
|
|
||||||
split_path(Path, Acc) ->
|
|
||||||
try
|
|
||||||
case binary:match(Path, <<"/">>) of
|
|
||||||
nomatch when Path =:= <<>> ->
|
|
||||||
lists:reverse([cowboy_http:urldecode(S) || S <- Acc]);
|
|
||||||
nomatch ->
|
|
||||||
lists:reverse([cowboy_http:urldecode(S) || S <- [Path|Acc]]);
|
|
||||||
{Pos, _} ->
|
|
||||||
<< Segment:Pos/binary, _:8, Rest/bits >> = Path,
|
|
||||||
split_path(Rest, [Segment|Acc])
|
|
||||||
end
|
|
||||||
catch
|
|
||||||
error:badarg ->
|
|
||||||
badrequest
|
|
||||||
end.
|
|
||||||
|
|
||||||
-spec list_match(tokens(), match_rule(), bindings())
|
|
||||||
-> {true, bindings(), undefined | tokens()} | false.
|
|
||||||
%% Atom '...' matches any trailing path, stop right now.
|
|
||||||
list_match(List, ['...'], Binds) ->
|
|
||||||
{true, Binds, List};
|
|
||||||
%% Atom '_' matches anything, continue.
|
|
||||||
list_match([_E|Tail], ['_'|TailMatch], Binds) ->
|
|
||||||
list_match(Tail, TailMatch, Binds);
|
|
||||||
%% Both values match, continue.
|
|
||||||
list_match([E|Tail], [E|TailMatch], Binds) ->
|
|
||||||
list_match(Tail, TailMatch, Binds);
|
|
||||||
%% Bind E to the variable name V and continue.
|
|
||||||
list_match([E|Tail], [V|TailMatch], Binds) when is_atom(V) ->
|
|
||||||
list_match(Tail, TailMatch, [{V, E}|Binds]);
|
|
||||||
%% Match complete.
|
|
||||||
list_match([], [], Binds) ->
|
|
||||||
{true, Binds, undefined};
|
|
||||||
%% Values don't match, stop.
|
|
||||||
list_match(_List, _Match, _Binds) ->
|
|
||||||
false.
|
|
||||||
|
|
||||||
%% Tests.
|
|
||||||
|
|
||||||
-ifdef(TEST).
|
|
||||||
|
|
||||||
split_host_test_() ->
|
|
||||||
%% {Host, Result}
|
|
||||||
Tests = [
|
|
||||||
{<<"">>, []},
|
|
||||||
{<<"*">>, [<<"*">>]},
|
|
||||||
{<<"cowboy.ninenines.eu">>,
|
|
||||||
[<<"eu">>, <<"ninenines">>, <<"cowboy">>]},
|
|
||||||
{<<"ninenines.eu">>,
|
|
||||||
[<<"eu">>, <<"ninenines">>]},
|
|
||||||
{<<"a.b.c.d.e.f.g.h.i.j.k.l.m.n.o.p.q.r.s.t.u.v.w.x.y.z">>,
|
|
||||||
[<<"z">>, <<"y">>, <<"x">>, <<"w">>, <<"v">>, <<"u">>, <<"t">>,
|
|
||||||
<<"s">>, <<"r">>, <<"q">>, <<"p">>, <<"o">>, <<"n">>, <<"m">>,
|
|
||||||
<<"l">>, <<"k">>, <<"j">>, <<"i">>, <<"h">>, <<"g">>, <<"f">>,
|
|
||||||
<<"e">>, <<"d">>, <<"c">>, <<"b">>, <<"a">>]}
|
|
||||||
],
|
|
||||||
[{H, fun() -> R = split_host(H) end} || {H, R} <- Tests].
|
|
||||||
|
|
||||||
split_path_test_() ->
|
|
||||||
%% {Path, Result, QueryString}
|
|
||||||
Tests = [
|
|
||||||
{<<"/">>, []},
|
|
||||||
{<<"/extend//cowboy">>, [<<"extend">>, <<>>, <<"cowboy">>]},
|
|
||||||
{<<"/users">>, [<<"users">>]},
|
|
||||||
{<<"/users/42/friends">>, [<<"users">>, <<"42">>, <<"friends">>]},
|
|
||||||
{<<"/users/a+b/c%21d">>, [<<"users">>, <<"a b">>, <<"c!d">>]}
|
|
||||||
],
|
|
||||||
[{P, fun() -> R = split_path(P) end} || {P, R} <- Tests].
|
|
||||||
|
|
||||||
match_test_() ->
|
|
||||||
Dispatch = [
|
|
||||||
{[<<"www">>, '_', <<"ninenines">>, <<"eu">>], [
|
|
||||||
{[<<"users">>, '_', <<"mails">>], match_any_subdomain_users, []}
|
|
||||||
]},
|
|
||||||
{[<<"ninenines">>, <<"eu">>], [
|
|
||||||
{[<<"users">>, id, <<"friends">>], match_extend_users_friends, []},
|
|
||||||
{'_', match_extend, []}
|
|
||||||
]},
|
|
||||||
{[<<"ninenines">>, var], [
|
|
||||||
{[<<"threads">>, var], match_duplicate_vars,
|
|
||||||
[we, {expect, two}, var, here]}
|
|
||||||
]},
|
|
||||||
{[<<"erlang">>, ext], [
|
|
||||||
{'_', match_erlang_ext, []}
|
|
||||||
]},
|
|
||||||
{'_', [
|
|
||||||
{[<<"users">>, id, <<"friends">>], match_users_friends, []},
|
|
||||||
{'_', match_any, []}
|
|
||||||
]}
|
|
||||||
],
|
|
||||||
%% {Host, Path, Result}
|
|
||||||
Tests = [
|
|
||||||
{<<"any">>, <<"/">>, {ok, match_any, [], []}},
|
|
||||||
{<<"www.any.ninenines.eu">>, <<"/users/42/mails">>,
|
|
||||||
{ok, match_any_subdomain_users, [], []}},
|
|
||||||
{<<"www.ninenines.eu">>, <<"/users/42/mails">>,
|
|
||||||
{ok, match_any, [], []}},
|
|
||||||
{<<"www.ninenines.eu">>, <<"/">>,
|
|
||||||
{ok, match_any, [], []}},
|
|
||||||
{<<"www.any.ninenines.eu">>, <<"/not_users/42/mails">>,
|
|
||||||
{error, notfound, path}},
|
|
||||||
{<<"ninenines.eu">>, <<"/">>,
|
|
||||||
{ok, match_extend, [], []}},
|
|
||||||
{<<"ninenines.eu">>, <<"/users/42/friends">>,
|
|
||||||
{ok, match_extend_users_friends, [], [{id, <<"42">>}]}},
|
|
||||||
{<<"erlang.fr">>, '_',
|
|
||||||
{ok, match_erlang_ext, [], [{ext, <<"fr">>}]}},
|
|
||||||
{<<"any">>, <<"/users/444/friends">>,
|
|
||||||
{ok, match_users_friends, [], [{id, <<"444">>}]}},
|
|
||||||
{<<"ninenines.fr">>, <<"/threads/987">>,
|
|
||||||
{ok, match_duplicate_vars, [we, {expect, two}, var, here],
|
|
||||||
[{var, <<"fr">>}, {var, <<"987">>}]}}
|
|
||||||
],
|
|
||||||
[{lists:flatten(io_lib:format("~p, ~p", [H, P])), fun() ->
|
|
||||||
{ok, Handler, Opts, Binds, undefined, undefined}
|
|
||||||
= match(Dispatch, H, P)
|
|
||||||
end} || {H, P, {ok, Handler, Opts, Binds}} <- Tests].
|
|
||||||
|
|
||||||
match_info_test_() ->
|
|
||||||
Dispatch = [
|
|
||||||
{[<<"www">>, <<"ninenines">>, <<"eu">>], [
|
|
||||||
{[<<"pathinfo">>, <<"is">>, <<"next">>, '...'], match_path, []}
|
|
||||||
]},
|
|
||||||
{['...', <<"ninenines">>, <<"eu">>], [
|
|
||||||
{'_', match_any, []}
|
|
||||||
]}
|
|
||||||
],
|
|
||||||
Tests = [
|
|
||||||
{<<"ninenines.eu">>, <<"/">>,
|
|
||||||
{ok, match_any, [], [], [], undefined}},
|
|
||||||
{<<"bugs.ninenines.eu">>, <<"/">>,
|
|
||||||
{ok, match_any, [], [], [<<"bugs">>], undefined}},
|
|
||||||
{<<"cowboy.bugs.ninenines.eu">>, <<"/">>,
|
|
||||||
{ok, match_any, [], [], [<<"cowboy">>, <<"bugs">>], undefined}},
|
|
||||||
{<<"www.ninenines.eu">>, <<"/pathinfo/is/next">>,
|
|
||||||
{ok, match_path, [], [], undefined, []}},
|
|
||||||
{<<"www.ninenines.eu">>, <<"/pathinfo/is/next/path_info">>,
|
|
||||||
{ok, match_path, [], [], undefined, [<<"path_info">>]}},
|
|
||||||
{<<"www.ninenines.eu">>, <<"/pathinfo/is/next/foo/bar">>,
|
|
||||||
{ok, match_path, [], [], undefined, [<<"foo">>, <<"bar">>]}}
|
|
||||||
],
|
|
||||||
[{lists:flatten(io_lib:format("~p, ~p", [H, P])), fun() ->
|
|
||||||
R = match(Dispatch, H, P)
|
|
||||||
end} || {H, P, R} <- Tests].
|
|
||||||
|
|
||||||
-endif.
|
|
|
@ -137,14 +137,14 @@
|
||||||
version = {1, 1} :: cowboy_http:version(),
|
version = {1, 1} :: cowboy_http:version(),
|
||||||
peer = undefined :: undefined | {inet:ip_address(), inet:port_number()},
|
peer = undefined :: undefined | {inet:ip_address(), inet:port_number()},
|
||||||
host = undefined :: undefined | binary(),
|
host = undefined :: undefined | binary(),
|
||||||
host_info = undefined :: undefined | cowboy_dispatcher:tokens(),
|
host_info = undefined :: undefined | cowboy_router:tokens(),
|
||||||
port = undefined :: undefined | inet:port_number(),
|
port = undefined :: undefined | inet:port_number(),
|
||||||
path = undefined :: binary(),
|
path = undefined :: binary(),
|
||||||
path_info = undefined :: undefined | cowboy_dispatcher:tokens(),
|
path_info = undefined :: undefined | cowboy_router:tokens(),
|
||||||
qs = undefined :: binary(),
|
qs = undefined :: binary(),
|
||||||
qs_vals = undefined :: undefined | list({binary(), binary() | true}),
|
qs_vals = undefined :: undefined | list({binary(), binary() | true}),
|
||||||
fragment = undefined :: binary(),
|
fragment = undefined :: binary(),
|
||||||
bindings = undefined :: undefined | cowboy_dispatcher:bindings(),
|
bindings = undefined :: undefined | cowboy_router:bindings(),
|
||||||
headers = [] :: cowboy_http:headers(),
|
headers = [] :: cowboy_http:headers(),
|
||||||
p_headers = [] :: [any()], %% @todo Improve those specs.
|
p_headers = [] :: [any()], %% @todo Improve those specs.
|
||||||
cookies = undefined :: undefined | [{binary(), binary()}],
|
cookies = undefined :: undefined | [{binary(), binary()}],
|
||||||
|
@ -256,7 +256,7 @@ host(Req) ->
|
||||||
%% @doc Return the extra host information obtained from partially matching
|
%% @doc Return the extra host information obtained from partially matching
|
||||||
%% the hostname using <em>'...'</em>.
|
%% the hostname using <em>'...'</em>.
|
||||||
-spec host_info(Req)
|
-spec host_info(Req)
|
||||||
-> {cowboy_dispatcher:tokens() | undefined, Req} when Req::req().
|
-> {cowboy_router:tokens() | undefined, Req} when Req::req().
|
||||||
host_info(Req) ->
|
host_info(Req) ->
|
||||||
{Req#http_req.host_info, Req}.
|
{Req#http_req.host_info, Req}.
|
||||||
|
|
||||||
|
@ -273,7 +273,7 @@ path(Req) ->
|
||||||
%% @doc Return the extra path information obtained from partially matching
|
%% @doc Return the extra path information obtained from partially matching
|
||||||
%% the patch using <em>'...'</em>.
|
%% the patch using <em>'...'</em>.
|
||||||
-spec path_info(Req)
|
-spec path_info(Req)
|
||||||
-> {cowboy_dispatcher:tokens() | undefined, Req} when Req::req().
|
-> {cowboy_router:tokens() | undefined, Req} when Req::req().
|
||||||
path_info(Req) ->
|
path_info(Req) ->
|
||||||
{Req#http_req.path_info, Req}.
|
{Req#http_req.path_info, Req}.
|
||||||
|
|
||||||
|
@ -1122,8 +1122,8 @@ set([{transport, Val}|Tail], Req) -> set(Tail, Req#http_req{transport=Val});
|
||||||
set([{version, Val}|Tail], Req) -> set(Tail, Req#http_req{version=Val}).
|
set([{version, Val}|Tail], Req) -> set(Tail, Req#http_req{version=Val}).
|
||||||
|
|
||||||
%% @private
|
%% @private
|
||||||
-spec set_bindings(cowboy_dispatcher:tokens(), cowboy_dispatcher:tokens(),
|
-spec set_bindings(cowboy_router:tokens(), cowboy_router:tokens(),
|
||||||
cowboy_dispatcher:bindings(), Req) -> Req when Req::req().
|
cowboy_router:bindings(), Req) -> Req when Req::req().
|
||||||
set_bindings(HostInfo, PathInfo, Bindings, Req) ->
|
set_bindings(HostInfo, PathInfo, Bindings, Req) ->
|
||||||
Req#http_req{host_info=HostInfo, path_info=PathInfo,
|
Req#http_req{host_info=HostInfo, path_info=PathInfo,
|
||||||
bindings=Bindings}.
|
bindings=Bindings}.
|
||||||
|
|
|
@ -22,13 +22,26 @@
|
||||||
%%
|
%%
|
||||||
%% If the route cannot be found, processing stops with either
|
%% If the route cannot be found, processing stops with either
|
||||||
%% a 400 or a 404 reply.
|
%% a 400 or a 404 reply.
|
||||||
%%
|
|
||||||
%% @see cowboy_dispatcher
|
|
||||||
-module(cowboy_router).
|
-module(cowboy_router).
|
||||||
-behaviour(cowboy_middleware).
|
-behaviour(cowboy_middleware).
|
||||||
|
|
||||||
-export([execute/2]).
|
-export([execute/2]).
|
||||||
|
|
||||||
|
-type bindings() :: [{atom(), binary()}].
|
||||||
|
-type tokens() :: [binary()].
|
||||||
|
-type match_rule() :: '_' | <<_:8>> | [binary() | '_' | '...' | atom()].
|
||||||
|
-type dispatch_path() :: [{match_rule(), module(), any()}].
|
||||||
|
-type dispatch_rule() :: {Host::match_rule(), Path::dispatch_path()}.
|
||||||
|
-type dispatch_rules() :: [dispatch_rule()].
|
||||||
|
|
||||||
|
-export_type([bindings/0]).
|
||||||
|
-export_type([tokens/0]).
|
||||||
|
-export_type([dispatch_rules/0]).
|
||||||
|
|
||||||
|
-ifdef(TEST).
|
||||||
|
-include_lib("eunit/include/eunit.hrl").
|
||||||
|
-endif.
|
||||||
|
|
||||||
%% @private
|
%% @private
|
||||||
-spec execute(Req, Env)
|
-spec execute(Req, Env)
|
||||||
-> {ok, Req, Env} | {error, 400 | 404, Req}
|
-> {ok, Req, Env} | {error, 400 | 404, Req}
|
||||||
|
@ -36,7 +49,7 @@
|
||||||
execute(Req, Env) ->
|
execute(Req, Env) ->
|
||||||
{_, Dispatch} = lists:keyfind(dispatch, 1, Env),
|
{_, Dispatch} = lists:keyfind(dispatch, 1, Env),
|
||||||
[Host, Path] = cowboy_req:get([host, path], Req),
|
[Host, Path] = cowboy_req:get([host, path], Req),
|
||||||
case cowboy_dispatcher:match(Dispatch, Host, Path) of
|
case match(Dispatch, Host, Path) of
|
||||||
{ok, Handler, HandlerOpts, Bindings, HostInfo, PathInfo} ->
|
{ok, Handler, HandlerOpts, Bindings, HostInfo, PathInfo} ->
|
||||||
Req2 = cowboy_req:set_bindings(HostInfo, PathInfo, Bindings, Req),
|
Req2 = cowboy_req:set_bindings(HostInfo, PathInfo, Bindings, Req),
|
||||||
{ok, Req2, [{handler, Handler}, {handler_opts, HandlerOpts}|Env]};
|
{ok, Req2, [{handler, Handler}, {handler_opts, HandlerOpts}|Env]};
|
||||||
|
@ -47,3 +60,257 @@ execute(Req, Env) ->
|
||||||
{error, notfound, path} ->
|
{error, notfound, path} ->
|
||||||
{error, 404, Req}
|
{error, 404, Req}
|
||||||
end.
|
end.
|
||||||
|
|
||||||
|
%% Internal.
|
||||||
|
|
||||||
|
%% @doc Match hostname tokens and path tokens against dispatch rules.
|
||||||
|
%%
|
||||||
|
%% It is typically used for matching tokens for the hostname and path of
|
||||||
|
%% the request against a global dispatch rule for your listener.
|
||||||
|
%%
|
||||||
|
%% Dispatch rules are a list of <em>{Hostname, PathRules}</em> tuples, with
|
||||||
|
%% <em>PathRules</em> being a list of <em>{Path, HandlerMod, HandlerOpts}</em>.
|
||||||
|
%%
|
||||||
|
%% <em>Hostname</em> and <em>Path</em> are match rules and can be either the
|
||||||
|
%% atom <em>'_'</em>, which matches everything, `<<"*">>', which match the
|
||||||
|
%% wildcard path, or a list of tokens.
|
||||||
|
%%
|
||||||
|
%% Each token can be either a binary, the atom <em>'_'</em>,
|
||||||
|
%% the atom '...' or a named atom. A binary token must match exactly,
|
||||||
|
%% <em>'_'</em> matches everything for a single token, <em>'...'</em> matches
|
||||||
|
%% everything for the rest of the tokens and a named atom will bind the
|
||||||
|
%% corresponding token value and return it.
|
||||||
|
%%
|
||||||
|
%% The list of hostname tokens is reversed before matching. For example, if
|
||||||
|
%% we were to match "www.ninenines.eu", we would first match "eu", then
|
||||||
|
%% "ninenines", then "www". This means that in the context of hostnames,
|
||||||
|
%% the <em>'...'</em> atom matches properly the lower levels of the domain
|
||||||
|
%% as would be expected.
|
||||||
|
%%
|
||||||
|
%% When a result is found, this function will return the handler module and
|
||||||
|
%% options found in the dispatch list, a key-value list of bindings and
|
||||||
|
%% the tokens that were matched by the <em>'...'</em> atom for both the
|
||||||
|
%% hostname and path.
|
||||||
|
-spec match(dispatch_rules(), Host::binary() | tokens(), Path::binary())
|
||||||
|
-> {ok, module(), any(), bindings(),
|
||||||
|
HostInfo::undefined | tokens(),
|
||||||
|
PathInfo::undefined | tokens()}
|
||||||
|
| {error, notfound, host} | {error, notfound, path}
|
||||||
|
| {error, badrequest, path}.
|
||||||
|
match([], _, _) ->
|
||||||
|
{error, notfound, host};
|
||||||
|
match([{'_', PathMatchs}|_Tail], _, Path) ->
|
||||||
|
match_path(PathMatchs, undefined, Path, []);
|
||||||
|
match([{HostMatch, PathMatchs}|Tail], Tokens, Path)
|
||||||
|
when is_list(Tokens) ->
|
||||||
|
case list_match(Tokens, lists:reverse(HostMatch), []) of
|
||||||
|
false ->
|
||||||
|
match(Tail, Tokens, Path);
|
||||||
|
{true, Bindings, undefined} ->
|
||||||
|
match_path(PathMatchs, undefined, Path, Bindings);
|
||||||
|
{true, Bindings, HostInfo} ->
|
||||||
|
match_path(PathMatchs, lists:reverse(HostInfo),
|
||||||
|
Path, Bindings)
|
||||||
|
end;
|
||||||
|
match(Dispatch, Host, Path) ->
|
||||||
|
match(Dispatch, split_host(Host), Path).
|
||||||
|
|
||||||
|
-spec match_path(dispatch_path(),
|
||||||
|
HostInfo::undefined | tokens(), binary() | tokens(), bindings())
|
||||||
|
-> {ok, module(), any(), bindings(),
|
||||||
|
HostInfo::undefined | tokens(),
|
||||||
|
PathInfo::undefined | tokens()}
|
||||||
|
| {error, notfound, path} | {error, badrequest, path}.
|
||||||
|
match_path([], _, _, _) ->
|
||||||
|
{error, notfound, path};
|
||||||
|
match_path([{'_', Handler, Opts}|_Tail], HostInfo, _, Bindings) ->
|
||||||
|
{ok, Handler, Opts, Bindings, HostInfo, undefined};
|
||||||
|
match_path([{<<"*">>, Handler, Opts}|_Tail], HostInfo, <<"*">>, Bindings) ->
|
||||||
|
{ok, Handler, Opts, Bindings, HostInfo, undefined};
|
||||||
|
match_path([{PathMatch, Handler, Opts}|Tail], HostInfo, Tokens,
|
||||||
|
Bindings) when is_list(Tokens) ->
|
||||||
|
case list_match(Tokens, PathMatch, []) of
|
||||||
|
false ->
|
||||||
|
match_path(Tail, HostInfo, Tokens, Bindings);
|
||||||
|
{true, PathBinds, PathInfo} ->
|
||||||
|
{ok, Handler, Opts, Bindings ++ PathBinds, HostInfo, PathInfo}
|
||||||
|
end;
|
||||||
|
match_path(_Dispatch, _HostInfo, badrequest, _Bindings) ->
|
||||||
|
{error, badrequest, path};
|
||||||
|
match_path(Dispatch, HostInfo, Path, Bindings) ->
|
||||||
|
match_path(Dispatch, HostInfo, split_path(Path), Bindings).
|
||||||
|
|
||||||
|
%% @doc Split a hostname into a list of tokens.
|
||||||
|
-spec split_host(binary()) -> tokens().
|
||||||
|
split_host(Host) ->
|
||||||
|
split_host(Host, []).
|
||||||
|
|
||||||
|
split_host(Host, Acc) ->
|
||||||
|
case binary:match(Host, <<".">>) of
|
||||||
|
nomatch when Host =:= <<>> ->
|
||||||
|
Acc;
|
||||||
|
nomatch ->
|
||||||
|
[Host|Acc];
|
||||||
|
{Pos, _} ->
|
||||||
|
<< Segment:Pos/binary, _:8, Rest/bits >> = Host,
|
||||||
|
false = byte_size(Segment) == 0,
|
||||||
|
split_host(Rest, [Segment|Acc])
|
||||||
|
end.
|
||||||
|
|
||||||
|
%% @doc Split a path into a list of path segments.
|
||||||
|
%%
|
||||||
|
%% Following RFC2396, this function may return path segments containing any
|
||||||
|
%% character, including <em>/</em> if, and only if, a <em>/</em> was escaped
|
||||||
|
%% and part of a path segment.
|
||||||
|
-spec split_path(binary()) -> tokens().
|
||||||
|
split_path(<< $/, Path/bits >>) ->
|
||||||
|
split_path(Path, []);
|
||||||
|
split_path(_) ->
|
||||||
|
badrequest.
|
||||||
|
|
||||||
|
split_path(Path, Acc) ->
|
||||||
|
try
|
||||||
|
case binary:match(Path, <<"/">>) of
|
||||||
|
nomatch when Path =:= <<>> ->
|
||||||
|
lists:reverse([cowboy_http:urldecode(S) || S <- Acc]);
|
||||||
|
nomatch ->
|
||||||
|
lists:reverse([cowboy_http:urldecode(S) || S <- [Path|Acc]]);
|
||||||
|
{Pos, _} ->
|
||||||
|
<< Segment:Pos/binary, _:8, Rest/bits >> = Path,
|
||||||
|
split_path(Rest, [Segment|Acc])
|
||||||
|
end
|
||||||
|
catch
|
||||||
|
error:badarg ->
|
||||||
|
badrequest
|
||||||
|
end.
|
||||||
|
|
||||||
|
-spec list_match(tokens(), match_rule(), bindings())
|
||||||
|
-> {true, bindings(), undefined | tokens()} | false.
|
||||||
|
%% Atom '...' matches any trailing path, stop right now.
|
||||||
|
list_match(List, ['...'], Binds) ->
|
||||||
|
{true, Binds, List};
|
||||||
|
%% Atom '_' matches anything, continue.
|
||||||
|
list_match([_E|Tail], ['_'|TailMatch], Binds) ->
|
||||||
|
list_match(Tail, TailMatch, Binds);
|
||||||
|
%% Both values match, continue.
|
||||||
|
list_match([E|Tail], [E|TailMatch], Binds) ->
|
||||||
|
list_match(Tail, TailMatch, Binds);
|
||||||
|
%% Bind E to the variable name V and continue.
|
||||||
|
list_match([E|Tail], [V|TailMatch], Binds) when is_atom(V) ->
|
||||||
|
list_match(Tail, TailMatch, [{V, E}|Binds]);
|
||||||
|
%% Match complete.
|
||||||
|
list_match([], [], Binds) ->
|
||||||
|
{true, Binds, undefined};
|
||||||
|
%% Values don't match, stop.
|
||||||
|
list_match(_List, _Match, _Binds) ->
|
||||||
|
false.
|
||||||
|
|
||||||
|
%% Tests.
|
||||||
|
|
||||||
|
-ifdef(TEST).
|
||||||
|
|
||||||
|
split_host_test_() ->
|
||||||
|
%% {Host, Result}
|
||||||
|
Tests = [
|
||||||
|
{<<"">>, []},
|
||||||
|
{<<"*">>, [<<"*">>]},
|
||||||
|
{<<"cowboy.ninenines.eu">>,
|
||||||
|
[<<"eu">>, <<"ninenines">>, <<"cowboy">>]},
|
||||||
|
{<<"ninenines.eu">>,
|
||||||
|
[<<"eu">>, <<"ninenines">>]},
|
||||||
|
{<<"a.b.c.d.e.f.g.h.i.j.k.l.m.n.o.p.q.r.s.t.u.v.w.x.y.z">>,
|
||||||
|
[<<"z">>, <<"y">>, <<"x">>, <<"w">>, <<"v">>, <<"u">>, <<"t">>,
|
||||||
|
<<"s">>, <<"r">>, <<"q">>, <<"p">>, <<"o">>, <<"n">>, <<"m">>,
|
||||||
|
<<"l">>, <<"k">>, <<"j">>, <<"i">>, <<"h">>, <<"g">>, <<"f">>,
|
||||||
|
<<"e">>, <<"d">>, <<"c">>, <<"b">>, <<"a">>]}
|
||||||
|
],
|
||||||
|
[{H, fun() -> R = split_host(H) end} || {H, R} <- Tests].
|
||||||
|
|
||||||
|
split_path_test_() ->
|
||||||
|
%% {Path, Result, QueryString}
|
||||||
|
Tests = [
|
||||||
|
{<<"/">>, []},
|
||||||
|
{<<"/extend//cowboy">>, [<<"extend">>, <<>>, <<"cowboy">>]},
|
||||||
|
{<<"/users">>, [<<"users">>]},
|
||||||
|
{<<"/users/42/friends">>, [<<"users">>, <<"42">>, <<"friends">>]},
|
||||||
|
{<<"/users/a+b/c%21d">>, [<<"users">>, <<"a b">>, <<"c!d">>]}
|
||||||
|
],
|
||||||
|
[{P, fun() -> R = split_path(P) end} || {P, R} <- Tests].
|
||||||
|
|
||||||
|
match_test_() ->
|
||||||
|
Dispatch = [
|
||||||
|
{[<<"www">>, '_', <<"ninenines">>, <<"eu">>], [
|
||||||
|
{[<<"users">>, '_', <<"mails">>], match_any_subdomain_users, []}
|
||||||
|
]},
|
||||||
|
{[<<"ninenines">>, <<"eu">>], [
|
||||||
|
{[<<"users">>, id, <<"friends">>], match_extend_users_friends, []},
|
||||||
|
{'_', match_extend, []}
|
||||||
|
]},
|
||||||
|
{[<<"ninenines">>, var], [
|
||||||
|
{[<<"threads">>, var], match_duplicate_vars,
|
||||||
|
[we, {expect, two}, var, here]}
|
||||||
|
]},
|
||||||
|
{[<<"erlang">>, ext], [
|
||||||
|
{'_', match_erlang_ext, []}
|
||||||
|
]},
|
||||||
|
{'_', [
|
||||||
|
{[<<"users">>, id, <<"friends">>], match_users_friends, []},
|
||||||
|
{'_', match_any, []}
|
||||||
|
]}
|
||||||
|
],
|
||||||
|
%% {Host, Path, Result}
|
||||||
|
Tests = [
|
||||||
|
{<<"any">>, <<"/">>, {ok, match_any, [], []}},
|
||||||
|
{<<"www.any.ninenines.eu">>, <<"/users/42/mails">>,
|
||||||
|
{ok, match_any_subdomain_users, [], []}},
|
||||||
|
{<<"www.ninenines.eu">>, <<"/users/42/mails">>,
|
||||||
|
{ok, match_any, [], []}},
|
||||||
|
{<<"www.ninenines.eu">>, <<"/">>,
|
||||||
|
{ok, match_any, [], []}},
|
||||||
|
{<<"www.any.ninenines.eu">>, <<"/not_users/42/mails">>,
|
||||||
|
{error, notfound, path}},
|
||||||
|
{<<"ninenines.eu">>, <<"/">>,
|
||||||
|
{ok, match_extend, [], []}},
|
||||||
|
{<<"ninenines.eu">>, <<"/users/42/friends">>,
|
||||||
|
{ok, match_extend_users_friends, [], [{id, <<"42">>}]}},
|
||||||
|
{<<"erlang.fr">>, '_',
|
||||||
|
{ok, match_erlang_ext, [], [{ext, <<"fr">>}]}},
|
||||||
|
{<<"any">>, <<"/users/444/friends">>,
|
||||||
|
{ok, match_users_friends, [], [{id, <<"444">>}]}},
|
||||||
|
{<<"ninenines.fr">>, <<"/threads/987">>,
|
||||||
|
{ok, match_duplicate_vars, [we, {expect, two}, var, here],
|
||||||
|
[{var, <<"fr">>}, {var, <<"987">>}]}}
|
||||||
|
],
|
||||||
|
[{lists:flatten(io_lib:format("~p, ~p", [H, P])), fun() ->
|
||||||
|
{ok, Handler, Opts, Binds, undefined, undefined}
|
||||||
|
= match(Dispatch, H, P)
|
||||||
|
end} || {H, P, {ok, Handler, Opts, Binds}} <- Tests].
|
||||||
|
|
||||||
|
match_info_test_() ->
|
||||||
|
Dispatch = [
|
||||||
|
{[<<"www">>, <<"ninenines">>, <<"eu">>], [
|
||||||
|
{[<<"pathinfo">>, <<"is">>, <<"next">>, '...'], match_path, []}
|
||||||
|
]},
|
||||||
|
{['...', <<"ninenines">>, <<"eu">>], [
|
||||||
|
{'_', match_any, []}
|
||||||
|
]}
|
||||||
|
],
|
||||||
|
Tests = [
|
||||||
|
{<<"ninenines.eu">>, <<"/">>,
|
||||||
|
{ok, match_any, [], [], [], undefined}},
|
||||||
|
{<<"bugs.ninenines.eu">>, <<"/">>,
|
||||||
|
{ok, match_any, [], [], [<<"bugs">>], undefined}},
|
||||||
|
{<<"cowboy.bugs.ninenines.eu">>, <<"/">>,
|
||||||
|
{ok, match_any, [], [], [<<"cowboy">>, <<"bugs">>], undefined}},
|
||||||
|
{<<"www.ninenines.eu">>, <<"/pathinfo/is/next">>,
|
||||||
|
{ok, match_path, [], [], undefined, []}},
|
||||||
|
{<<"www.ninenines.eu">>, <<"/pathinfo/is/next/path_info">>,
|
||||||
|
{ok, match_path, [], [], undefined, [<<"path_info">>]}},
|
||||||
|
{<<"www.ninenines.eu">>, <<"/pathinfo/is/next/foo/bar">>,
|
||||||
|
{ok, match_path, [], [], undefined, [<<"foo">>, <<"bar">>]}}
|
||||||
|
],
|
||||||
|
[{lists:flatten(io_lib:format("~p, ~p", [H, P])), fun() ->
|
||||||
|
R = match(Dispatch, H, P)
|
||||||
|
end} || {H, P, R} <- Tests].
|
||||||
|
|
||||||
|
-endif.
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue