filestream.destroy(): accept onclose callback instead of error

1 files changed, 8 insertions, 2 deletions

diff --git a/lib/file-stream.js b/lib/file-stream.js
index 9402573..55355ea 100644
--- a/lib/file-stream.js
+++ b/lib/file-stream.js
@@ -60,7 +60,7 @@ FileStream.prototype._notify = function () {
   self._torrent.store.get(p, function (err, buffer) {
     self._notifying = false
     if (self.destroyed) return
-    if (err) return self.destroy(err)
+    if (err) return self._destroy(err)
     debug('read %s (length %s) (err %s)', p, buffer.length, err && err.message)
 
     if (self._offset) {
@@ -82,9 +82,15 @@ FileStream.prototype._notify = function () {
   self._piece += 1
 }
 
-FileStream.prototype.destroy = function (err) {
+FileStream.prototype.destroy = function (onclose) {
+  this._destroy(null, onclose)
+}
+
+FileStream.prototype._destroy = function (err, onclose) {
   if (this.destroyed) return
+  if (onclose) this.once('close', onclose)
   this.destroyed = true
+
   if (!this._torrent.destroyed) {
     this._torrent.deselect(this._startPiece, this._endPiece, true)
   }