139 lines
3.9 KiB
Rust
139 lines
3.9 KiB
Rust
use std::{sync::Arc, time::Duration};
|
|
|
|
use reqwest::redirect;
|
|
|
|
use crate::{service::globals::resolver, utils::conduwuit_version, Config, Result};
|
|
|
|
pub(crate) struct Client {
|
|
pub(crate) default: reqwest::Client,
|
|
pub(crate) url_preview: reqwest::Client,
|
|
pub(crate) well_known: reqwest::Client,
|
|
pub(crate) federation: reqwest::Client,
|
|
pub(crate) sender: reqwest::Client,
|
|
pub(crate) appservice: reqwest::Client,
|
|
pub(crate) pusher: reqwest::Client,
|
|
}
|
|
|
|
impl Client {
|
|
pub(crate) fn new(config: &Config, resolver: &Arc<resolver::Resolver>) -> Client {
|
|
Client {
|
|
default: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.clone())
|
|
.build()
|
|
.unwrap(),
|
|
|
|
url_preview: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.clone())
|
|
.redirect(redirect::Policy::limited(3))
|
|
.build()
|
|
.unwrap(),
|
|
|
|
well_known: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.hooked.clone())
|
|
.connect_timeout(Duration::from_secs(config.well_known_conn_timeout))
|
|
.read_timeout(Duration::from_secs(config.well_known_timeout))
|
|
.timeout(Duration::from_secs(config.well_known_timeout))
|
|
.pool_max_idle_per_host(0)
|
|
.redirect(redirect::Policy::limited(4))
|
|
.build()
|
|
.unwrap(),
|
|
|
|
federation: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.hooked.clone())
|
|
.read_timeout(Duration::from_secs(config.federation_timeout))
|
|
.timeout(Duration::from_secs(config.federation_timeout))
|
|
.pool_max_idle_per_host(config.federation_idle_per_host.into())
|
|
.pool_idle_timeout(Duration::from_secs(config.federation_idle_timeout))
|
|
.redirect(redirect::Policy::limited(3))
|
|
.build()
|
|
.unwrap(),
|
|
|
|
sender: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.hooked.clone())
|
|
.read_timeout(Duration::from_secs(config.sender_timeout))
|
|
.timeout(Duration::from_secs(config.sender_timeout))
|
|
.pool_max_idle_per_host(1)
|
|
.pool_idle_timeout(Duration::from_secs(config.sender_idle_timeout))
|
|
.redirect(redirect::Policy::limited(2))
|
|
.build()
|
|
.unwrap(),
|
|
|
|
appservice: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.clone())
|
|
.connect_timeout(Duration::from_secs(5))
|
|
.read_timeout(Duration::from_secs(config.appservice_timeout))
|
|
.timeout(Duration::from_secs(config.appservice_timeout))
|
|
.pool_max_idle_per_host(1)
|
|
.pool_idle_timeout(Duration::from_secs(config.appservice_idle_timeout))
|
|
.redirect(redirect::Policy::limited(2))
|
|
.build()
|
|
.unwrap(),
|
|
|
|
pusher: Self::base(config)
|
|
.unwrap()
|
|
.dns_resolver(resolver.clone())
|
|
.pool_max_idle_per_host(1)
|
|
.pool_idle_timeout(Duration::from_secs(config.pusher_idle_timeout))
|
|
.redirect(redirect::Policy::limited(2))
|
|
.build()
|
|
.unwrap(),
|
|
}
|
|
}
|
|
|
|
fn base(config: &Config) -> Result<reqwest::ClientBuilder> {
|
|
let version = conduwuit_version();
|
|
|
|
let user_agent = format!("Conduwuit/{version}");
|
|
|
|
let mut builder = reqwest::Client::builder()
|
|
.hickory_dns(true)
|
|
.connect_timeout(Duration::from_secs(config.request_conn_timeout))
|
|
.read_timeout(Duration::from_secs(config.request_timeout))
|
|
.timeout(Duration::from_secs(config.request_total_timeout))
|
|
.pool_idle_timeout(Duration::from_secs(config.request_idle_timeout))
|
|
.pool_max_idle_per_host(config.request_idle_per_host.into())
|
|
.user_agent(user_agent)
|
|
.redirect(redirect::Policy::limited(6))
|
|
.connection_verbose(true);
|
|
|
|
#[cfg(feature = "gzip_compression")]
|
|
{
|
|
builder = if config.gzip_compression {
|
|
builder.gzip(true)
|
|
} else {
|
|
builder.gzip(false).no_gzip()
|
|
};
|
|
};
|
|
|
|
#[cfg(feature = "brotli_compression")]
|
|
{
|
|
builder = if config.brotli_compression {
|
|
builder.brotli(true)
|
|
} else {
|
|
builder.brotli(false).no_brotli()
|
|
};
|
|
};
|
|
|
|
#[cfg(not(feature = "gzip_compression"))]
|
|
{
|
|
builder = builder.no_gzip();
|
|
};
|
|
|
|
#[cfg(not(feature = "brotli_compression"))]
|
|
{
|
|
builder = builder.no_brotli();
|
|
};
|
|
|
|
if let Some(proxy) = config.proxy.to_proxy()? {
|
|
Ok(builder.proxy(proxy))
|
|
} else {
|
|
Ok(builder)
|
|
}
|
|
}
|
|
}
|