Add verify subcommand

src/bin/ledger-tool.rs

@@ -2,12 +2,14 @@ extern crate clap;
 extern crate serde_json;
 extern crate solana;
 
-use clap::{App, Arg};
+use clap::{App, Arg, SubCommand};
+use solana::bank::Bank;
 use solana::ledger::read_ledger;
 use std::io::{stdout, Write};
+use std::process::exit;
 
 fn main() {
-    let matches = App::new("ledger-view")
+    let matches = App::new("ledger-tool")
         .arg(
             Arg::with_name("ledger")
                 .short("l")
@@ -17,17 +19,38 @@ fn main() {
                 .required(true)
                 .help("use DIR for ledger location"),
         )
+        .subcommand(SubCommand::with_name("print").about("Print the ledger"))
+        .subcommand(SubCommand::with_name("verify").about("Verify the ledger"))
         .get_matches();
 
     let ledger_path = matches.value_of("ledger").unwrap();
+    let entries = match read_ledger(ledger_path) {
+        Ok(entries) => entries,
+        Err(err) => {
+            println!("Failed to open ledger at {}: {}", ledger_path, err);
+            exit(1);
+        }
+    };
 
-    let entries = read_ledger(ledger_path).expect("opening ledger");
+    match matches.subcommand() {
-
+        ("print", _) => {
-    stdout().write_all(b"{\"ledger\":[\n").expect("open array");
+            stdout().write_all(b"{\"ledger\":[\n").expect("open array");
-    for entry in entries {
+            for entry in entries {
-        let entry = entry.unwrap();
+                let entry = entry.unwrap();
-        serde_json::to_writer(stdout(), &entry).expect("serialize");
+                serde_json::to_writer(stdout(), &entry).expect("serialize");
-        stdout().write_all(b",\n").expect("newline");
+                stdout().write_all(b",\n").expect("newline");
-    }
+            }
-    stdout().write_all(b"\n]}\n").expect("close array");
+            stdout().write_all(b"\n]}\n").expect("close array");
+        }
+        ("verify", _) => {
+            let entries = entries.map(|entry| entry.unwrap());
+            let bank = Bank::default();
+            let entry_height = bank.process_ledger(entries).expect("process_ledger").0;
+            println!("Ledger is valid.  Height: {}", entry_height);
+        }
+        ("", _) => {
+            println!("{}", matches.usage());
+        }
+        _ => unreachable!(),
+    };
 }