X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Flib%2Fjob-queue%2Fhandlers%2Fvideo-live-ending.ts;h=8a3ee09a2290210354bedfb8b2895fdadd3ab020;hb=e15bc9dcd59ebb6102d1455450d2e8bbf9f49c27;hp=cd5bb1d1cdbbf864470802d105d4546b6cce050d;hpb=31c82cd914e13dbf53280d0aad0740d70c414441;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/lib/job-queue/handlers/video-live-ending.ts b/server/lib/job-queue/handlers/video-live-ending.ts index cd5bb1d1c..8a3ee09a2 100644 --- a/server/lib/job-queue/handlers/video-live-ending.ts +++ b/server/lib/job-queue/handlers/video-live-ending.ts @@ -1,33 +1,65 @@ -import * as Bull from 'bull' +import { Job } from 'bullmq' import { readdir, remove } from 'fs-extra' import { join } from 'path' -import { getVideoFileResolution, hlsPlaylistToFragmentedMP4 } from '@server/helpers/ffmpeg-utils' -import { getHLSDirectory } from '@server/lib/video-paths' -import { generateHlsPlaylist } from '@server/lib/video-transcoding' +import { ffprobePromise, getAudioStream, getVideoStreamDimensionsInfo } from '@server/helpers/ffmpeg' +import { getLocalVideoActivityPubUrl } from '@server/lib/activitypub/url' +import { federateVideoIfNeeded } from '@server/lib/activitypub/videos' +import { cleanupPermanentLive, cleanupTMPLiveFiles, cleanupUnsavedNormalLive } from '@server/lib/live' +import { generateHLSMasterPlaylistFilename, generateHlsSha256SegmentsFilename, getLiveReplayBaseDirectory } from '@server/lib/paths' +import { generateVideoMiniature } from '@server/lib/thumbnail' +import { generateHlsPlaylistResolutionFromTS } from '@server/lib/transcoding/transcoding' +import { moveToNextState } from '@server/lib/video-state' import { VideoModel } from '@server/models/video/video' +import { VideoBlacklistModel } from '@server/models/video/video-blacklist' +import { VideoFileModel } from '@server/models/video/video-file' import { VideoLiveModel } from '@server/models/video/video-live' +import { VideoLiveSessionModel } from '@server/models/video/video-live-session' import { VideoStreamingPlaylistModel } from '@server/models/video/video-streaming-playlist' -import { MStreamingPlaylist, MVideo, MVideoLive } from '@server/types/models' -import { VideoLiveEndingPayload, VideoState } from '@shared/models' -import { logger } from '../../../helpers/logger' +import { MVideo, MVideoLive, MVideoLiveSession, MVideoWithAllFiles } from '@server/types/models' +import { ThumbnailType, VideoLiveEndingPayload, VideoState } from '@shared/models' +import { logger, loggerTagsFactory } from '../../../helpers/logger' -async function processVideoLiveEnding (job: Bull.Job) { +const lTags = loggerTagsFactory('live', 'job') + +async function processVideoLiveEnding (job: Job) { const payload = job.data as VideoLiveEndingPayload + logger.info('Processing video live ending for %s.', payload.videoId, { payload, ...lTags() }) + + function logError () { + logger.warn('Video live %d does not exist anymore. Cannot process live ending.', payload.videoId, lTags()) + } + const video = await VideoModel.load(payload.videoId) const live = await VideoLiveModel.loadByVideoId(payload.videoId) + const liveSession = await VideoLiveSessionModel.load(payload.liveSessionId) + + const permanentLive = live.permanentLive - const streamingPlaylist = await VideoStreamingPlaylistModel.loadHLSPlaylistByVideo(video.id) - if (!video || !streamingPlaylist || !live) { - logger.warn('Video live %d does not exist anymore. Cannot process live ending.', payload.videoId) + if (!video || !live || !liveSession) { + logError() return } - if (live.saveReplay !== true) { - return cleanupLive(video, streamingPlaylist) + liveSession.endingProcessed = true + await liveSession.save() + + if (liveSession.saveReplay !== true) { + return cleanupLiveAndFederate({ permanentLive, video, streamingPlaylistId: payload.streamingPlaylistId }) + } + + if (permanentLive) { + await saveReplayToExternalVideo({ + liveVideo: video, + liveSession, + publishedAt: payload.publishedAt, + replayDirectory: payload.replayDirectory + }) + + return cleanupLiveAndFederate({ permanentLive, video, streamingPlaylistId: payload.streamingPlaylistId }) } - return saveLive(video, live) + return replaceLiveByReplay({ video, liveSession, live, permanentLive, replayDirectory: payload.replayDirectory }) } // --------------------------------------------------------------------------- @@ -38,84 +70,180 @@ export { // --------------------------------------------------------------------------- -async function saveLive (video: MVideo, live: MVideoLive) { - const hlsDirectory = getHLSDirectory(video, false) - const files = await readdir(hlsDirectory) - - const playlistFiles = files.filter(f => f.endsWith('.m3u8') && f !== 'master.m3u8') - const resolutions: number[] = [] +async function saveReplayToExternalVideo (options: { + liveVideo: MVideo + liveSession: MVideoLiveSession + publishedAt: string + replayDirectory: string +}) { + const { liveVideo, liveSession, publishedAt, replayDirectory } = options + + const replayVideo = new VideoModel({ + name: `${liveVideo.name} - ${new Date(publishedAt).toLocaleString()}`, + isLive: false, + state: VideoState.TO_TRANSCODE, + duration: 0, + + remote: liveVideo.remote, + category: liveVideo.category, + licence: liveVideo.licence, + language: liveVideo.language, + commentsEnabled: liveVideo.commentsEnabled, + downloadEnabled: liveVideo.downloadEnabled, + waitTranscoding: true, + nsfw: liveVideo.nsfw, + description: liveVideo.description, + support: liveVideo.support, + privacy: liveVideo.privacy, + channelId: liveVideo.channelId + }) as MVideoWithAllFiles + + replayVideo.Thumbnails = [] + replayVideo.VideoFiles = [] + replayVideo.VideoStreamingPlaylists = [] + + replayVideo.url = getLocalVideoActivityPubUrl(replayVideo) + + await replayVideo.save() + + liveSession.replayVideoId = replayVideo.id + await liveSession.save() + + // If live is blacklisted, also blacklist the replay + const blacklist = await VideoBlacklistModel.loadByVideoId(liveVideo.id) + if (blacklist) { + await VideoBlacklistModel.create({ + videoId: replayVideo.id, + unfederated: blacklist.unfederated, + reason: blacklist.reason, + type: blacklist.type + }) + } - for (const playlistFile of playlistFiles) { - const playlistPath = join(hlsDirectory, playlistFile) - const { videoFileResolution } = await getVideoFileResolution(playlistPath) + await assignReplayFilesToVideo({ video: replayVideo, replayDirectory }) - const mp4TmpName = buildMP4TmpName(videoFileResolution) + await remove(replayDirectory) - // Playlist name is for example 3.m3u8 - // Segments names are 3-0.ts 3-1.ts etc - const shouldStartWith = playlistFile.replace(/\.m3u8$/, '') + '-' + for (const type of [ ThumbnailType.MINIATURE, ThumbnailType.PREVIEW ]) { + const image = await generateVideoMiniature({ video: replayVideo, videoFile: replayVideo.getMaxQualityFile(), type }) + await replayVideo.addAndSaveThumbnail(image) + } - const segmentFiles = files.filter(f => f.startsWith(shouldStartWith) && f.endsWith('.ts')) - await hlsPlaylistToFragmentedMP4(hlsDirectory, segmentFiles, mp4TmpName) + await moveToNextState({ video: replayVideo, isNewVideo: true }) +} - resolutions.push(videoFileResolution) - } +async function replaceLiveByReplay (options: { + video: MVideo + liveSession: MVideoLiveSession + live: MVideoLive + permanentLive: boolean + replayDirectory: string +}) { + const { video, liveSession, live, permanentLive, replayDirectory } = options - await cleanupLiveFiles(hlsDirectory) + await cleanupTMPLiveFiles(video) await live.destroy() video.isLive = false + video.waitTranscoding = true video.state = VideoState.TO_TRANSCODE + await video.save() - const videoWithFiles = await VideoModel.loadWithFiles(video.id) + liveSession.replayVideoId = video.id + await liveSession.save() + + // Remove old HLS playlist video files + const videoWithFiles = await VideoModel.loadFull(video.id) - for (const resolution of resolutions) { - const videoInputPath = buildMP4TmpName(resolution) - const { isPortraitMode } = await getVideoFileResolution(videoInputPath) + const hlsPlaylist = videoWithFiles.getHLSPlaylist() + await VideoFileModel.removeHLSFilesOfVideoId(hlsPlaylist.id) - await generateHlsPlaylist({ + // Reset playlist + hlsPlaylist.VideoFiles = [] + hlsPlaylist.playlistFilename = generateHLSMasterPlaylistFilename() + hlsPlaylist.segmentsSha256Filename = generateHlsSha256SegmentsFilename() + await hlsPlaylist.save() + + await assignReplayFilesToVideo({ video: videoWithFiles, replayDirectory }) + + if (permanentLive) { // Remove session replay + await remove(replayDirectory) + } else { // We won't stream again in this live, we can delete the base replay directory + await remove(getLiveReplayBaseDirectory(videoWithFiles)) + } + + // Regenerate the thumbnail & preview? + if (videoWithFiles.getMiniature().automaticallyGenerated === true) { + const miniature = await generateVideoMiniature({ video: videoWithFiles, - videoInputPath, - resolution: resolution, - copyCodecs: true, - isPortraitMode + videoFile: videoWithFiles.getMaxQualityFile(), + type: ThumbnailType.MINIATURE }) + await videoWithFiles.addAndSaveThumbnail(miniature) } - video.state = VideoState.PUBLISHED - await video.save() + if (videoWithFiles.getPreview().automaticallyGenerated === true) { + const preview = await generateVideoMiniature({ + video: videoWithFiles, + videoFile: videoWithFiles.getMaxQualityFile(), + type: ThumbnailType.PREVIEW + }) + await videoWithFiles.addAndSaveThumbnail(preview) + } + + // We consider this is a new video + await moveToNextState({ video: videoWithFiles, isNewVideo: true }) } -async function cleanupLive (video: MVideo, streamingPlaylist: MStreamingPlaylist) { - const hlsDirectory = getHLSDirectory(video, false) +async function assignReplayFilesToVideo (options: { + video: MVideo + replayDirectory: string +}) { + const { video, replayDirectory } = options - await cleanupLiveFiles(hlsDirectory) + const concatenatedTsFiles = await readdir(replayDirectory) - streamingPlaylist.destroy() - .catch(err => logger.error('Cannot remove live streaming playlist.', { err })) + for (const concatenatedTsFile of concatenatedTsFiles) { + const concatenatedTsFilePath = join(replayDirectory, concatenatedTsFile) + + const probe = await ffprobePromise(concatenatedTsFilePath) + const { audioStream } = await getAudioStream(concatenatedTsFilePath, probe) + const { resolution } = await getVideoStreamDimensionsInfo(concatenatedTsFilePath, probe) + + await generateHlsPlaylistResolutionFromTS({ + video, + concatenatedTsFilePath, + resolution, + isAAC: audioStream?.codec_name === 'aac' + }) + } + + return video } -async function cleanupLiveFiles (hlsDirectory: string) { - const files = await readdir(hlsDirectory) - - for (const filename of files) { - if ( - filename.endsWith('.ts') || - filename.endsWith('.m3u8') || - filename.endsWith('.mpd') || - filename.endsWith('.m4s') || - filename.endsWith('.tmp') - ) { - const p = join(hlsDirectory, filename) - - remove(p) - .catch(err => logger.error('Cannot remove %s.', p, { err })) +async function cleanupLiveAndFederate (options: { + video: MVideo + permanentLive: boolean + streamingPlaylistId: number +}) { + const { permanentLive, video, streamingPlaylistId } = options + + const streamingPlaylist = await VideoStreamingPlaylistModel.loadWithVideo(streamingPlaylistId) + + if (streamingPlaylist) { + if (permanentLive) { + await cleanupPermanentLive(video, streamingPlaylist) + } else { + await cleanupUnsavedNormalLive(video, streamingPlaylist) } } -} -function buildMP4TmpName (resolution: number) { - return resolution + '-tmp.mp4' + try { + const fullVideo = await VideoModel.loadFull(video.id) + return federateVideoIfNeeded(fullVideo, false, undefined) + } catch (err) { + logger.warn('Cannot federate live after cleanup', { videoId: video.id, err }) + } }