#[macro_use]
extern crate error_chain;
extern crate clap;
extern crate env_logger;
extern crate geniza;

// TODO: more careful import
use geniza::*;
use clap::{App, SubCommand};

fn run() -> Result<()> {

    env_logger::init().unwrap();
 
    let matches = App::new("geniza-net")
        .version(env!("CARGO_PKG_VERSION"))
        .subcommand(SubCommand::with_name("connect")
            .about("Connects to a peer and exchanges handshake")
            .arg_from_usage("<host_port> 'peer host:port to connect to'")
            .arg_from_usage("<dat_key> 'dat key (public key) to register with'"))
        .subcommand(SubCommand::with_name("receive-all")
            .about("Connects to a peer, pulls all metadata and content")
            .arg_from_usage("<host_port> 'peer host:port to connect to'")
            .arg_from_usage("<dat_key> 'dat key (public key) to register with'"))
        .get_matches();


    match matches.subcommand() {
        ("connect", Some(subm)) => {
            let host_port = subm.value_of("host_port").unwrap();
            let dat_key = subm.value_of("dat_key").unwrap();
            if dat_key.len() != 32*2 {
                bail!("dat key not correct length");
            }
            let mut key_bytes = vec![];
            for i in 0..32 {
                let r = u8::from_str_radix(&dat_key[2*i .. 2*i+2], 16);
                match r {
                    Ok(b) => key_bytes.push(b),
                    Err(e) => bail!("Problem with hex: {}", e),
                };
            }
            DatConnection::connect(
                host_port,
                &key_bytes,
                false)?;
            println!("Done!");
        },
        ("receive-all", Some(subm)) => {
            let host_port = subm.value_of("host_port").unwrap();
            let dat_key = subm.value_of("dat_key").unwrap();
            if dat_key.len() != 32*2 {
                bail!("dat key not correct length");
            }
            let mut key_bytes = vec![];
            for i in 0..32 {
                let r = u8::from_str_radix(&dat_key[2*i .. 2*i+2], 16);
                match r {
                    Ok(b) => key_bytes.push(b),
                    Err(e) => bail!("Problem with hex: {}", e),
                };
            }
            let mut dc = DatConnection::connect(
                host_port,
                &key_bytes,
                false)?;
            dc.receive_all(false)?;
            dc.receive_all(true)?;
            println!("Done!");
        },
        _ => {
            println!("Missing or unimplemented command!");
            println!("{}", matches.usage());
            ::std::process::exit(-1);
        },
    }
    Ok(())
}

quick_main!(run);