-;(function () {
- 'use strict'
+'use strict'
+
+var config = require('config')
+var ipc = require('node-ipc')
+var pathUtils = require('path')
+var spawn = require('electron-spawn')
+
+var logger = require('../helpers/logger')
+
+var host = config.get('webserver.host')
+var port = config.get('webserver.port')
+var nodeKey = 'webtorrentnode' + port
+var processKey = 'webtorrentprocess' + port
+ipc.config.silent = true
+ipc.config.id = nodeKey
+
+var webtorrent = {
+ add: add,
+ app: null, // Pid of the app
+ create: create,
+ remove: remove,
+ seed: seed,
+ silent: false // Useful for beautiful tests
+}
+
+function create (options, callback) {
+ if (typeof options === 'function') {
+ callback = options
+ options = {}
+ }
- function webtorrent (args) {
- var WebTorrent = require('webtorrent')
- var ipc = require('node-ipc')
+ // Override options
+ if (options.host) host = options.host
+ if (options.port) {
+ port = options.port
+ nodeKey = 'webtorrentnode' + port
+ processKey = 'webtorrentprocess' + port
+ ipc.config.id = nodeKey
+ }
- if (args.length !== 3) {
- console.log('Wrong arguments number: ' + args.length + '/3')
- process.exit(-1)
- }
+ ipc.serve(function () {
+ if (!webtorrent.silent) logger.info('IPC server ready.')
- var host = args[1]
- var port = args[2]
- var nodeKey = 'webtorrentnode' + port
- var processKey = 'webtorrent' + port
+ // Run a timeout of 30s after which we exit the process
+ var timeout_webtorrent_process = setTimeout(function () {
+ logger.error('Timeout : cannot run the webtorrent process. Please ensure you have electron-prebuilt npm package installed with xvfb-run.')
+ process.exit()
+ }, 30000)
- ipc.config.silent = true
- ipc.config.id = processKey
+ ipc.server.on(processKey + '.ready', function () {
+ if (!webtorrent.silent) logger.info('Webtorrent process ready.')
+ clearTimeout(timeout_webtorrent_process)
+ callback()
+ })
- if (host === 'client' && port === '1') global.WEBTORRENT_ANNOUNCE = []
- else global.WEBTORRENT_ANNOUNCE = 'ws://' + host + ':' + port + '/tracker/socket'
- var wt = new WebTorrent({ dht: false })
+ ipc.server.on(processKey + '.exception', function (data) {
+ logger.error('Received exception error from webtorrent process.', { exception: data.exception })
+ process.exit()
+ })
- function seed (data) {
- var args = data.args
- var path = args.path
- var _id = data._id
+ var webtorrent_process = spawn(pathUtils.join(__dirname, 'webtorrentProcess.js'), host, port, { detached: true })
+ webtorrent_process.stderr.on('data', function (data) {
+ // logger.debug('Webtorrent process stderr: ', data.toString())
+ })
- wt.seed(path, { announceList: '' }, function (torrent) {
- var to_send = {
- magnetUri: torrent.magnetURI
- }
+ webtorrent_process.stdout.on('data', function (data) {
+ // logger.debug('Webtorrent process:', data.toString())
+ })
- ipc.of[nodeKey].emit(nodeKey + '.seedDone.' + _id, to_send)
- })
- }
+ webtorrent.app = webtorrent_process
+ })
+
+ ipc.server.start()
+}
- function add (data) {
- var args = data.args
- var magnetUri = args.magnetUri
- var _id = data._id
+function seed (path, callback) {
+ var extension = pathUtils.extname(path)
+ var basename = pathUtils.basename(path, extension)
+ var data = {
+ _id: basename,
+ args: {
+ path: path
+ }
+ }
- wt.add(magnetUri, function (torrent) {
- var to_send = {
- files: []
- }
+ if (!webtorrent.silent) logger.debug('Node wants to seed %s.', data._id)
- torrent.files.forEach(function (file) {
- to_send.files.push({ path: file.path })
- })
+ // Finish signal
+ var event_key = nodeKey + '.seedDone.' + data._id
+ ipc.server.on(event_key, function listener (received) {
+ if (!webtorrent.silent) logger.debug('Process seeded torrent %s.', received.magnetUri)
- ipc.of[nodeKey].emit(nodeKey + '.addDone.' + _id, to_send)
- })
+ // This is a fake object, we just use the magnetUri in this project
+ var torrent = {
+ magnetURI: received.magnetUri
}
- 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)
- }
+ ipc.server.off(event_key)
+ callback(torrent)
+ })
+
+ ipc.server.broadcast(processKey + '.seed', data)
+}
+
+function add (magnetUri, callback) {
+ var data = {
+ _id: magnetUri,
+ args: {
+ magnetUri: magnetUri
}
+ }
- console.log('Configuration: ' + host + ':' + port)
- console.log('Connecting to IPC...')
+ if (!webtorrent.silent) logger.debug('Node wants to add ' + data._id)
- 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)
+ // Finish signal
+ var event_key = nodeKey + '.addDone.' + data._id
+ ipc.server.on(event_key, function (received) {
+ if (!webtorrent.silent) logger.debug('Process added torrent.')
- ipc.of[nodeKey].emit(processKey + '.ready')
- console.log('Ready.')
- })
+ // This is a fake object, we just use the magnetUri in this project
+ var torrent = {
+ files: received.files
+ }
- process.on('uncaughtException', function (e) {
- ipc.of[nodeKey].emit(processKey + '.exception', { exception: e })
- })
+ ipc.server.off(event_key)
+ callback(torrent)
+ })
+
+ ipc.server.broadcast(processKey + '.add', data)
+}
+
+function remove (magnetUri, callback) {
+ var data = {
+ _id: magnetUri,
+ args: {
+ magnetUri: magnetUri
+ }
}
- // ---------------------------------------------------------------------------
+ if (!webtorrent.silent) logger.debug('Node wants to stop seeding %s.', data._id)
+
+ // Finish signal
+ var event_key = nodeKey + '.removeDone.' + data._id
+ ipc.server.on(event_key, function (received) {
+ if (!webtorrent.silent) logger.debug('Process removed torrent %s.', data._id)
+
+ var err = null
+ if (received.err) err = received.err
+
+ ipc.server.off(event_key)
+ callback(err)
+ })
+
+ ipc.server.broadcast(processKey + '.remove', data)
+}
+
+// ---------------------------------------------------------------------------
- module.exports = webtorrent
-})()
+module.exports = webtorrent
projects / github / Chocobozzz / PeerTube.git / blobdiff