Plume/src/routes/errors.rs

use rocket_contrib::Template;
use rocket::Request;
use rocket::request::FromRequest;
use plume_models::db_conn::DbConn;
use plume_models::users::User;

#[catch(404)]
fn not_found(req: &Request) -> Template {
    let conn = req.guard::<DbConn>().succeeded();
    let user = User::from_request(req).succeeded();
    Template::render("errors/404", json!({
        "error_message": "Page not found",
        "account": user.and_then(|u| conn.map(|conn| u.to_json(&*conn)))
    }))
}

#[catch(500)]
fn server_error(req: &Request) -> Template {
    let conn = req.guard::<DbConn>().succeeded();
    let user = User::from_request(req).succeeded();
    Template::render("errors/500", json!({
        "error_message": "Server error",
        "account": user.and_then(|u| conn.map(|conn| u.to_json(&*conn)))
    }))
}

#[derive(FromForm)]
pub struct Uri {
    target: String,
}

#[post("/csrf-violation?<uri>")]
fn csrf_violation(uri: Option<Uri>) -> Template {
    if let Some(uri) = uri {
        eprintln!("Csrf violation while acceding \"{}\"", uri.target)
    }
    Template::render("errors/csrf", json!({
        "error_message":""
    }))
}