]>
Commit | Line | Data |
---|---|---|
0c9668f7 C |
1 | import { pick } from 'lodash' |
2 | import { logger } from '@server/helpers/logger' | |
3 | import { VideoJobInfoModel } from '@server/models/video/video-job-info' | |
4 | import { MVideo } from '@server/types/models' | |
5 | import { MRunnerJob } from '@server/types/models/runners' | |
6 | import { buildUUID } from '@shared/extra-utils' | |
7 | import { getVideoStreamDuration } from '@shared/ffmpeg' | |
8 | import { | |
9 | RunnerJobUpdatePayload, | |
10 | RunnerJobVODAudioMergeTranscodingPayload, | |
11 | RunnerJobVODWebVideoTranscodingPrivatePayload, | |
12 | VODAudioMergeTranscodingSuccess | |
13 | } from '@shared/models' | |
14 | import { generateRunnerTranscodingVideoInputFileUrl, generateRunnerTranscodingVideoPreviewFileUrl } from '../runner-urls' | |
15 | import { AbstractVODTranscodingJobHandler } from './abstract-vod-transcoding-job-handler' | |
16 | import { loadTranscodingRunnerVideo, onVODWebVideoOrAudioMergeTranscodingJob } from './shared' | |
17 | ||
18 | type CreateOptions = { | |
19 | video: MVideo | |
20 | isNewVideo: boolean | |
21 | resolution: number | |
22 | fps: number | |
23 | priority: number | |
24 | dependsOnRunnerJob?: MRunnerJob | |
25 | } | |
26 | ||
27 | // eslint-disable-next-line max-len | |
28 | export class VODAudioMergeTranscodingJobHandler extends AbstractVODTranscodingJobHandler<CreateOptions, RunnerJobUpdatePayload, VODAudioMergeTranscodingSuccess> { | |
29 | ||
30 | async create (options: CreateOptions) { | |
31 | const { video, resolution, fps, priority, dependsOnRunnerJob } = options | |
32 | ||
33 | const jobUUID = buildUUID() | |
34 | const payload: RunnerJobVODAudioMergeTranscodingPayload = { | |
35 | input: { | |
36 | audioFileUrl: generateRunnerTranscodingVideoInputFileUrl(jobUUID, video.uuid), | |
37 | previewFileUrl: generateRunnerTranscodingVideoPreviewFileUrl(jobUUID, video.uuid) | |
38 | }, | |
39 | output: { | |
40 | resolution, | |
41 | fps | |
42 | } | |
43 | } | |
44 | ||
45 | const privatePayload: RunnerJobVODWebVideoTranscodingPrivatePayload = { | |
46 | ...pick(options, [ 'isNewVideo' ]), | |
47 | ||
48 | videoUUID: video.uuid | |
49 | } | |
50 | ||
51 | const job = await this.createRunnerJob({ | |
52 | type: 'vod-audio-merge-transcoding', | |
53 | jobUUID, | |
54 | payload, | |
55 | privatePayload, | |
56 | priority, | |
57 | dependsOnRunnerJob | |
58 | }) | |
59 | ||
60 | await VideoJobInfoModel.increaseOrCreate(video.uuid, 'pendingTranscode') | |
61 | ||
62 | return job | |
63 | } | |
64 | ||
65 | // --------------------------------------------------------------------------- | |
66 | ||
5e47f6ab | 67 | protected async specificComplete (options: { |
0c9668f7 C |
68 | runnerJob: MRunnerJob |
69 | resultPayload: VODAudioMergeTranscodingSuccess | |
70 | }) { | |
71 | const { runnerJob, resultPayload } = options | |
72 | const privatePayload = runnerJob.privatePayload as RunnerJobVODWebVideoTranscodingPrivatePayload | |
73 | ||
74 | const video = await loadTranscodingRunnerVideo(runnerJob, this.lTags) | |
75 | if (!video) return | |
76 | ||
77 | const videoFilePath = resultPayload.videoFile as string | |
78 | ||
79 | // ffmpeg generated a new video file, so update the video duration | |
80 | // See https://trac.ffmpeg.org/ticket/5456 | |
81 | video.duration = await getVideoStreamDuration(videoFilePath) | |
82 | await video.save() | |
83 | ||
84 | // We can remove the old audio file | |
85 | const oldAudioFile = video.VideoFiles[0] | |
86 | await video.removeWebTorrentFile(oldAudioFile) | |
87 | await oldAudioFile.destroy() | |
88 | video.VideoFiles = [] | |
89 | ||
90 | await onVODWebVideoOrAudioMergeTranscodingJob({ video, videoFilePath, privatePayload }) | |
91 | ||
92 | logger.info( | |
93 | 'Runner VOD audio merge transcoding job %s for %s ended.', | |
94 | runnerJob.uuid, video.uuid, this.lTags(video.uuid, runnerJob.uuid) | |
95 | ) | |
96 | } | |
97 | } |