]>
git.immae.eu Git - github/Chocobozzz/PeerTube.git/blob - src/pods.js
4 var async
= require('async')
5 var config
= require('config')
7 var request
= require('request')
9 var constants
= require('./constants')
10 var logger
= require('./logger')
11 var PodsDB
= require('./database').PodsDB
12 var poolRequests
= require('./poolRequests')
13 var utils
= require('./utils')
17 var http
= config
.get('webserver.https') ? 'https' : 'http'
18 var host
= config
.get('webserver.host')
19 var port
= config
.get('webserver.port')
21 // ----------- Private functions -----------
23 function getForeignPodsList (url
, callback
) {
24 var path
= '/api/' + constants
.API_VERSION
+ '/pods'
26 request
.get(url
+ path
, function (err
, response
, body
) {
28 callback(JSON
.parse(body
))
32 // ----------- Public functions -----------
34 pods
.list = function (callback
) {
35 PodsDB
.find(function (err
, pods_list
) {
37 logger
.error('Cannot get the list of the pods.', { error: err
})
41 return callback(null, pods_list
)
46 // TODO: check if the pod is not already a friend
47 pods
.add = function (data
, callback
) {
48 var videos
= require('./videos')
49 logger
.info('Adding pod: %s', data
.url
)
53 publicKey: data
.publicKey
,
54 score: constants
.FRIEND_BASE_SCORE
57 PodsDB
.create(params
, function (err
, pod
) {
59 logger
.error('Cannot insert the pod.', { error: err
})
63 videos
.addRemotes(data
.videos
)
65 fs
.readFile(utils
.certDir
+ 'peertube.pub', 'utf8', function (err
, cert
) {
67 logger
.error('Cannot read cert file.', { error: err
})
71 videos
.listOwned(function (err
, videos_list
) {
73 logger
.error('Cannot get the list of owned videos.', { error: err
})
77 return callback(null, { cert: cert
, videos: videos_list
})
83 pods
.remove = function (url
, callback
) {
84 var videos
= require('./videos')
85 logger
.info('Removing %s pod.', url
)
87 videos
.removeAllRemotesOf(url
, function (err
) {
88 if (err
) logger
.error('Cannot remove all remote videos of %s.', url
)
90 PodsDB
.remove({ url: url
}, function (err
) {
91 if (err
) return callback(err
)
93 logger
.info('%s pod removed.', url
)
99 pods
.addVideoToFriends = function (video
) {
100 // To avoid duplicates
101 var id
= video
.name
+ video
.magnetUri
102 poolRequests
.addToPoolRequests(id
, 'add', video
)
105 pods
.removeVideoToFriends = function (video
) {
106 // To avoid duplicates
107 var id
= video
.name
+ video
.magnetUri
108 poolRequests
.addToPoolRequests(id
, 'remove', video
)
111 pods
.makeFriends = function (callback
) {
112 var videos
= require('./videos')
115 logger
.info('Make friends!')
116 fs
.readFile(utils
.certDir
+ 'peertube.pub', 'utf8', function (err
, cert
) {
118 logger
.error('Cannot read public cert.', { error: err
})
122 var urls
= config
.get('network.friends')
124 async
.each(urls
, computeForeignPodsList
, function () {
125 logger
.debug('Pods scores computed.', { pods_score: pods_score
})
126 var pods_list
= computeWinningPods(urls
, pods_score
)
127 logger
.debug('Pods that we keep computed.', { pods_to_keep: pods_list
})
129 makeRequestsToWinningPods(cert
, pods_list
)
133 // -----------------------------------------------------------------------
135 function computeForeignPodsList (url
, callback
) {
136 // Let's give 1 point to the pod we ask the friends list
139 getForeignPodsList(url
, function (foreign_pods_list
) {
140 if (foreign_pods_list
.length
=== 0) return callback()
142 async
.each(foreign_pods_list
, function (foreign_pod
, callback_each
) {
143 var foreign_url
= foreign_pod
.url
145 if (pods_score
[foreign_url
]) pods_score
[foreign_url
]++
146 else pods_score
[foreign_url
] = 1
155 function computeWinningPods (urls
, pods_score
) {
156 // Build the list of pods to add
157 // Only add a pod if it exists in more than a half base pods
159 var base_score
= urls
.length
/ 2
160 Object
.keys(pods_score
).forEach(function (pod
) {
161 if (pods_score
[pod
] > base_score
) pods_list
.push({ url: pod
})
167 function makeRequestsToWinningPods (cert
, pods_list
) {
168 // Stop pool requests
169 poolRequests
.deactivate()
170 // Flush pool requests
171 poolRequests
.forceSend()
173 // Get the list of our videos to send to our new friends
174 videos
.listOwned(function (err
, videos_list
) {
178 url: http
+ '://' + host
+ ':' + port
,
183 utils
.makeMultipleRetryRequest(
184 { method: 'POST', path: '/api/' + constants
.API_VERSION
+ '/pods/', data: data
},
188 function eachRequest (err
, response
, body
, url
, pod
, callback_each_request
) {
189 // We add the pod if it responded correctly with its public certificate
190 if (!err
&& response
.statusCode
=== 200) {
191 pods
.add({ url: pod
.url
, publicKey: body
.cert
, score: constants
.FRIEND_BASE_SCORE
}, function (err
) {
192 if (err
) logger
.error('Error with adding %s pod.', pod
.url
, { error: err
})
194 videos
.addRemotes(body
.videos
, function (err
) {
195 if (err
) logger
.error('Error with adding videos of pod.', pod
.url
, { error: err
})
197 logger
.debug('Adding remote videos from %s.', pod
.url
, { videos: body
.videos
})
198 return callback_each_request()
202 logger
.error('Error with adding %s pod.', pod
.url
, { error: err
|| new Error('Status not 200') })
203 return callback_each_request()
207 function endRequests (err
) {
208 // Now we made new friends, we can re activate the pool of requests
209 poolRequests
.activate()
212 logger
.error('There was some errors when we wanted to make friends.', { error: err
})
216 logger
.debug('makeRequestsToWinningPods finished.')
217 return callback(null)
224 pods
.quitFriends = function (callback
) {
225 // Stop pool requests
226 poolRequests
.deactivate()
227 // Flush pool requests
228 poolRequests
.forceSend()
230 PodsDB
.find(function (err
, pods
) {
231 if (err
) return callback(err
)
235 path: '/api/' + constants
.API_VERSION
+ '/pods/remove',
239 url: 'me' // Fake data
243 // Announce we quit them
244 utils
.makeMultipleRetryRequest(request
, pods
, function () {
245 PodsDB
.remove(function (err
) {
246 poolRequests
.activate()
248 if (err
) return callback(err
)
250 logger
.info('Broke friends, so sad :(')
252 var videos
= require('./videos')
253 videos
.removeAllRemotes(function (err
) {
254 if (err
) return callback(err
)
256 logger
.info('Removed all remote videos.')
264 pods
.hasFriends = function (callback
) {
265 PodsDB
.count(function (err
, count
) {
266 if (err
) return callback(err
)
268 var has_friends
= (count
!== 0)
269 callback(null, has_friends
)
273 module
.exports
= pods