use super::SetOpt;
use curl::easy::Easy2;
use std::{net::IpAddr, time::Duration};
#[derive(Clone, Debug)]
pub enum DnsCache {
Disable,
Timeout(Duration),
Forever,
}
impl Default for DnsCache {
fn default() -> Self {
Duration::from_secs(60).into()
}
}
impl From<Duration> for DnsCache {
fn from(duration: Duration) -> Self {
DnsCache::Timeout(duration)
}
}
impl SetOpt for DnsCache {
#[allow(unsafe_code)]
fn set_opt<H>(&self, easy: &mut Easy2<H>) -> Result<(), curl::Error> {
let value = match self {
DnsCache::Disable => 0,
DnsCache::Timeout(duration) => duration.as_secs() as i64,
DnsCache::Forever => -1,
};
unsafe {
match curl_sys::curl_easy_setopt(easy.raw(), curl_sys::CURLOPT_DNS_CACHE_TIMEOUT, value)
{
curl_sys::CURLE_OK => Ok(()),
code => Err(curl::Error::new(code)),
}
}
}
}
#[derive(Clone, Debug, Default)]
pub struct ResolveMap(Vec<String>);
impl ResolveMap {
pub const fn new() -> Self {
ResolveMap(Vec::new())
}
#[must_use = "builders have no effect if unused"]
pub fn add<H, A>(mut self, host: H, port: u16, addr: A) -> Self
where
H: AsRef<str>,
A: Into<IpAddr>,
{
self.0
.push(format!("{}:{}:{}", host.as_ref(), port, addr.into()));
self
}
}
impl SetOpt for ResolveMap {
fn set_opt<H>(&self, easy: &mut curl::easy::Easy2<H>) -> Result<(), curl::Error> {
let mut list = curl::easy::List::new();
for entry in self.0.iter() {
list.append(entry)?;
}
easy.resolve(list)
}
}