use crate::framework::{
auth,
auth::{AuthClient, Credentials},
endpoint::Endpoint,
response::{ApiErrors, ApiFailure, ApiSuccess},
response::{ApiResponse, ApiResult},
Environment, HttpApiClientConfig,
};
use std::net::SocketAddr;
pub struct Client {
environment: Environment,
credentials: auth::Credentials,
http_client: reqwest::Client,
}
impl AuthClient for reqwest::RequestBuilder {
fn auth(mut self, credentials: &Credentials) -> Self {
for (k, v) in credentials.headers() {
self = self.header(k, v);
}
self
}
}
impl Client {
pub fn new(
credentials: auth::Credentials,
config: HttpApiClientConfig,
environment: Environment,
) -> Result<Client, crate::framework::Error> {
let mut builder = reqwest::Client::builder().default_headers(config.default_headers);
#[cfg(not(target_arch = "wasm32"))]
{
if let Some(address) = config.resolve_ip {
let url = url::Url::from(&environment);
builder = builder.resolve(
url.host_str()
.expect("Environment url should have a hostname"),
SocketAddr::new(address, 443),
);
}
builder = builder.timeout(config.http_timeout);
}
let http_client = builder.build()?;
Ok(Client {
environment,
credentials,
http_client,
})
}
pub async fn request<ResultType>(
&self,
endpoint: &(dyn Endpoint<ResultType> + Send + Sync),
) -> ApiResponse<ResultType>
where
ResultType: ApiResult,
{
let mut request = self
.http_client
.request(endpoint.method(), endpoint.url(&self.environment));
if let Some(body) = endpoint.body() {
request = request.body(body);
request = request.header(
reqwest::header::CONTENT_TYPE,
endpoint.content_type().as_ref(),
);
}
request = request.auth(&self.credentials);
let response = request.send().await?;
map_api_response(response).await
}
}
async fn map_api_response<ResultType: ApiResult>(
resp: reqwest::Response,
) -> ApiResponse<ResultType> {
let status = resp.status();
if status.is_success() {
let parsed: Result<ApiSuccess<ResultType>, reqwest::Error> = resp.json().await;
match parsed {
Ok(api_resp) => Ok(api_resp),
Err(e) => Err(ApiFailure::Invalid(e)),
}
} else {
let parsed: Result<ApiErrors, reqwest::Error> = resp.json().await;
let errors = parsed.unwrap_or_default();
Err(ApiFailure::Error(status, errors))
}
}