1 import { remove } from 'fs-extra'
2 import { throttle } from 'lodash'
3 import { ConfigManager, downloadFile, logger } from 'packages/peertube-runner/shared'
4 import { join } from 'path'
5 import { buildUUID } from '@shared/extra-utils'
6 import { FFmpegEdition, FFmpegLive, FFmpegVOD, getDefaultAvailableEncoders, getDefaultEncodersToTry } from '@shared/ffmpeg'
7 import { RunnerJob, RunnerJobPayload } from '@shared/models'
8 import { PeerTubeServer } from '@shared/server-commands'
9 import { getTranscodingLogger } from './transcoding-logger'
11 export type JobWithToken <T extends RunnerJobPayload = RunnerJobPayload> = RunnerJob<T> & { jobToken: string }
13 export type ProcessOptions <T extends RunnerJobPayload = RunnerJobPayload> = {
14 server: PeerTubeServer
19 export async function downloadInputFile (options: {
24 const { url, job, runnerToken } = options
25 const destination = join(ConfigManager.Instance.getTranscodingDirectory(), buildUUID())
28 await downloadFile({ url, jobToken: job.jobToken, runnerToken, destination })
31 .catch(err => logger.error({ err }, `Cannot remove ${destination}`))
39 export async function updateTranscodingProgress (options: {
40 server: PeerTubeServer
45 const { server, job, runnerToken, progress } = options
47 return server.runnerJobs.update({ jobToken: job.jobToken, jobUUID: job.uuid, runnerToken, progress })
50 // ---------------------------------------------------------------------------
52 export function buildFFmpegVOD (options: {
53 server: PeerTubeServer
57 const { server, job, runnerToken } = options
59 const updateInterval = ConfigManager.Instance.isTestInstance()
63 const updateJobProgress = throttle((progress: number) => {
64 if (progress < 0 || progress > 100) progress = undefined
66 updateTranscodingProgress({ server, job, runnerToken, progress })
67 .catch(err => logger.error({ err }, 'Cannot send job progress'))
68 }, updateInterval, { trailing: false })
70 return new FFmpegVOD({
71 ...getCommonFFmpegOptions(),
77 export function buildFFmpegLive () {
78 return new FFmpegLive(getCommonFFmpegOptions())
81 export function buildFFmpegEdition () {
82 return new FFmpegEdition(getCommonFFmpegOptions())
85 function getCommonFFmpegOptions () {
86 const config = ConfigManager.Instance.getConfig()
89 niceness: config.ffmpeg.nice,
90 threads: config.ffmpeg.threads,
91 tmpDirectory: ConfigManager.Instance.getTranscodingDirectory(),
94 available: getDefaultAvailableEncoders(),
95 encodersToTry: getDefaultEncodersToTry()
97 logger: getTranscodingLogger()