mirror of https://github.com/Chocobozzz/PeerTube
93 lines
2.3 KiB
JavaScript
93 lines
2.3 KiB
JavaScript
'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 })
|
|
})
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
module.exports = webtorrent
|