server/helpers/requests.js

   1 'use strict'
   2
   3 const async = require('async')
   4 const config = require('config')
   5 const request = require('request')
   6 const replay = require('request-replay')
   7
   8 const constants = require('../initializers/constants')
   9 const logger = require('./logger')
  10 const peertubeCrypto = require('./peertubeCrypto')
  11
  12 const http = config.get('webserver.https') ? 'https' : 'http'
  13 const host = config.get('webserver.host')
  14 const port = config.get('webserver.port')
  15
  16 const requests = {
  17   makeMultipleRetryRequest: makeMultipleRetryRequest
  18 }
  19
  20 function makeMultipleRetryRequest (allData, pods, callbackEach, callback) {
  21   if (!callback) {
  22     callback = callbackEach
  23     callbackEach = null
  24   }
  25
  26   const url = http + '://' + host + ':' + port
  27   let signature
  28
  29   // Add signature if it is specified in the params
  30   if (allData.method === 'POST' && allData.data && allData.sign === true) {
  31     signature = peertubeCrypto.sign(url)
  32   }
  33
  34   // Make a request for each pod
  35   async.each(pods, function (pod, callbackEachAsync) {
  36     function callbackEachRetryRequest (err, response, body, url, pod) {
  37       if (callbackEach !== null) {
  38         callbackEach(err, response, body, url, pod, function () {
  39           callbackEachAsync()
  40         })
  41       } else {
  42         callbackEachAsync()
  43       }
  44     }
  45
  46     const params = {
  47       url: pod.url + allData.path,
  48       method: allData.method
  49     }
  50
  51     // Add data with POST requst ?
  52     if (allData.method === 'POST' && allData.data) {
  53       // Encrypt data ?
  54       if (allData.encrypt === true) {
  55         peertubeCrypto.encrypt(pod.publicKey, JSON.stringify(allData.data), function (err, encrypted) {
  56           if (err) return callback(err)
  57
  58           params.json = {
  59             data: encrypted.data,
  60             key: encrypted.key
  61           }
  62
  63           makeRetryRequest(params, url, pod, signature, callbackEachRetryRequest)
  64         })
  65       } else {
  66         params.json = { data: allData.data }
  67         makeRetryRequest(params, url, pod, signature, callbackEachRetryRequest)
  68       }
  69     } else {
  70       makeRetryRequest(params, url, pod, signature, callbackEachRetryRequest)
  71     }
  72   }, callback)
  73 }
  74
  75 // ---------------------------------------------------------------------------
  76
  77 module.exports = requests
  78
  79 // ---------------------------------------------------------------------------
  80
  81 function makeRetryRequest (params, fromUrl, toPod, signature, callbackEach) {
  82   // Append the signature
  83   if (signature) {
  84     params.json.signature = {
  85       url: fromUrl,
  86       signature: signature
  87     }
  88   }
  89
  90   logger.debug('Make retry requests to %s.', toPod.url)
  91
  92   replay(
  93     request.post(params, function (err, response, body) {
  94       callbackEach(err, response, body, params.url, toPod)
  95     }),
  96     {
  97       retries: constants.REQUEST_RETRIES,
  98       factor: 3,
  99       maxTimeout: Infinity,
 100       errorCodes: [ 'EADDRINFO', 'ETIMEDOUT', 'ECONNRESET', 'ESOCKETTIMEDOUT', 'ENOTFOUND', 'ECONNREFUSED' ]
 101     }
 102   ).on('replay', function (replay) {
 103     logger.info('Replaying request to %s. Request failed: %d %s. Replay number: #%d. Will retry in: %d ms.',
 104       params.url, replay.error.code, replay.error.message, replay.number, replay.delay)
 105   })
 106 }