use rocket::handler::{Handler, Outcome};
use rocket::http::Method;
use rocket::response::Redirect;
use rocket::{Data, Request, Route};
#[derive(Clone)]
pub struct RedirectHandler {
dest: &'static str,
}
impl RedirectHandler {
pub fn to(dest: &'static str) -> Self {
Self {
dest: dest.trim_start_matches('/'),
}
}
pub fn into_route(self, path: impl AsRef<str>) -> Route {
Route::new(Method::Get, path, self)
}
}
impl Handler for RedirectHandler {
fn handle<'r>(&self, req: &'r Request, _: Data) -> Outcome<'r> {
let path = req.route().unwrap().base().trim_end_matches('/');
Outcome::from(req, Redirect::to(format!("{}/{}", path, self.dest)))
}
}