X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=scripts%2Fcreate-transcoding-job.ts;h=244c38fcdd21d0a95f59b98733865abc6b4ae391;hb=63a3d336f6cc9a293a07fdc12d6bdfb86cfc2fd5;hp=65e65b6162cff629bc1526e0df924c887830eb0c;hpb=cf21b2cbef61929177b9c09b5e017c3b7eb8535d;p=github%2FChocobozzz%2FPeerTube.git diff --git a/scripts/create-transcoding-job.ts b/scripts/create-transcoding-job.ts index 65e65b616..244c38fcd 100755 --- a/scripts/create-transcoding-job.ts +++ b/scripts/create-transcoding-job.ts @@ -5,10 +5,11 @@ 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 { VideoTranscodingPayload } from '@shared/models' +import { computeLowerResolutionsToTranscode } from '@server/helpers/ffprobe-utils' +import { VideoState, VideoTranscodingPayload } from '@shared/models' import { CONFIG } from '@server/initializers/config' -import { isUUIDValid } from '@server/helpers/custom-validators/misc' +import { isUUIDValid, toCompleteUUID } from '@server/helpers/custom-validators/misc' +import { addTranscodingJob } from '@server/lib/video' program .option('-v, --video [videoUUID]', 'Video UUID') @@ -38,22 +39,24 @@ run() async function run () { await initDatabaseModels(true) - if (isUUIDValid(options.video) === false) { + 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(options.video) + const video = await VideoModel.loadAndPopulateAccountAndServerAndTags(uuid) if (!video) throw new Error('Video not found.') const dataInput: VideoTranscodingPayload[] = [] - const { videoFileResolution } = await video.getMaxQualityResolution() + const maxResolution = video.getMaxQualityFile().resolution // Generate HLS files if (options.generateHls || CONFIG.TRANSCODING.WEBTORRENT.ENABLED === false) { const resolutionsEnabled = options.resolution - ? [ options.resolution ] - : computeResolutionsToTranscode(videoFileResolution, 'vod').concat([ videoFileResolution ]) + ? [ parseInt(options.resolution) ] + : computeLowerResolutionsToTranscode(maxResolution, 'vod').concat([ maxResolution ]) for (const resolution of resolutionsEnabled) { dataInput.push({ @@ -62,7 +65,9 @@ async function run () { resolution, isPortraitMode: false, copyCodecs: false, - isMaxQuality: false + isNewVideo: false, + isMaxQuality: maxResolution === resolution, + autoDeleteWebTorrentIfNeeded: false }) } } else { @@ -71,7 +76,7 @@ async function run () { type: 'new-resolution-to-webtorrent', videoUUID: video.uuid, isNewVideo: false, - resolution: options.resolution + resolution: parseInt(options.resolution) }) } else { if (video.VideoFiles.length === 0) { @@ -87,10 +92,13 @@ async function run () { } } - await JobQueue.Instance.init() + JobQueue.Instance.init(true) + + video.state = VideoState.TO_TRANSCODE + await video.save() for (const d of dataInput) { - await JobQueue.Instance.createJobWithPromise({ type: 'video-transcoding', payload: d }) + await addTranscodingJob(d, {}) console.log('Transcoding job for video %s created.', video.uuid) } }