forked from Chewing_Bever/rusty-bever
38 lines
1.3 KiB
Rust
38 lines
1.3 KiB
Rust
use rocket::request::Request;
|
|
use rocket::response::{self, Response, Responder};
|
|
use rocket::http::Status;
|
|
use std::io;
|
|
|
|
pub enum RBError {
|
|
/// When the login requests an unknown user
|
|
UnknownUser,
|
|
/// Invalid login password.
|
|
InvalidPassword,
|
|
/// When a non-admin user tries to use an admin endpoint
|
|
Unauthorized,
|
|
/// When an expired JWT token is used for auth.
|
|
JWTTokenExpired,
|
|
/// Umbrella error for when something goes wrong whilst creating a JWT token pair
|
|
JWTCreationError
|
|
}
|
|
|
|
impl<'r> Responder<'r, 'static> for RBError {
|
|
fn respond_to(self, _: &'r Request<'_>) -> response::Result<'static> {
|
|
let (status, message): (Status, &str) = match self {
|
|
RBError::UnknownUser => (Status::NotFound, "Unknown user"),
|
|
RBError::InvalidPassword => (Status::Unauthorized, "Invalid password"),
|
|
RBError::Unauthorized => (Status::Unauthorized, "Unauthorized"),
|
|
RBError::JWTTokenExpired => (Status::Unauthorized, "Token expired"),
|
|
RBError::JWTCreationError => (Status::InternalServerError, "Failed to create tokens."),
|
|
};
|
|
|
|
let mut res = Response::new();
|
|
res.set_status(status);
|
|
res.set_sized_body(message.len(), io::Cursor::new(message));
|
|
|
|
Ok(res)
|
|
}
|
|
}
|
|
|
|
pub type Result<T> = std::result::Result<T, RBError>;
|