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

use poem::{
    web::headers::{Authorization, HeaderMapExt},
    Request, Result,
};

use crate::{auth::BearerAuthorization, error::AuthorizationError};

/// Used to extract the token68 from the request.
pub struct Bearer {
    /// token
    pub token: String,
}

impl BearerAuthorization for Bearer {
    fn from_request(req: &Request) -> Result<Self> {
        if let Some(auth) = req
            .headers()
            .typed_get::<Authorization<poem::web::headers::authorization::Bearer>>()
        {
            return Ok(Bearer {
                token: auth.token().to_string(),
            });
        }

        Err(AuthorizationError.into())
    }
}