From e5a4e3c25cecb544f2ddba7bf7d2f9b0babbbb3e Mon Sep 17 00:00:00 2001
From: Mario Colque <colquemario@gmail.com>
Date: Fri, 18 Apr 2014 15:22:02 -0300
Subject: [PATCH 1/4] modev wallet validation from all controllers to router

---
 js/controllers/backup.js       |  7 +------
 js/controllers/header.js       |  4 ----
 js/controllers/home.js         |  4 ----
 js/controllers/peer.js         |  7 +------
 js/controllers/send.js         |  7 +------
 js/controllers/setup.js        |  1 +
 js/controllers/signin.js       |  4 ++--
 js/controllers/transactions.js | 12 +++---------
 js/routes.js                   |  8 ++++++++
 9 files changed, 17 insertions(+), 37 deletions(-)

diff --git a/js/controllers/backup.js b/js/controllers/backup.js
index bad72f097..309da15dc 100644
--- a/js/controllers/backup.js
+++ b/js/controllers/backup.js
@@ -2,12 +2,7 @@
 
 angular.module('copay.backup').controller('BackupController',
   function($scope, $rootScope, $location, $window, $timeout, Socket, controllerUtils) {
-    if (!$rootScope.wallet || !$rootScope.wallet.id) {
-      $location.path('signin');
-    }
-    else {
-      controllerUtils.handleTransactionByAddress($scope);
-    }
+    controllerUtils.handleTransactionByAddress($scope);
 
     $scope.title = 'Backup';
 
diff --git a/js/controllers/header.js b/js/controllers/header.js
index d6680e76b..9919d11fd 100644
--- a/js/controllers/header.js
+++ b/js/controllers/header.js
@@ -24,10 +24,6 @@ angular.module('copay.header').controller('HeaderController',
       'link': '#/backup'
     }];
 
-   if (!$rootScope.wallet || !$rootScope.wallet.id) {
-      $location.path('signin');
-    }
-
     $scope.isActive = function(item) {
       if (item.link && item.link.replace('#','') == $location.path()) {
         return true;
diff --git a/js/controllers/home.js b/js/controllers/home.js
index f17f92ac5..f103b200e 100644
--- a/js/controllers/home.js
+++ b/js/controllers/home.js
@@ -28,9 +28,5 @@ angular.module('copay.home').controller('HomeController',
       $scope.selectedAddr = addr;
     };
 
-    
-    if (!$rootScope.wallet || !$rootScope.wallet.id) {
-      $location.path('signin');
-    }
     _updateBalance();
   });
diff --git a/js/controllers/peer.js b/js/controllers/peer.js
index ad0537b4b..e3c7f8b7b 100644
--- a/js/controllers/peer.js
+++ b/js/controllers/peer.js
@@ -7,11 +7,6 @@ angular.module('copay.peer').controller('PeerController',
       //Network.connect($rootScope.masterId);
     };
 
-    if (!$rootScope.wallet || !$rootScope.wallet.id) {
-      $location.path('signin');
-    }
-    else {
-      controllerUtils.handleTransactionByAddress($scope);
-    }
+    controllerUtils.handleTransactionByAddress($scope);
   });
 
diff --git a/js/controllers/send.js b/js/controllers/send.js
index a808e3941..02cd74c37 100644
--- a/js/controllers/send.js
+++ b/js/controllers/send.js
@@ -4,12 +4,7 @@ angular.module('copay.send').controller('SendController',
   function($scope, $rootScope, $location, Socket, controllerUtils) {
     $scope.title = 'Send';
 
-    if (!$rootScope.wallet || !$rootScope.wallet.id) {
-      $location.path('signin');
-    }
-    else {
-      controllerUtils.handleTransactionByAddress($scope); 
-    }
+    controllerUtils.handleTransactionByAddress($scope); 
 
     $scope.sendTest = function() {
       var w    = $rootScope.wallet;
diff --git a/js/controllers/setup.js b/js/controllers/setup.js
index 58d5f158d..b13f96071 100644
--- a/js/controllers/setup.js
+++ b/js/controllers/setup.js
@@ -19,6 +19,7 @@ angular.module('copay.setup').controller('SetupController',
         }
       }
     };
+
     updateRCSelect($scope.totalCopayers);
 
     $scope.$watch('totalCopayers', function(tc) {
diff --git a/js/controllers/signin.js b/js/controllers/signin.js
index 260fbb07b..2d4e43e3d 100644
--- a/js/controllers/signin.js
+++ b/js/controllers/signin.js
@@ -1,12 +1,12 @@
 'use strict';
 
-
-
 angular.module('copay.signin').controller('SigninController',
   function($scope, $rootScope, $location, walletFactory, controllerUtils) {
 
 //    var peerData = Storage.get($rootScope.walletId, 'peerData');
 //    $rootScope.peerId = peerData ? peerData.peerId : null;
+//
+
     $scope.loading = false;
 
     $scope.walletIds = walletFactory.getWalletIds();
diff --git a/js/controllers/transactions.js b/js/controllers/transactions.js
index ccbed41a4..639890105 100644
--- a/js/controllers/transactions.js
+++ b/js/controllers/transactions.js
@@ -34,15 +34,9 @@ angular.module('copay.transactions').controller('TransactionsController',
       $scope.txs = ts;
     };
 
-
-    if (!$rootScope.wallet || !$rootScope.wallet.id) {
-      $location.path('signin');
-    }
-    else {
-      _updateTxs();
-      var socket = Socket($scope);
-      socket.on('connect', controllerUtils.handleTransactionByAddress($scope));
-    }
+    _updateTxs();
+    var socket = Socket($scope);
+    socket.on('connect', controllerUtils.handleTransactionByAddress($scope));
 
     $scope.sign = function (ntxid) {
       var w = $rootScope.wallet;
diff --git a/js/routes.js b/js/routes.js
index 5c655b208..6bca7f6a1 100644
--- a/js/routes.js
+++ b/js/routes.js
@@ -45,4 +45,12 @@ angular
     $locationProvider
       .html5Mode(false);
       //.hashPrefix('!');
+  })
+  .run(function($rootScope, $location) {
+    $rootScope.$on('$routeChangeStart', function() {
+      if (!$rootScope.wallet || !$rootScope.wallet.id) {
+        console.log('############ no wallet');
+        $location.path('signin');
+      }
+    });
   });

From 58a15502d1046b46cebfd5693295856d5d519c20 Mon Sep 17 00:00:00 2001
From: Mario Colque <colquemario@gmail.com>
Date: Fri, 18 Apr 2014 15:36:38 -0300
Subject: [PATCH 2/4] clean up peerController

---
 index.html             | 2 +-
 js/controllers/peer.js | 5 -----
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/index.html b/index.html
index 90191c5c3..4a0457bc2 100644
--- a/index.html
+++ b/index.html
@@ -170,7 +170,7 @@
     </script>
 
     <script type="text/ng-template" id="peer.html">
-      <div class="row" ng-controller="PeerController" ng-init="init()">
+      <div class="row" ng-controller="PeerController">
         <div class="large-6 columns p70r line-dashed-v">
           <h3>I am </h3>
           <div class="row">
diff --git a/js/controllers/peer.js b/js/controllers/peer.js
index e3c7f8b7b..5ce6125e1 100644
--- a/js/controllers/peer.js
+++ b/js/controllers/peer.js
@@ -2,11 +2,6 @@
 
 angular.module('copay.peer').controller('PeerController',
   function($scope, $rootScope, $location, $routeParams, Socket, controllerUtils) {
-    
-    $scope.init = function() {
-      //Network.connect($rootScope.masterId);
-    };
-
     controllerUtils.handleTransactionByAddress($scope);
   });
 

From 0455d5ee69063de86e24caf1a4b87b92e2ea89f3 Mon Sep 17 00:00:00 2001
From: Mario Colque <colquemario@gmail.com>
Date: Fri, 18 Apr 2014 19:08:01 -0300
Subject: [PATCH 3/4] refactory of the socket service

---
 index.html                     |   2 +-
 js/controllers/backup.js       |   4 +-
 js/controllers/header.js       |   6 ++
 js/controllers/home.js         |  17 +++---
 js/controllers/peer.js         |   3 +-
 js/controllers/send.js         |   4 +-
 js/controllers/setup.js        |   2 +-
 js/controllers/signin.js       |  35 +-----------
 js/controllers/transactions.js |   4 +-
 js/routes.js                   |  32 +++++++----
 js/services/controllerUtils.js |  21 ++++---
 js/services/socket.js          | 101 +++++++++++++--------------------
 12 files changed, 94 insertions(+), 137 deletions(-)

diff --git a/index.html b/index.html
index 4a0457bc2..3fe435694 100644
--- a/index.html
+++ b/index.html
@@ -211,7 +211,7 @@
         <h3>Address</h3>
         <div class="row">
           <div class="large-6 columns">
-            <a class="panel db" ng-repeat="addr in addrs" ng-click="selectAddr(addr)">{{addr}} <span class="right">{{balanceByAddr[addr] || 0}} BTC &gt;</span></a>
+            <a class="panel db" ng-repeat="addr in addrs" ng-click="selectAddr(addr)">{{addr}} <span class="right">{{$root.balanceByAddr[addr] || 0}} BTC &gt;</span></a>
           </div>
           <div class="large-3 columns line-dashed-v text-center">
             <qrcode size="160" data="{{selectedAddr}}"></qrcode>
diff --git a/js/controllers/backup.js b/js/controllers/backup.js
index 309da15dc..5c468fe24 100644
--- a/js/controllers/backup.js
+++ b/js/controllers/backup.js
@@ -1,9 +1,7 @@
 'use strict';
 
 angular.module('copay.backup').controller('BackupController',
-  function($scope, $rootScope, $location, $window, $timeout, Socket, controllerUtils) {
-    controllerUtils.handleTransactionByAddress($scope);
-
+  function($scope, $rootScope, $location, $window, $timeout) {
     $scope.title = 'Backup';
 
     var filename = $rootScope.wallet.id + '.json.aes';
diff --git a/js/controllers/header.js b/js/controllers/header.js
index 9919d11fd..36c67ebfd 100644
--- a/js/controllers/header.js
+++ b/js/controllers/header.js
@@ -24,6 +24,12 @@ angular.module('copay.header').controller('HeaderController',
       'link': '#/backup'
     }];
 
+    $rootScope.$watch('wallet', function(wallet) {
+      if (wallet) {
+        controllerUtils.setSocketHandlers(); 
+      }
+    });
+
     $scope.isActive = function(item) {
       if (item.link && item.link.replace('#','') == $location.path()) {
         return true;
diff --git a/js/controllers/home.js b/js/controllers/home.js
index f103b200e..0fc8e2b3c 100644
--- a/js/controllers/home.js
+++ b/js/controllers/home.js
@@ -1,7 +1,7 @@
 'use strict';
 
 angular.module('copay.home').controller('HomeController',
-  function($scope, $rootScope, $location, Socket, controllerUtils) {
+  function($scope, $rootScope, controllerUtils) {
     $scope.title = 'Home';
     $scope.oneAtATime = true;
     $scope.addrBalance = {};
@@ -10,18 +10,19 @@ angular.module('copay.home').controller('HomeController',
 
     var _updateBalance = function () {
       w.getBalance(function (balance, balanceByAddr) {
-        $scope.balanceByAddr = balanceByAddr;
-        $scope.addrs =  Object.keys(balanceByAddr);
-        $scope.selectedAddr = $scope.addrs[0];
-        $scope.$digest();
+        $rootScope.$apply(function() {
+          $rootScope.balanceByAddr = balanceByAddr;
+          $scope.addrs =  Object.keys(balanceByAddr);
+          $scope.selectedAddr = $scope.addrs[0];
+        });
       });
-      var socket = Socket($scope);
-      controllerUtils.handleTransactionByAddress($scope, _updateBalance);
     };
 
     $scope.newAddr = function() {
-      var a = w.generateAddress().toString();
+      w.generateAddress().toString();
       _updateBalance();
+
+      controllerUtils.setSocketHandlers();
     };
 
     $scope.selectAddr = function(addr) {
diff --git a/js/controllers/peer.js b/js/controllers/peer.js
index 5ce6125e1..5ae619020 100644
--- a/js/controllers/peer.js
+++ b/js/controllers/peer.js
@@ -1,7 +1,6 @@
 'use strict';
 
 angular.module('copay.peer').controller('PeerController',
-  function($scope, $rootScope, $location, $routeParams, Socket, controllerUtils) {
-    controllerUtils.handleTransactionByAddress($scope);
+  function($scope, $rootScope, $location, $routeParams) {
   });
 
diff --git a/js/controllers/send.js b/js/controllers/send.js
index 02cd74c37..5f1ea7267 100644
--- a/js/controllers/send.js
+++ b/js/controllers/send.js
@@ -1,11 +1,9 @@
 'use strict';
 
 angular.module('copay.send').controller('SendController',
-  function($scope, $rootScope, $location, Socket, controllerUtils) {
+  function($scope, $rootScope, $location) {
     $scope.title = 'Send';
 
-    controllerUtils.handleTransactionByAddress($scope); 
-
     $scope.sendTest = function() {
       var w    = $rootScope.wallet;
       w.createTx( 'mimoZNLcP2rrMRgdeX5PSnR7AjCqQveZZ4', '12345',function() {
diff --git a/js/controllers/setup.js b/js/controllers/setup.js
index b13f96071..db18809c1 100644
--- a/js/controllers/setup.js
+++ b/js/controllers/setup.js
@@ -24,7 +24,7 @@ angular.module('copay.setup').controller('SetupController',
 
     $scope.$watch('totalCopayers', function(tc) {
       updateRCSelect(tc);
-    })
+    });
 
     $scope.create = function(totalCopayers, requiredCopayers) {
       $scope.loading = true;
diff --git a/js/controllers/signin.js b/js/controllers/signin.js
index 2d4e43e3d..2336c3238 100644
--- a/js/controllers/signin.js
+++ b/js/controllers/signin.js
@@ -2,15 +2,8 @@
 
 angular.module('copay.signin').controller('SigninController',
   function($scope, $rootScope, $location, walletFactory, controllerUtils) {
-
-//    var peerData = Storage.get($rootScope.walletId, 'peerData');
-//    $rootScope.peerId = peerData ? peerData.peerId : null;
-//
-
     $scope.loading = false;
-
     $scope.walletIds = walletFactory.getWalletIds();
-
     $scope.selectedWalletId = $scope.walletIds.length ? $scope.walletIds[0]:null;
 
     $scope.create = function() {
@@ -25,39 +18,15 @@ angular.module('copay.signin').controller('SigninController',
       w.netStart();
     };
 
-    $scope.join = function(cid) {
+    $scope.join = function(peerId) {
       $scope.loading = true;
       walletFactory.network.on('openError', function() {
         controllerUtils.onError($scope); 
         $rootScope.$digest();
       });
-      walletFactory.connectTo(cid, function(w) {
+      walletFactory.connectTo(peerId, function(w) {
         controllerUtils.setupUxHandlers(w);
         w.netStart();
       });
     };
-
-
-//
-//       if (cid) {
-//         var w = walletFactory.(walletId);
-        //TODO
-        // Network.init(null, function() {
-        //   Network.connect(cid, 
-        //     function() {
-        //       $location.path('peer');
-        //       $rootScope.$digest();
-        //     }, function() {
-        //       $rootScope.flashMessage = { message: 'Connection refussed', type: 'error'};
-        //       $location.path('home');
-        //       $rootScope.$digest();
-        //   });
-        // });
-//      }
-
-    // if (peerData && peerData.peerId && peerData.connectedPeers.length > 0) {
-    //   $rootScope.peerId = peerData.peerId;
-    //   $scope.join(peerData.connectedPeers);
-    // }
   });
-
diff --git a/js/controllers/transactions.js b/js/controllers/transactions.js
index 639890105..44ca586d7 100644
--- a/js/controllers/transactions.js
+++ b/js/controllers/transactions.js
@@ -1,7 +1,7 @@
 'use strict';
 
 angular.module('copay.transactions').controller('TransactionsController',
-  function($scope, $rootScope, $location, Socket, controllerUtils) {
+  function($scope, $rootScope, $location) {
     $scope.title = 'Transactions';
 
     $scope.oneAtATime = true;
@@ -35,8 +35,6 @@ angular.module('copay.transactions').controller('TransactionsController',
     };
 
     _updateTxs();
-    var socket = Socket($scope);
-    socket.on('connect', controllerUtils.handleTransactionByAddress($scope));
 
     $scope.sign = function (ntxid) {
       var w = $rootScope.wallet;
diff --git a/js/routes.js b/js/routes.js
index 6bca7f6a1..fa17d4077 100644
--- a/js/routes.js
+++ b/js/routes.js
@@ -7,31 +7,40 @@ angular
 
     $routeProvider
       .when('/', {
-        templateUrl: 'signin.html'
+        templateUrl: 'signin.html',
+        validate: false
       })
       .when('/signin', {
-        templateUrl: 'signin.html' 
+        templateUrl: 'signin.html',
+        validate: false
       })
       .when('/setup', {
-        templateUrl: 'setup.html' 
+        templateUrl: 'setup.html',
+        validate: false
       })
       .when('/home', {
-        templateUrl: 'home.html'
+        templateUrl: 'home.html',
+        validate: true
       })
       .when('/join/:id', {
-        templateUrl: 'join.html'
+        templateUrl: 'join.html',
+        validate: true
       })
       .when('/peer', {
-        templateUrl: 'peer.html'
+        templateUrl: 'peer.html',
+        validate: true
       })
       .when('/transactions', {
-        templateUrl: 'transactions.html' 
+        templateUrl: 'transactions.html',
+        validate: true
       })
       .when('/send', {
-        templateUrl: 'send.html' 
+        templateUrl: 'send.html',
+        validate: true
       })
       .when('/backup', {
-        templateUrl: 'backup.html' 
+        templateUrl: 'backup.html',
+        validate: true
       })
       .otherwise({
         templateUrl: '404.html'
@@ -47,9 +56,8 @@ angular
       //.hashPrefix('!');
   })
   .run(function($rootScope, $location) {
-    $rootScope.$on('$routeChangeStart', function() {
-      if (!$rootScope.wallet || !$rootScope.wallet.id) {
-        console.log('############ no wallet');
+    $rootScope.$on('$routeChangeStart', function(event, next, current) {
+      if ((!$rootScope.wallet || !$rootScope.wallet.id) && next.validate) {
         $location.path('signin');
       }
     });
diff --git a/js/services/controllerUtils.js b/js/services/controllerUtils.js
index c3ffd373b..b208cef7a 100644
--- a/js/services/controllerUtils.js
+++ b/js/services/controllerUtils.js
@@ -21,9 +21,7 @@ angular.module('copay.controllerUtils').factory('controllerUtils', function ($ro
     $rootScope.$digest();
   }
 
-
   root.setupUxHandlers =  function(w) {
-
     w.on('badMessage', function(peerId) {
       $rootScope.flashMessage = {type:'error', message: 'Received wrong message from peer id:' + peerId};
     });
@@ -45,19 +43,24 @@ angular.module('copay.controllerUtils').factory('controllerUtils', function ($ro
     w.on('close', root.onErrorDigest);
   };
 
-  root.handleTransactionByAddress = function(scope, cb) {
-    var socket = Socket(scope);
+  root.setSocketHandlers = function(cb) {
+    Socket.removeAllListeners();
+
     var addrs = $rootScope.wallet.getAddressesStr();
-    for(var i=0;i<addrs.length;i++) {
-      socket.emit('subscribe', addrs[i]);
+    for(var i = 0; i < addrs.length; i++) {
+      console.log('### SUBSCRIBE TO', addrs[i]);
+      Socket.emit('subscribe', addrs[i]);
     }
+
     addrs.forEach(function(addr) {
-      socket.on(addr, function(txid) {
+      Socket.on(addr, function(txid) {
         console.log('Received!', txid);
-        $rootScope.wallet.getBalance(function(balance) {
-          scope.$apply(function() {
+        $rootScope.wallet.getBalance(function(balance, balanceByAddr) {
+          $rootScope.$apply(function() {
             $rootScope.totalBalance = balance;
+            $rootScope.balanceByAddr = balanceByAddr;
           });
+
           console.log('New balance:', balance);
           if (typeof cb === 'function') return cb();
         });
diff --git a/js/services/socket.js b/js/services/socket.js
index 9f2f1aa61..bd59bde8e 100644
--- a/js/services/socket.js
+++ b/js/services/socket.js
@@ -1,72 +1,49 @@
 'use strict';
 
-var ScopedSocket = function(socket, $rootScope) {
-  this.socket = socket;
-  this.$rootScope = $rootScope;
-  this.listeners = [];
-};
-
-ScopedSocket.prototype.removeAllListeners = function(opts) {
-  if (!opts) opts = {};
-  for (var i = 0; i < this.listeners.length; i++) {
-    var details = this.listeners[i];
-    if (opts.skipConnect && details.event === 'connect') {
-      continue;
-    }
-    this.socket.removeListener(details.event, details.fn);
-  }
-  this.listeners = [];
-};
-
-ScopedSocket.prototype.on = function(event, callback) {
-  var socket = this.socket;
-  var $rootScope = this.$rootScope;
-
-  var wrapped_callback = function() {
-    var args = arguments;
-    $rootScope.$apply(function() {
-      callback.apply(socket, args);
-    });
-  };
-  socket.on(event, wrapped_callback);
-
-  this.listeners.push({
-    event: event,
-    fn: wrapped_callback
-  });
-};
-
-ScopedSocket.prototype.emit = function(event, data, callback) {
-  var socket = this.socket;
-  var $rootScope = this.$rootScope;
-
-  socket.emit(event, data, function() {
-    var args = arguments;
-    $rootScope.$apply(function() {
-      if (callback) {
-        callback.apply(socket, args);
-      }
-    });
-  });
-};
-
 angular.module('copay.socket').factory('Socket',
   function($rootScope) {
-    var server = 'http://' + config.socket.host + ':' + config.socket.port;
-    var socket = io.connect(server, {
+    var listeners = [];
+    var url = 'http://' + config.socket.host + ':' + config.socket.port;
+    var socket = io.connect(url, {
       'reconnect': true,
       'reconnection delay': 500,
     });
-    return function(scope) {
-      var scopedSocket = new ScopedSocket(socket, $rootScope);
-      scope.$on('$destroy', function() {
-        scopedSocket.removeAllListeners();
-      });
-      socket.on('connect', function() {
-        scopedSocket.removeAllListeners({
-          skipConnect: true
+
+    return {
+      on: function(event, callback) {
+        var wrappedCallback = function() {
+          var args = arguments;
+          $rootScope.$apply(function() {
+            callback.apply(socket, args);
+          });
+        };
+
+        socket.on(event, wrappedCallback);
+
+        if (event !== 'connect') {
+          listeners.push({
+            event: event,
+            fn: wrappedCallback
+          });
+        }
+      },
+      emit: function(event, data, callback) {
+        socket.emit(event, data, function() {
+          var args = arguments;
+          $rootScope.$apply(function() {
+            if (callback) {
+              callback.apply(socket, args);
+            }
+          });
         });
-      });
-      return scopedSocket;
+      },
+      removeAllListeners: function() {
+        for (var i = 0; i < listeners.length; i++) {
+          var details = listeners[i];
+          socket.removeListener(details.event, details.fn);
+        }
+
+        listeners = [];
+      }
     };
   });

From e928ca2b81a2c9c09923e7c581d866b0e272ec17 Mon Sep 17 00:00:00 2001
From: Mario Colque <colquemario@gmail.com>
Date: Mon, 21 Apr 2014 11:32:51 -0300
Subject: [PATCH 4/4] several improvements

---
 js/controllers/addresses.js | 12 ++++++------
 js/controllers/header.js    |  7 ++++++-
 js/routes.js                |  3 ++-
 3 files changed, 14 insertions(+), 8 deletions(-)

diff --git a/js/controllers/addresses.js b/js/controllers/addresses.js
index bfe216d41..4f644208c 100644
--- a/js/controllers/addresses.js
+++ b/js/controllers/addresses.js
@@ -9,21 +9,21 @@ angular.module('copay.addresses').controller('AddressesController',
     var w = $rootScope.wallet;
 
     var _updateBalance = function () {
+      controllerUtils.setSocketHandlers();
+
       w.getBalance(function (balance, balanceByAddr) {
         if (balanceByAddr  && Object.keys(balanceByAddr).length) {
-          $scope.balanceByAddr = balanceByAddr;
+          $rootScope.balanceByAddr = balanceByAddr;
           $scope.addrs =  Object.keys(balanceByAddr);
           $scope.selectedAddr = $scope.addrs[0];
-          $scope.$digest();
+          $rootScope.$digest();
         }
       });
     };
 
     $scope.newAddr = function() {
-      w.generateAddress().toString();
+      w.generateAddress();
       _updateBalance();
-
-      controllerUtils.setSocketHandlers();
     };
 
     $scope.selectAddr = function(addr) {
@@ -31,5 +31,5 @@ angular.module('copay.addresses').controller('AddressesController',
     };
 
     _updateBalance();
-    w.on('refresh',_updateBalance);
+    w.on('refresh', _updateBalance);
   });
diff --git a/js/controllers/header.js b/js/controllers/header.js
index 2f1bf79eb..ac800178f 100644
--- a/js/controllers/header.js
+++ b/js/controllers/header.js
@@ -26,7 +26,12 @@ angular.module('copay.header').controller('HeaderController',
 
     $rootScope.$watch('wallet', function(wallet) {
       if (wallet) {
-        controllerUtils.setSocketHandlers(); 
+        controllerUtils.setSocketHandlers();
+        $rootScope.wallet.getBalance(function(balance) {
+          $rootScope.$apply(function() {
+            $rootScope.totalBalance = balance;
+          });
+        });
       }
     });
 
diff --git a/js/routes.js b/js/routes.js
index 83b158d1b..3970dd047 100644
--- a/js/routes.js
+++ b/js/routes.js
@@ -19,7 +19,8 @@ angular
         validate: false
       })
       .when('/addresses', {
-        templateUrl: 'addresses.html'
+        templateUrl: 'addresses.html',
+        validate: true
       })
       .when('/join/:id', {
         templateUrl: 'join.html',