1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

#[macro_export]
macro_rules! router {
    ($($input:tt)*) => {{
            use $crate::HttpRouter;
            let mut router = $crate::Router::new();

            _router_inner!(router $($input)*)
    }}
}

#[doc(hidden)]
#[macro_export]
macro_rules! _router_inner {
    ($router:ident)
        => { $router }; // Base case
    ($router:ident $method:ident $path:expr => |$req:tt, mut $res:ident| { $($b:tt)* } $($rest:tt)*)
        => {{
            $router.$method($path, middleware!(|$req, mut $res| $($b)*));

            _router_inner!($router $($rest)*)
        }};
    ($router:ident $method:ident $path:expr => |$req:tt, $res:ident| { $($b:tt)* } $($rest:tt)*)
        => {{
            $router.$method($path, middleware!(|$req, $res| $($b)*));

            _router_inner!($router $($rest)*)
        }};
    ($router:ident $method:ident $path:expr => |$req:tt| { $($b:tt)* } $($rest:tt)*)
        => {
            _router_inner!($router $method $path => |$req, _res| { $($b)* } $($rest)*)
        };
    ($router:ident $method:ident $path:expr => { $($b:tt)* } $($rest:tt)*)
        => {
            _router_inner!($router $method $path => |_, _res| { $($b)* } $($rest)*)
        };
}