add torrent.createServer() method

1 files changed, 4 insertions, 0 deletions

diff --git a/lib/storage.js b/lib/storage.js
index 69f4bd5..5c96402 100644
--- a/lib/storage.js
+++ b/lib/storage.js
@@ -10,6 +10,7 @@ var extend = require('extend.js')
 var FileStream = require('./file-stream')
 var inherits = require('inherits')
 var MultiStream = require('multistream')
+var once = require('once')
 var sha1 = require('git-sha1')
 var stream = require('stream')
 
@@ -325,6 +326,9 @@ Storage.BLOCK_LENGTH = BLOCK_LENGTH
 Storage.prototype.load = function (streams, cb) {
   var self = this
   if (!Array.isArray(streams)) streams = [ streams ]
+  if (!cb) cb = function () {}
+  cb = once(cb)
+
   var pieceIndex = 0
   ;(new MultiStream(streams))
     .pipe(new BlockStream(self.pieceLength, { nopad: true }))