107 lines
2.9 KiB
TypeScript
107 lines
2.9 KiB
TypeScript
import { remove } from 'fs-extra'
|
|
import { ConfigManager, downloadFile, logger } from 'packages/peertube-runner/shared'
|
|
import { join } from 'path'
|
|
import { buildUUID } from '@shared/extra-utils'
|
|
import { FFmpegEdition, FFmpegLive, FFmpegVOD, getDefaultAvailableEncoders, getDefaultEncodersToTry } from '@shared/ffmpeg'
|
|
import { RunnerJob, RunnerJobPayload } from '@shared/models'
|
|
import { PeerTubeServer } from '@shared/server-commands'
|
|
import { getTranscodingLogger } from './transcoding-logger'
|
|
|
|
export type JobWithToken <T extends RunnerJobPayload = RunnerJobPayload> = RunnerJob<T> & { jobToken: string }
|
|
|
|
export type ProcessOptions <T extends RunnerJobPayload = RunnerJobPayload> = {
|
|
server: PeerTubeServer
|
|
job: JobWithToken<T>
|
|
runnerToken: string
|
|
}
|
|
|
|
export async function downloadInputFile (options: {
|
|
url: string
|
|
job: JobWithToken
|
|
runnerToken: string
|
|
}) {
|
|
const { url, job, runnerToken } = options
|
|
const destination = join(ConfigManager.Instance.getTranscodingDirectory(), buildUUID())
|
|
|
|
try {
|
|
await downloadFile({ url, jobToken: job.jobToken, runnerToken, destination })
|
|
} catch (err) {
|
|
remove(destination)
|
|
.catch(err => logger.error({ err }, `Cannot remove ${destination}`))
|
|
|
|
throw err
|
|
}
|
|
|
|
return destination
|
|
}
|
|
|
|
export function scheduleTranscodingProgress (options: {
|
|
server: PeerTubeServer
|
|
runnerToken: string
|
|
job: JobWithToken
|
|
progressGetter: () => number
|
|
}) {
|
|
const { job, server, progressGetter, runnerToken } = options
|
|
|
|
const updateInterval = ConfigManager.Instance.isTestInstance()
|
|
? 500
|
|
: 60000
|
|
|
|
const update = () => {
|
|
server.runnerJobs.update({ jobToken: job.jobToken, jobUUID: job.uuid, runnerToken, progress: progressGetter() })
|
|
.catch(err => logger.error({ err }, 'Cannot send job progress'))
|
|
}
|
|
|
|
const interval = setInterval(() => {
|
|
update()
|
|
}, updateInterval)
|
|
|
|
update()
|
|
|
|
return interval
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------
|
|
|
|
export function buildFFmpegVOD (options: {
|
|
onJobProgress: (progress: number) => void
|
|
}) {
|
|
const { onJobProgress } = options
|
|
|
|
return new FFmpegVOD({
|
|
...getCommonFFmpegOptions(),
|
|
|
|
updateJobProgress: arg => {
|
|
const progress = arg < 0 || arg > 100
|
|
? undefined
|
|
: arg
|
|
|
|
onJobProgress(progress)
|
|
}
|
|
})
|
|
}
|
|
|
|
export function buildFFmpegLive () {
|
|
return new FFmpegLive(getCommonFFmpegOptions())
|
|
}
|
|
|
|
export function buildFFmpegEdition () {
|
|
return new FFmpegEdition(getCommonFFmpegOptions())
|
|
}
|
|
|
|
function getCommonFFmpegOptions () {
|
|
const config = ConfigManager.Instance.getConfig()
|
|
|
|
return {
|
|
niceness: config.ffmpeg.nice,
|
|
threads: config.ffmpeg.threads,
|
|
tmpDirectory: ConfigManager.Instance.getTranscodingDirectory(),
|
|
profile: 'default',
|
|
availableEncoders: {
|
|
available: getDefaultAvailableEncoders(),
|
|
encodersToTry: getDefaultEncodersToTry()
|
|
},
|
|
logger: getTranscodingLogger()
|
|
}
|
|
}
|