support streaming into video tag

close #144

1 files changed, 18 insertions, 0 deletions

diff --git a/lib/file-stream.js b/lib/file-stream.js
index 6d86ea8..cbf1319 100644
--- a/lib/file-stream.js
+++ b/lib/file-stream.js
@@ -2,7 +2,9 @@ module.exports = FileStream
 
 var debug = require('debug')('webtorrent:file-stream')
 var inherits = require('inherits')
+var path = require('path')
 var stream = require('stream')
+var VideoStream = require('./video-stream')
 
 inherits(FileStream, stream.Readable)
 
@@ -32,6 +34,7 @@ function FileStream (file, opts) {
   self.startPiece = offset / pieceLength | 0
   self.endPiece = (opts.end + file.offset) / pieceLength | 0
 
+  self._extname = path.extname(file.name)
   self._storage = file.storage
   self._piece = self.startPiece
   self._missing = self.length
@@ -92,6 +95,21 @@ FileStream.prototype.notify = function () {
   })
 }
 
+FileStream.prototype.pipe = function (dst) {
+  var self = this
+  var pipe = stream.Readable.prototype.pipe
+
+  if (dst && dst.nodeName === 'VIDEO') { // <video> tag
+    var type = self._extname === '.webm'
+      ? 'video/webm; codecs="vorbis,vp8"'
+      : self._extname === '.mp4'
+        ? 'video/mp4; codecs="avc1.42c01e,mp4a.40.2"'
+        : undefined
+    return pipe.call(self, new VideoStream(dst, { type: type }))
+  } else
+    return pipe.call(self, dst)
+}
+
 FileStream.prototype.destroy = function () {
   var self = this
   if (self._destroyed) return