From 66b1ee42a5318c09b005f253dfddef371e6955f4 Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Tue, 5 Dec 2017 23:18:41 -0800
Subject: geniza-net cleanups

---
 src/bin/geniza-net.rs | 27 ++++-----------------------
 1 file changed, 4 insertions(+), 23 deletions(-)

(limited to 'src/bin')

diff --git a/src/bin/geniza-net.rs b/src/bin/geniza-net.rs
index e6fa448..11bcee8 100644
--- a/src/bin/geniza-net.rs
+++ b/src/bin/geniza-net.rs
@@ -40,8 +40,8 @@ fn run() -> Result<()> {
                 .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")
+            SubCommand::with_name("receive-some")
+                .about("Connects to a peer, pulls some 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'"),
         )
@@ -80,26 +80,13 @@ fn run() -> Result<()> {
         ("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),
-                };
-            }
+            let key_bytes = parse_dat_key(&dat_key)?;
             DatConnection::connect(host_port, &key_bytes, false)?;
             println!("Done!");
         }
-        ("receive-all", Some(subm)) => {
+        ("receive-some", 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 key_bytes = parse_dat_key(&dat_key)?;
             let mut dc = DatConnection::connect(host_port, &key_bytes, false)?;
             // XXX: number here totally arbitrary
@@ -109,9 +96,6 @@ fn run() -> Result<()> {
         }
         ("discovery-key", Some(subm)) => {
             let dat_key = subm.value_of("dat_key").unwrap();
-            if dat_key.len() != 32 * 2 {
-                bail!("dat key not correct length");
-            }
             let key_bytes = parse_dat_key(&dat_key)?;
             let disc_key = make_discovery_key(&key_bytes);
             for b in disc_key {
@@ -121,9 +105,6 @@ fn run() -> Result<()> {
         }
         ("discovery-dns-name", Some(subm)) => {
             let dat_key = subm.value_of("dat_key").unwrap();
-            if dat_key.len() != 32 * 2 {
-                bail!("dat key not correct length");
-            }
             let key_bytes = parse_dat_key(&dat_key)?;
             let disc_key = make_discovery_key(&key_bytes);
             for b in 0..20 {
-- 
cgit v1.2.3