server/controllers/api/videos/transcoding.ts

   1 import Bluebird from 'bluebird'
   2 import express from 'express'
   3 import { computeResolutionsToTranscode } from '@server/helpers/ffmpeg'
   4 import { logger, loggerTagsFactory } from '@server/helpers/logger'
   5 import { JobQueue } from '@server/lib/job-queue'
   6 import { Hooks } from '@server/lib/plugins/hooks'
   7 import { buildTranscodingJob } from '@server/lib/video'
   8 import { HttpStatusCode, UserRight, VideoState, VideoTranscodingCreate } from '@shared/models'
   9 import { asyncMiddleware, authenticate, createTranscodingValidator, ensureUserHasRight } from '../../../middlewares'
  10
  11 const lTags = loggerTagsFactory('api', 'video')
  12 const transcodingRouter = express.Router()
  13
  14 transcodingRouter.post('/:videoId/transcoding',
  15   authenticate,
  16   ensureUserHasRight(UserRight.RUN_VIDEO_TRANSCODING),
  17   asyncMiddleware(createTranscodingValidator),
  18   asyncMiddleware(createTranscoding)
  19 )
  20
  21 // ---------------------------------------------------------------------------
  22
  23 export {
  24   transcodingRouter
  25 }
  26
  27 // ---------------------------------------------------------------------------
  28
  29 async function createTranscoding (req: express.Request, res: express.Response) {
  30   const video = res.locals.videoAll
  31   logger.info('Creating %s transcoding job for %s.', req.body.transcodingType, video.url, lTags())
  32
  33   const body: VideoTranscodingCreate = req.body
  34
  35   const { resolution: maxResolution, hasAudio } = await video.probeMaxQualityFile()
  36
  37   const resolutions = await Hooks.wrapObject(
  38     computeResolutionsToTranscode({ input: maxResolution, type: 'vod', includeInput: true, strictLower: false, hasAudio }),
  39     'filter:transcoding.manual.resolutions-to-transcode.result',
  40     body
  41   )
  42
  43   if (resolutions.length === 0) {
  44     return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
  45   }
  46
  47   video.state = VideoState.TO_TRANSCODE
  48   await video.save()
  49
  50   const childrenResolutions = resolutions.filter(r => r !== maxResolution)
  51
  52   const children = await Bluebird.mapSeries(childrenResolutions, resolution => {
  53     if (body.transcodingType === 'hls') {
  54       return buildHLSJobOption({
  55         videoUUID: video.uuid,
  56         hasAudio,
  57         resolution,
  58         isMaxQuality: false
  59       })
  60     }
  61
  62     if (body.transcodingType === 'webtorrent') {
  63       return buildWebTorrentJobOption({
  64         videoUUID: video.uuid,
  65         hasAudio,
  66         resolution
  67       })
  68     }
  69   })
  70
  71   const parent = body.transcodingType === 'hls'
  72     ? await buildHLSJobOption({
  73       videoUUID: video.uuid,
  74       hasAudio,
  75       resolution: maxResolution,
  76       isMaxQuality: false
  77     })
  78     : await buildWebTorrentJobOption({
  79       videoUUID: video.uuid,
  80       hasAudio,
  81       resolution: maxResolution
  82     })
  83
  84   // Porcess the last resolution after the other ones to prevent concurrency issue
  85   // Because low resolutions use the biggest one as ffmpeg input
  86   await JobQueue.Instance.createJobWithChildren(parent, children)
  87
  88   return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
  89 }
  90
  91 function buildHLSJobOption (options: {
  92   videoUUID: string
  93   hasAudio: boolean
  94   resolution: number
  95   isMaxQuality: boolean
  96 }) {
  97   const { videoUUID, hasAudio, resolution, isMaxQuality } = options
  98
  99   return buildTranscodingJob({
 100     type: 'new-resolution-to-hls',
 101     videoUUID,
 102     resolution,
 103     hasAudio,
 104     copyCodecs: false,
 105     isNewVideo: false,
 106     autoDeleteWebTorrentIfNeeded: false,
 107     isMaxQuality
 108   })
 109 }
 110
 111 function buildWebTorrentJobOption (options: {
 112   videoUUID: string
 113   hasAudio: boolean
 114   resolution: number
 115 }) {
 116   const { videoUUID, hasAudio, resolution } = options
 117
 118   return buildTranscodingJob({
 119     type: 'new-resolution-to-webtorrent',
 120     videoUUID,
 121     isNewVideo: false,
 122     resolution,
 123     hasAudio,
 124     createHLSIfNeeded: false
 125   })
 126 }