]>
Commit | Line | Data |
---|---|---|
6b2ef589 | 1 | import * as Bluebird from 'bluebird' |
c9d5c64f | 2 | import { createWriteStream } from 'fs-extra' |
571389d4 | 3 | import * as request from 'request' |
6040f87d | 4 | import { ACTIVITY_PUB, CONFIG } from '../initializers' |
58d515e3 | 5 | import { processImage } from './image-utils' |
6040f87d | 6 | import { join } from 'path' |
da854ddd | 7 | |
5c6d985f | 8 | function doRequest <T> ( |
7006bc63 | 9 | requestOptions: request.CoreOptions & request.UriOptions & { activityPub?: boolean } |
633868bc | 10 | ): Bluebird<{ response: request.RequestResponse, body: any }> { |
da854ddd C |
11 | if (requestOptions.activityPub === true) { |
12 | if (!Array.isArray(requestOptions.headers)) requestOptions.headers = {} | |
13 | requestOptions.headers['accept'] = ACTIVITY_PUB.ACCEPT_HEADER | |
14 | } | |
e4f97bab | 15 | |
5c6d985f | 16 | return new Bluebird<{ response: request.RequestResponse, body: T }>((res, rej) => { |
e4f97bab C |
17 | request(requestOptions, (err, response, body) => err ? rej(err) : res({ response, body })) |
18 | }) | |
19 | } | |
dac0a531 | 20 | |
0d0e8dd0 | 21 | function doRequestAndSaveToFile (requestOptions: request.CoreOptions & request.UriOptions, destPath: string) { |
02988fdc C |
22 | return new Bluebird<void>((res, rej) => { |
23 | const file = createWriteStream(destPath) | |
24 | file.on('finish', () => res()) | |
25 | ||
0d0e8dd0 | 26 | request(requestOptions) |
0d0e8dd0 | 27 | .on('error', err => rej(err)) |
02988fdc | 28 | .pipe(file) |
0d0e8dd0 C |
29 | }) |
30 | } | |
31 | ||
6040f87d C |
32 | async function downloadImage (url: string, destDir: string, destName: string, size: { width: number, height: number }) { |
33 | const tmpPath = join(CONFIG.STORAGE.TMP_DIR, 'pending-' + destName) | |
58d515e3 C |
34 | await doRequestAndSaveToFile({ method: 'GET', uri: url }, tmpPath) |
35 | ||
6040f87d | 36 | const destPath = join(destDir, destName) |
58d515e3 C |
37 | await processImage({ path: tmpPath }, destPath, size) |
38 | } | |
39 | ||
9f10b292 | 40 | // --------------------------------------------------------------------------- |
dac0a531 | 41 | |
65fcc311 | 42 | export { |
e4f97bab | 43 | doRequest, |
58d515e3 C |
44 | doRequestAndSaveToFile, |
45 | downloadImage | |
65fcc311 | 46 | } |