From 1b870835793b7864c95b61734eb580082010376e Mon Sep 17 00:00:00 2001
From: Manuel Araoz <manuelaraoz@gmail.com>
Date: Fri, 9 May 2014 14:44:05 -0300
Subject: [PATCH] disconnect working

---
 js/models/core/Wallet.js       | 3 ++-
 js/models/network/WebRTC.js    | 6 +++---
 js/services/controllerUtils.js | 3 +++
 3 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/js/models/core/Wallet.js b/js/models/core/Wallet.js
index a906c20ed..5bab60fe8 100644
--- a/js/models/core/Wallet.js
+++ b/js/models/core/Wallet.js
@@ -134,7 +134,8 @@ Wallet.prototype._handleConnect = function(newCopayerId) {
   this.emit('connect', peerID);
 };
 
-Wallet.prototype._handleDisconnect = function(copayerID) {
+Wallet.prototype._handleDisconnect = function(peerID) {
+  this.emit('disconnect', peerID);
 };
 
 Wallet.prototype._optsToObj = function() {
diff --git a/js/models/network/WebRTC.js b/js/models/network/WebRTC.js
index 59f520935..5877f45a2 100644
--- a/js/models/network/WebRTC.js
+++ b/js/models/network/WebRTC.js
@@ -115,9 +115,9 @@ Network.prototype._deletePeer = function(peerId) {
   this.connectedPeers = Network._arrayRemove(peerId, this.connectedPeers);
 };
 
-Network.prototype._onClose = function(peerId) {
-  this._deletePeer(peerId);
-  this.emit('disconnect');
+Network.prototype._onClose = function(peerID) {
+  this._deletePeer(peerID);
+  this.emit('disconnect', peerID);
 };
 
 Network.prototype.connectToCopayers = function(copayerIds) {
diff --git a/js/services/controllerUtils.js b/js/services/controllerUtils.js
index 76302af12..e3574c5fe 100644
--- a/js/services/controllerUtils.js
+++ b/js/services/controllerUtils.js
@@ -96,6 +96,9 @@ angular.module('copay.controllerUtils')
         }
         $rootScope.$digest();
       });
+      w.on('disconnect', function(peerID) {
+        $rootScope.$digest();
+      });
       w.on('close', root.onErrorDigest);
       w.netStart();
     };