123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103 |
- import { registerTSPaths } from '../server/helpers/register-ts-paths'
- registerTSPaths()
- import { program } from 'commander'
- import { VideoModel } from '../server/models/video/video'
- import { initDatabaseModels } from '../server/initializers/database'
- import { JobQueue } from '../server/lib/job-queue'
- import { computeResolutionsToTranscode } from '@server/helpers/ffprobe-utils'
- import { VideoState, VideoTranscodingPayload } from '@shared/models'
- import { CONFIG } from '@server/initializers/config'
- import { isUUIDValid, toCompleteUUID } from '@server/helpers/custom-validators/misc'
- import { addTranscodingJob } from '@server/lib/video'
- program
- .option('-v, --video [videoUUID]', 'Video UUID')
- .option('-r, --resolution [resolution]', 'Video resolution (integer)')
- .option('--generate-hls', 'Generate HLS playlist')
- .parse(process.argv)
- const options = program.opts()
- if (options.video === undefined) {
- console.error('All parameters are mandatory.')
- process.exit(-1)
- }
- if (options.resolution !== undefined && Number.isNaN(+options.resolution)) {
- console.error('The resolution must be an integer (example: 1080).')
- process.exit(-1)
- }
- run()
- .then(() => process.exit(0))
- .catch(err => {
- console.error(err)
- process.exit(-1)
- })
- async function run () {
- await initDatabaseModels(true)
- const uuid = toCompleteUUID(options.video)
- if (isUUIDValid(uuid) === false) {
- console.error('%s is not a valid video UUID.', options.video)
- return
- }
- const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(uuid)
- if (!video) throw new Error('Video not found.')
- const dataInput: VideoTranscodingPayload[] = []
- const resolution = video.getMaxQualityFile().resolution
- // Generate HLS files
- if (options.generateHls || CONFIG.TRANSCODING.WEBTORRENT.ENABLED === false) {
- const resolutionsEnabled = options.resolution
- ? [ options.resolution ]
- : computeResolutionsToTranscode(resolution, 'vod').concat([ resolution ])
- for (const resolution of resolutionsEnabled) {
- dataInput.push({
- type: 'new-resolution-to-hls',
- videoUUID: video.uuid,
- resolution,
- isPortraitMode: false,
- copyCodecs: false,
- isNewVideo: false,
- isMaxQuality: false
- })
- }
- } else {
- if (options.resolution !== undefined) {
- dataInput.push({
- type: 'new-resolution-to-webtorrent',
- videoUUID: video.uuid,
- isNewVideo: false,
- resolution: options.resolution
- })
- } else {
- if (video.VideoFiles.length === 0) {
- console.error('Cannot regenerate webtorrent files with a HLS only video.')
- return
- }
- dataInput.push({
- type: 'optimize-to-webtorrent',
- videoUUID: video.uuid,
- isNewVideo: false
- })
- }
- }
- JobQueue.Instance.init()
- video.state = VideoState.TO_TRANSCODE
- await video.save()
- for (const d of dataInput) {
- await addTranscodingJob(d, {})
- console.log('Transcoding job for video %s created.', video.uuid)
- }
- }
|