server/lib/live/shared/transcoding-wrapper/ffmpeg-transcoding-wrapper.ts

   1 import { FfmpegCommand } from 'fluent-ffmpeg'
   2 import { getFFmpegCommandWrapperOptions } from '@server/helpers/ffmpeg'
   3 import { logger } from '@server/helpers/logger'
   4 import { CONFIG } from '@server/initializers/config'
   5 import { VIDEO_LIVE } from '@server/initializers/constants'
   6 import { VideoTranscodingProfilesManager } from '@server/lib/transcoding/default-transcoding-profiles'
   7 import { FFmpegLive } from '@shared/ffmpeg'
   8 import { getLiveSegmentTime } from '../../live-utils'
   9 import { AbstractTranscodingWrapper } from './abstract-transcoding-wrapper'
  10
  11 export class FFmpegTranscodingWrapper extends AbstractTranscodingWrapper {
  12   private ffmpegCommand: FfmpegCommand
  13
  14   private aborted = false
  15   private errored = false
  16   private ended = false
  17
  18   async run () {
  19     this.ffmpegCommand = CONFIG.LIVE.TRANSCODING.ENABLED
  20       ? await this.buildFFmpegLive().getLiveTranscodingCommand({
  21         inputUrl: this.inputLocalUrl,
  22
  23         outPath: this.outDirectory,
  24         masterPlaylistName: this.streamingPlaylist.playlistFilename,
  25
  26         segmentListSize: this.segmentListSize,
  27         segmentDuration: this.segmentDuration,
  28
  29         toTranscode: this.toTranscode,
  30
  31         bitrate: this.bitrate,
  32         ratio: this.ratio,
  33
  34         hasAudio: this.hasAudio
  35       })
  36       : this.buildFFmpegLive().getLiveMuxingCommand({
  37         inputUrl: this.inputLocalUrl,
  38         outPath: this.outDirectory,
  39
  40         masterPlaylistName: this.streamingPlaylist.playlistFilename,
  41
  42         segmentListSize: VIDEO_LIVE.SEGMENTS_LIST_SIZE,
  43         segmentDuration: getLiveSegmentTime(this.videoLive.latencyMode)
  44       })
  45
  46     logger.info('Running local live muxing/transcoding for %s.', this.videoUUID, this.lTags())
  47
  48     this.ffmpegCommand.run()
  49
  50     let ffmpegShellCommand: string
  51     this.ffmpegCommand.on('start', cmdline => {
  52       ffmpegShellCommand = cmdline
  53
  54       logger.debug('Running ffmpeg command for live', { ffmpegShellCommand, ...this.lTags() })
  55     })
  56
  57     this.ffmpegCommand.on('error', (err, stdout, stderr) => {
  58       this.onFFmpegError({ err, stdout, stderr, ffmpegShellCommand })
  59     })
  60
  61     this.ffmpegCommand.on('end', () => {
  62       this.onFFmpegEnded()
  63     })
  64
  65     this.ffmpegCommand.run()
  66   }
  67
  68   abort () {
  69     if (this.ended || this.errored || this.aborted) return
  70
  71     this.ffmpegCommand.kill('SIGINT')
  72
  73     this.aborted = true
  74     this.emit('end')
  75   }
  76
  77   private onFFmpegError (options: {
  78     err: any
  79     stdout: string
  80     stderr: string
  81     ffmpegShellCommand: string
  82   }) {
  83     const { err, stdout, stderr, ffmpegShellCommand } = options
  84
  85     // Don't care that we killed the ffmpeg process
  86     if (err?.message?.includes('Exiting normally')) return
  87     if (this.ended || this.errored || this.aborted) return
  88
  89     logger.error('FFmpeg transcoding error.', { err, stdout, stderr, ffmpegShellCommand, ...this.lTags() })
  90
  91     this.errored = true
  92     this.emit('error', { err })
  93   }
  94
  95   private onFFmpegEnded () {
  96     if (this.ended || this.errored || this.aborted) return
  97
  98     this.ended = true
  99     this.emit('end')
 100   }
 101
 102   private buildFFmpegLive () {
 103     return new FFmpegLive(getFFmpegCommandWrapperOptions('live', VideoTranscodingProfilesManager.Instance.getAvailableEncoders()))
 104   }
 105 }