server/helpers/ffmpeg/ffmpeg-options.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

import { logger } from '@server/helpers/logger'
import { CONFIG } from '@server/initializers/config'
import { FFMPEG_NICE } from '@server/initializers/constants'
import { FFmpegCommandWrapperOptions } from '@shared/ffmpeg'
import { AvailableEncoders } from '@shared/models'

type CommandType = 'live' | 'vod' | 'thumbnail'

export function getFFmpegCommandWrapperOptions (type: CommandType, availableEncoders?: AvailableEncoders): FFmpegCommandWrapperOptions {
  return {
    availableEncoders,
    profile: getProfile(type),

    niceness: FFMPEG_NICE[type.toUpperCase()],
    tmpDirectory: CONFIG.STORAGE.TMP_DIR,
    threads: getThreads(type),

    logger: {
      debug: logger.debug.bind(logger),
      info: logger.info.bind(logger),
      warn: logger.warn.bind(logger),
      error: logger.error.bind(logger)
    },
    lTags: { tags: [ 'ffmpeg' ] }
  }
}

// ---------------------------------------------------------------------------
// Private
// ---------------------------------------------------------------------------

function getThreads (type: CommandType) {
  if (type === 'live') return CONFIG.LIVE.TRANSCODING.THREADS
  if (type === 'vod') return CONFIG.TRANSCODING.THREADS

  // Auto
  return 0
}

function getProfile (type: CommandType) {
  if (type === 'live') return CONFIG.LIVE.TRANSCODING.PROFILE
  if (type === 'vod') return CONFIG.TRANSCODING.PROFILE

  return undefined
}