scripts/create-transcoding-job.ts

   1 import { registerTSPaths } from '../server/helpers/register-ts-paths'
   2 registerTSPaths()
   3
   4 import { program } from 'commander'
   5 import { VideoModel } from '../server/models/video/video'
   6 import { initDatabaseModels } from '../server/initializers/database'
   7 import { JobQueue } from '../server/lib/job-queue'
   8 import { computeLowerResolutionsToTranscode } from '@server/helpers/ffprobe-utils'
   9 import { VideoState, VideoTranscodingPayload } from '@shared/models'
  10 import { CONFIG } from '@server/initializers/config'
  11 import { isUUIDValid, toCompleteUUID } from '@server/helpers/custom-validators/misc'
  12 import { addTranscodingJob } from '@server/lib/video'
  13
  14 program
  15   .option('-v, --video [videoUUID]', 'Video UUID')
  16   .option('-r, --resolution [resolution]', 'Video resolution (integer)')
  17   .option('--generate-hls', 'Generate HLS playlist')
  18   .parse(process.argv)
  19
  20 const options = program.opts()
  21
  22 if (options.video === undefined) {
  23   console.error('All parameters are mandatory.')
  24   process.exit(-1)
  25 }
  26
  27 if (options.resolution !== undefined && Number.isNaN(+options.resolution)) {
  28   console.error('The resolution must be an integer (example: 1080).')
  29   process.exit(-1)
  30 }
  31
  32 run()
  33   .then(() => process.exit(0))
  34   .catch(err => {
  35     console.error(err)
  36     process.exit(-1)
  37   })
  38
  39 async function run () {
  40   await initDatabaseModels(true)
  41
  42   const uuid = toCompleteUUID(options.video)
  43
  44   if (isUUIDValid(uuid) === false) {
  45     console.error('%s is not a valid video UUID.', options.video)
  46     return
  47   }
  48
  49   const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(uuid)
  50   if (!video) throw new Error('Video not found.')
  51
  52   const dataInput: VideoTranscodingPayload[] = []
  53   const maxResolution = video.getMaxQualityFile().resolution
  54
  55   // Generate HLS files
  56   if (options.generateHls || CONFIG.TRANSCODING.WEBTORRENT.ENABLED === false) {
  57     const resolutionsEnabled = options.resolution
  58       ? [ parseInt(options.resolution) ]
  59       : computeLowerResolutionsToTranscode(maxResolution, 'vod').concat([ maxResolution ])
  60
  61     for (const resolution of resolutionsEnabled) {
  62       dataInput.push({
  63         type: 'new-resolution-to-hls',
  64         videoUUID: video.uuid,
  65         resolution,
  66         isPortraitMode: false,
  67         copyCodecs: false,
  68         isNewVideo: false,
  69         isMaxQuality: maxResolution === resolution,
  70         autoDeleteWebTorrentIfNeeded: false
  71       })
  72     }
  73   } else {
  74     if (options.resolution !== undefined) {
  75       dataInput.push({
  76         type: 'new-resolution-to-webtorrent',
  77         videoUUID: video.uuid,
  78         isNewVideo: false,
  79         resolution: parseInt(options.resolution)
  80       })
  81     } else {
  82       if (video.VideoFiles.length === 0) {
  83         console.error('Cannot regenerate webtorrent files with a HLS only video.')
  84         return
  85       }
  86
  87       dataInput.push({
  88         type: 'optimize-to-webtorrent',
  89         videoUUID: video.uuid,
  90         isNewVideo: false
  91       })
  92     }
  93   }
  94
  95   JobQueue.Instance.init(true)
  96
  97   video.state = VideoState.TO_TRANSCODE
  98   await video.save()
  99
 100   for (const d of dataInput) {
 101     await addTranscodingJob(d, {})
 102     console.log('Transcoding job for video %s created.', video.uuid)
 103   }
 104 }