cmd: --vlc keeps the event loop alive during "graceful exit" stage (fix #388)

1 files changed, 17 insertions, 1 deletions

diff --git a/lib/server.js b/lib/server.js
index a9c2542..540794e 100644
--- a/lib/server.js
+++ b/lib/server.js
@@ -1,3 +1,5 @@
+module.exports = Server
+
 var debug = require('debug')('webtorrent:server')
 var http = require('http')
 var mime = require('mime')
@@ -5,13 +7,27 @@ var pump = require('pump')
 var rangeParser = require('range-parser')
 var url = require('url')
 
-module.exports = function Server (torrent, opts) {
+function Server (torrent, opts) {
   var server = http.createServer(opts)
 
+  var sockets = []
+
   server.on('connection', function (socket) {
     socket.setTimeout(36000000)
+    sockets.push(socket)
+    socket.on('close', function () {
+      var index = sockets.indexOf(socket)
+      if (index >= 0) sockets.splice(index, 1)
+    })
   })
 
+  server.destroy = function (cb) {
+    sockets.forEach(function (socket) {
+      socket.destroy()
+    })
+    server.close(cb)
+  }
+
   server.on('request', function (req, res) {
     debug('onRequest')