import * as memoizee from 'memoizee' import { join } from 'path' import { Op } from 'sequelize' import { AllowNull, BelongsTo, Column, CreatedAt, DataType, Default, ForeignKey, HasMany, Is, Model, Table, UpdatedAt } from 'sequelize-typescript' import { getHLSPublicFileUrl } from '@server/lib/object-storage' import { VideoFileModel } from '@server/models/video/video-file' import { MStreamingPlaylist, MVideo } from '@server/types/models' import { AttributesOnly } from '@shared/core-utils' import { VideoStorage } from '@shared/models' import { VideoStreamingPlaylistType } from '../../../shared/models/videos/video-streaming-playlist.type' import { sha1 } from '../../helpers/core-utils' import { isActivityPubUrlValid } from '../../helpers/custom-validators/activitypub/misc' 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, WEBSERVER } from '../../initializers/constants' import { VideoRedundancyModel } from '../redundancy/video-redundancy' import { doesExist } from '../shared' import { throwIfNotValid } from '../utils' import { VideoModel } from './video' @Table({ tableName: 'videoStreamingPlaylist', indexes: [ { fields: [ 'videoId' ] }, { fields: [ 'videoId', 'type' ], unique: true }, { fields: [ 'p2pMediaLoaderInfohashes' ], using: 'gin' } ] }) export class VideoStreamingPlaylistModel extends Model>> { @CreatedAt createdAt: Date @UpdatedAt updatedAt: Date @AllowNull(false) @Column type: VideoStreamingPlaylistType @AllowNull(false) @Column playlistFilename: string @AllowNull(true) @Is('PlaylistUrl', value => throwIfNotValid(value, isActivityPubUrlValid, 'playlist url', true)) @Column(DataType.STRING(CONSTRAINTS_FIELDS.VIDEOS.URL.max)) playlistUrl: string @AllowNull(false) @Is('VideoStreamingPlaylistInfoHashes', value => throwIfNotValid(value, v => isArrayOf(v, isVideoFileInfoHashValid), 'info hashes')) @Column(DataType.ARRAY(DataType.STRING)) p2pMediaLoaderInfohashes: string[] @AllowNull(false) @Column p2pMediaLoaderPeerVersion: number @AllowNull(false) @Column segmentsSha256Filename: string @AllowNull(true) @Is('VideoStreamingSegmentsSha256Url', value => throwIfNotValid(value, isActivityPubUrlValid, 'segments sha256 url', true)) @Column segmentsSha256Url: string @ForeignKey(() => VideoModel) @Column videoId: number @AllowNull(false) @Default(VideoStorage.FILE_SYSTEM) @Column storage: VideoStorage @BelongsTo(() => VideoModel, { foreignKey: { allowNull: false }, onDelete: 'CASCADE' }) Video: VideoModel @HasMany(() => VideoFileModel, { foreignKey: { allowNull: true }, onDelete: 'CASCADE' }) VideoFiles: VideoFileModel[] @HasMany(() => VideoRedundancyModel, { foreignKey: { allowNull: false }, onDelete: 'CASCADE', hooks: true }) RedundancyVideos: VideoRedundancyModel[] static doesInfohashExistCached = memoizee(VideoStreamingPlaylistModel.doesInfohashExist, { promise: true, max: MEMOIZE_LENGTH.INFO_HASH_EXISTS, maxAge: MEMOIZE_TTL.INFO_HASH_EXISTS }) static doesInfohashExist (infoHash: string) { const query = 'SELECT 1 FROM "videoStreamingPlaylist" WHERE $infoHash = ANY("p2pMediaLoaderInfohashes") LIMIT 1' return doesExist(query, { infoHash }) } static buildP2PMediaLoaderInfoHashes (playlistUrl: string, files: unknown[]) { const hashes: string[] = [] // 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++) { hashes.push(sha1(`${P2P_MEDIA_LOADER_PEER_VERSION}${playlistUrl}+V${i}`)) } return hashes } static listByIncorrectPeerVersion () { const query = { where: { p2pMediaLoaderPeerVersion: { [Op.ne]: P2P_MEDIA_LOADER_PEER_VERSION } }, include: [ { model: VideoModel.unscoped(), required: true } ] } return VideoStreamingPlaylistModel.findAll(query) } static loadWithVideo (id: number) { const options = { include: [ { model: VideoModel.unscoped(), required: true } ] } return VideoStreamingPlaylistModel.findByPk(id, options) } static loadHLSPlaylistByVideo (videoId: number): Promise { const options = { where: { type: VideoStreamingPlaylistType.HLS, videoId } } return VideoStreamingPlaylistModel.findOne(options) } static async loadOrGenerate (video: MVideo) { let playlist = await VideoStreamingPlaylistModel.loadHLSPlaylistByVideo(video.id) if (!playlist) playlist = new VideoStreamingPlaylistModel() return Object.assign(playlist, { videoId: video.id, Video: video }) } assignP2PMediaLoaderInfoHashes (video: MVideo, files: unknown[]) { const masterPlaylistUrl = this.getMasterPlaylistUrl(video) this.p2pMediaLoaderInfohashes = VideoStreamingPlaylistModel.buildP2PMediaLoaderInfoHashes(masterPlaylistUrl, files) } getMasterPlaylistUrl (video: MVideo) { if (this.storage === VideoStorage.OBJECT_STORAGE) { return getHLSPublicFileUrl(this.playlistUrl) } if (video.isOwned()) return WEBSERVER.URL + this.getMasterPlaylistStaticPath(video.uuid) return this.playlistUrl } getSha256SegmentsUrl (video: MVideo) { if (this.storage === VideoStorage.OBJECT_STORAGE) { return getHLSPublicFileUrl(this.segmentsSha256Url) } if (video.isOwned()) return WEBSERVER.URL + this.getSha256SegmentsStaticPath(video.uuid, video.isLive) return this.segmentsSha256Url } getStringType () { if (this.type === VideoStreamingPlaylistType.HLS) return 'hls' return 'unknown' } getTrackerUrls (baseUrlHttp: string, baseUrlWs: string) { return [ baseUrlWs + '/tracker/socket', baseUrlHttp + '/tracker/announce' ] } hasSameUniqueKeysThan (other: MStreamingPlaylist) { return this.type === other.type && this.videoId === other.videoId } private getMasterPlaylistStaticPath (videoUUID: string) { return join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, videoUUID, this.playlistFilename) } private getSha256SegmentsStaticPath (videoUUID: string, isLive: boolean) { if (isLive) return join('/live', 'segments-sha256', videoUUID) return join(STATIC_PATHS.STREAMING_PLAYLISTS.HLS, videoUUID, this.segmentsSha256Filename) } }