mirror of https://github.com/w4/bin.git
58 lines
1.7 KiB
Rust
58 lines
1.7 KiB
Rust
use std::ops::Deref;
|
|
|
|
use actix_web::{
|
|
dev::Payload,
|
|
http::header::{self, HeaderValue},
|
|
FromRequest, HttpMessage, HttpRequest,
|
|
};
|
|
use futures::future::ok;
|
|
|
|
/// Holds a value that determines whether or not this request wanted a plaintext response.
|
|
///
|
|
/// We assume anything with the text/plain Accept or Content-Type headers want plaintext,
|
|
/// and also anything calling us from the console or that we can't identify.
|
|
pub struct IsPlaintextRequest(pub bool);
|
|
|
|
impl Deref for IsPlaintextRequest {
|
|
type Target = bool;
|
|
|
|
fn deref(&self) -> &bool {
|
|
&self.0
|
|
}
|
|
}
|
|
|
|
impl FromRequest for IsPlaintextRequest {
|
|
type Error = actix_web::Error;
|
|
type Future = futures::future::Ready<Result<Self, Self::Error>>;
|
|
|
|
fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
|
|
if req.content_type() == "text/plain" {
|
|
return ok(IsPlaintextRequest(true));
|
|
}
|
|
|
|
match req
|
|
.headers()
|
|
.get(header::USER_AGENT)
|
|
.and_then(|u| u.to_str().unwrap().split('/').next())
|
|
{
|
|
None | Some("Wget" | "curl" | "HTTPie") => ok(IsPlaintextRequest(true)),
|
|
_ => ok(IsPlaintextRequest(false)),
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Gets the Host header from the request.
|
|
///
|
|
/// The inner value of this `HostHeader` will be `None` if there was no Host header
|
|
/// on the request.
|
|
pub struct HostHeader(pub Option<HeaderValue>);
|
|
|
|
impl FromRequest for HostHeader {
|
|
type Error = actix_web::Error;
|
|
type Future = futures::future::Ready<Result<Self, Self::Error>>;
|
|
|
|
fn from_request(req: &HttpRequest, _payload: &mut Payload) -> Self::Future {
|
|
ok(Self(req.headers().get(header::HOST).cloned()))
|
|
}
|
|
}
|