From 144812c09316d9016a87d6a8e5096ddb7fbf5f63 Mon Sep 17 00:00:00 2001
From: Arnaldas Augutis <info@untitled.lt>
Date: Thu, 22 Nov 2018 15:38:05 +0200
Subject: ability to close and restore streamin server

---
 lib/server.js | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

(limited to 'lib')

diff --git a/lib/server.js b/lib/server.js
index 60595a8..2988f27 100644
--- a/lib/server.js
+++ b/lib/server.js
@@ -12,11 +12,16 @@ function Server (torrent, opts = {}) {
   const sockets = []
   const pendingReady = []
   let closed = false
+  const _listen = server.listen
+  const _close = server.close
 
-  server.on('connection', onConnection)
-  server.on('request', onRequest)
+  server.listen = port => {
+    closed = false
+    server.on('connection', onConnection)
+    server.on('request', onRequest)
+    _listen.call(server, port)
+  }
 
-  const _close = server.close
   server.close = cb => {
     closed = true
     server.removeListener('connection', onConnection)
@@ -25,7 +30,6 @@ function Server (torrent, opts = {}) {
       const onReady = pendingReady.pop()
       torrent.removeListener('ready', onReady)
     }
-    torrent = null
     _close.call(server, cb)
   }
 
-- 
cgit v1.2.3