Fix: main.rs to take keyfile and token as command line args

1 files changed, 9 insertions, 10 deletions

diff --git a/src/main.rs b/src/main.rs
index fcae3b9..c6363cc 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -12,28 +12,27 @@ use jwt::{decode, Validation, Algorithm};
 #[derive(Debug, Serialize, Deserialize)]
 struct Claims {
     aud: String,
-    exp: f64,
+    exp: u64,
     iss: String,
     sub: String,
     scope: String,
 }
 
 fn main() {
-    println!("reading key");
-    let mut f = File::open("foo.der").unwrap();
+    let args: Vec<String> = env::args().collect();
+
+    let keyfile = &args[1];
+    println!("reading key from {}", keyfile);
+    let mut f = File::open(keyfile).unwrap();
 
     let mut key = Vec::new();
     f.read_to_end(&mut key).unwrap();
     println!("key len {}", key.len());
 
     println!("finding tokens on command line");
-    let mut tokens = env::args();
-    tokens.next();
-
-    for token in tokens {
-        println!("token {}", token);
-        decode_token(&token, &key);
-    }
+    let token = &args[2];
+    println!("token {}", token);
+    decode_token(&token, &key);
 }