X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Frequests.ts;h=ce185a2c0248b3999e0902f2c08cb73abaf3956a;hb=da854ddd502cd70685ef779c673b9e63757b8aa0;hp=8ded529728ea53927db1b9a89f7042afa8aee24f;hpb=65fcc3119c334b75dd13bcfdebf186afdc580a8f;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/requests.ts b/server/helpers/requests.ts index 8ded52972..ce185a2c0 100644 --- a/server/helpers/requests.ts +++ b/server/helpers/requests.ts @@ -1,65 +1,31 @@ -import replay = require('request-replay') -import request = require('request') - -import { - RETRY_REQUESTS, - REMOTE_SCHEME, - CONFIG -} from '../initializers' -import { sign } from './peertube-crypto' - -function makeRetryRequest (params, callback) { - replay( - request(params, callback), - { - retries: RETRY_REQUESTS, - factor: 3, - maxTimeout: Infinity, - errorCodes: [ 'EADDRINFO', 'ETIMEDOUT', 'ECONNRESET', 'ESOCKETTIMEDOUT', 'ENOTFOUND', 'ECONNREFUSED' ] - } - ) -} - -function makeSecureRequest (params, callback) { - const requestParams = { - url: REMOTE_SCHEME.HTTP + '://' + params.toPod.host + params.path, - json: {} - } - - if (params.method !== 'POST') { - return callback(new Error('Cannot make a secure request with a non POST method.')) +import * as Promise from 'bluebird' +import { createWriteStream } from 'fs' +import * as request from 'request' +import { ACTIVITY_PUB } from '../initializers' + +function doRequest (requestOptions: request.CoreOptions & request.UriOptions & { activityPub?: boolean }) { + if (requestOptions.activityPub === true) { + if (!Array.isArray(requestOptions.headers)) requestOptions.headers = {} + requestOptions.headers['accept'] = ACTIVITY_PUB.ACCEPT_HEADER } - // 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 - } + return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => { + request(requestOptions, (err, response, body) => err ? rej(err) : res({ response, body })) + }) +} - request.post(requestParams, callback) +function doRequestAndSaveToFile (requestOptions: request.CoreOptions & request.UriOptions, destPath: string) { + return new Promise((res, rej) => { + request(requestOptions) + .on('response', response => res(response as request.RequestResponse)) + .on('error', err => rej(err)) + .pipe(createWriteStream(destPath)) + }) } // --------------------------------------------------------------------------- export { - makeRetryRequest, - makeSecureRequest + doRequest, + doRequestAndSaveToFile }