server/helpers/requests.ts

   1 import * as Promise from 'bluebird'
   2 import { createWriteStream } from 'fs'
   3 import * as request from 'request'
   4 import { ACTIVITY_PUB } from '../initializers'
   5
   6 function doRequest (requestOptions: request.CoreOptions & request.UriOptions & { activityPub?: boolean }) {
   7   if (requestOptions.activityPub === true) {
   8     if (!Array.isArray(requestOptions.headers)) requestOptions.headers = {}
   9     requestOptions.headers['accept'] = ACTIVITY_PUB.ACCEPT_HEADER
  10   }
  11
  12   return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => {
  13     request(requestOptions, (err, response, body) => err ? rej(err) : res({ response, body }))
  14   })
  15 }
  16
  17 function doRequestAndSaveToFile (requestOptions: request.CoreOptions & request.UriOptions, destPath: string) {
  18   return new Promise<request.RequestResponse>((res, rej) => {
  19     request(requestOptions)
  20       .on('response', response => res(response as request.RequestResponse))
  21       .on('error', err => rej(err))
  22       .pipe(createWriteStream(destPath))
  23   })
  24 }
  25
  26 // ---------------------------------------------------------------------------
  27
  28 export {
  29   doRequest,
  30   doRequestAndSaveToFile
  31 }