X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Frequests.ts;h=0756beb16fb1c226f4f860a83a6dd8ff2f35fe2b;hb=6c38f40d966eb25e53ac38174153cb1fef0475f5;hp=fc77ebd357f5f1d8d8f57f2a42f90a810a56dc55;hpb=3455c2656e257ae3d9b4169af58b6889d9904148;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/requests.ts b/server/helpers/requests.ts index fc77ebd35..0756beb16 100644 --- a/server/helpers/requests.ts +++ b/server/helpers/requests.ts @@ -1,11 +1,8 @@ import { createWriteStream, remove } from 'fs-extra' import got, { CancelableRequest, NormalizedOptions, Options as GotOptions, RequestError, Response } from 'got' import { HttpProxyAgent, HttpsProxyAgent } from 'hpagent' -import { join } from 'path' -import { CONFIG } from '../initializers/config' -import { ACTIVITY_PUB, BINARY_CONTENT_TYPES, PEERTUBE_VERSION, REQUEST_TIMEOUT, WEBSERVER } from '../initializers/constants' +import { ACTIVITY_PUB, BINARY_CONTENT_TYPES, PEERTUBE_VERSION, REQUEST_TIMEOUTS, WEBSERVER } from '../initializers/constants' import { pipelinePromise } from './core-utils' -import { processImage } from './image-utils' import { logger, loggerTagsFactory } from './logger' import { getProxy, isProxyEnabled } from './proxy' @@ -20,6 +17,7 @@ export interface PeerTubeRequestError extends Error { } type PeerTubeRequestOptions = { + timeout?: number activityPub?: boolean bodyKBLimit?: number // 1MB httpSignature?: { @@ -87,11 +85,14 @@ const peertubeGot = got.extend({ } httpSignature.signRequest({ - getHeader: function (header) { - return options.headers[header] + getHeader: function (header: string) { + const value = options.headers[header.toLowerCase()] + + if (!value) logger.warn('Unknown header requested by http-signature.', { headers: options.headers, header }) + return value }, - setHeader: function (header, value) { + setHeader: function (header: string, value: string) { options.headers[header] = value }, @@ -129,7 +130,7 @@ async function doRequestAndSaveToFile ( destPath: string, options: PeerTubeRequestOptions = {} ) { - const gotOptions = buildGotOptions(options) + const gotOptions = buildGotOptions({ ...options, timeout: options.timeout ?? REQUEST_TIMEOUTS.FILE }) const outFile = createWriteStream(destPath) @@ -146,21 +147,6 @@ async function doRequestAndSaveToFile ( } } -async function downloadImage (url: string, destDir: string, destName: string, size: { width: number, height: number }) { - const tmpPath = join(CONFIG.STORAGE.TMP_DIR, 'pending-' + destName) - await doRequestAndSaveToFile(url, tmpPath) - - const destPath = join(destDir, destName) - - try { - await processImage(tmpPath, destPath, size) - } catch (err) { - await remove(tmpPath) - - throw err - } -} - function getAgent () { if (!isProxyEnabled()) return {} @@ -210,7 +196,7 @@ export { doJSONRequest, doRequestAndSaveToFile, isBinaryResponse, - downloadImage, + getAgent, findLatestRedirection, peertubeGot } @@ -235,7 +221,7 @@ function buildGotOptions (options: PeerTubeRequestOptions) { return { method: options.method, dnsCache: true, - timeout: REQUEST_TIMEOUT, + timeout: options.timeout ?? REQUEST_TIMEOUTS.DEFAULT, json: options.json, searchParams: options.searchParams, retry: 2,