allow for progress on write

webtorrent · ftreesmilo · Dec 10, 2019 · Dec 10, 2019 · Dec 10, 2019 · Dec 10, 2019
commit 6c9d48c5da6fb94d10cbb206eb445349eeacb065
diff --git a/lib/torrent.js b/lib/torrent.js
@@ -113,6 +113,8 @@ class Torrent extends EventEmitter {
     // for cleanup
     this._servers = []
     this._xsRequests = []
+
+    this._writeCb = opts.writeCb,
 
     // TODO: remove this and expose a hook instead
     // optimization: don't recheck every file if it hasn't changed
@@ -1597,7 +1599,11 @@ class Torrent extends EventEmitter {
     if (!cb) cb = noop
 
     const readable = new MultiStream(streams)
+    let read = 0;
     const writable = new ChunkStoreWriteStream(this.store, this.pieceLength)
+    if (this._writeCb) {
+      writable._blockstream.on('data', data => this._writeCb(this.length, read += data.length))
+    }
 
     pump(readable, writable, err => {
       if (err) return cb(err)