libp2p/builder/phase/
dns.rs

1use std::marker::PhantomData;
2
3use super::*;
4use crate::SwarmBuilder;
5
6pub struct DnsPhase<T> {
7    pub(crate) transport: T,
8}
9
10#[cfg(all(not(target_arch = "wasm32"), feature = "tokio", feature = "dns"))]
11impl<T: AuthenticatedMultiplexedTransport> SwarmBuilder<super::provider::Tokio, DnsPhase<T>> {
12    pub fn with_dns(
13        self,
14    ) -> Result<
15        SwarmBuilder<
16            super::provider::Tokio,
17            WebsocketPhase<impl AuthenticatedMultiplexedTransport>,
18        >,
19        std::io::Error,
20    > {
21        Ok(SwarmBuilder {
22            keypair: self.keypair,
23            phantom: PhantomData,
24            phase: WebsocketPhase {
25                transport: libp2p_dns::tokio::Transport::system(self.phase.transport)?,
26            },
27        })
28    }
29}
30
31#[cfg(all(not(target_arch = "wasm32"), feature = "tokio", feature = "dns"))]
32impl<T: AuthenticatedMultiplexedTransport> SwarmBuilder<super::provider::Tokio, DnsPhase<T>> {
33    pub fn with_dns_config(
34        self,
35        cfg: libp2p_dns::ResolverConfig,
36        opts: libp2p_dns::ResolverOpts,
37    ) -> SwarmBuilder<super::provider::Tokio, WebsocketPhase<impl AuthenticatedMultiplexedTransport>>
38    {
39        SwarmBuilder {
40            keypair: self.keypair,
41            phantom: PhantomData,
42            phase: WebsocketPhase {
43                transport: libp2p_dns::tokio::Transport::custom(self.phase.transport, cfg, opts),
44            },
45        }
46    }
47}
48
49impl<Provider, T> SwarmBuilder<Provider, DnsPhase<T>> {
50    pub(crate) fn without_dns(self) -> SwarmBuilder<Provider, WebsocketPhase<T>> {
51        SwarmBuilder {
52            keypair: self.keypair,
53            phantom: PhantomData,
54            phase: WebsocketPhase {
55                transport: self.phase.transport,
56            },
57        }
58    }
59}
60
61// Shortcuts
62impl<Provider, T: AuthenticatedMultiplexedTransport> SwarmBuilder<Provider, DnsPhase<T>> {
63    pub fn with_behaviour<B, R: TryIntoBehaviour<B>>(
64        self,
65        constructor: impl FnOnce(&libp2p_identity::Keypair) -> R,
66    ) -> Result<SwarmBuilder<Provider, SwarmPhase<T, B>>, R::Error> {
67        self.without_dns()
68            .without_websocket()
69            .without_relay()
70            .with_behaviour(constructor)
71    }
72}
libp2p/builder/phase/dns.rs

libp2p/builder/phase/
dns.rs
