PeerTube/server/controllers/api/videos/transcoding.ts

import express from 'express'
import { computeResolutionsToTranscode } from '@server/helpers/ffmpeg'
import { logger, loggerTagsFactory } from '@server/helpers/logger'
import { addTranscodingJob } from '@server/lib/video'
import { HttpStatusCode, UserRight, VideoState, VideoTranscodingCreate } from '@shared/models'
import { asyncMiddleware, authenticate, createTranscodingValidator, ensureUserHasRight } from '../../../middlewares'
import { Hooks } from '@server/lib/plugins/hooks'

const lTags = loggerTagsFactory('api', 'video')
const transcodingRouter = express.Router()

transcodingRouter.post('/:videoId/transcoding',
  authenticate,
  ensureUserHasRight(UserRight.RUN_VIDEO_TRANSCODING),
  asyncMiddleware(createTranscodingValidator),
  asyncMiddleware(createTranscoding)
)

// ---------------------------------------------------------------------------

export {
  transcodingRouter
}

// ---------------------------------------------------------------------------

async function createTranscoding (req: express.Request, res: express.Response) {
  const video = res.locals.videoAll
  logger.info('Creating %s transcoding job for %s.', req.body.transcodingType, video.url, lTags())

  const body: VideoTranscodingCreate = req.body

  const { resolution: maxResolution, audioStream } = await video.probeMaxQualityFile()
  const resolutions = await Hooks.wrapObject(
    computeResolutionsToTranscode({ input: maxResolution, type: 'vod', includeInput: true, strictLower: false }),
    'filter:transcoding.manual.resolutions-to-transcode.result',
    body
  )

  if (resolutions.length === 0) {
    return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
  }

  video.state = VideoState.TO_TRANSCODE
  await video.save()

  for (const resolution of resolutions) {
    if (body.transcodingType === 'hls') {
      await addTranscodingJob({
        type: 'new-resolution-to-hls',
        videoUUID: video.uuid,
        resolution,
        hasAudio: !!audioStream,
        copyCodecs: false,
        isNewVideo: false,
        autoDeleteWebTorrentIfNeeded: false,
        isMaxQuality: maxResolution === resolution
      })
    } else if (body.transcodingType === 'webtorrent') {
      await addTranscodingJob({
        type: 'new-resolution-to-webtorrent',
        videoUUID: video.uuid,
        isNewVideo: false,
        resolution,
        hasAudio: !!audioStream,
        createHLSIfNeeded: false
      })
    }
  }

  return res.sendStatus(HttpStatusCode.NO_CONTENT_204)
}