X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Frequests.ts;h=64e3ce663b13f8668483296047577f956e171bdb;hb=5d08a6a74e83f2e4dfe2f3ba7f5a39371e1bc89e;hp=8c4c983f7ac0f61ddb4e9ea207f0f049e890e1a6;hpb=e4f97babf701481b55cc10fb3448feab5f97c867;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/requests.ts b/server/helpers/requests.ts index 8c4c983f7..64e3ce663 100644 --- a/server/helpers/requests.ts +++ b/server/helpers/requests.ts @@ -1,86 +1,29 @@ -import * as replay from 'request-replay' +import * as Bluebird from 'bluebird' +import { createWriteStream } from 'fs' 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 { signObject } from './peertube-crypto' +import { ACTIVITY_PUB } from '../initializers' + +function doRequest ( + requestOptions: request.CoreOptions & request.UriOptions & { activityPub?: boolean } +): Bluebird<{ response: request.RequestResponse, body: any }> { + if (requestOptions.activityPub === true) { + if (!Array.isArray(requestOptions.headers)) requestOptions.headers = {} + requestOptions.headers['accept'] = ACTIVITY_PUB.ACCEPT_HEADER + } -function doRequest (requestOptions: request.CoreOptions & request.UriOptions) { - return new Promise<{ response: request.RequestResponse, body: any }>((res, rej) => { + return new Bluebird<{ response: request.RequestResponse, body: any }>((res, rej) => { request(requestOptions, (err, response, body) => err ? rej(err) : res({ response, body })) }) } -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 = { - toPod: PodInstance - path: string - data?: Object -} -function makeSecureRequest (params: MakeSecureRequestParams) { - const requestParams: { - method: 'POST', - uri: string, - json: { - signature: PodSignature, - data: any - } - } = { - method: 'POST', - uri: REMOTE_SCHEME.HTTP + '://' + params.toPod.host + params.path, - json: { - signature: null, - data: null - } - } - - 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 - } +function doRequestAndSaveToFile (requestOptions: request.CoreOptions & request.UriOptions, destPath: string) { + return new Bluebird((res, rej) => { + const file = createWriteStream(destPath) + file.on('finish', () => res()) - return doRequest(requestParams) + request(requestOptions) + .on('error', err => rej(err)) + .pipe(file) }) } @@ -88,6 +31,5 @@ function makeSecureRequest (params: MakeSecureRequestParams) { export { doRequest, - makeRetryRequest, - makeSecureRequest + doRequestAndSaveToFile }