Chewing_Bever/rusty-bever
Archived
1
0
Fork 1
Code Issues 41 Pull requests 1 Projects Releases Activity

Separated JWT header into own guard

Browse source
This commit is contained in:
Jef Roosens 2021-08-21 22:21:42 +02:00
parent 35fb38de9e
commit dab90bc4a9
Signed by: Jef Roosens
GPG key ID: B580B976584B5F30
4 changed files with 30 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

19
src/rbs/guards.rs
View file

@ -8,10 +8,10 @@ use jwt::VerifyWithKey;
use rb::auth::Claims;
use sha2::Sha256;
pub struct User(Claims);
pub struct JWT(String);
#[rocket::async_trait]
impl<'r> FromRequest<'r> for User {
impl<'r> FromRequest<'r> for JWT {
type Error = rb::errors::RBError;
async fn from_request(req: &'r Request<'_>) -> Outcome<Self, Self::Error> {
@ -28,9 +28,22 @@ impl<'r> FromRequest<'r> for User {
// Extract the jwt token from the header
let jwt_token = match header.get(7..) {
Some(token) => token,
None => return Outcome::Failure((Status::Unauthorized, Self::Error::JWTError)),
None => return Outcome::Forward(()),
};
Outcome::Success(Self(jwt_token.to_string()))
}
}
pub struct User(Claims);
#[rocket::async_trait]
impl<'r> FromRequest<'r> for User {
type Error = rb::errors::RBError;
async fn from_request(req: &'r Request<'_>) -> Outcome<Self, Self::Error> {
let jwt_token = try_outcome!(req.guard::<JWT>().await).0;
// Get secret & key
let secret = match std::env::var("JWT_KEY") {
Ok(key) => key,