X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Frequests.ts;h=ee9e804040e8364a00723f2312d3917c221fd830;hb=6360e12616d54c2321ee5e3a6523a8b60d70c0a7;hp=ce185a2c0248b3999e0902f2c08cb73abaf3956a;hpb=da854ddd502cd70685ef779c673b9e63757b8aa0;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/requests.ts b/server/helpers/requests.ts index ce185a2c0..ee9e80404 100644 --- a/server/helpers/requests.ts +++ b/server/helpers/requests.ts @@ -1,25 +1,29 @@ -import * as Promise from 'bluebird' -import { createWriteStream } from 'fs' +import * as Bluebird from 'bluebird' +import { createWriteStream } from 'fs-extra' import * as request from 'request' import { ACTIVITY_PUB } from '../initializers' -function doRequest (requestOptions: request.CoreOptions & request.UriOptions & { activityPub?: boolean }) { +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 } - 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 })) }) } function doRequestAndSaveToFile (requestOptions: request.CoreOptions & request.UriOptions, destPath: string) { - return new Promise((res, rej) => { + return new Bluebird((res, rej) => { + const file = createWriteStream(destPath) + file.on('finish', () => res()) + request(requestOptions) - .on('response', response => res(response as request.RequestResponse)) .on('error', err => rej(err)) - .pipe(createWriteStream(destPath)) + .pipe(file) }) }