axum-rh 0.2.8

A helper library for the axum router
Documentation 
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

use crate::router::utils::{self, session_manager::SessionObject};
use axum::{body::Body, http};

pub async fn auth_middleware<T>(
    mut session: SessionObject<T>,
    req: axum::http::Request<Body>,
    next: axum::middleware::Next,
) -> axum::http::Response<Body>
where
    T: utils::session_manager::SessionTrait,
{
    let token = match req
        .headers()
        .get(http::header::AUTHORIZATION)
        .map(|h| h.to_str().unwrap_or_default().to_string())
    {
        Some(token) => token,
        None => {
            return axum::http::Response::builder()
                .status(401)
                .body(Body::empty())
                .expect("failed to build response");
        }
    };
    if session.data.key().is_some() && utils::auth::decode_jwt(token.to_string()).is_ok() {
        return next.run(req).await;
    }

    let token = token.split(' ').collect::<Vec<&str>>();
    if token.len() != 2 || token[0] != "Bearer" {
        return axum::http::Response::builder()
            .status(401)
            .body(Body::empty())
            .expect("failed to build response");
    }
    let token = token[1];

    match utils::auth::decode_jwt(token.to_string()) {
        Ok(decoded_token) => {
            session.update_key(decoded_token.claims.user_id).await;
        }
        Err(e) => {
            log::error!("Error decoding token: {e:?}");
            return axum::http::Response::builder()
                .status(401)
                .body(Body::empty())
                .expect("failed to build response");
        }
    }

    next.run(req).await
}