server/lib/job-queue/handlers/video-live-ending.ts

   1 import * as Bull from 'bull'
   2 import { readdir, remove } from 'fs-extra'
   3 import { join } from 'path'
   4 import { getVideoFileResolution, hlsPlaylistToFragmentedMP4 } from '@server/helpers/ffmpeg-utils'
   5 import { getHLSDirectory } from '@server/lib/video-paths'
   6 import { generateHlsPlaylist } from '@server/lib/video-transcoding'
   7 import { VideoModel } from '@server/models/video/video'
   8 import { VideoLiveModel } from '@server/models/video/video-live'
   9 import { VideoStreamingPlaylistModel } from '@server/models/video/video-streaming-playlist'
  10 import { MStreamingPlaylist, MVideo } from '@server/types/models'
  11 import { VideoLiveEndingPayload, VideoState } from '@shared/models'
  12 import { logger } from '../../../helpers/logger'
  13
  14 async function processVideoLiveEnding (job: Bull.Job) {
  15   const payload = job.data as VideoLiveEndingPayload
  16
  17   const video = await VideoModel.load(payload.videoId)
  18   const live = await VideoLiveModel.loadByVideoId(payload.videoId)
  19
  20   const streamingPlaylist = await VideoStreamingPlaylistModel.loadHLSPlaylistByVideo(video.id)
  21   if (!video || !streamingPlaylist || !live) {
  22     logger.warn('Video live %d does not exist anymore. Cannot process live ending.', payload.videoId)
  23     return
  24   }
  25
  26   if (live.saveReplay !== true) {
  27     return cleanupLive(video, streamingPlaylist)
  28   }
  29
  30   return saveLive(video, streamingPlaylist)
  31 }
  32
  33 // ---------------------------------------------------------------------------
  34
  35 export {
  36   processVideoLiveEnding
  37 }
  38
  39 // ---------------------------------------------------------------------------
  40
  41 async function saveLive (video: MVideo, streamingPlaylist: MStreamingPlaylist) {
  42   const videoFiles = await streamingPlaylist.get('VideoFiles')
  43   const hlsDirectory = getHLSDirectory(video, false)
  44
  45   for (const videoFile of videoFiles) {
  46     const playlistPath = join(hlsDirectory, VideoStreamingPlaylistModel.getHlsPlaylistFilename(videoFile.resolution))
  47
  48     const mp4TmpName = buildMP4TmpName(videoFile.resolution)
  49     await hlsPlaylistToFragmentedMP4(playlistPath, mp4TmpName)
  50   }
  51
  52   await cleanupLiveFiles(hlsDirectory)
  53
  54   video.isLive = false
  55   video.state = VideoState.TO_TRANSCODE
  56   await video.save()
  57
  58   const videoWithFiles = await VideoModel.loadWithFiles(video.id)
  59
  60   for (const videoFile of videoFiles) {
  61     const videoInputPath = buildMP4TmpName(videoFile.resolution)
  62     const { isPortraitMode } = await getVideoFileResolution(videoInputPath)
  63
  64     await generateHlsPlaylist({
  65       video: videoWithFiles,
  66       videoInputPath,
  67       resolution: videoFile.resolution,
  68       copyCodecs: true,
  69       isPortraitMode
  70     })
  71   }
  72
  73   video.state = VideoState.PUBLISHED
  74   await video.save()
  75 }
  76
  77 async function cleanupLive (video: MVideo, streamingPlaylist: MStreamingPlaylist) {
  78   const hlsDirectory = getHLSDirectory(video, false)
  79
  80   await cleanupLiveFiles(hlsDirectory)
  81
  82   streamingPlaylist.destroy()
  83     .catch(err => logger.error('Cannot remove live streaming playlist.', { err }))
  84 }
  85
  86 async function cleanupLiveFiles (hlsDirectory: string) {
  87   const files = await readdir(hlsDirectory)
  88
  89   for (const filename of files) {
  90     if (
  91       filename.endsWith('.ts') ||
  92       filename.endsWith('.m3u8') ||
  93       filename.endsWith('.mpd') ||
  94       filename.endsWith('.m4s') ||
  95       filename.endsWith('.tmp')
  96     ) {
  97       const p = join(hlsDirectory, filename)
  98
  99       remove(p)
 100         .catch(err => logger.error('Cannot remove %s.', p, { err }))
 101     }
 102   }
 103 }
 104
 105 function buildMP4TmpName (resolution: number) {
 106   return resolution + 'tmp.mp4'
 107 }