shared/server-commands/videos/live.ts

   1 import ffmpeg, { FfmpegCommand } from 'fluent-ffmpeg'
   2 import { truncate } from 'lodash'
   3 import { buildAbsoluteFixturePath, wait } from '@shared/core-utils'
   4 import { VideoDetails, VideoInclude, VideoPrivacy } from '@shared/models'
   5 import { PeerTubeServer } from '../server/server'
   6
   7 function sendRTMPStream (options: {
   8   rtmpBaseUrl: string
   9   streamKey: string
  10   fixtureName?: string // default video_short.mp4
  11   copyCodecs?: boolean // default false
  12 }) {
  13   const { rtmpBaseUrl, streamKey, fixtureName = 'video_short.mp4', copyCodecs = false } = options
  14
  15   const fixture = buildAbsoluteFixturePath(fixtureName)
  16
  17   const command = ffmpeg(fixture)
  18   command.inputOption('-stream_loop -1')
  19   command.inputOption('-re')
  20
  21   if (copyCodecs) {
  22     command.outputOption('-c copy')
  23   } else {
  24     command.outputOption('-c:v libx264')
  25     command.outputOption('-g 120')
  26     command.outputOption('-x264-params "no-scenecut=1"')
  27     command.outputOption('-r 60')
  28   }
  29
  30   command.outputOption('-f flv')
  31
  32   const rtmpUrl = rtmpBaseUrl + '/' + streamKey
  33   command.output(rtmpUrl)
  34
  35   command.on('error', err => {
  36     if (err?.message?.includes('Exiting normally')) return
  37
  38     if (process.env.DEBUG) console.error(err)
  39   })
  40
  41   if (process.env.DEBUG) {
  42     command.on('stderr', data => console.log(data))
  43     command.on('stdout', data => console.log(data))
  44   }
  45
  46   command.run()
  47
  48   return command
  49 }
  50
  51 function waitFfmpegUntilError (command: FfmpegCommand, successAfterMS = 10000) {
  52   return new Promise<void>((res, rej) => {
  53     command.on('error', err => {
  54       return rej(err)
  55     })
  56
  57     setTimeout(() => {
  58       res()
  59     }, successAfterMS)
  60   })
  61 }
  62
  63 async function testFfmpegStreamError (command: FfmpegCommand, shouldHaveError: boolean) {
  64   let error: Error
  65
  66   try {
  67     await waitFfmpegUntilError(command, 45000)
  68   } catch (err) {
  69     error = err
  70   }
  71
  72   await stopFfmpeg(command)
  73
  74   if (shouldHaveError && !error) throw new Error('Ffmpeg did not have an error')
  75   if (!shouldHaveError && error) throw error
  76 }
  77
  78 async function stopFfmpeg (command: FfmpegCommand) {
  79   command.kill('SIGINT')
  80
  81   await wait(500)
  82 }
  83
  84 async function waitUntilLivePublishedOnAllServers (servers: PeerTubeServer[], videoId: string) {
  85   for (const server of servers) {
  86     await server.live.waitUntilPublished({ videoId })
  87   }
  88 }
  89
  90 async function waitUntilLiveWaitingOnAllServers (servers: PeerTubeServer[], videoId: string) {
  91   for (const server of servers) {
  92     await server.live.waitUntilWaiting({ videoId })
  93   }
  94 }
  95
  96 async function waitUntilLiveReplacedByReplayOnAllServers (servers: PeerTubeServer[], videoId: string) {
  97   for (const server of servers) {
  98     await server.live.waitUntilReplacedByReplay({ videoId })
  99   }
 100 }
 101
 102 async function findExternalSavedVideo (server: PeerTubeServer, liveDetails: VideoDetails) {
 103   const include = VideoInclude.BLACKLISTED
 104   const privacyOneOf = [ VideoPrivacy.INTERNAL, VideoPrivacy.PRIVATE, VideoPrivacy.PUBLIC, VideoPrivacy.UNLISTED ]
 105
 106   const { data } = await server.videos.list({ token: server.accessToken, sort: '-publishedAt', include, privacyOneOf })
 107
 108   const videoNameSuffix = ` - ${new Date(liveDetails.publishedAt).toLocaleString()}`
 109   const truncatedVideoName = truncate(liveDetails.name, {
 110     length: 120 - videoNameSuffix.length
 111   })
 112   const toFind = truncatedVideoName + videoNameSuffix
 113
 114   return data.find(v => v.name === toFind)
 115 }
 116
 117 export {
 118   sendRTMPStream,
 119   waitFfmpegUntilError,
 120   testFfmpegStreamError,
 121   stopFfmpeg,
 122
 123   waitUntilLivePublishedOnAllServers,
 124   waitUntilLiveReplacedByReplayOnAllServers,
 125   waitUntilLiveWaitingOnAllServers,
 126
 127   findExternalSavedVideo
 128 }