// 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.");
	}
}