continuwuity/src/service/globals/client.rs

use std::{sync::Arc, time::Duration};

use reqwest::redirect;

use crate::{service::globals::resolver, Config, Result};

pub struct Client {
	pub default: reqwest::Client,
	pub url_preview: reqwest::Client,
	pub well_known: reqwest::Client,
	pub federation: reqwest::Client,
	pub sender: reqwest::Client,
	pub appservice: reqwest::Client,
	pub pusher: reqwest::Client,
}

impl Client {
	pub 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))
				.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())
				.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())
				.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))
				.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 = match option_env!("CONDUIT_VERSION_EXTRA") {
			Some(extra) => format!("{} ({})", env!("CARGO_PKG_VERSION"), extra),
			None => env!("CARGO_PKG_VERSION").to_owned(),
		};

		let mut builder = reqwest::Client::builder()
			.hickory_dns(true)
			.timeout(Duration::from_secs(config.request_timeout))
			.connect_timeout(Duration::from_secs(config.request_conn_timeout))
			.pool_max_idle_per_host(config.request_idle_per_host.into())
			.pool_idle_timeout(Duration::from_secs(config.request_idle_timeout))
			.user_agent("Conduwuit".to_owned() + "/" + &version)
			.redirect(redirect::Policy::limited(6));

		#[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)
		}
	}
}