PeerTube/server/models/video/video-live.ts

120 lines
2.5 KiB
TypeScript
Raw Normal View History

import { AllowNull, BelongsTo, Column, CreatedAt, DataType, DefaultScope, ForeignKey, Model, Table, UpdatedAt } from 'sequelize-typescript'
import { WEBSERVER } from '@server/initializers/constants'
import { MVideoLive, MVideoLiveVideo } from '@server/types/models'
2021-05-12 14:09:04 +02:00
import { AttributesOnly } from '@shared/core-utils'
import { LiveVideo, VideoState } from '@shared/models'
import { VideoModel } from './video'
import { VideoBlacklistModel } from './video-blacklist'
2021-11-05 11:36:03 +01:00
import { CONFIG } from '@server/initializers/config'
@DefaultScope(() => ({
include: [
{
model: VideoModel,
required: true,
include: [
{
model: VideoBlacklistModel,
required: false
}
]
}
]
}))
@Table({
tableName: 'videoLive',
indexes: [
{
fields: [ 'videoId' ],
unique: true
}
]
})
2021-05-12 14:09:04 +02:00
export class VideoLiveModel extends Model<Partial<AttributesOnly<VideoLiveModel>>> {
2020-09-25 16:19:35 +02:00
@AllowNull(true)
@Column(DataType.STRING)
streamKey: string
2020-09-25 16:19:35 +02:00
@AllowNull(false)
@Column
saveReplay: boolean
2020-12-03 14:10:54 +01:00
@AllowNull(false)
@Column
permanentLive: boolean
@CreatedAt
createdAt: Date
@UpdatedAt
updatedAt: Date
@ForeignKey(() => VideoModel)
@Column
videoId: number
@BelongsTo(() => VideoModel, {
foreignKey: {
allowNull: false
},
onDelete: 'cascade'
})
Video: VideoModel
static loadByStreamKey (streamKey: string) {
const query = {
where: {
streamKey
},
include: [
{
model: VideoModel.unscoped(),
required: true,
where: {
state: VideoState.WAITING_FOR_LIVE
},
include: [
{
model: VideoBlacklistModel.unscoped(),
required: false
}
]
}
]
}
return VideoLiveModel.findOne<MVideoLiveVideo>(query)
}
static loadByVideoId (videoId: number) {
const query = {
where: {
videoId
}
}
return VideoLiveModel.findOne<MVideoLive>(query)
}
toFormattedJSON (): LiveVideo {
2021-11-05 11:36:03 +01:00
let rtmpUrl: string = null
let rtmpsUrl: string = null
// If we don't have a stream key, it means this is a remote live so we don't specify the rtmp URL
if (this.streamKey) {
if (CONFIG.LIVE.RTMP.ENABLED) rtmpUrl = WEBSERVER.RTMP_URL
if (CONFIG.LIVE.RTMPS.ENABLED) rtmpsUrl = WEBSERVER.RTMPS_URL
}
return {
2021-11-05 11:36:03 +01:00
rtmpUrl,
rtmpsUrl,
2020-11-02 15:43:44 +01:00
2020-10-26 16:44:23 +01:00
streamKey: this.streamKey,
2020-12-03 14:10:54 +01:00
permanentLive: this.permanentLive,
2020-10-26 16:44:23 +01:00
saveReplay: this.saveReplay
}
}
}