use std::marker::PhantomData;
use std::net::SocketAddr;
use std::sync::Arc;
use rustls::ClientConfig;
use trust_dns_proto::https::{HttpsClientConnect, HttpsClientStream, HttpsClientStreamBuilder};
use trust_dns_proto::tcp::Connect;
use crate::client::{ClientConnection, Signer};
#[derive(Clone)]
pub struct HttpsClientConnection<T> {
name_server: SocketAddr,
bind_addr: Option<SocketAddr>,
dns_name: String,
client_config: Arc<ClientConfig>,
marker: PhantomData<T>,
}
impl<T> HttpsClientConnection<T> {
#[allow(clippy::new_ret_no_self)]
pub fn new(
name_server: SocketAddr,
dns_name: String,
client_config: Arc<ClientConfig>,
) -> Self {
Self::new_with_bind_addr(name_server, None, dns_name, client_config)
}
#[allow(clippy::new_ret_no_self)]
pub fn new_with_bind_addr(
name_server: SocketAddr,
bind_addr: Option<SocketAddr>,
dns_name: String,
client_config: Arc<ClientConfig>,
) -> Self {
Self {
name_server,
bind_addr,
dns_name,
client_config,
marker: PhantomData,
}
}
}
impl<T> ClientConnection for HttpsClientConnection<T>
where
T: Connect,
{
type Sender = HttpsClientStream;
type SenderFuture = HttpsClientConnect<T>;
fn new_stream(
&self,
_signer: Option<Arc<Signer>>,
) -> Self::SenderFuture {
let mut https_builder =
HttpsClientStreamBuilder::with_client_config(Arc::clone(&self.client_config));
if let Some(bind_addr) = self.bind_addr {
https_builder.bind_addr(bind_addr);
}
https_builder.build(self.name_server, self.dns_name.clone())
}
}