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
37
38
39
40
41
42
43

extern crate iron;

use iron::{Handler, IronResult, BeforeMiddleware, Request, Response, Chain};

pub struct SelectiveMiddleWare {
    chain: Chain,
}

impl SelectiveMiddleWare {
    /// Create a new SelectiveMiddleWare instance with the given BeforeMiddleware.
    pub fn new<H: Handler, M: BeforeMiddleware>(handler: H, m: Vec<M>) -> Self {

        let mut chain = Chain::new(handler);
        for item in m.into_iter() {
            chain.link_before(item);
        }

        SelectiveMiddleWare {
            chain: chain
        }
    }
}

impl Handler for SelectiveMiddleWare {
    fn handle(&self, req: &mut Request) -> IronResult<Response> {
        self.chain.handle(req)
    }
}

#[macro_export]
macro_rules! with_middleware {
    ($handler:ident, before=[$($y:ident),+] ) => {
        {
            use selective_middleware::SelectiveMiddleWare;
            let before = vec!($( $y  ),*);

            SelectiveMiddleWare::new($handler, before)
        }
    };
    ($handler:ident, [$($y:ident),+] ) => {
        with_middleware!($handler, before=[$( $y  ),*])
    }
}