use std::env;
use std::str::FromStr;
use domain::base::Rtype;
use domain::base::name::Dname;
use domain::rdata::AllRecordData;
use domain_resolv::StubResolver;

fn main() {
    let mut args = env::args().skip(1);
    let name = args.next().and_then(|arg| Dname::<Vec<_>>::from_str(&arg).ok());
    let rtype = args.next().and_then(|arg| Rtype::from_str(&arg).ok());
    let (name, rtype) = match (name, rtype) {
        (Some(name), Some(rtype)) => (name, rtype),
        _ => {
            println!("Usage: sync <domain> <record type>");
            return
        }
    };
    
    let res = StubResolver::run(move |stub| async move {
        stub.query((name, rtype)).await
    });
    let res = res.unwrap();
    let res = res.answer().unwrap().limit_to::<AllRecordData<_, _>>();
    for record in res {
        let record = record.unwrap();
        println!("{}", record);
    }
}