Import upstream v0.16.22, stripped
Upstream commit: 474dd0229cb20cf513036619781ed97bd8073c3f Enterprise-only files removed or emptied: 63 Enterprise-only snippets removed: 117 in 50 files Dangling module declarations removed: 5 Cargo edits turning enterprise off: 14 Verification: clean Enterprise feature gates left for rebuilt features: 19 in 18 files Produced by tools/fork/strip.py. The full report is in docs/fork/strip-reports/ on main.
This commit is contained in:
@@ -0,0 +1,189 @@
|
||||
/*
|
||||
* SPDX-FileCopyrightText: 2020 Stalwart Labs LLC <[email protected]>
|
||||
*
|
||||
* SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-SEL
|
||||
*/
|
||||
|
||||
use hyper::Method;
|
||||
use reqwest::header::HeaderMap;
|
||||
use serde::{Serialize, de::DeserializeOwned};
|
||||
use std::time::Duration;
|
||||
|
||||
pub struct HttpRequest {
|
||||
pub port: u16,
|
||||
pub username: Option<String>,
|
||||
pub password: Option<String>,
|
||||
}
|
||||
|
||||
pub struct HttpResponseFull {
|
||||
pub status: reqwest::StatusCode,
|
||||
pub headers: HeaderMap,
|
||||
pub body: String,
|
||||
}
|
||||
|
||||
impl HttpResponseFull {
|
||||
pub fn header(&self, name: &str) -> Option<&str> {
|
||||
self.headers.get(name).and_then(|v| v.to_str().ok())
|
||||
}
|
||||
|
||||
pub fn rate_limit_policy(&self) -> Option<&str> {
|
||||
self.header("RateLimit-Policy")
|
||||
}
|
||||
|
||||
pub fn rate_limit(&self) -> Option<&str> {
|
||||
self.header("RateLimit")
|
||||
}
|
||||
|
||||
pub fn retry_after(&self) -> Option<u64> {
|
||||
self.header("Retry-After").and_then(|v| v.parse().ok())
|
||||
}
|
||||
}
|
||||
|
||||
impl Default for HttpRequest {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
port: 8899,
|
||||
username: None,
|
||||
password: None,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl HttpRequest {
|
||||
pub fn new() -> Self {
|
||||
Self::default()
|
||||
}
|
||||
|
||||
pub fn with_credentials(port: u16, username: &str, password: &str) -> Self {
|
||||
Self {
|
||||
port,
|
||||
username: Some(username.to_string()),
|
||||
password: Some(password.to_string()),
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn post<T: DeserializeOwned>(
|
||||
&self,
|
||||
query: &str,
|
||||
body: &impl Serialize,
|
||||
) -> Result<T, String> {
|
||||
self.request_raw(
|
||||
Method::POST,
|
||||
query,
|
||||
Some(serde_json::to_string(body).unwrap()),
|
||||
)
|
||||
.await
|
||||
.map(|result| {
|
||||
serde_json::from_str::<T>(&result).unwrap_or_else(|err| panic!("{err}: {result}"))
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn patch<T: DeserializeOwned>(
|
||||
&self,
|
||||
query: &str,
|
||||
body: &impl Serialize,
|
||||
) -> Result<T, String> {
|
||||
self.request_raw(
|
||||
Method::PATCH,
|
||||
query,
|
||||
Some(serde_json::to_string(body).unwrap()),
|
||||
)
|
||||
.await
|
||||
.map(|result| {
|
||||
serde_json::from_str::<T>(&result).unwrap_or_else(|err| panic!("{err}: {result}"))
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn delete<T: DeserializeOwned>(&self, query: &str) -> Result<T, String> {
|
||||
self.request_raw(Method::DELETE, query, None)
|
||||
.await
|
||||
.map(|result| {
|
||||
serde_json::from_str::<T>(&result).unwrap_or_else(|err| panic!("{err}: {result}"))
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn get<T: DeserializeOwned>(&self, query: &str) -> Result<T, String> {
|
||||
self.request_raw(Method::GET, query, None)
|
||||
.await
|
||||
.map(|result| {
|
||||
serde_json::from_str::<T>(&result).unwrap_or_else(|err| panic!("{err}: {result}"))
|
||||
})
|
||||
}
|
||||
pub async fn request<T: DeserializeOwned>(
|
||||
&self,
|
||||
method: Method,
|
||||
query: &str,
|
||||
) -> Result<T, String> {
|
||||
self.request_raw(method, query, None).await.map(|result| {
|
||||
serde_json::from_str::<T>(&result).unwrap_or_else(|err| panic!("{err}: {result}"))
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn send_full(
|
||||
&self,
|
||||
method: Method,
|
||||
query: &str,
|
||||
body: Option<Vec<u8>>,
|
||||
content_type: Option<&str>,
|
||||
) -> HttpResponseFull {
|
||||
let mut request = reqwest::Client::builder()
|
||||
.timeout(Duration::from_secs(5))
|
||||
.danger_accept_invalid_certs(true)
|
||||
.build()
|
||||
.unwrap()
|
||||
.request(method, format!("https://127.0.0.1:{}{query}", self.port));
|
||||
|
||||
if let Some(body) = body {
|
||||
request = request.body(body);
|
||||
}
|
||||
|
||||
if let Some(ct) = content_type {
|
||||
request = request.header(hyper::header::CONTENT_TYPE, ct);
|
||||
}
|
||||
|
||||
if let (Some(username), Some(password)) = (&self.username, &self.password) {
|
||||
request = request.basic_auth(username, Some(password));
|
||||
}
|
||||
|
||||
let response = request.send().await.expect("HTTP request failed");
|
||||
let status = response.status();
|
||||
let headers = response.headers().clone();
|
||||
let body = response.text().await.unwrap_or_default();
|
||||
HttpResponseFull {
|
||||
status,
|
||||
headers,
|
||||
body,
|
||||
}
|
||||
}
|
||||
|
||||
async fn request_raw(
|
||||
&self,
|
||||
method: Method,
|
||||
query: &str,
|
||||
body: Option<String>,
|
||||
) -> Result<String, String> {
|
||||
let mut request = reqwest::Client::builder()
|
||||
.timeout(Duration::from_millis(500))
|
||||
.danger_accept_invalid_certs(true)
|
||||
.build()
|
||||
.unwrap()
|
||||
.request(method, format!("https://127.0.0.1:{}{query}", self.port));
|
||||
|
||||
if let Some(body) = body {
|
||||
request = request.body(body);
|
||||
}
|
||||
|
||||
if let (Some(username), Some(password)) = (&self.username, &self.password) {
|
||||
request = request.basic_auth(username, Some(password));
|
||||
}
|
||||
|
||||
request
|
||||
.send()
|
||||
.await
|
||||
.map_err(|err| err.to_string())?
|
||||
.bytes()
|
||||
.await
|
||||
.map(|bytes| String::from_utf8(bytes.to_vec()).unwrap())
|
||||
.map_err(|err| err.to_string())
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user