peerflix-server/server/engine.js

Streaming torrent client for node.js with web ui.

'use strict';

var torrentStream = require('torrent-stream');
var _ = require('lodash');
var net = require('net');

var BITTORRENT_PORT = 6881;
var tryPort = new Promise(function (resolve) {
  function findPort(port) {
    var s = net.createServer();
    s.on('error', function() {
      findPort(0);
    });

    s.listen(port, function() {
      var port = s.address().port;
      s.close(function() {
        resolve(port);
      });
    });
  }
  findPort(BITTORRENT_PORT);
});

module.exports = function (torrent, opts) {
  var engine = torrentStream(torrent, _.cloneDeep(opts));

  engine.once('verifying', function () {
    var totalPieces = engine.torrent.pieces.length;
    var verifiedPieces = 0;

    console.log('verifying ' + engine.infoHash);
    engine.files.forEach(function (file, i) {
      console.log(i + ' ' + file.name);
    });

    engine.on('verify', function () {
      if (++verifiedPieces === totalPieces) {
        engine.emit('finished');
        console.log('finished ' + engine.infoHash);
      }
    });
  });

  engine.once('ready', function () {
    console.log('ready ' + engine.infoHash);
    engine.ready = true;

    // select the largest file
    var file = engine.files.reduce(function (a, b) {
      return a.length > b.length ? a : b;
    });
    file.select();
  });

  engine.on('uninterested', function () {
    console.log('uninterested ' + engine.infoHash);
  });

  engine.on('interested', function () {
    console.log('interested ' + engine.infoHash);
  });

  engine.on('idle', function () {
    console.log('idle ' + engine.infoHash);
  });

  engine.on('error', function (e) {
    console.log('error ' + engine.infoHash + ': ' + e);
  });

  engine.once('destroyed', function () {
    console.log('destroyed ' + engine.infoHash);
    engine.removeAllListeners();
  });

  tryPort.then(function (port) {
    engine.listen(port, function () {
      console.log('listening ' + engine.infoHash + ' on port ' + engine.port);
    });
  });

  return engine;
};