aboutsummaryrefslogtreecommitdiffhomepage
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rw-r--r--lib/friends.js9
-rw-r--r--lib/poolRequests.js30
2 files changed, 20 insertions, 19 deletions
diff --git a/lib/friends.js b/lib/friends.js
index e093c85c4..b0086a38b 100644
--- a/lib/friends.js
+++ b/lib/friends.js
@@ -8,10 +8,11 @@
8 8
9 var constants = require('../initializers/constants') 9 var constants = require('../initializers/constants')
10 var logger = require('../helpers/logger') 10 var logger = require('../helpers/logger')
11 var peertubeCrypto = require('../helpers/peertubeCrypto')
11 var Pods = require('../models/pods') 12 var Pods = require('../models/pods')
12 var PoolRequests = require('../models/poolRequests') 13 var PoolRequests = require('../models/poolRequests')
13 var poolRequests = require('../lib/poolRequests') 14 var poolRequests = require('../lib/poolRequests')
14 var utils = require('../helpers/utils') 15 var requests = require('../helpers/requests')
15 var Videos = require('../models/videos') 16 var Videos = require('../models/videos')
16 17
17 var http = config.get('webserver.https') ? 'https' : 'http' 18 var http = config.get('webserver.https') ? 'https' : 'http'
@@ -48,7 +49,7 @@
48 var pods_score = {} 49 var pods_score = {}
49 50
50 logger.info('Make friends!') 51 logger.info('Make friends!')
51 fs.readFile(utils.getCertDir() + 'peertube.pub', 'utf8', function (err, cert) { 52 fs.readFile(peertubeCrypto.getCertDir() + 'peertube.pub', 'utf8', function (err, cert) {
52 if (err) { 53 if (err) {
53 logger.error('Cannot read public cert.', { error: err }) 54 logger.error('Cannot read public cert.', { error: err })
54 return callback(err) 55 return callback(err)
@@ -115,7 +116,7 @@
115 videos: videos_list 116 videos: videos_list
116 } 117 }
117 118
118 utils.makeMultipleRetryRequest( 119 requests.makeMultipleRetryRequest(
119 { method: 'POST', path: '/api/' + constants.API_VERSION + '/pods/', data: data }, 120 { method: 'POST', path: '/api/' + constants.API_VERSION + '/pods/', data: data },
120 121
121 pods_list, 122 pods_list,
@@ -176,7 +177,7 @@
176 } 177 }
177 178
178 // Announce we quit them 179 // Announce we quit them
179 utils.makeMultipleRetryRequest(request, pods, function () { 180 requests.makeMultipleRetryRequest(request, pods, function () {
180 Pods.removeAll(function (err) { 181 Pods.removeAll(function (err) {
181 poolRequests.activate() 182 poolRequests.activate()
182 183
diff --git a/lib/poolRequests.js b/lib/poolRequests.js
index 796f06149..f4ab434ad 100644
--- a/lib/poolRequests.js
+++ b/lib/poolRequests.js
@@ -8,7 +8,7 @@
8 var logger = require('../helpers/logger') 8 var logger = require('../helpers/logger')
9 var Pods = require('../models/pods') 9 var Pods = require('../models/pods')
10 var PoolRequests = require('../models/poolRequests') 10 var PoolRequests = require('../models/poolRequests')
11 var utils = require('../helpers/utils') 11 var requests = require('../helpers/requests')
12 var Videos = require('../models/videos') 12 var Videos = require('../models/videos')
13 13
14 var timer = null 14 var timer = null
@@ -40,7 +40,7 @@
40 40
41 // --------------------------------------------------------------------------- 41 // ---------------------------------------------------------------------------
42 42
43 function makePoolRequest (type, requests, callback) { 43 function makePoolRequest (type, requests_to_make, callback) {
44 if (!callback) callback = function () {} 44 if (!callback) callback = function () {}
45 45
46 Pods.list(function (err, pods) { 46 Pods.list(function (err, pods) {
@@ -51,7 +51,7 @@
51 sign: true, 51 sign: true,
52 method: 'POST', 52 method: 'POST',
53 path: null, 53 path: null,
54 data: requests 54 data: requests_to_make
55 } 55 }
56 56
57 if (type === 'add') { 57 if (type === 'add') {
@@ -65,7 +65,7 @@
65 var bad_pods = [] 65 var bad_pods = []
66 var good_pods = [] 66 var good_pods = []
67 67
68 utils.makeMultipleRetryRequest(params, pods, callbackEachPodFinished, callbackAllPodsFinished) 68 requests.makeMultipleRetryRequest(params, pods, callbackEachPodFinished, callbackAllPodsFinished)
69 69
70 function callbackEachPodFinished (err, response, body, url, pod, callback_each_pod_finished) { 70 function callbackEachPodFinished (err, response, body, url, pod, callback_each_pod_finished) {
71 if (err || (response.statusCode !== 200 && response.statusCode !== 204)) { 71 if (err || (response.statusCode !== 200 && response.statusCode !== 204)) {
@@ -95,7 +95,7 @@
95 95
96 if (pool_requests.length === 0) return 96 if (pool_requests.length === 0) return
97 97
98 var requests = { 98 var requests_to_make = {
99 add: { 99 add: {
100 ids: [], 100 ids: [],
101 requests: [] 101 requests: []
@@ -108,11 +108,11 @@
108 108
109 async.each(pool_requests, function (pool_request, callback_each) { 109 async.each(pool_requests, function (pool_request, callback_each) {
110 if (pool_request.type === 'add') { 110 if (pool_request.type === 'add') {
111 requests.add.requests.push(pool_request.request) 111 requests_to_make.add.requests.push(pool_request.request)
112 requests.add.ids.push(pool_request._id) 112 requests_to_make.add.ids.push(pool_request._id)
113 } else if (pool_request.type === 'remove') { 113 } else if (pool_request.type === 'remove') {
114 requests.remove.requests.push(pool_request.request) 114 requests_to_make.remove.requests.push(pool_request.request)
115 requests.remove.ids.push(pool_request._id) 115 requests_to_make.remove.ids.push(pool_request._id)
116 } else { 116 } else {
117 throw new Error('Unkown pool request type.') 117 throw new Error('Unkown pool request type.')
118 } 118 }
@@ -120,20 +120,20 @@
120 callback_each() 120 callback_each()
121 }, function () { 121 }, function () {
122 // Send the add requests 122 // Send the add requests
123 if (requests.add.requests.length !== 0) { 123 if (requests_to_make.add.requests.length !== 0) {
124 makePoolRequest('add', requests.add.requests, function (err) { 124 makePoolRequest('add', requests_to_make.add.requests, function (err) {
125 if (err) logger.error('Errors when sent add pool requests.', { error: err }) 125 if (err) logger.error('Errors when sent add pool requests.', { error: err })
126 126
127 PoolRequests.removeRequests(requests.add.ids) 127 PoolRequests.removeRequests(requests_to_make.add.ids)
128 }) 128 })
129 } 129 }
130 130
131 // Send the remove requests 131 // Send the remove requests
132 if (requests.remove.requests.length !== 0) { 132 if (requests_to_make.remove.requests.length !== 0) {
133 makePoolRequest('remove', requests.remove.requests, function (err) { 133 makePoolRequest('remove', requests_to_make.remove.requests, function (err) {
134 if (err) logger.error('Errors when sent remove pool requests.', { error: err }) 134 if (err) logger.error('Errors when sent remove pool requests.', { error: err })
135 135
136 PoolRequests.removeRequests(requests.remove.ids) 136 PoolRequests.removeRequests(requests_to_make.remove.ids)
137 }) 137 })
138 } 138 }
139 }) 139 })