aboutsummaryrefslogtreecommitdiffhomepage
path: root/lib/webtorrent.js
blob: d1ca3c9f2fe62a784784416c2c38a8e3a36259ab (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
;(function () {
  'use strict'

  function webtorrent (args) {
    var WebTorrent = require('webtorrent')
    var ipc = require('node-ipc')

    if (args.length !== 3) {
      console.log('Wrong arguments number: ' + args.length + '/3')
      process.exit(-1)
    }

    var host = args[1]
    var port = args[2]
    var nodeKey = 'webtorrentnode' + port
    var processKey = 'webtorrent' + port

    ipc.config.silent = true
    ipc.config.id = processKey

    if (host === 'client' && port === '1') global.WEBTORRENT_ANNOUNCE = []
    else global.WEBTORRENT_ANNOUNCE = 'ws://' + host + ':' + port + '/tracker/socket'
    var wt = new WebTorrent({ dht: false })

    function seed (data) {
      var args = data.args
      var path = args.path
      var _id = data._id

      wt.seed(path, { announceList: '' }, function (torrent) {
        var to_send = {
          magnetUri: torrent.magnetURI
        }

        ipc.of[nodeKey].emit(nodeKey + '.seedDone.' + _id, to_send)
      })
    }

    function add (data) {
      var args = data.args
      var magnetUri = args.magnetUri
      var _id = data._id

      wt.add(magnetUri, function (torrent) {
        var to_send = {
          files: []
        }

        torrent.files.forEach(function (file) {
          to_send.files.push({ path: file.path })
        })

        ipc.of[nodeKey].emit(nodeKey + '.addDone.' + _id, to_send)
      })
    }

    function remove (data) {
      var args = data.args
      var magnetUri = args.magnetUri
      var _id = data._id

      try {
        wt.remove(magnetUri, callback)
      } catch (err) {
        console.log('Cannot remove the torrent from WebTorrent')
        return callback(null)
      }

      function callback () {
        var to_send = {}
        ipc.of[nodeKey].emit(nodeKey + '.removeDone.' + _id, to_send)
      }
    }

    console.log('Configuration: ' + host + ':' + port)
    console.log('Connecting to IPC...')

    ipc.connectTo(nodeKey, function () {
      ipc.of[nodeKey].on(processKey + '.seed', seed)
      ipc.of[nodeKey].on(processKey + '.add', add)
      ipc.of[nodeKey].on(processKey + '.remove', remove)

      ipc.of[nodeKey].emit(processKey + '.ready')
      console.log('Ready.')
    })

    process.on('uncaughtException', function (e) {
      ipc.of[nodeKey].emit(processKey + '.exception', { exception: e })
    })
  }

  // ---------------------------------------------------------------------------

  module.exports = webtorrent
})()