aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/lib/webtorrentProcess.js
blob: 7889e7128fb3174bc4cd9f43a9b29c71b39e41d8 (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
'use strict'

const WebTorrent = require('webtorrent')
const ipc = require('node-ipc')

function webtorrent (args) {
  if (args.length !== 3) {
    throw new Error('Wrong arguments number: ' + args.length + '/3')
  }

  const host = args[1]
  const port = args[2]
  const nodeKey = 'webtorrentnode' + port
  const processKey = 'webtorrentprocess' + 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'
  const wt = new WebTorrent({ dht: false })

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

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

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

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

    wt.add(magnetUri, function (torrent) {
      const 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) {
    const args = data.args
    const magnetUri = args.magnetUri
    const _id = data._id

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

    function callback () {
      const 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.toString() })
  })
}

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

module.exports = webtorrent