mirror of https://github.com/Chocobozzz/PeerTube
107 lines
3.1 KiB
JavaScript
107 lines
3.1 KiB
JavaScript
'use strict'
|
|
|
|
const async = require('async')
|
|
const config = require('config')
|
|
const request = require('request')
|
|
const replay = require('request-replay')
|
|
|
|
const constants = require('../initializers/constants')
|
|
const logger = require('./logger')
|
|
const peertubeCrypto = require('./peertubeCrypto')
|
|
|
|
const http = config.get('webserver.https') ? 'https' : 'http'
|
|
const host = config.get('webserver.host')
|
|
const port = config.get('webserver.port')
|
|
|
|
const requests = {
|
|
makeMultipleRetryRequest: makeMultipleRetryRequest
|
|
}
|
|
|
|
function makeMultipleRetryRequest (all_data, pods, callbackEach, callback) {
|
|
if (!callback) {
|
|
callback = callbackEach
|
|
callbackEach = null
|
|
}
|
|
|
|
const url = http + '://' + host + ':' + port
|
|
let signature
|
|
|
|
// Add signature if it is specified in the params
|
|
if (all_data.method === 'POST' && all_data.data && all_data.sign === true) {
|
|
signature = peertubeCrypto.sign(url)
|
|
}
|
|
|
|
// Make a request for each pod
|
|
async.each(pods, function (pod, callback_each_async) {
|
|
function callbackEachRetryRequest (err, response, body, url, pod) {
|
|
if (callbackEach !== null) {
|
|
callbackEach(err, response, body, url, pod, function () {
|
|
callback_each_async()
|
|
})
|
|
} else {
|
|
callback_each_async()
|
|
}
|
|
}
|
|
|
|
const params = {
|
|
url: pod.url + all_data.path,
|
|
method: all_data.method
|
|
}
|
|
|
|
// Add data with POST requst ?
|
|
if (all_data.method === 'POST' && all_data.data) {
|
|
// Encrypt data ?
|
|
if (all_data.encrypt === true) {
|
|
peertubeCrypto.encrypt(pod.publicKey, JSON.stringify(all_data.data), function (err, encrypted) {
|
|
if (err) return callback(err)
|
|
|
|
params.json = {
|
|
data: encrypted.data,
|
|
key: encrypted.key
|
|
}
|
|
|
|
makeRetryRequest(params, url, pod, signature, callbackEachRetryRequest)
|
|
})
|
|
} else {
|
|
params.json = { data: all_data.data }
|
|
makeRetryRequest(params, url, pod, signature, callbackEachRetryRequest)
|
|
}
|
|
} else {
|
|
makeRetryRequest(params, url, pod, signature, callbackEachRetryRequest)
|
|
}
|
|
}, callback)
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
module.exports = requests
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
function makeRetryRequest (params, from_url, to_pod, signature, callbackEach) {
|
|
// Append the signature
|
|
if (signature) {
|
|
params.json.signature = {
|
|
url: from_url,
|
|
signature: signature
|
|
}
|
|
}
|
|
|
|
logger.debug('Make retry requests to %s.', to_pod.url)
|
|
|
|
replay(
|
|
request.post(params, function (err, response, body) {
|
|
callbackEach(err, response, body, params.url, to_pod)
|
|
}),
|
|
{
|
|
retries: constants.REQUEST_RETRIES,
|
|
factor: 3,
|
|
maxTimeout: Infinity,
|
|
errorCodes: [ 'EADDRINFO', 'ETIMEDOUT', 'ECONNRESET', 'ESOCKETTIMEDOUT', 'ENOTFOUND', 'ECONNREFUSED' ]
|
|
}
|
|
).on('replay', function (replay) {
|
|
logger.info('Replaying request to %s. Request failed: %d %s. Replay number: #%d. Will retry in: %d ms.',
|
|
params.url, replay.error.code, replay.error.message, replay.number, replay.delay)
|
|
})
|
|
}
|