mirror of https://github.com/rwf2/Rocket.git
38 lines
1.0 KiB
Rust
38 lines
1.0 KiB
Rust
use response::{Outcome, Responder};
|
|
use http::hyper::{header, FreshHyperResponse, StatusCode};
|
|
|
|
#[derive(Debug)]
|
|
pub struct Redirect(StatusCode, String);
|
|
|
|
impl Redirect {
|
|
pub fn to(uri: &str) -> Redirect {
|
|
Redirect(StatusCode::Found, String::from(uri))
|
|
}
|
|
|
|
pub fn created(uri: &str) -> Redirect {
|
|
Redirect(StatusCode::Created, String::from(uri))
|
|
}
|
|
|
|
pub fn other(uri: &str) -> Redirect {
|
|
Redirect(StatusCode::SeeOther, String::from(uri))
|
|
}
|
|
|
|
pub fn permanent(uri: &str) -> Redirect {
|
|
Redirect(StatusCode::PermanentRedirect, String::from(uri))
|
|
}
|
|
|
|
pub fn temporary(uri: &str) -> Redirect {
|
|
Redirect(StatusCode::TemporaryRedirect, String::from(uri))
|
|
}
|
|
}
|
|
|
|
impl<'a> Responder for Redirect {
|
|
fn respond<'b>(&mut self, mut res: FreshHyperResponse<'b>) -> Outcome<'b> {
|
|
res.headers_mut().set(header::ContentLength(0));
|
|
res.headers_mut().set(header::Location(self.1.clone()));
|
|
*(res.status_mut()) = self.0;
|
|
res.send(b"").unwrap();
|
|
Outcome::Complete
|
|
}
|
|
}
|