diff --git a/js/controllers/backup.js b/js/controllers/backup.js index e851e602c..3f78d1ad5 100644 --- a/js/controllers/backup.js +++ b/js/controllers/backup.js @@ -1,13 +1,15 @@ 'use strict'; angular.module('copay.backup').controller('BackupController', - function($scope, $rootScope, $location) { + function($scope, $rootScope, $location, Socket, controllerUtils) { - - if (!$rootScope.wallet.id) { + if (!$rootScope.wallet || !$rootScope.wallet.id) { $location.path('signin'); } - + else { + var socket = Socket($scope); + socket.on('connect', controllerUtils.handleTransactionByAddress($scope)); + } $scope.title = 'Backup'; }); diff --git a/js/controllers/home.js b/js/controllers/home.js index 8c7798428..75f059d01 100644 --- a/js/controllers/home.js +++ b/js/controllers/home.js @@ -1,12 +1,12 @@ 'use strict'; angular.module('copay.home').controller('HomeController', - function($scope, $rootScope, $location) { + function($scope, $rootScope, $location, Socket, controllerUtils) { $scope.title = 'Home'; $scope.oneAtATime = true; $scope.addrBalance = {}; - + var _getBalance = function() { $scope.addrs.forEach(function(addr) { $rootScope.wallet.blockchain.listUnspent([addr], function(unspent) { @@ -24,6 +24,9 @@ angular.module('copay.home').controller('HomeController', $scope.selectedAddr = $scope.addrs[0]; _getBalance(); + + var socket = Socket($scope); + socket.on('connect', controllerUtils.handleTransactionByAddress($scope)); } $scope.newAddr = function() { @@ -31,6 +34,8 @@ angular.module('copay.home').controller('HomeController', $scope.addrs.push(a); _getBalance(); + var socket = Socket($scope); + socket.on('connect', controllerUtils.handleTransactionByAddress($scope)); }; $scope.selectAddr = function(addr) { diff --git a/js/controllers/peer.js b/js/controllers/peer.js index f072745e9..a3098aca4 100644 --- a/js/controllers/peer.js +++ b/js/controllers/peer.js @@ -1,31 +1,18 @@ 'use strict'; angular.module('copay.peer').controller('PeerController', - function($scope, $rootScope, $location, $routeParams, Socket) { + function($scope, $rootScope, $location, $routeParams, Socket, controllerUtils) { $scope.init = function() { //Network.connect($rootScope.masterId); }; - var socket = Socket($scope); - socket.on('connect', function() { - var addrs = $rootScope.wallet.getAddressesStr(); - socket.emit('subscribe', 'inv'); - for(var i=0;i