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