From 61dfec7a0d7abe96aca189fd0e520b2745523377 Mon Sep 17 00:00:00 2001
From: Gustavo Cortez <cmgustavo83@gmail.com>
Date: Wed, 21 May 2014 18:53:17 -0300
Subject: [PATCH] Fixes: * Reload address list when click on refresh * Angular
 error when do not have any address and click on refresh

---
 index.html                     | 10 +++++-----
 js/controllers/addresses.js    |  9 ++++++---
 js/controllers/header.js       |  8 +++++---
 js/services/controllerUtils.js |  4 +---
 4 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/index.html b/index.html
index e62cab0c0..0437a522e 100644
--- a/index.html
+++ b/index.html
@@ -331,16 +331,16 @@
                   <span>{{addr.address}}</span>
                   <span ng-if="addr.isChange">(change)</span>
                   <span class="right">
-                    <span ng-if="$root.receivedFund[1] == addr.address">
+                    <span ng-if="addrWithFund == addr.address">
                       <span ng-if="$root.updatingBalance">
                         <i class="fi-bitcoin-circle icon-rotate spinner"></i>
                       </span>
                       <span ng-if="!$root.updatingBalance">
-                        {{$root.balanceByAddr[addr.address]}}
+                        {{$root.balanceByAddr[addr.address] || 0}}
                         <i class="fi-bitcoin"></i>
                       </span>
                     </span>
-                    <span ng-if="$root.receivedFund[1] != addr.address">
+                    <span ng-if="addrWithFund != addr.address">
                       {{addr.balance || 0}}
                       <i class="fi-bitcoin"></i>
                     </span>
@@ -351,7 +351,7 @@
                 <qrcode size="160" data="{{selectedAddr.address}}"></qrcode>
                 <p class="m10t">
                   <strong>
-                    <span ng-if="$root.receivedFund[1] == selectedAddr.address">
+                    <span ng-if="addrWithFund == selectedAddr.address">
                       <span ng-if="$root.updatingBalance">
                         <i class="fi-bitcoin-circle icon-rotate spinner"></i>
                       </span>
@@ -359,7 +359,7 @@
                         <i class="fi-bitcoin"></i>
                       </span>
                     </span>
-                    <span ng-if="$root.receivedFund[1] != selectedAddr.address">
+                    <span ng-if="addrWithFund != selectedAddr.address">
                       {{selectedAddr.balance || 0}}
                       <i class="fi-bitcoin"></i>
                     </span>
diff --git a/js/controllers/addresses.js b/js/controllers/addresses.js
index c34931dd2..943b5a6f4 100644
--- a/js/controllers/addresses.js
+++ b/js/controllers/addresses.js
@@ -20,12 +20,13 @@ angular.module('copay.addresses').controller('AddressesController',
       $scope.selectedAddr = addr;
     };
 
-    $rootScope.$watch('addrInfos', function(addrInfos) {
-      $scope.addressList(addrInfos);
+    $rootScope.$watch('addrInfos', function() {
+      $scope.addressList();
     });   
 
-    $scope.addressList = function (addrInfos) {
+    $scope.addressList = function () {
       $scope.addresses = [];
+      var addrInfos = $rootScope.addrInfos;
       if (addrInfos) {
         for(var i=0;i<addrInfos.length;i++) {
           var addrinfo = addrInfos[i];
@@ -36,6 +37,8 @@ angular.module('copay.addresses').controller('AddressesController',
           });
         }
         $scope.selectedAddr = $scope.addresses[0];
+        $scope.addrWithFund = $rootScope.receivedFund ? $rootScope.receivedFund[1] : null;
+        $rootScope.receivedFund = null;
       }
     }
 
diff --git a/js/controllers/header.js b/js/controllers/header.js
index 8c4a9b58f..af2b86fab 100644
--- a/js/controllers/header.js
+++ b/js/controllers/header.js
@@ -63,9 +63,11 @@ angular.module('copay.header').controller('HeaderController',
     $scope.refresh = function() {
       var w = $rootScope.wallet;
       w.connectToAll();
-      controllerUtils.updateBalance(function() {
-        $rootScope.$digest();
-      });
+      if ($rootScope.addrInfos.length > 0 ) {
+        controllerUtils.updateBalance(function() {
+          $rootScope.$digest();
+        });
+      }
     };
 
     $scope.clearFlashMessage = function() {
diff --git a/js/services/controllerUtils.js b/js/services/controllerUtils.js
index d0298f0d7..61afc52b1 100644
--- a/js/services/controllerUtils.js
+++ b/js/services/controllerUtils.js
@@ -107,10 +107,7 @@ angular.module('copay.controllerUtils')
 
     root.updateBalance = function(cb) {
       console.log('Updating balance...');
-      root.updateAddressList();
       var w = $rootScope.wallet;
-      if ($rootScope.addrInfos.length === 0) 
-        return cb?cb():null;
 
       $rootScope.balanceByAddr = {};
       $rootScope.updatingBalance = true;
@@ -118,6 +115,7 @@ angular.module('copay.controllerUtils')
         $rootScope.totalBalance = balance;
         $rootScope.balanceByAddr = balanceByAddr;
         $rootScope.availableBalance = safeBalance;
+        root.updateAddressList();
         $rootScope.updatingBalance = false;
         console.log('Done updating balance.'); //TODO
         return cb?cb():null;