fix $ ui error

2 files changed, 13 insertions, 2 deletions

diff --git a/lib/Torrent.js b/lib/Torrent.js
index b0b2599..6d047a0 100644
--- a/lib/Torrent.js
+++ b/lib/Torrent.js
@@ -35,8 +35,6 @@ function Torrent (uri, opts) {
   })
 
   this.swarm.on('wire', function (wire) {
-    $('.connectedPeers span').text(this.swarm.wires.length)
-
     // Send KEEP-ALIVE (every 60s) so peers will not disconnect the wire
     wire.setKeepAlive(true)
 
@@ -153,6 +151,12 @@ function Torrent (uri, opts) {
   }.bind(this))
 }
 
+Object.defineProperty(Torrent.prototype, 'numPeers', {
+  get: function () {
+    return this.swarm.wires.length
+  }
+})
+
 /**
  * Add a peer to the swarm
  * @param {string} addr
diff --git a/lib/TorrentManager.js b/lib/TorrentManager.js
index cac04eb..265ef26 100644
--- a/lib/TorrentManager.js
+++ b/lib/TorrentManager.js
@@ -84,5 +84,12 @@ TorrentManager.prototype.updateUI = function () {
 
   $('.dhtNodes span').text(Object.keys(this.dht.nodes).length)
   $('.dhtPeers span').text(Object.keys(this.dht.peers).length)
+
+  var connectedPeers = 0
+  for (var infoHash in this.torrents) {
+    var torrent = this.torrents[infoHash]
+    connectedPeers += torrent.numPeers
+  }
+  $('.connectedPeers span').text(connectedPeers)
 }