From 2fc3a8e8faf01956b9de02efa86753e907dd7182 Mon Sep 17 00:00:00 2001
From: Matias Alejo Garcia <ematiu@gmail.com>
Date: Mon, 21 Apr 2014 12:37:32 -0300
Subject: [PATCH] safe unspent

---
 index.html                     |  6 ++++--
 js/controllers/addresses.js    |  3 +--
 js/models/core/Wallet.js       | 27 ++++++++++++++++++++++-----
 js/services/controllerUtils.js | 18 +++++++++++-------
 test/test.TxProposals.js       |  3 ---
 5 files changed, 38 insertions(+), 19 deletions(-)

diff --git a/index.html b/index.html
index fb498ba3a..cb440eeab 100644
--- a/index.html
+++ b/index.html
@@ -18,8 +18,10 @@
           </figure>
           <div class="text-right" ng-show="$root.wallet">
             <h5 ng-show="$root.wallet.id">Wallet ID: {{$root.wallet.id}}</h5>
-            <p ng-show="totalBalance">{{totalBalance}} <i class="fi-bitcoin"></i></p>
-            <p ng-show="!totalBalance">0 <i class="fi-bitcoin"></i></p>
+            <p>
+                Balance: {{totalBalance || 0}} <i class="fi-bitcoin"></i><br>
+                Available to Spend:{{availableBalance || 0}} <i class="fi-bitcoin"></i>
+            </p>
           </div>
         </div>
 
diff --git a/js/controllers/addresses.js b/js/controllers/addresses.js
index c92c7ea76..af3a7ee01 100644
--- a/js/controllers/addresses.js
+++ b/js/controllers/addresses.js
@@ -10,8 +10,7 @@ angular.module('copay.addresses').controller('AddressesController',
 
     var _updateBalance = function () {
       controllerUtils.setSocketHandlers();
-
-      w.getBalance(function (balance, balanceByAddr, isMain) {
+      w.getBalance(true, function (balance, balanceByAddr, isMain) {
         if (balanceByAddr  && Object.keys(balanceByAddr).length) {
           $rootScope.balanceByAddr = balanceByAddr;
           $scope.isMain = isMain;
diff --git a/js/models/core/Wallet.js b/js/models/core/Wallet.js
index d86e3adf1..707e7f3db 100644
--- a/js/models/core/Wallet.js
+++ b/js/models/core/Wallet.js
@@ -410,7 +410,7 @@ Wallet.prototype.addressIsOwn = function(addrStr) {
   return ret;
 };
 
-Wallet.prototype.getBalance = function(cb) {
+Wallet.prototype.getBalance = function(safe, cb) {
   var balance = 0;
   var balanceByAddr = {};
   var isMain = {};
@@ -424,7 +424,8 @@ Wallet.prototype.getBalance = function(cb) {
     balanceByAddr[a]=0;
     isMain[a]=1;
   });
-  this.getUnspent(function(utxos) {
+  var f = safe ?  this.getSafeUnspent.bind(this):this.getUnspent.bind(this);
+  f(function(utxos) {
     for(var i=0;i<utxos.length; i++) {
       var u= utxos[i];
       var amt = u.amount * COIN;
@@ -444,8 +445,24 @@ Wallet.prototype.getUnspent = function(cb) {
   });
 };
 
+Wallet.prototype.getSafeUnspent = function(cb) {
+  var self = this;
+  this.blockchain.getUnspent(this.getAddressesStr(), function(unspentList) {
+
+    var ret=[];
+    var uu = self.txProposals.getUsedUnspent();
+
+    for(var i in unspentList){
+      if (uu.indexOf(unspentList[i].txid) === -1) 
+        ret.push(unspentList[i]);
+    }
+
+    return cb(ret);
+  });
+};
+
+
 Wallet.prototype.createTx = function(toAddress, amountSatStr, opts, cb) {
-console.log('[Wallet.js.447:createTx:]'); //TODO
   var self = this;
   if (typeof opts === 'function') {
     cb = opts;
@@ -456,7 +473,8 @@ console.log('[Wallet.js.447:createTx:]'); //TODO
   if (typeof opts.spendUnconfirmed === 'undefined') {
     opts.spendUnconfirmed = this.spendUnconfirmed;
   }
-  self.getUnspent(function(unspentList) {
+
+  self.getSafeUnspent(function(unspentList) {
     // TODO check enough funds, etc.
     self.createTxSync(toAddress, amountSatStr, unspentList, opts);
     self.sendPublicKeyRing();   // Change Address
@@ -480,7 +498,6 @@ Wallet.prototype.createTxSync = function(toAddress, amountSatStr, utxos, opts) {
   if (!opts.remainderOut) {
     opts.remainderOut ={ address: this.generateAddress(true).toString() };
   }
-console.log('[Wallet.js.480:opts: CREATETXSYNC]',opts); //TODO
 
   var b = new Builder(opts)
     .setUnspent(utxos)
diff --git a/js/services/controllerUtils.js b/js/services/controllerUtils.js
index 1cab6f6dd..13604c1ef 100644
--- a/js/services/controllerUtils.js
+++ b/js/services/controllerUtils.js
@@ -31,9 +31,13 @@ console.log('[controllerUtils.js.30:created:] RECV '); //TODO
       $location.path('peer');
       $rootScope.wallet = w;
       
-      $rootScope.wallet.getBalance(function(balance) {
+      $rootScope.wallet.getBalance(false,function(balance) {
         $rootScope.totalBalance = balance;
       });
+      $rootScope.wallet.getBalance(true,function(balance) {
+        $rootScope.availableBalance = balance;
+      });
+
     });
     w.on('refresh', function() {
       console.log('[controllerUtils.js] Refreshing'); //TODO
@@ -47,8 +51,9 @@ console.log('[controllerUtils.js.45] CALLING NETSTART FROM setupUxHandlers'); //
 console.log('[controllerUtils.js.45] setupUxHandlers END'); //TODO
   };
 
-  root.setSocketHandlers = function(cb) {
+  root.setSocketHandlers = function() {
     Socket.removeAllListeners();
+    var w = $rootScope.wallet;
 
     var addrs = $rootScope.wallet.getAddressesStr();
     for(var i = 0; i < addrs.length; i++) {
@@ -59,14 +64,13 @@ console.log('[controllerUtils.js.45] setupUxHandlers END'); //TODO
     addrs.forEach(function(addr) {
       Socket.on(addr, function(txid) {
         console.log('Received!', txid);
-        $rootScope.wallet.getBalance(function(balance, balanceByAddr) {
-          $rootScope.$apply(function() {
+        w.getBalance(false,function(balance, balanceByAddr) {
             $rootScope.totalBalance = balance;
             $rootScope.balanceByAddr = balanceByAddr;
+            console.log('New balance:', balance);
+            w.getBalance(true,function(balance) {
+              $rootScope.availableBalance = balance;
           });
-
-          console.log('New balance:', balance);
-          if (typeof cb === 'function') return cb();
         });
       });
     });
diff --git a/test/test.TxProposals.js b/test/test.TxProposals.js
index 1b9739cdb..f49c92319 100644
--- a/test/test.TxProposals.js
+++ b/test/test.TxProposals.js
@@ -128,9 +128,6 @@ describe('TxProposals model', function() {
     w.getUsedUnspent()[0].should.equal(unspentTest[0].txid);
   });
 
-
-
-
   it('#merge with self', function () {
     var priv = new PrivateKey(config);
     var w = new TxProposals({