2021-11-26 22:21:27 +00:00
|
|
|
use hyper::{Body, Client, Request, Response, StatusCode};
|
2021-06-20 21:09:27 +00:00
|
|
|
use hyper_proxy::{Intercept, Proxy, ProxyConnector};
|
2021-11-26 22:21:27 +00:00
|
|
|
use hyper_rustls::HttpsConnector;
|
|
|
|
|
use thiserror::Error;
|
2021-06-20 21:09:27 +00:00
|
|
|
use url::Url;
|
|
|
|
|
|
|
|
|
|
pub struct HttpClient {
|
|
|
|
|
proxy: Option<Url>,
|
|
|
|
|
}
|
|
|
|
|
|
2021-11-26 22:21:27 +00:00
|
|
|
#[derive(Error, Debug)]
|
|
|
|
|
pub enum HttpClientError {
|
|
|
|
|
#[error("could not parse request: {0}")]
|
|
|
|
|
Parsing(#[from] http::uri::InvalidUri),
|
|
|
|
|
#[error("could not send request: {0}")]
|
|
|
|
|
Request(hyper::Error),
|
|
|
|
|
#[error("could not read response: {0}")]
|
|
|
|
|
Response(hyper::Error),
|
|
|
|
|
#[error("could not build proxy connector: {0}")]
|
|
|
|
|
ProxyBuilder(#[from] std::io::Error),
|
|
|
|
|
}
|
|
|
|
|
|
2021-06-20 21:09:27 +00:00
|
|
|
impl HttpClient {
|
|
|
|
|
pub fn new(proxy: Option<&Url>) -> Self {
|
|
|
|
|
Self {
|
|
|
|
|
proxy: proxy.cloned(),
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2021-11-26 22:21:27 +00:00
|
|
|
pub async fn request(&self, req: Request<Body>) -> Result<Response<Body>, HttpClientError> {
|
|
|
|
|
let connector = HttpsConnector::with_native_roots();
|
|
|
|
|
let uri = req.uri().clone();
|
|
|
|
|
|
|
|
|
|
let response = if let Some(url) = &self.proxy {
|
|
|
|
|
let uri = url.to_string().parse()?;
|
2021-06-20 21:09:27 +00:00
|
|
|
let proxy = Proxy::new(Intercept::All, uri);
|
2021-11-26 22:21:27 +00:00
|
|
|
let proxy_connector = ProxyConnector::from_proxy(connector, proxy)?;
|
|
|
|
|
|
|
|
|
|
Client::builder()
|
|
|
|
|
.build(proxy_connector)
|
|
|
|
|
.request(req)
|
|
|
|
|
.await
|
|
|
|
|
.map_err(HttpClientError::Request)
|
2021-06-20 21:09:27 +00:00
|
|
|
} else {
|
2021-11-26 22:21:27 +00:00
|
|
|
Client::builder()
|
|
|
|
|
.build(connector)
|
|
|
|
|
.request(req)
|
|
|
|
|
.await
|
|
|
|
|
.map_err(HttpClientError::Request)
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if let Ok(response) = &response {
|
|
|
|
|
if response.status() != StatusCode::OK {
|
|
|
|
|
debug!("{} returned status {}", uri, response.status());
|
|
|
|
|
}
|
2021-06-20 21:09:27 +00:00
|
|
|
}
|
2021-11-26 22:21:27 +00:00
|
|
|
|
|
|
|
|
response
|
2021-06-20 21:09:27 +00:00
|
|
|
}
|
|
|
|
|
|
2021-11-26 22:21:27 +00:00
|
|
|
pub async fn request_body(&self, req: Request<Body>) -> Result<bytes::Bytes, HttpClientError> {
|
2021-06-20 21:09:27 +00:00
|
|
|
let response = self.request(req).await?;
|
2021-11-26 22:21:27 +00:00
|
|
|
hyper::body::to_bytes(response.into_body())
|
|
|
|
|
.await
|
|
|
|
|
.map_err(HttpClientError::Response)
|
2021-06-20 21:09:27 +00:00
|
|
|
}
|
|
|
|
|
}
|
