mirror of https://github.com/rwf2/Rocket.git
Add a default template for every known error code.
This commit is contained in:
Sergio Benitez
parent
37e6a367b8
commit
59623d936f
|
|
@ -20,19 +20,15 @@ pub struct Catcher {
|
||||||
|
|
||||||
impl Catcher {
|
impl Catcher {
|
||||||
pub fn new(code: u16, handler: ErrorHandler) -> Catcher {
|
pub fn new(code: u16, handler: ErrorHandler) -> Catcher {
|
||||||
Catcher::new_with_default(code, handler, false)
|
Catcher { code: code, handler: handler, is_default: false }
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn handle<'b, 'r>(&self, err: Error, request: &'b Request<'r>) -> Response<'b> {
|
pub fn handle<'b, 'r>(&self, err: Error, request: &'b Request<'r>) -> Response<'b> {
|
||||||
(self.handler)(err, request)
|
(self.handler)(err, request)
|
||||||
}
|
}
|
||||||
|
|
||||||
fn new_with_default(code: u16, handler: ErrorHandler, default: bool) -> Catcher {
|
fn new_default(code: u16, handler: ErrorHandler) -> Catcher {
|
||||||
Catcher {
|
Catcher { code: code, handler: handler, is_default: true, }
|
||||||
code: code,
|
|
||||||
handler: handler,
|
|
||||||
is_default: default,
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn is_default(&self) -> bool {
|
pub fn is_default(&self) -> bool {
|
||||||
|
|
@ -52,6 +48,48 @@ impl fmt::Display for Catcher {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
macro_rules! error_page_template {
|
||||||
|
($code:expr, $name:expr, $description:expr) => (
|
||||||
|
concat!(r#"
|
||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<title>"#, $code, " ", $name, r#"</title>
|
||||||
|
</head>
|
||||||
|
<body align="center">
|
||||||
|
<div align="center">
|
||||||
|
<h1>"#, $code, " ", $name, r#"</hi>
|
||||||
|
<p>"#, $description, r#"</p>
|
||||||
|
<hr />
|
||||||
|
<small>Rocket</small>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
"#
|
||||||
|
)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
macro_rules! default_errors {
|
||||||
|
($($code:expr, $name:expr, $description:expr, $fn_name:ident),+) => (
|
||||||
|
let mut map = HashMap::new();
|
||||||
|
|
||||||
|
$(
|
||||||
|
fn $fn_name<'r>(_: Error, _r: &'r Request) -> Response<'r> {
|
||||||
|
Response::with_status(
|
||||||
|
StatusCode::Unregistered($code),
|
||||||
|
data::HTML(error_page_template!($code, $name, $description))
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
map.insert($code, Catcher::new_default($code, $fn_name));
|
||||||
|
)+
|
||||||
|
|
||||||
|
map
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
pub mod defaults {
|
pub mod defaults {
|
||||||
use super::Catcher;
|
use super::Catcher;
|
||||||
|
|
||||||
|
|
@ -63,52 +101,69 @@ pub mod defaults {
|
||||||
use error::Error;
|
use error::Error;
|
||||||
use http::hyper::StatusCode;
|
use http::hyper::StatusCode;
|
||||||
|
|
||||||
pub fn not_found<'r>(_error: Error, _request: &'r Request) -> Response<'r> {
|
|
||||||
Response::with_status(StatusCode::NotFound, data::HTML(r#"
|
|
||||||
<!DOCTYPE html>
|
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<meta charset="utf-8">
|
|
||||||
<title>404: Not Found</title>
|
|
||||||
</head>
|
|
||||||
<body>
|
|
||||||
<div align="center">
|
|
||||||
<h1>404: Not Found</h1>
|
|
||||||
<p>The page you were looking for could not be found.<p>
|
|
||||||
<hr />
|
|
||||||
<small>Rocket</small>
|
|
||||||
</div>
|
|
||||||
</body>
|
|
||||||
</html>
|
|
||||||
"#))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn internal_error<'r>(_error: Error,
|
|
||||||
_request: &'r Request)
|
|
||||||
-> Response<'r> {
|
|
||||||
Response::with_status(StatusCode::InternalServerError, data::HTML(r#"
|
|
||||||
<!DOCTYPE html>
|
|
||||||
<html>
|
|
||||||
<head>
|
|
||||||
<meta charset="utf-8">
|
|
||||||
<title>500: Internal Server Error</title>
|
|
||||||
</head>
|
|
||||||
<body align="center">
|
|
||||||
<div align="center">
|
|
||||||
<h1>500: Internal Server Error</h1>
|
|
||||||
<p>The server encountered a problem processing your request.<p>
|
|
||||||
<hr />
|
|
||||||
<small>Rocket</small>
|
|
||||||
</div>
|
|
||||||
</body>
|
|
||||||
</html>
|
|
||||||
"#))
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn get() -> HashMap<u16, Catcher> {
|
pub fn get() -> HashMap<u16, Catcher> {
|
||||||
let mut map = HashMap::new();
|
default_errors! {
|
||||||
map.insert(404, Catcher::new_with_default(404, not_found, true));
|
400, "Bad Request", "The request could not be understood by the server due
|
||||||
map.insert(500, Catcher::new_with_default(500, internal_error, true));
|
to malformed syntax.", handle_400,
|
||||||
map
|
401, "Unauthorized", "The request requires user authentication.",
|
||||||
|
handle_401,
|
||||||
|
403, "Forbidden", "The request was forbidden by the server.
|
||||||
|
Check authentication.", handle_403,
|
||||||
|
402, "Payment Required", "The request could not be processed due to lack of
|
||||||
|
payment.", handle_402,
|
||||||
|
404, "Not Found", "The requested resource could not be found.", handle_404,
|
||||||
|
405, "Method Not Allowed", "The request method is not supported for the
|
||||||
|
requested resource.", handle_405,
|
||||||
|
406, "Not Acceptable", "The requested resource is capable of generating
|
||||||
|
only content not acceptable according to the Accept headers sent in the
|
||||||
|
request.", handle_406,
|
||||||
|
407, "Proxy Authentication Required", "Authentication with the proxy is
|
||||||
|
required.", handle_407,
|
||||||
|
408, "Request Timeout", "The server timed out waiting for the
|
||||||
|
request.", handle_408,
|
||||||
|
409, "Conflict", "The request could not be processed because of a conflict
|
||||||
|
in the request.", handle_409,
|
||||||
|
410, "Gone", "The resource requested is no longer available and will not be
|
||||||
|
available again.", handle_410,
|
||||||
|
411, "Length Required", "The request did not specify the length of its
|
||||||
|
content, which is required by the requested resource.", handle_411,
|
||||||
|
412, "Precondition Failed", "The server does not meet one of the
|
||||||
|
preconditions specified in the request.", handle_412,
|
||||||
|
413, "Payload Too Large", "The request is larger than the server is
|
||||||
|
willing or able to process.", handle_413,
|
||||||
|
414, "URI Too Long", "The URI provided was too long for the server to
|
||||||
|
process.", handle_414,
|
||||||
|
415, "Unsupported Media Type", "The request entity has a media type which
|
||||||
|
the server or resource does not support.", handle_415,
|
||||||
|
416, "Range Not Satisfiable", "The portion of the requested file cannot be
|
||||||
|
supplied by the server.", handle_416,
|
||||||
|
417, "Expectation Failed", "The server cannot meet the requirements of the
|
||||||
|
Expect request-header field.", handle_417,
|
||||||
|
418, "I'm a teapot", "I was requested to brew coffee, and I am a
|
||||||
|
teapot.", handle_418,
|
||||||
|
421, "Misdirected Request", "The server cannot produce a response for this
|
||||||
|
request.", handle_421,
|
||||||
|
426, "Upgrade Required", "Switching to the protocol in the Upgrade header
|
||||||
|
field is required.", handle_426,
|
||||||
|
428, "Precondition Required", "The server requires the request to be
|
||||||
|
conditional.", handle_428,
|
||||||
|
429, "Too Many Requests", "Too many requests have been received
|
||||||
|
recently.", handle_429,
|
||||||
|
431, "Request Header Fields Too Large", "The server is unwilling to process
|
||||||
|
the request because either an individual header field, or all
|
||||||
|
the header fields collectively, are too large.", handle_431,
|
||||||
|
451, "Unavailable For Legal Reasons", "The requested resource is
|
||||||
|
unavailable due to a legal demand to deny access to this
|
||||||
|
resource.", handle_451,
|
||||||
|
500, "Internal Server Error", "The server encountered an internal error
|
||||||
|
while processing this request.", handle_500,
|
||||||
|
501, "Not Implemented", "The server either does not recognize the request
|
||||||
|
method, or it lacks the ability to fulfill the request.", handle_501,
|
||||||
|
503, "Service Unavailable", "The server is currently unavailable.",
|
||||||
|
handle_503,
|
||||||
|
510, "Not Extended", "Further extensions to the request are required for
|
||||||
|
the server to fulfill it.", handle_510
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue