56 lines
1.6 KiB
Rust
56 lines
1.6 KiB
Rust
use crate::users::User;
|
|
use rocket::{
|
|
http::Status,
|
|
request::{self, FromRequest, Request},
|
|
Outcome,
|
|
};
|
|
|
|
/// Wrapper around User to use as a request guard on pages exclusively reserved to admins.
|
|
pub struct Admin(pub User);
|
|
|
|
impl<'a, 'r> FromRequest<'a, 'r> for Admin {
|
|
type Error = ();
|
|
|
|
fn from_request(request: &'a Request<'r>) -> request::Outcome<Admin, ()> {
|
|
let user = request.guard::<User>()?;
|
|
if user.is_admin() {
|
|
Outcome::Success(Admin(user))
|
|
} else {
|
|
Outcome::Failure((Status::Unauthorized, ()))
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Same as `Admin` but it forwards to next guard if the user is not an admin.
|
|
/// It's useful when there are multiple implementations of routes for admin and moderator.
|
|
pub struct InclusiveAdmin(pub User);
|
|
|
|
impl<'a, 'r> FromRequest<'a, 'r> for InclusiveAdmin {
|
|
type Error = ();
|
|
|
|
fn from_request(request: &'a Request<'r>) -> request::Outcome<InclusiveAdmin, ()> {
|
|
let user = request.guard::<User>()?;
|
|
if user.is_admin() {
|
|
Outcome::Success(InclusiveAdmin(user))
|
|
} else {
|
|
Outcome::Forward(())
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Same as `Admin` but for moderators.
|
|
pub struct Moderator(pub User);
|
|
|
|
impl<'a, 'r> FromRequest<'a, 'r> for Moderator {
|
|
type Error = ();
|
|
|
|
fn from_request(request: &'a Request<'r>) -> request::Outcome<Moderator, ()> {
|
|
let user = request.guard::<User>()?;
|
|
if user.is_moderator() {
|
|
Outcome::Success(Moderator(user))
|
|
} else {
|
|
Outcome::Failure((Status::Unauthorized, ()))
|
|
}
|
|
}
|
|
}
|