grin/wallet/src/info.rs
Yeastplume b831192335
added wallet info/outputs commands with pretty printing (#254)
* added wallet info/outputs commands with pretty printing

* added confirmed but locked to display output
2017-11-10 14:03:31 +00:00

83 lines
2.8 KiB
Rust

// Copyright 2017 The Grin Developers
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
use checker;
use keychain::Keychain;
use core::core::amount_to_hr_string;
use types::{WalletConfig, WalletData, OutputStatus};
use prettytable;
use term;
use std::io::prelude::*;
pub fn show_info(config: &WalletConfig, keychain: &Keychain) {
let result = checker::refresh_outputs(&config, &keychain);
let _ = WalletData::read_wallet(&config.data_file_dir, |wallet_data| {
let current_height = match checker::get_tip_from_node(config) {
Ok(tip) => tip.height,
Err(_) => match wallet_data.outputs.values().map(|out| out.height).max() {
Some(height) => height,
None => 0,
},
};
let mut unspent_total=0;
let mut unspent_but_locked_total=0;
let mut unconfirmed_total=0;
let mut locked_total=0;
for out in wallet_data
.outputs
.values()
.filter(|out| out.root_key_id == keychain.root_key_id())
{
if out.status == OutputStatus::Unspent {
unspent_total+=out.value;
if out.lock_height > current_height {
unspent_but_locked_total+=out.value;
}
}
if out.status == OutputStatus::Unconfirmed && !out.is_coinbase {
unconfirmed_total+=out.value;
}
if out.status == OutputStatus::Locked {
locked_total+=out.value;
}
};
println!();
let title=format!("Wallet Summary Info - Block Height: {}", current_height);
let mut t = term::stdout().unwrap();
t.fg(term::color::MAGENTA).unwrap();
writeln!(t, "{}", title).unwrap();
writeln!(t, "--------------------------").unwrap();
t.reset().unwrap();
let mut table = table!(
[bFG->"Total", FG->amount_to_hr_string(unspent_total+unconfirmed_total)],
[bFY->"Awaiting Confirmation", FY->amount_to_hr_string(unconfirmed_total)],
[bFY->"Confirmed but Still Locked", FY->amount_to_hr_string(unspent_but_locked_total)],
[bFG->"Currently Spendable", FG->amount_to_hr_string(unspent_total-unspent_but_locked_total)],
[Fw->"---------", Fw->"---------"],
[Fr->"(Locked by previous transaction)", Fr->amount_to_hr_string(locked_total)]
);
table.set_format(*prettytable::format::consts::FORMAT_NO_BORDER_LINE_SEPARATOR);
table.printstd();
println!();
});
if let Err(_) = result {
println!("WARNING - Showing local data only - Wallet was unable to contact a node to update and verify the info shown here.");
}
}