scripts/create-transcoding-job.ts

   1 import * as program from 'commander'
   2 import { createReadStream } from 'fs'
   3 import { join } from 'path'
   4 import { createInterface } from 'readline'
   5 import { VideoModel } from '../server/models/video/video'
   6 import { initDatabaseModels } from '../server/initializers'
   7 import { JobQueue } from '../server/lib/job-queue'
   8
   9 program
  10   .option('-v, --video [videoUUID]', 'Video UUID')
  11   .option('-r, --resolution [resolution]', 'Video resolution (integer)')
  12   .parse(process.argv)
  13
  14 if (program['video'] === undefined) {
  15   console.error('All parameters are mandatory.')
  16   process.exit(-1)
  17 }
  18
  19 if (program.resolution !== undefined && Number.isNaN(+program.resolution)) {
  20   console.error('The resolution must be an integer (example: 1080).')
  21   process.exit(-1)
  22 }
  23
  24 run()
  25   .then(() => process.exit(0))
  26   .catch(err => {
  27     console.error(err)
  28     process.exit(-1)
  29   })
  30
  31 async function run () {
  32   await initDatabaseModels(true)
  33
  34   const video = await VideoModel.loadByUUID(program['video'])
  35   if (!video) throw new Error('Video not found.')
  36
  37   const dataInput = {
  38     videoUUID: video.uuid,
  39     isNewVideo: false,
  40     resolution: undefined
  41   }
  42
  43   if (program.resolution !== undefined) {
  44     dataInput.resolution = program.resolution
  45   }
  46
  47   await JobQueue.Instance.init()
  48   await JobQueue.Instance.createJob({ type: 'video-file', payload: dataInput })
  49   console.log('Transcoding job for video %s created.', video.uuid)
  50 }