use std::default::Default;
use std::env;
use std::time::Duration;
use crate::http::headers::Headers;
const DEFAULT_TIMEOUT: u32 = 60;
#[derive(Clone, Debug)]
pub struct SquareHttpClientConfig {
pub timeout: u32,
pub user_agent: String,
pub default_headers: Headers,
pub retry_configuration: RetryConfig,
}
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct RetryConfig {
pub retries_count: u32,
pub min_retry_interval: Duration,
pub max_retry_interval: Duration,
}
impl Default for RetryConfig {
fn default() -> Self {
Self {
retries_count: 0,
min_retry_interval: Duration::from_secs(1),
max_retry_interval: Duration::from_secs(30 * 60),
}
}
}
#[derive(Default)]
pub struct SquareHttpClientConfigBuilder {
timeout: Option<u32>,
user_agent: Option<String>,
default_headers: Option<Headers>,
retry_configuration: Option<RetryConfig>,
}
impl SquareHttpClientConfigBuilder {
pub fn timeout(&mut self, timeout: u32) -> &mut Self {
self.timeout = Some(timeout);
self
}
pub fn user_agent(&mut self, user_agent: String) -> &mut Self {
self.user_agent = Some(user_agent);
self
}
pub fn default_headers(&mut self, default_headers: Headers) -> &mut Self {
self.default_headers = Some(default_headers);
self
}
pub fn retry_configuration(&mut self, retry_configuration: RetryConfig) -> &mut Self {
self.retry_configuration = Some(retry_configuration);
self
}
pub fn build(&self) -> SquareHttpClientConfig {
SquareHttpClientConfig {
timeout: self.timeout.unwrap_or(DEFAULT_TIMEOUT),
user_agent: self.user_agent.clone().unwrap_or_else(SquareHttpClientConfig::default_user_agent),
default_headers: self.default_headers.clone().unwrap_or_default(),
retry_configuration: self.retry_configuration.clone().unwrap_or_default(),
}
}
}
impl SquareHttpClientConfig {
pub fn builder() -> SquareHttpClientConfigBuilder {
SquareHttpClientConfigBuilder::default()
}
pub(crate) fn default_user_agent() -> String {
let sdk_version = env!("CARGO_PKG_VERSION");
let engine = "Rust";
let rust_version = rustc_version_runtime::version();
let os = env::consts::OS;
format!(
"Rust Square API Client Lib/({}) {}/{} ({})",
sdk_version, engine, rust_version, os
)
}
}
impl Default for SquareHttpClientConfig {
fn default() -> Self {
Self {
timeout: DEFAULT_TIMEOUT,
user_agent: Self::default_user_agent(),
default_headers: Default::default(),
retry_configuration: Default::default(),
}
}
}
#[cfg(test)]
mod tests {
use crate::http::client::config::{SquareHttpClientConfig, RetryConfig};
use crate::http::headers::Headers;
#[test]
fn http_client_configuration_new_with_default_headers() {
let http_client_configuration =
SquareHttpClientConfig::builder()
.timeout(15)
.user_agent(String::from("some_user_agent"))
.default_headers(Headers::default())
.build();
assert_eq!(15, http_client_configuration.timeout);
assert_eq!(String::from("some_user_agent"), http_client_configuration.user_agent);
assert_eq!(Headers::default(), http_client_configuration.default_headers);
assert_eq!(RetryConfig::default(), http_client_configuration.retry_configuration);
}
#[test]
fn http_client_configuration_new_with_different_user_agent_in_headers() {
let http_client_configuration =
SquareHttpClientConfig::builder()
.timeout(15)
.user_agent(String::from("some_user_agent"))
.retry_configuration(RetryConfig::default())
.build();
assert_eq!(15, http_client_configuration.timeout);
assert_eq!(String::from("some_user_agent"), http_client_configuration.user_agent);
assert_eq!(Headers::default(), http_client_configuration.default_headers);
assert_eq!(RetryConfig::default(), http_client_configuration.retry_configuration);
}
}