path: root/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 { 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
  sign: boolean
  data?: Object
}
function makeSecureRequest (params: MakeSecureRequestParams) {
  return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
    const requestParams = {
      url: REMOTE_SCHEME.HTTP + '://' + params.toPod.host + params.path,
      json: {}
    }

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

    // Add signature if it is specified in the params
    if (params.sign === true) {
      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
      }

      requestParams.json['signature'] = {
        host, // Which host we pretend to be
        signature: sign(dataToSign)
      }
    }

    // If there are data informations
    if (params.data) {
      requestParams.json['data'] = params.data
    }

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

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

export {
  makeRetryRequest,
  makeSecureRequest
}