forked from Plume/Plume
96 lines
2.5 KiB
Rust
96 lines
2.5 KiB
Rust
use plume_models::users::User;
|
|
use plume_models::{db_conn::DbConn, Error};
|
|
use rocket::{
|
|
request::FromRequest,
|
|
response::{self, Responder},
|
|
Request,
|
|
};
|
|
use rocket_i18n::I18n;
|
|
use template_utils::Ructe;
|
|
|
|
#[derive(Debug)]
|
|
pub struct ErrorPage(Error);
|
|
|
|
impl From<Error> for ErrorPage {
|
|
fn from(err: Error) -> ErrorPage {
|
|
ErrorPage(err)
|
|
}
|
|
}
|
|
|
|
impl<'r> Responder<'r> for ErrorPage {
|
|
fn respond_to(self, req: &Request) -> response::Result<'r> {
|
|
let conn = req.guard::<DbConn>().succeeded();
|
|
let intl = req.guard::<I18n>().succeeded();
|
|
let user = User::from_request(req).succeeded();
|
|
|
|
match self.0 {
|
|
Error::NotFound => render!(errors::not_found(&(
|
|
&*conn.unwrap(),
|
|
&intl.unwrap().catalog,
|
|
user
|
|
)))
|
|
.respond_to(req),
|
|
Error::Unauthorized => render!(errors::not_found(&(
|
|
&*conn.unwrap(),
|
|
&intl.unwrap().catalog,
|
|
user
|
|
)))
|
|
.respond_to(req),
|
|
_ => render!(errors::not_found(&(
|
|
&*conn.unwrap(),
|
|
&intl.unwrap().catalog,
|
|
user
|
|
)))
|
|
.respond_to(req),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[catch(404)]
|
|
pub fn not_found(req: &Request) -> Ructe {
|
|
let conn = req.guard::<DbConn>().succeeded();
|
|
let intl = req.guard::<I18n>().succeeded();
|
|
let user = User::from_request(req).succeeded();
|
|
render!(errors::not_found(&(
|
|
&*conn.unwrap(),
|
|
&intl.unwrap().catalog,
|
|
user
|
|
)))
|
|
}
|
|
|
|
#[catch(422)]
|
|
pub fn unprocessable_entity(req: &Request) -> Ructe {
|
|
let conn = req.guard::<DbConn>().succeeded();
|
|
let intl = req.guard::<I18n>().succeeded();
|
|
let user = User::from_request(req).succeeded();
|
|
render!(errors::unprocessable_entity(&(
|
|
&*conn.unwrap(),
|
|
&intl.unwrap().catalog,
|
|
user
|
|
)))
|
|
}
|
|
|
|
#[catch(500)]
|
|
pub fn server_error(req: &Request) -> Ructe {
|
|
let conn = req.guard::<DbConn>().succeeded();
|
|
let intl = req.guard::<I18n>().succeeded();
|
|
let user = User::from_request(req).succeeded();
|
|
render!(errors::server_error(&(
|
|
&*conn.unwrap(),
|
|
&intl.unwrap().catalog,
|
|
user
|
|
)))
|
|
}
|
|
|
|
#[post("/csrf-violation?<target>")]
|
|
pub fn csrf_violation(
|
|
target: Option<String>,
|
|
conn: DbConn,
|
|
intl: I18n,
|
|
user: Option<User>,
|
|
) -> Ructe {
|
|
if let Some(uri) = target {
|
|
eprintln!("Csrf violation while acceding \"{}\"", uri)
|
|
}
|
|
render!(errors::csrf(&(&*conn, &intl.catalog, user)))
|
|
}
|