1 import { remove } from 'fs-extra'
2 import { pick } from 'lodash'
3 import { logger } from 'packages/peertube-runner/shared'
4 import { extname, join } from 'path'
5 import { buildUUID } from '@shared/extra-utils'
7 RunnerJobVideoEditionTranscodingPayload,
8 VideoEditionTranscodingSuccess,
10 VideoStudioTaskCutPayload,
11 VideoStudioTaskIntroPayload,
12 VideoStudioTaskOutroPayload,
13 VideoStudioTaskPayload,
14 VideoStudioTaskWatermarkPayload
15 } from '@shared/models'
16 import { ConfigManager } from '../../../shared/config-manager'
17 import { buildFFmpegEdition, downloadInputFile, JobWithToken, ProcessOptions } from './common'
19 export async function processStudioTranscoding (options: ProcessOptions<RunnerJobVideoEditionTranscodingPayload>) {
20 const { server, job, runnerToken } = options
21 const payload = job.payload
23 let outputPath: string
24 const inputPath = await downloadInputFile({ url: payload.input.videoFileUrl, runnerToken, job })
25 let tmpInputFilePath = inputPath
28 for (const task of payload.tasks) {
29 const outputFilename = 'output-edition-' + buildUUID() + '.mp4'
30 outputPath = join(ConfigManager.Instance.getTranscodingDirectory(), outputFilename)
33 inputPath: tmpInputFilePath,
40 if (tmpInputFilePath) await remove(tmpInputFilePath)
42 // For the next iteration
43 tmpInputFilePath = outputPath
46 const successBody: VideoEditionTranscodingSuccess = {
50 await server.runnerJobs.success({
51 jobToken: job.jobToken,
57 await remove(tmpInputFilePath)
58 await remove(outputPath)
62 // ---------------------------------------------------------------------------
64 // ---------------------------------------------------------------------------
66 type TaskProcessorOptions <T extends VideoStudioTaskPayload = VideoStudioTaskPayload> = {
74 const taskProcessors: { [id in VideoStudioTask['name']]: (options: TaskProcessorOptions) => Promise<any> } = {
75 'add-intro': processAddIntroOutro,
76 'add-outro': processAddIntroOutro,
78 'add-watermark': processAddWatermark
81 async function processTask (options: TaskProcessorOptions) {
82 const { task } = options
84 const processor = taskProcessors[options.task.name]
85 if (!process) throw new Error('Unknown task ' + task.name)
87 return processor(options)
90 async function processAddIntroOutro (options: TaskProcessorOptions<VideoStudioTaskIntroPayload | VideoStudioTaskOutroPayload>) {
91 const { inputPath, task, runnerToken, job } = options
93 logger.debug('Adding intro/outro to ' + inputPath)
95 const introOutroPath = await downloadInputFile({ url: task.options.file, runnerToken, job })
97 return buildFFmpegEdition().addIntroOutro({
98 ...pick(options, [ 'inputPath', 'outputPath' ]),
101 type: task.name === 'add-intro'
107 function processCut (options: TaskProcessorOptions<VideoStudioTaskCutPayload>) {
108 const { inputPath, task } = options
110 logger.debug(`Cutting ${inputPath}`)
112 return buildFFmpegEdition().cutVideo({
113 ...pick(options, [ 'inputPath', 'outputPath' ]),
115 start: task.options.start,
116 end: task.options.end
120 async function processAddWatermark (options: TaskProcessorOptions<VideoStudioTaskWatermarkPayload>) {
121 const { inputPath, task, runnerToken, job } = options
123 logger.debug('Adding watermark to ' + inputPath)
125 const watermarkPath = await downloadInputFile({ url: task.options.file, runnerToken, job })
127 return buildFFmpegEdition().addWatermark({
128 ...pick(options, [ 'inputPath', 'outputPath' ]),
133 watermarkSizeRatio: task.options.watermarkSizeRatio,
134 horitonzalMarginRatio: task.options.horitonzalMarginRatio,
135 verticalMarginRatio: task.options.verticalMarginRatio