PeerTube/server/helpers/requests.ts

import * as replay from 'request-replay'
import * as request from 'request'
import * as Promise from 'bluebird'

import {
  RETRY_REQUESTS,
  REMOTE_SCHEME,
  CONFIG
} from '../initializers'
import { PodInstance } from '../models'
import { PodSignature } from '../../shared'
import { sign } from './peertube-crypto'

type MakeRetryRequestParams = {
  url: string,
  method: 'GET' | 'POST',
  json: Object
}
function makeRetryRequest (params: MakeRetryRequestParams) {
  return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
    replay(
      request(params, (err, response, body) => err ? rej(err) : res({ response, body })),
      {
        retries: RETRY_REQUESTS,
        factor: 3,
        maxTimeout: Infinity,
        errorCodes: [ 'EADDRINFO', 'ETIMEDOUT', 'ECONNRESET', 'ESOCKETTIMEDOUT', 'ENOTFOUND', 'ECONNREFUSED' ]
      }
    )
  })
}

type MakeSecureRequestParams = {
  method: 'GET' | 'POST'
  toPod: PodInstance
  path: string
  data?: Object
}
function makeSecureRequest (params: MakeSecureRequestParams) {
  return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
    const requestParams: {
      url: string,
      json: {
        signature: PodSignature,
        data: any
      }
    } = {
      url: REMOTE_SCHEME.HTTP + '://' + params.toPod.host + params.path,
      json: {
        signature: null,
        data: null
      }
    }

    if (params.method !== 'POST') {
      return rej(new Error('Cannot make a secure request with a non POST method.'))
    }

    const host = CONFIG.WEBSERVER.HOST

    let dataToSign
    if (params.data) {
      dataToSign = params.data
    } else {
      // We do not have data to sign so we just take our host
      // It is not ideal but the connection should be in HTTPS
      dataToSign = host
    }

    sign(dataToSign).then(signature => {
      requestParams.json.signature = {
        host, // Which host we pretend to be
        signature
      }

      // If there are data information
      if (params.data) {
        requestParams.json.data = params.data
      }

      request.post(requestParams, (err, response, body) => err ? rej(err) : res({ response, body }))
    })
  })
}

// ---------------------------------------------------------------------------

export {
  makeRetryRequest,
  makeSecureRequest
}