X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Flib%2Fjob-queue%2Fhandlers%2Fvideo-file-import.ts;h=d950f64072e20db266a2989641322358db7e9bef;hb=e15bc9dcd59ebb6102d1455450d2e8bbf9f49c27;hp=8297a15716819c5fba66efeb3ef7d2be5f0f8f17;hpb=7d9ba5c08999c6482f0bc5e0c09c6f55b7724090;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/lib/job-queue/handlers/video-file-import.ts b/server/lib/job-queue/handlers/video-file-import.ts index 8297a1571..d950f6407 100644 --- a/server/lib/job-queue/handlers/video-file-import.ts +++ b/server/lib/job-queue/handlers/video-file-import.ts @@ -1,43 +1,38 @@ -import * as Bull from 'bull' +import { Job } from 'bullmq' import { copy, stat } from 'fs-extra' -import { extname } from 'path' import { createTorrentAndSetInfoHash } from '@server/helpers/webtorrent' -import { generateVideoFilename, getVideoFilePath } from '@server/lib/video-paths' -import { UserModel } from '@server/models/user/user' +import { CONFIG } from '@server/initializers/config' +import { federateVideoIfNeeded } from '@server/lib/activitypub/videos' +import { generateWebTorrentVideoFilename } from '@server/lib/paths' +import { buildMoveToObjectStorageJob } from '@server/lib/video' +import { VideoPathManager } from '@server/lib/video-path-manager' +import { VideoModel } from '@server/models/video/video' +import { VideoFileModel } from '@server/models/video/video-file' import { MVideoFullLight } from '@server/types/models' -import { VideoFileImportPayload } from '@shared/models' -import { getVideoFileFPS, getVideoFileResolution } from '../../../helpers/ffprobe-utils' +import { getLowercaseExtension } from '@shared/core-utils' +import { VideoFileImportPayload, VideoStorage } from '@shared/models' +import { getVideoStreamFPS, getVideoStreamDimensionsInfo } from '../../../helpers/ffmpeg' import { logger } from '../../../helpers/logger' -import { VideoModel } from '../../../models/video/video' -import { VideoFileModel } from '../../../models/video/video-file' -import { onNewWebTorrentFileResolution } from './video-transcoding' +import { JobQueue } from '../job-queue' -async function processVideoFileImport (job: Bull.Job) { +async function processVideoFileImport (job: Job) { const payload = job.data as VideoFileImportPayload - logger.info('Processing video file import in job %d.', job.id) + logger.info('Processing video file import in job %s.', job.id) - const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(payload.videoUUID) + const video = await VideoModel.loadFull(payload.videoUUID) // No video, maybe deleted? if (!video) { logger.info('Do not process job %d, video does not exist.', job.id) return undefined } - const data = await getVideoFileResolution(payload.filePath) - await updateVideoFile(video, payload.filePath) - const user = await UserModel.loadByChannelActorId(video.VideoChannel.actorId) - - const newResolutionPayload = { - type: 'new-resolution-to-webtorrent' as 'new-resolution-to-webtorrent', - videoUUID: video.uuid, - resolution: data.videoFileResolution, - isPortraitMode: data.isPortraitMode, - copyCodecs: false, - isNewVideo: false + if (CONFIG.OBJECT_STORAGE.ENABLED) { + await JobQueue.Instance.createJob(await buildMoveToObjectStorageJob({ video, previousVideoState: video.state })) + } else { + await federateVideoIfNeeded(video, false) } - await onNewWebTorrentFileResolution(video, user, newResolutionPayload) return video } @@ -51,18 +46,17 @@ export { // --------------------------------------------------------------------------- async function updateVideoFile (video: MVideoFullLight, inputFilePath: string) { - const { videoFileResolution } = await getVideoFileResolution(inputFilePath) + const { resolution } = await getVideoStreamDimensionsInfo(inputFilePath) const { size } = await stat(inputFilePath) - const fps = await getVideoFileFPS(inputFilePath) + const fps = await getVideoStreamFPS(inputFilePath) - const fileExt = extname(inputFilePath) + const fileExt = getLowercaseExtension(inputFilePath) - const currentVideoFile = video.VideoFiles.find(videoFile => videoFile.resolution === videoFileResolution) + const currentVideoFile = video.VideoFiles.find(videoFile => videoFile.resolution === resolution) if (currentVideoFile) { // Remove old file and old torrent - await video.removeFile(currentVideoFile) - await currentVideoFile.removeTorrent() + await video.removeWebTorrentFile(currentVideoFile) // Remove the old video file from the array video.VideoFiles = video.VideoFiles.filter(f => f !== currentVideoFile) @@ -70,15 +64,16 @@ async function updateVideoFile (video: MVideoFullLight, inputFilePath: string) { } const newVideoFile = new VideoFileModel({ - resolution: videoFileResolution, + resolution, extname: fileExt, - filename: generateVideoFilename(video, false, videoFileResolution, fileExt), + filename: generateWebTorrentVideoFilename(resolution, fileExt), + storage: VideoStorage.FILE_SYSTEM, size, fps, videoId: video.id }) - const outputPath = getVideoFilePath(video, newVideoFile) + const outputPath = VideoPathManager.Instance.getFSVideoFileOutputPath(video, newVideoFile) await copy(inputFilePath, outputPath) video.VideoFiles.push(newVideoFile)