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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
use std::ops::{BitAnd, BitOr, BitXor};

use crate::Context;

/// A guard for Context
pub trait Guard: Send + Sync + 'static {
    /// Checks the Context
    fn check(&self, _: &Context) -> bool;
}

impl<F> Guard for F
where
    F: Send + Sync + 'static + Fn(&Context) -> bool,
{
    fn check(&self, cx: &Context) -> bool {
        (self)(cx)
    }
}

impl Guard for Box<dyn Guard> {
    fn check(&self, cx: &Context) -> bool {
        (**self).check(cx)
    }
}

impl BitAnd for Box<dyn Guard> {
    type Output = Self;

    fn bitand(self, rhs: Self) -> Self::Output {
        Box::new(move |cx: &Context| self.check(cx) && rhs.check(cx))
    }
}

impl BitOr for Box<dyn Guard> {
    type Output = Self;

    fn bitor(self, rhs: Self) -> Self::Output {
        Box::new(move |cx: &Context| self.check(cx) || rhs.check(cx))
    }
}

impl BitXor for Box<dyn Guard> {
    type Output = Self;

    fn bitxor(self, rhs: Self) -> Self::Output {
        Box::new(move |cx: &Context| self.check(cx) ^ rhs.check(cx))
    }
}

impl<F> From<F> for Box<dyn Guard>
where
    F: Send + Sync + 'static + Fn(&Context) -> bool,
{
    fn from(f: F) -> Self {
        Box::new(f)
    }
}

#[cfg(test)]
mod tests {
    use crate::{http, Context, Guard};

    #[test]
    fn guard() {
        let a: Box<dyn Guard> = Box::new(|_: &Context| true);
        let b: Box<dyn Guard> = Box::new(|_: &Context| false);
        let c: Box<dyn Guard> = Box::new(|_: &Context| true);
        let d: Box<dyn Guard> = Box::new(|_: &Context| false);
        let e: Box<dyn Guard> = (a & b) ^ (c | d);
        let cx = Context::from(http::Request::new("hello world".into()));
        let res = e.check(&cx);
        assert!(res);
    }
}