1 files changed, 102 insertions, 0 deletions

diff --git a/shared/extra-utils/videos/live.ts b/shared/extra-utils/videos/live.ts
new file mode 100644
index 000000000..f500fdc3e
--- /dev/null
+++ b/shared/extra-utils/videos/live.ts
@@ -0,0 +1,102 @@
+import * as ffmpeg from 'fluent-ffmpeg'
+import { LiveVideoCreate, LiveVideoUpdate, VideoDetails, VideoState } from '@shared/models'
+import { buildAbsoluteFixturePath, wait } from '../miscs/miscs'
+import { makeGetRequest, makePutBodyRequest, makeUploadRequest } from '../requests/requests'
+import { ServerInfo } from '../server/servers'
+import { getVideo, getVideoWithToken } from './videos'
+
+function getLive (url: string, token: string, videoId: number | string, statusCodeExpected = 200) {
+  const path = '/api/v1/videos/live'
+
+  return makeGetRequest({
+    url,
+    token,
+    path: path + '/' + videoId,
+    statusCodeExpected
+  })
+}
+
+function updateLive (url: string, token: string, videoId: number | string, fields: LiveVideoUpdate, statusCodeExpected = 204) {
+  const path = '/api/v1/videos/live'
+
+  return makePutBodyRequest({
+    url,
+    token,
+    path: path + '/' + videoId,
+    fields,
+    statusCodeExpected
+  })
+}
+
+function createLive (url: string, token: string, fields: LiveVideoCreate, statusCodeExpected = 200) {
+  const path = '/api/v1/videos/live'
+
+  let attaches: any = {}
+  if (fields.thumbnailfile) attaches = { thumbnailfile: fields.thumbnailfile }
+  if (fields.previewfile) attaches = { previewfile: fields.previewfile }
+
+  return makeUploadRequest({
+    url,
+    path,
+    token,
+    attaches,
+    fields,
+    statusCodeExpected
+  })
+}
+
+function sendRTMPStream (rtmpBaseUrl: string, streamKey: string) {
+  const fixture = buildAbsoluteFixturePath('video_short.mp4')
+
+  const command = ffmpeg(fixture)
+  command.inputOption('-stream_loop -1')
+  command.inputOption('-re')
+
+  command.outputOption('-c copy')
+  command.outputOption('-f flv')
+
+  const rtmpUrl = rtmpBaseUrl + '/' + streamKey
+  command.output(rtmpUrl)
+
+  command.on('error', err => {
+    if (err?.message?.includes('Exiting normally')) return
+
+    console.error('Cannot send RTMP stream.', { err })
+  })
+
+  if (process.env.DEBUG) {
+    command.on('stderr', data => console.log(data))
+  }
+
+  command.run()
+
+  return command
+}
+
+async function stopFfmpeg (command: ffmpeg.FfmpegCommand) {
+  command.kill('SIGINT')
+
+  await wait(500)
+}
+
+async function waitUntilLiveStarts (url: string, token: string, videoId: number | string) {
+  let video: VideoDetails
+
+  do {
+    const res = await getVideoWithToken(url, token, videoId)
+    video = res.body
+
+    await wait(500)
+  } while (video.state.id === VideoState.WAITING_FOR_LIVE)
+}
+
+// ---------------------------------------------------------------------------
+
+export {
+  getLive,
+  updateLive,
+  waitUntilLiveStarts,
+  createLive,
+  stopFfmpeg,
+  sendRTMPStream
+}