]>
Commit | Line | Data |
---|---|---|
1 | ;(function () { | |
2 | 'use strict' | |
3 | ||
4 | function webtorrent (args) { | |
5 | var WebTorrent = require('webtorrent') | |
6 | var ipc = require('node-ipc') | |
7 | ||
8 | if (args.length !== 3) { | |
9 | console.log('Wrong arguments number: ' + args.length + '/3') | |
10 | process.exit(-1) | |
11 | } | |
12 | ||
13 | var host = args[1] | |
14 | var port = args[2] | |
15 | var nodeKey = 'webtorrentnode' + port | |
16 | var processKey = 'webtorrent' + port | |
17 | ||
18 | ipc.config.silent = true | |
19 | ipc.config.id = processKey | |
20 | ||
21 | if (host === 'client' && port === '1') global.WEBTORRENT_ANNOUNCE = [] | |
22 | else global.WEBTORRENT_ANNOUNCE = 'ws://' + host + ':' + port + '/tracker/socket' | |
23 | var wt = new WebTorrent({ dht: false }) | |
24 | ||
25 | function seed (data) { | |
26 | var args = data.args | |
27 | var path = args.path | |
28 | var _id = data._id | |
29 | ||
30 | wt.seed(path, { announceList: '' }, function (torrent) { | |
31 | var to_send = { | |
32 | magnetUri: torrent.magnetURI | |
33 | } | |
34 | ||
35 | ipc.of[nodeKey].emit(nodeKey + '.seedDone.' + _id, to_send) | |
36 | }) | |
37 | } | |
38 | ||
39 | function add (data) { | |
40 | var args = data.args | |
41 | var magnetUri = args.magnetUri | |
42 | var _id = data._id | |
43 | ||
44 | wt.add(magnetUri, function (torrent) { | |
45 | var to_send = { | |
46 | files: [] | |
47 | } | |
48 | ||
49 | torrent.files.forEach(function (file) { | |
50 | to_send.files.push({ path: file.path }) | |
51 | }) | |
52 | ||
53 | ipc.of[nodeKey].emit(nodeKey + '.addDone.' + _id, to_send) | |
54 | }) | |
55 | } | |
56 | ||
57 | function remove (data) { | |
58 | var args = data.args | |
59 | var magnetUri = args.magnetUri | |
60 | var _id = data._id | |
61 | ||
62 | try { | |
63 | wt.remove(magnetUri, callback) | |
64 | } catch (err) { | |
65 | console.log('Cannot remove the torrent from WebTorrent.') | |
66 | return callback(null) | |
67 | } | |
68 | ||
69 | function callback () { | |
70 | var to_send = {} | |
71 | ipc.of[nodeKey].emit(nodeKey + '.removeDone.' + _id, to_send) | |
72 | } | |
73 | } | |
74 | ||
75 | console.log('Configuration: ' + host + ':' + port) | |
76 | console.log('Connecting to IPC...') | |
77 | ||
78 | ipc.connectTo(nodeKey, function () { | |
79 | ipc.of[nodeKey].on(processKey + '.seed', seed) | |
80 | ipc.of[nodeKey].on(processKey + '.add', add) | |
81 | ipc.of[nodeKey].on(processKey + '.remove', remove) | |
82 | ||
83 | ipc.of[nodeKey].emit(processKey + '.ready') | |
84 | console.log('Ready.') | |
85 | }) | |
86 | ||
87 | process.on('uncaughtException', function (e) { | |
88 | ipc.of[nodeKey].emit(processKey + '.exception', { exception: e }) | |
89 | }) | |
90 | } | |
91 | ||
92 | // --------------------------------------------------------------------------- | |
93 | ||
94 | module.exports = webtorrent | |
95 | })() |