server/lib/runners/job-handlers/vod-web-video-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 {
   8   RunnerJobUpdatePayload,
   9   RunnerJobVODWebVideoTranscodingPayload,
  10   RunnerJobVODWebVideoTranscodingPrivatePayload,
  11   VODWebVideoTranscodingSuccess
  12 } from '@shared/models'
  13 import { generateRunnerTranscodingVideoInputFileUrl } from '../runner-urls'
  14 import { AbstractVODTranscodingJobHandler } from './abstract-vod-transcoding-job-handler'
  15 import { loadTranscodingRunnerVideo, onVODWebVideoOrAudioMergeTranscodingJob } from './shared'
  16
  17 type CreateOptions = {
  18   video: MVideo
  19   isNewVideo: boolean
  20   resolution: number
  21   fps: number
  22   priority: number
  23   dependsOnRunnerJob?: MRunnerJob
  24 }
  25
  26 // eslint-disable-next-line max-len
  27 export class VODWebVideoTranscodingJobHandler extends AbstractVODTranscodingJobHandler<CreateOptions, RunnerJobUpdatePayload, VODWebVideoTranscodingSuccess> {
  28
  29   async create (options: CreateOptions) {
  30     const { video, resolution, fps, priority, dependsOnRunnerJob } = options
  31
  32     const jobUUID = buildUUID()
  33     const payload: RunnerJobVODWebVideoTranscodingPayload = {
  34       input: {
  35         videoFileUrl: generateRunnerTranscodingVideoInputFileUrl(jobUUID, video.uuid)
  36       },
  37       output: {
  38         resolution,
  39         fps
  40       }
  41     }
  42
  43     const privatePayload: RunnerJobVODWebVideoTranscodingPrivatePayload = {
  44       ...pick(options, [ 'isNewVideo' ]),
  45
  46       videoUUID: video.uuid
  47     }
  48
  49     const job = await this.createRunnerJob({
  50       type: 'vod-web-video-transcoding',
  51       jobUUID,
  52       payload,
  53       privatePayload,
  54       dependsOnRunnerJob,
  55       priority
  56     })
  57
  58     await VideoJobInfoModel.increaseOrCreate(video.uuid, 'pendingTranscode')
  59
  60     return job
  61   }
  62
  63   // ---------------------------------------------------------------------------
  64
  65   protected async specificComplete (options: {
  66     runnerJob: MRunnerJob
  67     resultPayload: VODWebVideoTranscodingSuccess
  68   }) {
  69     const { runnerJob, resultPayload } = options
  70     const privatePayload = runnerJob.privatePayload as RunnerJobVODWebVideoTranscodingPrivatePayload
  71
  72     const video = await loadTranscodingRunnerVideo(runnerJob, this.lTags)
  73     if (!video) return
  74
  75     const videoFilePath = resultPayload.videoFile as string
  76
  77     await onVODWebVideoOrAudioMergeTranscodingJob({ video, videoFilePath, privatePayload })
  78
  79     logger.info(
  80       'Runner VOD web video transcoding job %s for %s ended.',
  81       runnerJob.uuid, video.uuid, this.lTags(video.uuid, runnerJob.uuid)
  82     )
  83   }
  84 }