use std::env;
use std::net::SocketAddr;
use mhost::nameserver::NameServerConfig;
use mhost::resolver::{MultiQuery, ResolverGroupBuilder, UniQuery};
use mhost::RecordType;
#[tokio::main]
async fn main() {
let name = env::args().nth(1).unwrap_or_else(|| "www.example.com".to_string());
let sock_addr: SocketAddr = "8.8.8.8:53".parse().unwrap();
let resolvers = ResolverGroupBuilder::new()
.nameserver(NameServerConfig::udp(sock_addr))
.build()
.await
.expect("Failed to create resolver");
let query = UniQuery::new(name, RecordType::A).expect("Failed to create query");
let one_lookup = resolvers.lookup(query).await.expect("failed to execute lookups");
println!("Lookup result: #{} {:?}", one_lookup.len(), &one_lookup);
let mq = MultiQuery::multi_record(
"www.example.com",
vec![RecordType::A, RecordType::AAAA, RecordType::TXT],
)
.expect("Failed to create multi-query");
let lookups = resolvers.lookup(mq).await.expect("failed to execute lookups");
let successes = lookups.iter().filter(|x| x.result().is_response()).count();
println!("Multi-Lookup successful results: {}/{}", successes, lookups.len());
let failures: Vec<_> = lookups.iter().filter(|x| !x.result().is_response()).collect();
println!("Multi-Lookup failed results: {:#?}", failures);
}