aboutsummaryrefslogtreecommitdiffhomepage
path: root/server/lib
diff options
context:
space:
mode:
authorChocobozzz <florian.bigard@gmail.com>2016-07-26 22:30:46 +0200
committerChocobozzz <florian.bigard@gmail.com>2016-07-27 21:21:57 +0200
commit052937db8a8d282eccdbdf38d487ed8d85d3c0a7 (patch)
treeda84e08fbe3569c2122c9b733e22768980750744 /server/lib
parent71d3476b825571376b06498326d138685493fb62 (diff)
downloadPeerTube-052937db8a8d282eccdbdf38d487ed8d85d3c0a7.tar.gz
PeerTube-052937db8a8d282eccdbdf38d487ed8d85d3c0a7.tar.zst
PeerTube-052937db8a8d282eccdbdf38d487ed8d85d3c0a7.zip
First draft using only webseed for server
Diffstat (limited to 'server/lib')
-rw-r--r--server/lib/webtorrent-process.js92
-rw-r--r--server/lib/webtorrent.js161
2 files changed, 0 insertions, 253 deletions
diff --git a/server/lib/webtorrent-process.js b/server/lib/webtorrent-process.js
deleted file mode 100644
index be7ac5bb4..000000000
--- a/server/lib/webtorrent-process.js
+++ /dev/null
@@ -1,92 +0,0 @@
1'use strict'
2
3const WebTorrent = require('webtorrent')
4const ipc = require('node-ipc')
5
6function webtorrent (args) {
7 if (args.length !== 3) {
8 throw new Error('Wrong arguments number: ' + args.length + '/3')
9 }
10
11 const host = args[1]
12 const port = args[2]
13 const nodeKey = 'webtorrentnode' + port
14 const processKey = 'webtorrentprocess' + port
15
16 ipc.config.silent = true
17 ipc.config.id = processKey
18
19 if (host === 'client' && port === '1') global.WEBTORRENT_ANNOUNCE = []
20 else global.WEBTORRENT_ANNOUNCE = 'ws://' + host + ':' + port + '/tracker/socket'
21 const wt = new WebTorrent({ dht: false })
22
23 function seed (data) {
24 const args = data.args
25 const path = args.path
26 const _id = data._id
27
28 wt.seed(path, { announceList: '' }, function (torrent) {
29 const toSend = {
30 magnetUri: torrent.magnetURI
31 }
32
33 ipc.of[nodeKey].emit(nodeKey + '.seedDone.' + _id, toSend)
34 })
35 }
36
37 function add (data) {
38 const args = data.args
39 const magnetUri = args.magnetUri
40 const _id = data._id
41
42 wt.add(magnetUri, function (torrent) {
43 const toSend = {
44 files: []
45 }
46
47 torrent.files.forEach(function (file) {
48 toSend.files.push({ path: file.path })
49 })
50
51 ipc.of[nodeKey].emit(nodeKey + '.addDone.' + _id, toSend)
52 })
53 }
54
55 function remove (data) {
56 const args = data.args
57 const magnetUri = args.magnetUri
58 const _id = data._id
59
60 try {
61 wt.remove(magnetUri, callback)
62 } catch (err) {
63 console.log('Cannot remove the torrent from WebTorrent.')
64 return callback(null)
65 }
66
67 function callback () {
68 const toSend = {}
69 ipc.of[nodeKey].emit(nodeKey + '.removeDone.' + _id, toSend)
70 }
71 }
72
73 console.log('Configuration: ' + host + ':' + port)
74 console.log('Connecting to IPC...')
75
76 ipc.connectTo(nodeKey, function () {
77 ipc.of[nodeKey].on(processKey + '.seed', seed)
78 ipc.of[nodeKey].on(processKey + '.add', add)
79 ipc.of[nodeKey].on(processKey + '.remove', remove)
80
81 ipc.of[nodeKey].emit(processKey + '.ready')
82 console.log('Ready.')
83 })
84
85 process.on('uncaughtException', function (e) {
86 ipc.of[nodeKey].emit(processKey + '.exception', { exception: e.toString() })
87 })
88}
89
90// ---------------------------------------------------------------------------
91
92module.exports = webtorrent
diff --git a/server/lib/webtorrent.js b/server/lib/webtorrent.js
deleted file mode 100644
index bcd30139e..000000000
--- a/server/lib/webtorrent.js
+++ /dev/null
@@ -1,161 +0,0 @@
1'use strict'
2
3const config = require('config')
4const ipc = require('node-ipc')
5const pathUtils = require('path')
6const spawn = require('electron-spawn')
7
8const logger = require('../helpers/logger')
9
10const electronDebug = config.get('electron.debug')
11let host = config.get('webserver.host')
12let port = config.get('webserver.port')
13let nodeKey = 'webtorrentnode' + port
14let processKey = 'webtorrentprocess' + port
15ipc.config.silent = true
16ipc.config.id = nodeKey
17
18const webtorrent = {
19 add: add,
20 app: null, // Pid of the app
21 create: create,
22 remove: remove,
23 seed: seed,
24 silent: false // Useful for beautiful tests
25}
26
27function create (options, callback) {
28 if (typeof options === 'function') {
29 callback = options
30 options = {}
31 }
32
33 // Override options
34 if (options.host) host = options.host
35 if (options.port) {
36 port = options.port
37 nodeKey = 'webtorrentnode' + port
38 processKey = 'webtorrentprocess' + port
39 ipc.config.id = nodeKey
40 }
41
42 ipc.serve(function () {
43 if (!webtorrent.silent) logger.info('IPC server ready.')
44
45 // Run a timeout of 30s after which we exit the process
46 const timeoutWebtorrentProcess = setTimeout(function () {
47 throw new Error('Timeout : cannot run the webtorrent process. Please ensure you have electron-prebuilt npm package installed with xvfb-run.')
48 }, 30000)
49
50 ipc.server.on(processKey + '.ready', function () {
51 if (!webtorrent.silent) logger.info('Webtorrent process ready.')
52 clearTimeout(timeoutWebtorrentProcess)
53 callback()
54 })
55
56 ipc.server.on(processKey + '.exception', function (data) {
57 throw new Error('Received exception error from webtorrent process : ' + data.exception)
58 })
59
60 const webtorrentProcess = spawn(pathUtils.join(__dirname, 'webtorrent-process.js'), host, port, { detached: true })
61
62 if (electronDebug === true) {
63 webtorrentProcess.stderr.on('data', function (data) {
64 logger.debug('Webtorrent process stderr: ', data.toString())
65 })
66
67 webtorrentProcess.stdout.on('data', function (data) {
68 logger.debug('Webtorrent process:', data.toString())
69 })
70 }
71
72 webtorrent.app = webtorrentProcess
73 })
74
75 ipc.server.start()
76}
77
78function seed (path, callback) {
79 const extension = pathUtils.extname(path)
80 const basename = pathUtils.basename(path, extension)
81 const data = {
82 _id: basename,
83 args: {
84 path: path
85 }
86 }
87
88 if (!webtorrent.silent) logger.debug('Node wants to seed %s.', data._id)
89
90 // Finish signal
91 const eventKey = nodeKey + '.seedDone.' + data._id
92 ipc.server.on(eventKey, function listener (received) {
93 if (!webtorrent.silent) logger.debug('Process seeded torrent %s.', received.magnetUri)
94
95 // This is a fake object, we just use the magnetUri in this project
96 const torrent = {
97 magnetURI: received.magnetUri
98 }
99
100 ipc.server.off(eventKey, '*')
101 callback(torrent)
102 })
103
104 ipc.server.broadcast(processKey + '.seed', data)
105}
106
107function add (magnetUri, callback) {
108 const data = {
109 _id: magnetUri,
110 args: {
111 magnetUri: magnetUri
112 }
113 }
114
115 if (!webtorrent.silent) logger.debug('Node wants to add ' + data._id)
116
117 // Finish signal
118 const eventKey = nodeKey + '.addDone.' + data._id
119 ipc.server.on(eventKey, function (received) {
120 if (!webtorrent.silent) logger.debug('Process added torrent.')
121
122 // This is a fake object, we just use the magnetUri in this project
123 const torrent = {
124 files: received.files
125 }
126
127 ipc.server.off(eventKey, '*')
128 callback(torrent)
129 })
130
131 ipc.server.broadcast(processKey + '.add', data)
132}
133
134function remove (magnetUri, callback) {
135 const data = {
136 _id: magnetUri,
137 args: {
138 magnetUri: magnetUri
139 }
140 }
141
142 if (!webtorrent.silent) logger.debug('Node wants to stop seeding %s.', data._id)
143
144 // Finish signal
145 const eventKey = nodeKey + '.removeDone.' + data._id
146 ipc.server.on(eventKey, function (received) {
147 if (!webtorrent.silent) logger.debug('Process removed torrent %s.', data._id)
148
149 let err = null
150 if (received.err) err = received.err
151
152 ipc.server.off(eventKey, '*')
153 callback(err)
154 })
155
156 ipc.server.broadcast(processKey + '.remove', data)
157}
158
159// ---------------------------------------------------------------------------
160
161module.exports = webtorrent