server/lib/runners/job-handlers/vod-audio-merge-transcoding-job-handler.ts

   1 import { pick } from 'lodash'
   2 import { logger } from '@server/helpers/logger'
   3 import { VideoJobInfoModel } from '@server/models/video/video-job-info'
   4 import { MVideo } from '@server/types/models'
   5 import { MRunnerJob } from '@server/types/models/runners'
   6 import { buildUUID } from '@shared/extra-utils'
   7 import { getVideoStreamDuration } from '@shared/ffmpeg'
   8 import {
   9   RunnerJobUpdatePayload,
  10   RunnerJobVODAudioMergeTranscodingPayload,
  11   RunnerJobVODWebVideoTranscodingPrivatePayload,
  12   VODAudioMergeTranscodingSuccess
  13 } from '@shared/models'
  14 import { generateRunnerTranscodingVideoInputFileUrl, generateRunnerTranscodingVideoPreviewFileUrl } from '../runner-urls'
  15 import { AbstractVODTranscodingJobHandler } from './abstract-vod-transcoding-job-handler'
  16 import { loadTranscodingRunnerVideo, onVODWebVideoOrAudioMergeTranscodingJob } from './shared'
  17
  18 type CreateOptions = {
  19   video: MVideo
  20   isNewVideo: boolean
  21   resolution: number
  22   fps: number
  23   priority: number
  24   dependsOnRunnerJob?: MRunnerJob
  25 }
  26
  27 // eslint-disable-next-line max-len
  28 export class VODAudioMergeTranscodingJobHandler extends AbstractVODTranscodingJobHandler<CreateOptions, RunnerJobUpdatePayload, VODAudioMergeTranscodingSuccess> {
  29
  30   async create (options: CreateOptions) {
  31     const { video, resolution, fps, priority, dependsOnRunnerJob } = options
  32
  33     const jobUUID = buildUUID()
  34     const payload: RunnerJobVODAudioMergeTranscodingPayload = {
  35       input: {
  36         audioFileUrl: generateRunnerTranscodingVideoInputFileUrl(jobUUID, video.uuid),
  37         previewFileUrl: generateRunnerTranscodingVideoPreviewFileUrl(jobUUID, video.uuid)
  38       },
  39       output: {
  40         resolution,
  41         fps
  42       }
  43     }
  44
  45     const privatePayload: RunnerJobVODWebVideoTranscodingPrivatePayload = {
  46       ...pick(options, [ 'isNewVideo' ]),
  47
  48       videoUUID: video.uuid
  49     }
  50
  51     const job = await this.createRunnerJob({
  52       type: 'vod-audio-merge-transcoding',
  53       jobUUID,
  54       payload,
  55       privatePayload,
  56       priority,
  57       dependsOnRunnerJob
  58     })
  59
  60     await VideoJobInfoModel.increaseOrCreate(video.uuid, 'pendingTranscode')
  61
  62     return job
  63   }
  64
  65   // ---------------------------------------------------------------------------
  66
  67   async specificComplete (options: {
  68     runnerJob: MRunnerJob
  69     resultPayload: VODAudioMergeTranscodingSuccess
  70   }) {
  71     const { runnerJob, resultPayload } = options
  72     const privatePayload = runnerJob.privatePayload as RunnerJobVODWebVideoTranscodingPrivatePayload
  73
  74     const video = await loadTranscodingRunnerVideo(runnerJob, this.lTags)
  75     if (!video) return
  76
  77     const videoFilePath = resultPayload.videoFile as string
  78
  79     // ffmpeg generated a new video file, so update the video duration
  80     // See https://trac.ffmpeg.org/ticket/5456
  81     video.duration = await getVideoStreamDuration(videoFilePath)
  82     await video.save()
  83
  84     // We can remove the old audio file
  85     const oldAudioFile = video.VideoFiles[0]
  86     await video.removeWebTorrentFile(oldAudioFile)
  87     await oldAudioFile.destroy()
  88     video.VideoFiles = []
  89
  90     await onVODWebVideoOrAudioMergeTranscodingJob({ video, videoFilePath, privatePayload })
  91
  92     logger.info(
  93       'Runner VOD audio merge transcoding job %s for %s ended.',
  94       runnerJob.uuid, video.uuid, this.lTags(video.uuid, runnerJob.uuid)
  95     )
  96   }
  97 }