PeerTube/server/models/video/video-streaming-playlist.ts

199 lines
5.7 KiB
TypeScript
Raw Normal View History

import * as memoizee from 'memoizee'
import { join } from 'path'
import { Op, QueryTypes } from 'sequelize'
2019-08-15 11:53:26 +02:00
import { AllowNull, BelongsTo, Column, CreatedAt, DataType, ForeignKey, HasMany, Is, Model, Table, UpdatedAt } from 'sequelize-typescript'
import { VideoFileModel } from '@server/models/video/video-file'
import { MStreamingPlaylist } from '@server/types/models'
2019-01-29 08:37:25 +01:00
import { VideoStreamingPlaylistType } from '../../../shared/models/videos/video-streaming-playlist.type'
import { sha1 } from '../../helpers/core-utils'
import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc'
2019-01-29 08:37:25 +01:00
import { isArrayOf } from '../../helpers/custom-validators/misc'
import { isVideoFileInfoHashValid } from '../../helpers/custom-validators/videos'
import { CONSTRAINTS_FIELDS, MEMOIZE_LENGTH, MEMOIZE_TTL, P2P_MEDIA_LOADER_PEER_VERSION, STATIC_PATHS } from '../../initializers/constants'
import { VideoRedundancyModel } from '../redundancy/video-redundancy'
import { throwIfNotValid } from '../utils'
import { VideoModel } from './video'
2021-05-12 14:09:04 +02:00
import { AttributesOnly } from '@shared/core-utils'
2019-01-29 08:37:25 +01:00
@Table({
tableName: 'videoStreamingPlaylist',
indexes: [
{
fields: [ 'videoId' ]
},
{
fields: [ 'videoId', 'type' ],
unique: true
},
{
fields: [ 'p2pMediaLoaderInfohashes' ],
using: 'gin'
}
2019-04-23 09:50:57 +02:00
]
2019-01-29 08:37:25 +01:00
})
2021-05-12 14:09:04 +02:00
export class VideoStreamingPlaylistModel extends Model<Partial<AttributesOnly<VideoStreamingPlaylistModel>>> {
2019-01-29 08:37:25 +01:00
@CreatedAt
createdAt: Date
@UpdatedAt
updatedAt: Date
@AllowNull(false)
@Column
type: VideoStreamingPlaylistType
@AllowNull(false)
@Is('PlaylistUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'playlist url'))
@Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEOS.URL.max))
playlistUrl: string
@AllowNull(false)
@Is('VideoStreamingPlaylistInfoHashes', value => throwIfNotValid(value, v => isArrayOf(v, isVideoFileInfoHashValid), 'info hashes'))
2019-04-23 09:50:57 +02:00
@Column(DataType.ARRAY(DataType.STRING))
2019-01-29 08:37:25 +01:00
p2pMediaLoaderInfohashes: string[]
2019-04-08 11:13:49 +02:00
@AllowNull(false)
@Column
p2pMediaLoaderPeerVersion: number
2019-01-29 08:37:25 +01:00
@AllowNull(false)
@Is('VideoStreamingSegmentsSha256Url', value => throwIfNotValid(value, isActivityPubUrlValid, 'segments sha256 url'))
@Column
segmentsSha256Url: string
@ForeignKey(() => VideoModel)
@Column
videoId: number
@BelongsTo(() => VideoModel, {
foreignKey: {
allowNull: false
},
onDelete: 'CASCADE'
})
Video: VideoModel
@HasMany(() => VideoFileModel, {
foreignKey: {
allowNull: true
},
onDelete: 'CASCADE'
})
VideoFiles: VideoFileModel[]
2019-01-29 08:37:25 +01:00
@HasMany(() => VideoRedundancyModel, {
foreignKey: {
allowNull: false
},
onDelete: 'CASCADE',
hooks: true
})
RedundancyVideos: VideoRedundancyModel[]
2020-01-03 13:47:45 +01:00
static doesInfohashExistCached = memoizee(VideoStreamingPlaylistModel.doesInfohashExist, {
promise: true,
max: MEMOIZE_LENGTH.INFO_HASH_EXISTS,
maxAge: MEMOIZE_TTL.INFO_HASH_EXISTS
})
2019-01-29 08:37:25 +01:00
static doesInfohashExist (infoHash: string) {
const query = 'SELECT 1 FROM "videoStreamingPlaylist" WHERE $infoHash = ANY("p2pMediaLoaderInfohashes") LIMIT 1'
const options = {
2019-04-18 11:28:17 +02:00
type: QueryTypes.SELECT as QueryTypes.SELECT,
2019-01-29 08:37:25 +01:00
bind: { infoHash },
raw: true
}
2019-04-23 09:50:57 +02:00
return VideoModel.sequelize.query<object>(query, options)
2019-04-18 11:28:17 +02:00
.then(results => results.length === 1)
2019-01-29 08:37:25 +01:00
}
static buildP2PMediaLoaderInfoHashes (playlistUrl: string, files: unknown[]) {
2019-01-29 08:37:25 +01:00
const hashes: string[] = []
2019-04-08 11:13:49 +02:00
// https://github.com/Novage/p2p-media-loader/blob/master/p2p-media-loader-core/lib/p2p-media-manager.ts#L115
for (let i = 0; i < files.length; i++) {
2019-04-08 11:13:49 +02:00
hashes.push(sha1(`${P2P_MEDIA_LOADER_PEER_VERSION}${playlistUrl}+V${i}`))
2019-01-29 08:37:25 +01:00
}
return hashes
}
2019-04-08 11:13:49 +02:00
static listByIncorrectPeerVersion () {
const query = {
where: {
p2pMediaLoaderPeerVersion: {
2019-04-18 11:28:17 +02:00
[Op.ne]: P2P_MEDIA_LOADER_PEER_VERSION
2019-04-08 11:13:49 +02:00
}
}
}
return VideoStreamingPlaylistModel.findAll(query)
}
2019-01-29 08:37:25 +01:00
static loadWithVideo (id: number) {
const options = {
include: [
{
model: VideoModel.unscoped(),
required: true
}
]
}
2019-02-21 14:28:06 +01:00
return VideoStreamingPlaylistModel.findByPk(id, options)
2019-01-29 08:37:25 +01:00
}
static loadHLSPlaylistByVideo (videoId: number) {
const options = {
where: {
type: VideoStreamingPlaylistType.HLS,
videoId
}
}
return VideoStreamingPlaylistModel.findOne(options)
}
2019-01-29 08:37:25 +01:00
static getHlsPlaylistFilename (resolution: number) {
return resolution + '.m3u8'
}
static getMasterHlsPlaylistFilename () {
return 'master.m3u8'
}
static getHlsSha256SegmentsFilename () {
return 'segments-sha256.json'
}
static getHlsMasterPlaylistStaticPath (videoUUID: string) {
return join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, videoUUID, VideoStreamingPlaylistModel.getMasterHlsPlaylistFilename())
2019-01-29 08:37:25 +01:00
}
static getHlsPlaylistStaticPath (videoUUID: string, resolution: number) {
return join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, videoUUID, VideoStreamingPlaylistModel.getHlsPlaylistFilename(resolution))
2019-01-29 08:37:25 +01:00
}
static getHlsSha256SegmentsStaticPath (videoUUID: string, isLive: boolean) {
if (isLive) return join('/live', 'segments-sha256', videoUUID)
return join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, videoUUID, VideoStreamingPlaylistModel.getHlsSha256SegmentsFilename())
2019-01-29 08:37:25 +01:00
}
getStringType () {
if (this.type === VideoStreamingPlaylistType.HLS) return 'hls'
return 'unknown'
}
getTrackerUrls (baseUrlHttp: string, baseUrlWs: string) {
return [ baseUrlWs + '/tracker/socket', baseUrlHttp + '/tracker/announce' ]
}
2019-08-15 11:53:26 +02:00
hasSameUniqueKeysThan (other: MStreamingPlaylist) {
2019-01-29 08:37:25 +01:00
return this.type === other.type &&
this.videoId === other.videoId
}
}