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

use re_log::error;

use tonic::{metadata::errors::InvalidMetadataValue, service::Interceptor, Request, Status};

use crate::Jwt;

use super::{AUTHORIZATION_KEY, TOKEN_PREFIX};

#[derive(Default)]
pub struct AuthDecorator {
    jwt: Option<Jwt>,
}

impl AuthDecorator {
    pub fn new(jwt: Option<Jwt>) -> Self {
        Self { jwt }
    }
}

impl Interceptor for AuthDecorator {
    fn call(&mut self, req: Request<()>) -> Result<Request<()>, Status> {
        if let Some(jwt) = self.jwt.as_ref() {
            let token = format!("{TOKEN_PREFIX}{}", jwt.0).parse().map_err(
                |err: InvalidMetadataValue| {
                    error!("malformed token: {}", err.to_string());
                    Status::invalid_argument("malformed token")
                },
            )?;

            let mut req = req;
            req.metadata_mut().insert(AUTHORIZATION_KEY, token);

            Ok(req)
        } else {
            Ok(req)
        }
    }
}