From f11285087fc4e2d03d4cd14c241820da605f754a Mon Sep 17 00:00:00 2001
From: Tanguy Pruvot <tanguy.pruvot@gmail.com>
Date: Tue, 3 Jan 2017 08:15:40 +0100
Subject: [PATCH] wallet: update db balances after manual send

---
 web/yaamp/core/backend/blocks.php             | 14 +++++++++++++-
 web/yaamp/modules/market/MarketController.php |  1 +
 web/yaamp/modules/site/SiteController.php     |  1 +
 3 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/web/yaamp/core/backend/blocks.php b/web/yaamp/core/backend/blocks.php
index 1c1dd37..0506d2b 100644
--- a/web/yaamp/core/backend/blocks.php
+++ b/web/yaamp/core/backend/blocks.php
@@ -343,7 +343,19 @@ function BackendUpdatePoolBalances($coinid = NULL)
 {
 	$t1 = microtime(true);
 
-	$sqlFilter = $coinid ? "id=".intval($coinid) : 'enable=1';
+	$sqlFilter = 'enable=1';
+
+	if ($coinid) { // used from wallet manual send
+		$sqlFilter = "id=".intval($coinid);
+		// refresh balance field from the wallet info
+		$coin = getdbo('db_coins', $coinid);
+		$remote = new WalletRPC($coin);
+		$info = $remote->getinfo();
+		if(isset($info['balance'])) {
+			$coin->balance = $info['balance'];
+			$coin->save();
+		}
+	}
 
 	$coins = getdbolist('db_coins', $sqlFilter);
 	foreach($coins as $coin)
diff --git a/web/yaamp/modules/market/MarketController.php b/web/yaamp/modules/market/MarketController.php
index 1f287fe..5b11158 100644
--- a/web/yaamp/modules/market/MarketController.php
+++ b/web/yaamp/modules/market/MarketController.php
@@ -85,6 +85,7 @@ class MarketController extends CommonController
 		} else {
 			$market->lastsent = time();
 			$market->save();
+			BackendUpdatePoolBalances($coin->id);
 		}
 
 		$exchange = new db_exchange;
diff --git a/web/yaamp/modules/site/SiteController.php b/web/yaamp/modules/site/SiteController.php
index 4fd2647..731e71c 100644
--- a/web/yaamp/modules/site/SiteController.php
+++ b/web/yaamp/modules/site/SiteController.php
@@ -252,6 +252,7 @@ class SiteController extends CommonController
 				debuglog("sent $amount {$coin->symbol} to bookmark {$bookmark->address}");
 				$bookmark->lastused = time();
 				$bookmark->save();
+				BackendUpdatePoolBalances($coin->id);
 			}
 		}