X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Fyoutube-dl.ts;h=577a59dbf4f69e69c86dea675e1c1a2453f18893;hb=43482ec712d4298e18e4c7642d115ed78a345681;hp=87a0d0584e174ed73ee7c0da721893e60904f38e;hpb=687c6180bc5f48b9159bbb229ec5404cc205919e;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/youtube-dl.ts b/server/helpers/youtube-dl.ts index 87a0d0584..577a59dbf 100644 --- a/server/helpers/youtube-dl.ts +++ b/server/helpers/youtube-dl.ts @@ -6,6 +6,7 @@ import { peertubeTruncate, root } from './core-utils' import { ensureDir, remove, writeFile } from 'fs-extra' import * as request from 'request' import { createWriteStream } from 'fs' +import { CONFIG } from '@server/initializers/config' export type YoutubeDLInfo = { name?: string @@ -24,7 +25,8 @@ const processOptions = { function getYoutubeDLInfo (url: string, opts?: string[]): Promise { return new Promise(async (res, rej) => { - const args = opts || [ '-j', '--flat-playlist' ] + let args = opts || [ '-j', '--flat-playlist' ] + args = wrapWithProxyOptions(args) const youtubeDL = await safeGetYoutubeDL() youtubeDL.getInfo(url, args, processOptions, (err, info) => { @@ -45,11 +47,11 @@ function downloadYoutubeDLVideo (url: string, timeout: number) { logger.info('Importing youtubeDL video %s', url) - const options = [ '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best', '-o', path ] + let options = [ '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best', '-o', path ] + options = wrapWithProxyOptions(options) if (process.env.FFMPEG_PATH) { - options.push('--ffmpeg-location') - options.push(process.env.FFMPEG_PATH) + options = options.concat([ '--ffmpeg-location', process.env.FFMPEG_PATH ]) } return new Promise(async (res, rej) => { @@ -264,3 +266,13 @@ function getCategory (categories: string[]) { return undefined } + +function wrapWithProxyOptions (options: string[]) { + if (CONFIG.IMPORT.VIDEOS.HTTP.PROXY.ENABLED) { + logger.debug('Using proxy for YoutubeDL') + + return [ '--proxy', CONFIG.IMPORT.VIDEOS.HTTP.PROXY.URL ].concat(options) + } + + return options +}