Something went wrong. Try again.
A lexicon-driven AppView for ATProto.
Something went wrong. Try again.
Rust
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758use atrium_identity::handle::DnsTxtResolver;use hickory_resolver::TokioResolver;use hickory_resolver::proto::rr::RData;
/// Native DNS TXT resolver using system DNS configuration./// Implements atrium's `DnsTxtResolver` trait for use with `AtprotoHandleResolver`,/// and provides a `lookup_txt` method for direct TXT record queries.#[derive(Clone)]pub struct NativeDnsResolver { resolver: TokioResolver,}
impl Default for NativeDnsResolver { fn default() -> Self { Self::new() }}
impl NativeDnsResolver { pub fn new() -> Self { Self { resolver: TokioResolver::builder_tokio() .expect("Failed to read system DNS config") .build() .expect("Failed to build DNS resolver"), } }
/// Look up TXT records for a given name, returning all record values. pub async fn lookup_txt(&self, name: &str) -> Result<Vec<String>, String> { let response = self .resolver .txt_lookup(name) .await .map_err(|e| format!("DNS TXT lookup failed for {name}: {e}"))?;
Ok(response .answers() .iter() .filter_map(|r| match &r.data { RData::TXT(txt) => Some(txt.to_string()), _ => None, }) .collect()) }}
impl DnsTxtResolver for NativeDnsResolver { async fn resolve( &self, query: &str, ) -> core::result::Result<Vec<String>, Box<dyn std::error::Error + Send + Sync + 'static>> { self.lookup_txt(query).await.map_err(|e| { Box::new(std::io::Error::other(e)) as Box<dyn std::error::Error + Send + Sync> }) }}