X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Fhelpers%2Fyoutube-dl.ts;h=b3079370f0bd71ccebb7fd28172b708f4b881bce;hb=b6a1dd4d1b3b0032f8b968e72cbd074f646e8827;hp=4d4b37ef938fcd74249df092b71ada6c0950c7b6;hpb=4e553a41fdc67c4da186502522ac99a5ba230453;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/helpers/youtube-dl.ts b/server/helpers/youtube-dl.ts index 4d4b37ef9..b3079370f 100644 --- a/server/helpers/youtube-dl.ts +++ b/server/helpers/youtube-dl.ts @@ -1,5 +1,5 @@ import { truncate } from 'lodash' -import { CONSTRAINTS_FIELDS, VIDEO_CATEGORIES } from '../initializers' +import { CONSTRAINTS_FIELDS, VIDEO_CATEGORIES } from '../initializers/constants' import { logger } from './logger' import { generateVideoImportTmpPath } from './utils' import { join } from 'path' @@ -16,7 +16,7 @@ export type YoutubeDLInfo = { nsfw?: boolean tags?: string[] thumbnailUrl?: string - originallyPublishedAt?: string + originallyPublishedAt?: Date } const processOptions = { @@ -48,6 +48,11 @@ function downloadYoutubeDLVideo (url: string, timeout: number) { const options = [ '-f', 'bestvideo[ext=mp4]+bestaudio[ext=m4a]/best', '-o', path ] + if (process.env.FFMPEG_PATH) { + options.push('--ffmpeg-location') + options.push(process.env.FFMPEG_PATH) + } + return new Promise(async (res, rej) => { const youtubeDL = await safeGetYoutubeDL() youtubeDL.exec(url, options, processOptions, err => { @@ -143,13 +148,33 @@ async function safeGetYoutubeDL () { return youtubeDL } +function buildOriginallyPublishedAt (obj: any) { + let originallyPublishedAt: Date = null + + const uploadDateMatcher = /^(\d{4})(\d{2})(\d{2})$/.exec(obj.upload_date) + if (uploadDateMatcher) { + originallyPublishedAt = new Date() + originallyPublishedAt.setHours(0, 0, 0, 0) + + const year = parseInt(uploadDateMatcher[1], 10) + // Month starts from 0 + const month = parseInt(uploadDateMatcher[2], 10) - 1 + const day = parseInt(uploadDateMatcher[3], 10) + + originallyPublishedAt.setFullYear(year, month, day) + } + + return originallyPublishedAt +} + // --------------------------------------------------------------------------- export { updateYoutubeDLBinary, downloadYoutubeDLVideo, getYoutubeDLInfo, - safeGetYoutubeDL + safeGetYoutubeDL, + buildOriginallyPublishedAt } // --------------------------------------------------------------------------- @@ -174,9 +199,6 @@ function normalizeObject (obj: any) { } function buildVideoInfo (obj: any) { - - const date = obj.upload_date.slice(0,4)+","+obj.upload_date.slice(4,6)+","+obj.upload_date.slice(6,8) - return { name: titleTruncation(obj.title), description: descriptionTruncation(obj.description), @@ -185,7 +207,7 @@ function buildVideoInfo (obj: any) { nsfw: isNSFW(obj), tags: getTags(obj.tags), thumbnailUrl: obj.thumbnail || undefined, - originallyPublishedAt: new Date(date).toISOString() + originallyPublishedAt: buildOriginallyPublishedAt(obj) } }