X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=shared%2Fextra-utils%2Fvideos%2Flive.ts;h=c0384769bda247f4dc13ce8a38d70ac2c8879259;hb=d4a8e7a65f97bb3257facc13e1ae8ffdbad61ddb;hp=65942db0a1dda5b67ac4755ecefe2ebc1868de24;hpb=af4ae64f6faf38f8179f2e07d3cd4ad60006be92;p=github%2FChocobozzz%2FPeerTube.git diff --git a/shared/extra-utils/videos/live.ts b/shared/extra-utils/videos/live.ts index 65942db0a..c0384769b 100644 --- a/shared/extra-utils/videos/live.ts +++ b/shared/extra-utils/videos/live.ts @@ -1,11 +1,18 @@ +/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */ + +import { expect } from 'chai' import * as ffmpeg from 'fluent-ffmpeg' -import { LiveVideoCreate, LiveVideoUpdate, VideoDetails, VideoState } from '@shared/models' -import { buildAbsoluteFixturePath, wait } from '../miscs/miscs' +import { pathExists, readdir } from 'fs-extra' +import { omit } from 'lodash' +import { join } from 'path' +import { LiveVideo, LiveVideoCreate, LiveVideoUpdate, VideoDetails, VideoState } from '@shared/models' +import { HttpStatusCode } from '../../../shared/core-utils/miscs/http-error-codes' +import { buildAbsoluteFixturePath, buildServerDirectory, wait } from '../miscs/miscs' import { makeGetRequest, makePutBodyRequest, makeUploadRequest } from '../requests/requests' +import { ServerInfo, waitUntilLog } from '../server/servers' import { getVideoWithToken } from './videos' -import { omit } from 'lodash' -function getLive (url: string, token: string, videoId: number | string, statusCodeExpected = 200) { +function getLive (url: string, token: string, videoId: number | string, statusCodeExpected = HttpStatusCode.OK_200) { const path = '/api/v1/videos/live' return makeGetRequest({ @@ -16,7 +23,13 @@ function getLive (url: string, token: string, videoId: number | string, statusCo }) } -function updateLive (url: string, token: string, videoId: number | string, fields: LiveVideoUpdate, statusCodeExpected = 204) { +function updateLive ( + url: string, + token: string, + videoId: number | string, + fields: LiveVideoUpdate, + statusCodeExpected = HttpStatusCode.NO_CONTENT_204 +) { const path = '/api/v1/videos/live' return makePutBodyRequest({ @@ -28,7 +41,7 @@ function updateLive (url: string, token: string, videoId: number | string, field }) } -function createLive (url: string, token: string, fields: LiveVideoCreate, statusCodeExpected = 200) { +function createLive (url: string, token: string, fields: LiveVideoCreate, statusCodeExpected = HttpStatusCode.OK_200) { const path = '/api/v1/videos/live' const attaches: any = {} @@ -47,14 +60,23 @@ function createLive (url: string, token: string, fields: LiveVideoCreate, status }) } -function sendRTMPStream (rtmpBaseUrl: string, streamKey: string) { - const fixture = buildAbsoluteFixturePath('video_short.mp4') +async function sendRTMPStreamInVideo (url: string, token: string, videoId: number | string, fixtureName?: string) { + const res = await getLive(url, token, videoId) + const videoLive = res.body as LiveVideo + + return sendRTMPStream(videoLive.rtmpUrl, videoLive.streamKey, fixtureName) +} + +function sendRTMPStream (rtmpBaseUrl: string, streamKey: string, fixtureName = 'video_short.mp4') { + const fixture = buildAbsoluteFixturePath(fixtureName) const command = ffmpeg(fixture) command.inputOption('-stream_loop -1') command.inputOption('-re') - - command.outputOption('-c copy') + command.outputOption('-c:v libx264') + command.outputOption('-g 50') + command.outputOption('-keyint_min 2') + command.outputOption('-r 60') command.outputOption('-f flv') const rtmpUrl = rtmpBaseUrl + '/' + streamKey @@ -63,7 +85,7 @@ function sendRTMPStream (rtmpBaseUrl: string, streamKey: string) { command.on('error', err => { if (err?.message?.includes('Exiting normally')) return - console.error('Cannot send RTMP stream.', { err }) + if (process.env.DEBUG) console.error(err) }) if (process.env.DEBUG) { @@ -75,13 +97,74 @@ function sendRTMPStream (rtmpBaseUrl: string, streamKey: string) { return command } +function waitFfmpegUntilError (command: ffmpeg.FfmpegCommand, successAfterMS = 10000) { + return new Promise((res, rej) => { + command.on('error', err => { + return rej(err) + }) + + setTimeout(() => { + res() + }, successAfterMS) + }) +} + +async function runAndTestFfmpegStreamError (url: string, token: string, videoId: number | string, shouldHaveError: boolean) { + const command = await sendRTMPStreamInVideo(url, token, videoId) + + return testFfmpegStreamError(command, shouldHaveError) +} + +async function testFfmpegStreamError (command: ffmpeg.FfmpegCommand, shouldHaveError: boolean) { + let error: Error + + try { + await waitFfmpegUntilError(command, 35000) + } catch (err) { + error = err + } + + await stopFfmpeg(command) + + if (shouldHaveError && !error) throw new Error('Ffmpeg did not have an error') + if (!shouldHaveError && error) throw error +} + async function stopFfmpeg (command: ffmpeg.FfmpegCommand) { command.kill('SIGINT') await wait(500) } -async function waitUntilLiveStarts (url: string, token: string, videoId: number | string) { +function waitUntilLivePublished (url: string, token: string, videoId: number | string) { + return waitUntilLiveState(url, token, videoId, VideoState.PUBLISHED) +} + +function waitUntilLiveWaiting (url: string, token: string, videoId: number | string) { + return waitUntilLiveState(url, token, videoId, VideoState.WAITING_FOR_LIVE) +} + +function waitUntilLiveEnded (url: string, token: string, videoId: number | string) { + return waitUntilLiveState(url, token, videoId, VideoState.LIVE_ENDED) +} + +function waitUntilLiveSegmentGeneration (server: ServerInfo, videoUUID: string, resolutionNum: number, segmentNum: number) { + const segmentName = `${resolutionNum}-00000${segmentNum}.ts` + return waitUntilLog(server, `${videoUUID}/${segmentName}`, 2, false) +} + +async function waitUntilLiveState (url: string, token: string, videoId: number | string, state: VideoState) { + let video: VideoDetails + + do { + const res = await getVideoWithToken(url, token, videoId) + video = res.body + + await wait(500) + } while (video.state.id !== state) +} + +async function waitUntilLiveSaved (url: string, token: string, videoId: number | string) { let video: VideoDetails do { @@ -89,16 +172,67 @@ async function waitUntilLiveStarts (url: string, token: string, videoId: number video = res.body await wait(500) - } while (video.state.id === VideoState.WAITING_FOR_LIVE) + } while (video.isLive === true && video.state.id !== VideoState.PUBLISHED) +} + +async function waitUntilLivePublishedOnAllServers (servers: ServerInfo[], videoId: string) { + for (const server of servers) { + await waitUntilLivePublished(server.url, server.accessToken, videoId) + } +} + +async function checkLiveCleanup (server: ServerInfo, videoUUID: string, resolutions: number[] = []) { + const basePath = buildServerDirectory(server, 'streaming-playlists') + const hlsPath = join(basePath, 'hls', videoUUID) + + if (resolutions.length === 0) { + const result = await pathExists(hlsPath) + expect(result).to.be.false + + return + } + + const files = await readdir(hlsPath) + + // fragmented file and playlist per resolution + master playlist + segments sha256 json file + expect(files).to.have.lengthOf(resolutions.length * 2 + 2) + + for (const resolution of resolutions) { + expect(files).to.contain(`${videoUUID}-${resolution}-fragmented.mp4`) + expect(files).to.contain(`${resolution}.m3u8`) + } + + expect(files).to.contain('master.m3u8') + expect(files).to.contain('segments-sha256.json') +} + +async function getPlaylistsCount (server: ServerInfo, videoUUID: string) { + const basePath = buildServerDirectory(server, 'streaming-playlists') + const hlsPath = join(basePath, 'hls', videoUUID) + + const files = await readdir(hlsPath) + + return files.filter(f => f.endsWith('.m3u8')).length } // --------------------------------------------------------------------------- export { getLive, + getPlaylistsCount, + waitUntilLiveSaved, + waitUntilLivePublished, updateLive, - waitUntilLiveStarts, createLive, + runAndTestFfmpegStreamError, + checkLiveCleanup, + waitUntilLiveSegmentGeneration, stopFfmpeg, - sendRTMPStream + waitUntilLiveWaiting, + sendRTMPStreamInVideo, + waitUntilLiveEnded, + waitFfmpegUntilError, + waitUntilLivePublishedOnAllServers, + sendRTMPStream, + testFfmpegStreamError }