From 945e3f16a738b6e7904c647d63593f3b171366d6 Mon Sep 17 00:00:00 2001
From: SergiySW <sergiysw@gmail.com>
Date: Wed, 21 Jun 2017 17:54:10 +0300
Subject: [PATCH] Balance in account viewer

---
 rai/qt/qt.cpp | 16 ++++++++++++++++
 rai/qt/qt.hpp |  3 +++
 2 files changed, 19 insertions(+)

diff --git a/rai/qt/qt.cpp b/rai/qt/qt.cpp
index 2fc99a2a..e41e7132 100644
--- a/rai/qt/qt.cpp
+++ b/rai/qt/qt.cpp
@@ -578,6 +578,9 @@ layout (new QVBoxLayout),
 account_label (new QLabel ("Account:")),
 account_line (new QLineEdit),
 refresh (new QPushButton ("Refresh")),
+balance_window (new QWidget),
+balance_layout (new QHBoxLayout),
+balance_label (new QLabel),
 history (wallet_a.wallet_m->node.ledger, account, wallet_a.rendering_ratio),
 back (new QPushButton ("Back")),
 account (wallet_a.account),
@@ -586,6 +589,11 @@ wallet (wallet_a)
 	layout->addWidget (account_label);
 	layout->addWidget (account_line);
 	layout->addWidget (refresh);
+	balance_layout->addWidget (balance_label);
+	balance_layout->addStretch ();
+	balance_layout->setContentsMargins (0, 0, 0, 0);
+	balance_window->setLayout (balance_layout);
+	layout->addWidget (balance_window);
 	layout->addWidget (history.window);
 	layout->addWidget (back);
 	window->setLayout (layout);
@@ -600,10 +608,18 @@ wallet (wallet_a)
 		{
 			show_line_ok (*account_line);
 			this->history.refresh ();
+			auto balance (wallet.node.balance_pending (account));
+			auto final_text (std::string ("Balance (XRB): ") + (balance.first / wallet.rendering_ratio).convert_to <std::string> ());
+			if (!balance.second.is_zero ())
+			{
+				final_text += "\nPending: " + (balance.second / wallet.rendering_ratio).convert_to <std::string> ();
+			}
+			balance_label->setText (QString (final_text.c_str ()));
 		}
 		else
 		{
 			show_line_error (*account_line);
+			balance_label->clear ();
 		}
 	});
 }
diff --git a/rai/qt/qt.hpp b/rai/qt/qt.hpp
index c3a3164d..8b58c0ea 100644
--- a/rai/qt/qt.hpp
+++ b/rai/qt/qt.hpp
@@ -248,6 +248,9 @@ namespace rai_qt {
 		QLabel * account_label;
 		QLineEdit * account_line;
 		QPushButton * refresh;
+		QWidget * balance_window;
+		QHBoxLayout * balance_layout;
+		QLabel * balance_label;
 		rai_qt::history history;
 		QPushButton * back;
 		rai::account account;