2021-07-08 03:18:40 -05:00
|
|
|
/* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */
|
|
|
|
|
|
|
|
import { readdir } from 'fs-extra'
|
|
|
|
import { omit } from 'lodash'
|
|
|
|
import { join } from 'path'
|
2021-07-16 07:27:30 -05:00
|
|
|
import { HttpStatusCode, LiveVideo, LiveVideoCreate, LiveVideoUpdate, VideoCreateResult, VideoDetails, VideoState } from '@shared/models'
|
2021-07-13 02:43:59 -05:00
|
|
|
import { wait } from '../miscs'
|
2021-07-08 03:18:40 -05:00
|
|
|
import { unwrapBody } from '../requests'
|
|
|
|
import { AbstractCommand, OverrideCommandOptions } from '../shared'
|
|
|
|
import { sendRTMPStream, testFfmpegStreamError } from './live'
|
|
|
|
|
|
|
|
export class LiveCommand extends AbstractCommand {
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
get (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
}) {
|
|
|
|
const path = '/api/v1/videos/live'
|
|
|
|
|
|
|
|
return this.getRequestBody<LiveVideo>({
|
|
|
|
...options,
|
|
|
|
|
|
|
|
path: path + '/' + options.videoId,
|
2021-07-08 03:55:16 -05:00
|
|
|
implicitToken: true,
|
2021-07-08 03:18:40 -05:00
|
|
|
defaultExpectedStatus: HttpStatusCode.OK_200
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
update (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
fields: LiveVideoUpdate
|
|
|
|
}) {
|
|
|
|
const { videoId, fields } = options
|
|
|
|
const path = '/api/v1/videos/live'
|
|
|
|
|
|
|
|
return this.putBodyRequest({
|
|
|
|
...options,
|
|
|
|
|
|
|
|
path: path + '/' + videoId,
|
|
|
|
fields,
|
2021-07-08 03:55:16 -05:00
|
|
|
implicitToken: true,
|
2021-07-08 03:18:40 -05:00
|
|
|
defaultExpectedStatus: HttpStatusCode.NO_CONTENT_204
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
async create (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
fields: LiveVideoCreate
|
|
|
|
}) {
|
|
|
|
const { fields } = options
|
|
|
|
const path = '/api/v1/videos/live'
|
|
|
|
|
|
|
|
const attaches: any = {}
|
|
|
|
if (fields.thumbnailfile) attaches.thumbnailfile = fields.thumbnailfile
|
|
|
|
if (fields.previewfile) attaches.previewfile = fields.previewfile
|
|
|
|
|
|
|
|
const body = await unwrapBody<{ video: VideoCreateResult }>(this.postUploadRequest({
|
|
|
|
...options,
|
|
|
|
|
|
|
|
path,
|
|
|
|
attaches,
|
|
|
|
fields: omit(fields, 'thumbnailfile', 'previewfile'),
|
2021-07-08 03:55:16 -05:00
|
|
|
implicitToken: true,
|
2021-07-08 03:18:40 -05:00
|
|
|
defaultExpectedStatus: HttpStatusCode.OK_200
|
|
|
|
}))
|
|
|
|
|
|
|
|
return body.video
|
|
|
|
}
|
|
|
|
|
|
|
|
async sendRTMPStreamInVideo (options: OverrideCommandOptions & {
|
|
|
|
videoId: number | string
|
|
|
|
fixtureName?: string
|
|
|
|
}) {
|
|
|
|
const { videoId, fixtureName } = options
|
2021-07-08 03:25:50 -05:00
|
|
|
const videoLive = await this.get({ videoId })
|
2021-07-08 03:18:40 -05:00
|
|
|
|
|
|
|
return sendRTMPStream(videoLive.rtmpUrl, videoLive.streamKey, fixtureName)
|
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
async runAndTestStreamError (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
shouldHaveError: boolean
|
|
|
|
}) {
|
|
|
|
const command = await this.sendRTMPStreamInVideo(options)
|
|
|
|
|
|
|
|
return testFfmpegStreamError(command, options.shouldHaveError)
|
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
waitUntilPublished (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
}) {
|
|
|
|
const { videoId } = options
|
2021-07-08 03:25:50 -05:00
|
|
|
return this.waitUntilState({ videoId, state: VideoState.PUBLISHED })
|
2021-07-08 03:18:40 -05:00
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
waitUntilWaiting (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
}) {
|
|
|
|
const { videoId } = options
|
2021-07-08 03:25:50 -05:00
|
|
|
return this.waitUntilState({ videoId, state: VideoState.WAITING_FOR_LIVE })
|
2021-07-08 03:18:40 -05:00
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
waitUntilEnded (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
}) {
|
|
|
|
const { videoId } = options
|
2021-07-08 03:25:50 -05:00
|
|
|
return this.waitUntilState({ videoId, state: VideoState.LIVE_ENDED })
|
2021-07-08 03:18:40 -05:00
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
waitUntilSegmentGeneration (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoUUID: string
|
|
|
|
resolution: number
|
|
|
|
segment: number
|
|
|
|
}) {
|
|
|
|
const { resolution, segment, videoUUID } = options
|
|
|
|
const segmentName = `${resolution}-00000${segment}.ts`
|
|
|
|
|
2021-07-16 02:04:35 -05:00
|
|
|
return this.server.servers.waitUntilLog(`${videoUUID}/${segmentName}`, 2, false)
|
2021-07-08 03:18:40 -05:00
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
async waitUntilSaved (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
}) {
|
|
|
|
let video: VideoDetails
|
|
|
|
|
|
|
|
do {
|
2021-07-16 02:04:35 -05:00
|
|
|
video = await this.server.videos.getWithToken({ token: options.token, id: options.videoId })
|
2021-07-08 03:18:40 -05:00
|
|
|
|
|
|
|
await wait(500)
|
|
|
|
} while (video.isLive === true && video.state.id !== VideoState.PUBLISHED)
|
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
async countPlaylists (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoUUID: string
|
|
|
|
}) {
|
2021-07-16 02:04:35 -05:00
|
|
|
const basePath = this.server.servers.buildDirectory('streaming-playlists')
|
2021-07-08 03:18:40 -05:00
|
|
|
const hlsPath = join(basePath, 'hls', options.videoUUID)
|
|
|
|
|
|
|
|
const files = await readdir(hlsPath)
|
|
|
|
|
|
|
|
return files.filter(f => f.endsWith('.m3u8')).length
|
|
|
|
}
|
|
|
|
|
2021-07-08 03:25:50 -05:00
|
|
|
private async waitUntilState (options: OverrideCommandOptions & {
|
2021-07-08 03:18:40 -05:00
|
|
|
videoId: number | string
|
|
|
|
state: VideoState
|
|
|
|
}) {
|
|
|
|
let video: VideoDetails
|
|
|
|
|
|
|
|
do {
|
2021-07-16 02:04:35 -05:00
|
|
|
video = await this.server.videos.getWithToken({ token: options.token, id: options.videoId })
|
2021-07-08 03:18:40 -05:00
|
|
|
|
|
|
|
await wait(500)
|
|
|
|
} while (video.state.id !== options.state)
|
|
|
|
}
|
|
|
|
}
|