axum-security 0.0.2

A security toolbox for the Axum library
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

use axum::{Router, routing::MethodRouter};

use crate::oauth2::{
    OAuth2Context, OAuth2Handler,
    redirect::{OAuth2LoginService, OAuth2RedirectService},
};

/// Extension trait on [`Router`] to register OAuth 2.0 login and callback routes.
pub trait OAuth2Ext {
    /// Add the login redirect and callback routes for the given [`OAuth2Context`].
    fn with_oauth2<H: OAuth2Handler>(self, context: OAuth2Context<H>) -> Self;
}

impl<S> OAuth2Ext for Router<S>
where
    S: Clone + Send + Sync + 'static,
{
    fn with_oauth2<H: OAuth2Handler>(mut self, context: OAuth2Context<H>) -> Self {
        if let Some(start_challenge_path) = context.get_start_challenge_path() {
            let challenge_route =
                MethodRouter::new().get_service(OAuth2LoginService::new(context.clone()));

            self = self.route(start_challenge_path, challenge_route);
        }

        let route = MethodRouter::new().get_service(OAuth2RedirectService::new(context.clone()));

        self.route(context.callback_url(), route)
    }
}