shield_tower/
layer.rs

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
use shield::{Shield, User};
use tower_layer::Layer;

use crate::service::ShieldService;

pub const SESSION_KEY: &str = "shield";

#[derive(Clone)]
pub struct ShieldLayer<U: User> {
    shield: Shield<U>,
    session_key: &'static str,
}

impl<U: User> ShieldLayer<U> {
    pub fn new(shield: Shield<U>) -> Self {
        Self::new_with_session_key(shield, SESSION_KEY)
    }

    pub fn new_with_session_key(shield: Shield<U>, session_key: &'static str) -> Self {
        Self {
            shield,
            session_key,
        }
    }
}

impl<S, U: User + Clone> Layer<S> for ShieldLayer<U> {
    type Service = ShieldService<S, U>;

    fn layer(&self, inner: S) -> Self::Service {
        ShieldService::new(inner, self.shield.clone(), self.session_key)
    }
}