30 lines
858 B
Rust
30 lines
858 B
Rust
use rocket::request::{self, FromRequest, Request};
|
|
use rocket::{
|
|
http::{Header, HeaderMap},
|
|
Outcome,
|
|
};
|
|
|
|
pub struct Headers<'r>(pub HeaderMap<'r>);
|
|
|
|
impl<'a, 'r> FromRequest<'a, 'r> for Headers<'r> {
|
|
type Error = ();
|
|
|
|
fn from_request(request: &'a Request<'r>) -> request::Outcome<Self, ()> {
|
|
let mut headers = HeaderMap::new();
|
|
for header in request.headers().clone().into_iter() {
|
|
headers.add(header);
|
|
}
|
|
let ori = request.uri();
|
|
let uri = if let Some(query) = ori.query() {
|
|
format!("{}?{}", ori.path(), query)
|
|
} else {
|
|
ori.path().to_owned()
|
|
};
|
|
headers.add(Header::new(
|
|
"(request-target)",
|
|
format!("{} {}", request.method().as_str().to_lowercase(), uri),
|
|
));
|
|
Outcome::Success(Headers(headers))
|
|
}
|
|
}
|