server/helpers/requests.ts

   1 import * as replay from 'request-replay'
   2 import * as request from 'request'
   3 import * as Promise from 'bluebird'
   4
   5 import {
   6   RETRY_REQUESTS,
   7   REMOTE_SCHEME,
   8   CONFIG
   9 } from '../initializers'
  10 import { PodInstance } from '../models'
  11 import { PodSignature } from '../../shared'
  12 import { signObject } from './peertube-crypto'
  13 import { createWriteStream } from 'fs'
  14
  15 function doRequest (requestOptions: request.CoreOptions & request.UriOptions) {
  16   return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
  17     request(requestOptions, (err, response, body) => err ? rej(err) : res({ response, body }))
  18   })
  19 }
  20
  21 function doRequestAndSaveToFile (requestOptions: request.CoreOptions & request.UriOptions, destPath: string) {
  22   return new Promise<request.RequestResponse>((res, rej) => {
  23     request(requestOptions)
  24       .on('response', response => res(response as request.RequestResponse))
  25       .on('error', err => rej(err))
  26       .pipe(createWriteStream(destPath))
  27   })
  28 }
  29
  30 type MakeRetryRequestParams = {
  31   url: string,
  32   method: 'GET' | 'POST',
  33   json: Object
  34 }
  35 function makeRetryRequest (params: MakeRetryRequestParams) {
  36   return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
  37     replay(
  38       request(params, (err, response, body) => err ? rej(err) : res({ response, body })),
  39       {
  40         retries: RETRY_REQUESTS,
  41         factor: 3,
  42         maxTimeout: Infinity,
  43         errorCodes: [ 'EADDRINFO', 'ETIMEDOUT', 'ECONNRESET', 'ESOCKETTIMEDOUT', 'ENOTFOUND', 'ECONNREFUSED' ]
  44       }
  45     )
  46   })
  47 }
  48
  49 type MakeSecureRequestParams = {
  50   toPod: PodInstance
  51   path: string
  52   data?: Object
  53 }
  54 function makeSecureRequest (params: MakeSecureRequestParams) {
  55   const requestParams: {
  56     method: 'POST',
  57     uri: string,
  58     json: {
  59       signature: PodSignature,
  60       data: any
  61     }
  62   } = {
  63     method: 'POST',
  64     uri: REMOTE_SCHEME.HTTP + '://' + params.toPod.host + params.path,
  65     json: {
  66       signature: null,
  67       data: null
  68     }
  69   }
  70
  71   const host = CONFIG.WEBSERVER.HOST
  72
  73   let dataToSign
  74   if (params.data) {
  75     dataToSign = params.data
  76   } else {
  77     // We do not have data to sign so we just take our host
  78     // It is not ideal but the connection should be in HTTPS
  79     dataToSign = host
  80   }
  81
  82   sign(dataToSign).then(signature => {
  83     requestParams.json.signature = {
  84       host, // Which host we pretend to be
  85       signature
  86     }
  87
  88     // If there are data information
  89     if (params.data) {
  90       requestParams.json.data = params.data
  91     }
  92
  93     return doRequest(requestParams)
  94   })
  95 }
  96
  97 // ---------------------------------------------------------------------------
  98
  99 export {
 100   doRequest,
 101   doRequestAndSaveToFile,
 102   makeRetryRequest,
 103   makeSecureRequest
 104 }