]>
Commit | Line | Data |
---|---|---|
4ec52d04 | 1 | import { readdir, readFile } from 'fs-extra' |
8ebf2a5d | 2 | import { createServer, Server } from 'net' |
4ec52d04 | 3 | import { join } from 'path' |
df1db951 | 4 | import { createServer as createServerTLS, Server as ServerTLS } from 'tls' |
8ebf2a5d C |
5 | import { logger, loggerTagsFactory } from '@server/helpers/logger' |
6 | import { CONFIG, registerConfigChangedHandler } from '@server/initializers/config' | |
28705705 | 7 | import { VIDEO_LIVE, WEBSERVER } from '@server/initializers/constants' |
0c9668f7 | 8 | import { sequelizeTypescript } from '@server/initializers/database' |
5a05c145 | 9 | import { RunnerJobModel } from '@server/models/runner/runner-job' |
8ebf2a5d C |
10 | import { UserModel } from '@server/models/user/user' |
11 | import { VideoModel } from '@server/models/video/video' | |
12 | import { VideoLiveModel } from '@server/models/video/video-live' | |
0c9668f7 | 13 | import { VideoLiveReplaySettingModel } from '@server/models/video/video-live-replay-setting' |
26e3e98f | 14 | import { VideoLiveSessionModel } from '@server/models/video/video-live-session' |
8ebf2a5d | 15 | import { VideoStreamingPlaylistModel } from '@server/models/video/video-streaming-playlist' |
05a60d85 | 16 | import { MVideo, MVideoLiveSession, MVideoLiveVideo, MVideoLiveVideoWithSetting } from '@server/types/models' |
1ce4256a | 17 | import { pick, wait } from '@shared/core-utils' |
0c9668f7 | 18 | import { ffprobePromise, getVideoStreamBitrate, getVideoStreamDimensionsInfo, getVideoStreamFPS, hasAudioStream } from '@shared/ffmpeg' |
afb371d9 | 19 | import { LiveVideoError, VideoState } from '@shared/models' |
8ebf2a5d C |
20 | import { federateVideoIfNeeded } from '../activitypub/videos' |
21 | import { JobQueue } from '../job-queue' | |
afb371d9 | 22 | import { getLiveReplayBaseDirectory } from '../paths' |
df1db951 | 23 | import { PeerTubeSocket } from '../peertube-socket' |
84cae54e | 24 | import { Hooks } from '../plugins/hooks' |
0c9668f7 | 25 | import { computeResolutionsToTranscode } from '../transcoding/transcoding-resolutions' |
8ebf2a5d | 26 | import { LiveQuotaStore } from './live-quota-store' |
0c9668f7 | 27 | import { cleanupAndDestroyPermanentLive, getLiveSegmentTime } from './live-utils' |
8ebf2a5d C |
28 | import { MuxingSession } from './shared' |
29 | ||
7a397c7f C |
30 | const NodeRtmpSession = require('node-media-server/src/node_rtmp_session') |
31 | const context = require('node-media-server/src/node_core_ctx') | |
32 | const nodeMediaServerLogger = require('node-media-server/src/node_core_logger') | |
8ebf2a5d C |
33 | |
34 | // Disable node media server logs | |
35 | nodeMediaServerLogger.setLogType(0) | |
36 | ||
37 | const config = { | |
38 | rtmp: { | |
39 | port: CONFIG.LIVE.RTMP.PORT, | |
40 | chunk_size: VIDEO_LIVE.RTMP.CHUNK_SIZE, | |
41 | gop_cache: VIDEO_LIVE.RTMP.GOP_CACHE, | |
42 | ping: VIDEO_LIVE.RTMP.PING, | |
43 | ping_timeout: VIDEO_LIVE.RTMP.PING_TIMEOUT | |
8ebf2a5d C |
44 | } |
45 | } | |
46 | ||
47 | const lTags = loggerTagsFactory('live') | |
48 | ||
49 | class LiveManager { | |
50 | ||
51 | private static instance: LiveManager | |
52 | ||
53 | private readonly muxingSessions = new Map<string, MuxingSession>() | |
0c9668f7 | 54 | private readonly videoSessions = new Map<string, string>() |
8ebf2a5d C |
55 | |
56 | private rtmpServer: Server | |
df1db951 C |
57 | private rtmpsServer: ServerTLS |
58 | ||
59 | private running = false | |
8ebf2a5d C |
60 | |
61 | private constructor () { | |
62 | } | |
63 | ||
64 | init () { | |
65 | const events = this.getContext().nodeEvent | |
66 | events.on('postPublish', (sessionId: string, streamPath: string) => { | |
67 | logger.debug('RTMP received stream', { id: sessionId, streamPath, ...lTags(sessionId) }) | |
68 | ||
69 | const splittedPath = streamPath.split('/') | |
70 | if (splittedPath.length !== 3 || splittedPath[1] !== VIDEO_LIVE.RTMP.BASE_PATH) { | |
71 | logger.warn('Live path is incorrect.', { streamPath, ...lTags(sessionId) }) | |
72 | return this.abortSession(sessionId) | |
73 | } | |
74 | ||
df1db951 | 75 | const session = this.getContext().sessions.get(sessionId) |
28705705 C |
76 | const inputLocalUrl = session.inputOriginLocalUrl + streamPath |
77 | const inputPublicUrl = session.inputOriginPublicUrl + streamPath | |
df1db951 | 78 | |
28705705 | 79 | this.handleSession({ sessionId, inputPublicUrl, inputLocalUrl, streamKey: splittedPath[2] }) |
8ebf2a5d C |
80 | .catch(err => logger.error('Cannot handle sessions.', { err, ...lTags(sessionId) })) |
81 | }) | |
82 | ||
83 | events.on('donePublish', sessionId => { | |
84 | logger.info('Live session ended.', { sessionId, ...lTags(sessionId) }) | |
5a05c145 C |
85 | |
86 | // Force session aborting, so we kill ffmpeg even if it still has data to process (slow CPU) | |
87 | setTimeout(() => this.abortSession(sessionId), 2000) | |
8ebf2a5d C |
88 | }) |
89 | ||
90 | registerConfigChangedHandler(() => { | |
df1db951 C |
91 | if (!this.running && CONFIG.LIVE.ENABLED === true) { |
92 | this.run().catch(err => logger.error('Cannot run live server.', { err })) | |
8ebf2a5d C |
93 | return |
94 | } | |
95 | ||
df1db951 | 96 | if (this.running && CONFIG.LIVE.ENABLED === false) { |
8ebf2a5d C |
97 | this.stop() |
98 | } | |
99 | }) | |
100 | ||
101 | // Cleanup broken lives, that were terminated by a server restart for example | |
102 | this.handleBrokenLives() | |
103 | .catch(err => logger.error('Cannot handle broken lives.', { err, ...lTags() })) | |
8ebf2a5d C |
104 | } |
105 | ||
df1db951 C |
106 | async run () { |
107 | this.running = true | |
8ebf2a5d | 108 | |
df1db951 C |
109 | if (CONFIG.LIVE.RTMP.ENABLED) { |
110 | logger.info('Running RTMP server on port %d', CONFIG.LIVE.RTMP.PORT, lTags()) | |
8ebf2a5d | 111 | |
df1db951 C |
112 | this.rtmpServer = createServer(socket => { |
113 | const session = new NodeRtmpSession(config, socket) | |
8ebf2a5d | 114 | |
28705705 C |
115 | session.inputOriginLocalUrl = 'rtmp://127.0.0.1:' + CONFIG.LIVE.RTMP.PORT |
116 | session.inputOriginPublicUrl = WEBSERVER.RTMP_URL | |
df1db951 C |
117 | session.run() |
118 | }) | |
119 | ||
120 | this.rtmpServer.on('error', err => { | |
121 | logger.error('Cannot run RTMP server.', { err, ...lTags() }) | |
122 | }) | |
123 | ||
cfbe6be5 | 124 | this.rtmpServer.listen(CONFIG.LIVE.RTMP.PORT, CONFIG.LIVE.RTMP.HOSTNAME) |
df1db951 | 125 | } |
8ebf2a5d | 126 | |
df1db951 C |
127 | if (CONFIG.LIVE.RTMPS.ENABLED) { |
128 | logger.info('Running RTMPS server on port %d', CONFIG.LIVE.RTMPS.PORT, lTags()) | |
129 | ||
130 | const [ key, cert ] = await Promise.all([ | |
131 | readFile(CONFIG.LIVE.RTMPS.KEY_FILE), | |
132 | readFile(CONFIG.LIVE.RTMPS.CERT_FILE) | |
133 | ]) | |
134 | const serverOptions = { key, cert } | |
135 | ||
136 | this.rtmpsServer = createServerTLS(serverOptions, socket => { | |
137 | const session = new NodeRtmpSession(config, socket) | |
138 | ||
28705705 C |
139 | session.inputOriginLocalUrl = 'rtmps://127.0.0.1:' + CONFIG.LIVE.RTMPS.PORT |
140 | session.inputOriginPublicUrl = WEBSERVER.RTMPS_URL | |
df1db951 C |
141 | session.run() |
142 | }) | |
143 | ||
144 | this.rtmpsServer.on('error', err => { | |
145 | logger.error('Cannot run RTMPS server.', { err, ...lTags() }) | |
146 | }) | |
147 | ||
cfbe6be5 | 148 | this.rtmpsServer.listen(CONFIG.LIVE.RTMPS.PORT, CONFIG.LIVE.RTMPS.HOSTNAME) |
df1db951 | 149 | } |
8ebf2a5d C |
150 | } |
151 | ||
152 | stop () { | |
df1db951 C |
153 | this.running = false |
154 | ||
5037e0e4 C |
155 | if (this.rtmpServer) { |
156 | logger.info('Stopping RTMP server.', lTags()) | |
8ebf2a5d | 157 | |
5037e0e4 C |
158 | this.rtmpServer.close() |
159 | this.rtmpServer = undefined | |
160 | } | |
161 | ||
162 | if (this.rtmpsServer) { | |
163 | logger.info('Stopping RTMPS server.', lTags()) | |
164 | ||
165 | this.rtmpsServer.close() | |
166 | this.rtmpsServer = undefined | |
167 | } | |
8ebf2a5d C |
168 | |
169 | // Sessions is an object | |
170 | this.getContext().sessions.forEach((session: any) => { | |
171 | if (session instanceof NodeRtmpSession) { | |
172 | session.stop() | |
173 | } | |
174 | }) | |
175 | } | |
176 | ||
177 | isRunning () { | |
178 | return !!this.rtmpServer | |
179 | } | |
180 | ||
0c9668f7 C |
181 | stopSessionOf (videoUUID: string, error: LiveVideoError | null) { |
182 | const sessionId = this.videoSessions.get(videoUUID) | |
183 | if (!sessionId) { | |
184 | logger.debug('No live session to stop for video %s', videoUUID, lTags(sessionId, videoUUID)) | |
185 | return | |
186 | } | |
8ebf2a5d | 187 | |
0c9668f7 | 188 | logger.info('Stopping live session of video %s', videoUUID, { error, ...lTags(sessionId, videoUUID) }) |
26e3e98f | 189 | |
0c9668f7 C |
190 | this.saveEndingSession(videoUUID, error) |
191 | .catch(err => logger.error('Cannot save ending session.', { err, ...lTags(sessionId, videoUUID) })) | |
192 | ||
193 | this.videoSessions.delete(videoUUID) | |
8ebf2a5d C |
194 | this.abortSession(sessionId) |
195 | } | |
196 | ||
8ebf2a5d C |
197 | private getContext () { |
198 | return context | |
199 | } | |
200 | ||
201 | private abortSession (sessionId: string) { | |
202 | const session = this.getContext().sessions.get(sessionId) | |
203 | if (session) { | |
204 | session.stop() | |
205 | this.getContext().sessions.delete(sessionId) | |
206 | } | |
207 | ||
208 | const muxingSession = this.muxingSessions.get(sessionId) | |
609a4442 | 209 | if (muxingSession) { |
e466544f | 210 | // Muxing session will fire and event so we correctly cleanup the session |
609a4442 | 211 | muxingSession.abort() |
609a4442 C |
212 | |
213 | this.muxingSessions.delete(sessionId) | |
214 | } | |
8ebf2a5d C |
215 | } |
216 | ||
28705705 C |
217 | private async handleSession (options: { |
218 | sessionId: string | |
219 | inputLocalUrl: string | |
220 | inputPublicUrl: string | |
221 | streamKey: string | |
222 | }) { | |
223 | const { inputLocalUrl, inputPublicUrl, sessionId, streamKey } = options | |
224 | ||
8ebf2a5d C |
225 | const videoLive = await VideoLiveModel.loadByStreamKey(streamKey) |
226 | if (!videoLive) { | |
227 | logger.warn('Unknown live video with stream key %s.', streamKey, lTags(sessionId)) | |
228 | return this.abortSession(sessionId) | |
229 | } | |
230 | ||
231 | const video = videoLive.Video | |
232 | if (video.isBlacklisted()) { | |
233 | logger.warn('Video is blacklisted. Refusing stream %s.', streamKey, lTags(sessionId, video.uuid)) | |
234 | return this.abortSession(sessionId) | |
235 | } | |
236 | ||
0c9668f7 C |
237 | if (this.videoSessions.has(video.uuid)) { |
238 | logger.warn('Video %s has already a live session. Refusing stream %s.', video.uuid, streamKey, lTags(sessionId, video.uuid)) | |
239 | return this.abortSession(sessionId) | |
240 | } | |
241 | ||
92083e42 | 242 | // Cleanup old potential live (could happen with a permanent live) |
8ebf2a5d C |
243 | const oldStreamingPlaylist = await VideoStreamingPlaylistModel.loadHLSPlaylistByVideo(video.id) |
244 | if (oldStreamingPlaylist) { | |
5333788c C |
245 | if (!videoLive.permanentLive) throw new Error('Found previous session in a non permanent live: ' + video.uuid) |
246 | ||
cfd57d2c | 247 | await cleanupAndDestroyPermanentLive(video, oldStreamingPlaylist) |
8ebf2a5d C |
248 | } |
249 | ||
0c9668f7 | 250 | this.videoSessions.set(video.uuid, sessionId) |
8ebf2a5d | 251 | |
c826f34a | 252 | const now = Date.now() |
28705705 | 253 | const probe = await ffprobePromise(inputLocalUrl) |
c826f34a | 254 | |
1ce4256a | 255 | const [ { resolution, ratio }, fps, bitrate, hasAudio ] = await Promise.all([ |
28705705 C |
256 | getVideoStreamDimensionsInfo(inputLocalUrl, probe), |
257 | getVideoStreamFPS(inputLocalUrl, probe), | |
258 | getVideoStreamBitrate(inputLocalUrl, probe), | |
259 | hasAudioStream(inputLocalUrl, probe) | |
8ebf2a5d C |
260 | ]) |
261 | ||
c826f34a C |
262 | logger.info( |
263 | '%s probing took %d ms (bitrate: %d, fps: %d, resolution: %d)', | |
28705705 | 264 | inputLocalUrl, Date.now() - now, bitrate, fps, resolution, lTags(sessionId, video.uuid) |
c826f34a C |
265 | ) |
266 | ||
ebb9e53a | 267 | const allResolutions = await Hooks.wrapObject( |
a32bf8cd | 268 | this.buildAllResolutionsToTranscode(resolution, hasAudio), |
64fd6158 | 269 | 'filter:transcoding.auto.resolutions-to-transcode.result', |
ebb9e53a C |
270 | { video } |
271 | ) | |
8ebf2a5d C |
272 | |
273 | logger.info( | |
0c9668f7 | 274 | 'Handling live video of original resolution %d.', resolution, |
8ebf2a5d C |
275 | { allResolutions, ...lTags(sessionId, video.uuid) } |
276 | ) | |
277 | ||
8ebf2a5d C |
278 | return this.runMuxingSession({ |
279 | sessionId, | |
280 | videoLive, | |
1ce4256a | 281 | |
28705705 C |
282 | inputLocalUrl, |
283 | inputPublicUrl, | |
8ebf2a5d | 284 | fps, |
c826f34a | 285 | bitrate, |
679c12e6 | 286 | ratio, |
1ce4256a C |
287 | allResolutions, |
288 | hasAudio | |
8ebf2a5d C |
289 | }) |
290 | } | |
291 | ||
292 | private async runMuxingSession (options: { | |
293 | sessionId: string | |
05a60d85 | 294 | videoLive: MVideoLiveVideoWithSetting |
1ce4256a | 295 | |
28705705 C |
296 | inputLocalUrl: string |
297 | inputPublicUrl: string | |
298 | ||
8ebf2a5d | 299 | fps: number |
c826f34a | 300 | bitrate: number |
679c12e6 | 301 | ratio: number |
8ebf2a5d | 302 | allResolutions: number[] |
1ce4256a | 303 | hasAudio: boolean |
8ebf2a5d | 304 | }) { |
1ce4256a | 305 | const { sessionId, videoLive } = options |
8ebf2a5d C |
306 | const videoUUID = videoLive.Video.uuid |
307 | const localLTags = lTags(sessionId, videoUUID) | |
308 | ||
26e3e98f C |
309 | const liveSession = await this.saveStartingSession(videoLive) |
310 | ||
8ebf2a5d C |
311 | const user = await UserModel.loadByLiveId(videoLive.id) |
312 | LiveQuotaStore.Instance.addNewLive(user.id, videoLive.id) | |
313 | ||
314 | const muxingSession = new MuxingSession({ | |
315 | context: this.getContext(), | |
8ebf2a5d C |
316 | sessionId, |
317 | videoLive, | |
1ce4256a C |
318 | user, |
319 | ||
28705705 | 320 | ...pick(options, [ 'inputLocalUrl', 'inputPublicUrl', 'bitrate', 'ratio', 'fps', 'allResolutions', 'hasAudio' ]) |
8ebf2a5d C |
321 | }) |
322 | ||
cfd57d2c | 323 | muxingSession.on('live-ready', () => this.publishAndFederateLive(videoLive, localLTags)) |
8ebf2a5d | 324 | |
0c9668f7 | 325 | muxingSession.on('bad-socket-health', ({ videoUUID }) => { |
8ebf2a5d C |
326 | logger.error( |
327 | 'Too much data in client socket stream (ffmpeg is too slow to transcode the video).' + | |
328 | ' Stopping session of video %s.', videoUUID, | |
329 | localLTags | |
330 | ) | |
331 | ||
0c9668f7 | 332 | this.stopSessionOf(videoUUID, LiveVideoError.BAD_SOCKET_HEALTH) |
8ebf2a5d C |
333 | }) |
334 | ||
0c9668f7 | 335 | muxingSession.on('duration-exceeded', ({ videoUUID }) => { |
8ebf2a5d C |
336 | logger.info('Stopping session of %s: max duration exceeded.', videoUUID, localLTags) |
337 | ||
0c9668f7 | 338 | this.stopSessionOf(videoUUID, LiveVideoError.DURATION_EXCEEDED) |
8ebf2a5d C |
339 | }) |
340 | ||
0c9668f7 | 341 | muxingSession.on('quota-exceeded', ({ videoUUID }) => { |
8ebf2a5d C |
342 | logger.info('Stopping session of %s: user quota exceeded.', videoUUID, localLTags) |
343 | ||
0c9668f7 | 344 | this.stopSessionOf(videoUUID, LiveVideoError.QUOTA_EXCEEDED) |
26e3e98f C |
345 | }) |
346 | ||
0c9668f7 C |
347 | muxingSession.on('transcoding-error', ({ videoUUID }) => { |
348 | this.stopSessionOf(videoUUID, LiveVideoError.FFMPEG_ERROR) | |
8ebf2a5d C |
349 | }) |
350 | ||
0c9668f7 C |
351 | muxingSession.on('transcoding-end', ({ videoUUID }) => { |
352 | this.onMuxingFFmpegEnd(videoUUID, sessionId) | |
8ebf2a5d C |
353 | }) |
354 | ||
0c9668f7 | 355 | muxingSession.on('after-cleanup', ({ videoUUID }) => { |
8ebf2a5d C |
356 | this.muxingSessions.delete(sessionId) |
357 | ||
9a82ce24 C |
358 | LiveQuotaStore.Instance.removeLive(user.id, videoLive.id) |
359 | ||
609a4442 C |
360 | muxingSession.destroy() |
361 | ||
0c9668f7 | 362 | return this.onAfterMuxingCleanup({ videoUUID, liveSession }) |
8ebf2a5d C |
363 | .catch(err => logger.error('Error in end transmuxing.', { err, ...localLTags })) |
364 | }) | |
365 | ||
366 | this.muxingSessions.set(sessionId, muxingSession) | |
367 | ||
368 | muxingSession.runMuxing() | |
369 | .catch(err => { | |
370 | logger.error('Cannot run muxing.', { err, ...localLTags }) | |
371 | this.abortSession(sessionId) | |
372 | }) | |
373 | } | |
374 | ||
375 | private async publishAndFederateLive (live: MVideoLiveVideo, localLTags: { tags: string[] }) { | |
376 | const videoId = live.videoId | |
377 | ||
378 | try { | |
4fae2b1f | 379 | const video = await VideoModel.loadFull(videoId) |
8ebf2a5d C |
380 | |
381 | logger.info('Will publish and federate live %s.', video.url, localLTags) | |
382 | ||
383 | video.state = VideoState.PUBLISHED | |
7137377d | 384 | video.publishedAt = new Date() |
8ebf2a5d C |
385 | await video.save() |
386 | ||
387 | live.Video = video | |
388 | ||
4ec52d04 | 389 | await wait(getLiveSegmentTime(live.latencyMode) * 1000 * VIDEO_LIVE.EDGE_LIVE_DELAY_SEGMENTS_NOTIFICATION) |
8ebf2a5d | 390 | |
4ec52d04 C |
391 | try { |
392 | await federateVideoIfNeeded(video, false) | |
393 | } catch (err) { | |
394 | logger.error('Cannot federate live video %s.', video.url, { err, ...localLTags }) | |
395 | } | |
396 | ||
397 | PeerTubeSocket.Instance.sendVideoLiveNewState(video) | |
8ebf2a5d C |
398 | } catch (err) { |
399 | logger.error('Cannot save/federate live video %d.', videoId, { err, ...localLTags }) | |
400 | } | |
401 | } | |
402 | ||
0c9668f7 | 403 | private onMuxingFFmpegEnd (videoUUID: string, sessionId: string) { |
c08a7f16 C |
404 | // Session already cleaned up |
405 | if (!this.videoSessions.has(videoUUID)) return | |
406 | ||
0c9668f7 | 407 | this.videoSessions.delete(videoUUID) |
26e3e98f | 408 | |
0c9668f7 | 409 | this.saveEndingSession(videoUUID, null) |
26e3e98f | 410 | .catch(err => logger.error('Cannot save ending session.', { err, ...lTags(sessionId) })) |
8ebf2a5d C |
411 | } |
412 | ||
4ec52d04 | 413 | private async onAfterMuxingCleanup (options: { |
0c9668f7 | 414 | videoUUID: string |
26e3e98f | 415 | liveSession?: MVideoLiveSession |
4ec52d04 C |
416 | cleanupNow?: boolean // Default false |
417 | }) { | |
0c9668f7 C |
418 | const { videoUUID, liveSession: liveSessionArg, cleanupNow = false } = options |
419 | ||
420 | logger.debug('Live of video %s has been cleaned up. Moving to its next state.', videoUUID, lTags(videoUUID)) | |
4ec52d04 | 421 | |
8ebf2a5d | 422 | try { |
0c9668f7 | 423 | const fullVideo = await VideoModel.loadFull(videoUUID) |
8ebf2a5d C |
424 | if (!fullVideo) return |
425 | ||
426 | const live = await VideoLiveModel.loadByVideoId(fullVideo.id) | |
427 | ||
46f7cd68 | 428 | const liveSession = liveSessionArg ?? await VideoLiveSessionModel.findLatestSessionOf(fullVideo.id) |
26e3e98f C |
429 | |
430 | // On server restart during a live | |
431 | if (!liveSession.endDate) { | |
432 | liveSession.endDate = new Date() | |
433 | await liveSession.save() | |
434 | } | |
435 | ||
bd911b54 | 436 | JobQueue.Instance.createJobAsync({ |
4ec52d04 C |
437 | type: 'video-live-ending', |
438 | payload: { | |
439 | videoId: fullVideo.id, | |
26e3e98f | 440 | |
4ec52d04 C |
441 | replayDirectory: live.saveReplay |
442 | ? await this.findReplayDirectory(fullVideo) | |
443 | : undefined, | |
26e3e98f C |
444 | |
445 | liveSessionId: liveSession.id, | |
cdd83816 | 446 | streamingPlaylistId: fullVideo.getHLSPlaylist()?.id, |
26e3e98f | 447 | |
4ec52d04 | 448 | publishedAt: fullVideo.publishedAt.toISOString() |
bd911b54 C |
449 | }, |
450 | ||
451 | delay: cleanupNow | |
452 | ? 0 | |
453 | : VIDEO_LIVE.CLEANUP_DELAY | |
454 | }) | |
4ec52d04 C |
455 | |
456 | fullVideo.state = live.permanentLive | |
457 | ? VideoState.WAITING_FOR_LIVE | |
458 | : VideoState.LIVE_ENDED | |
8ebf2a5d C |
459 | |
460 | await fullVideo.save() | |
461 | ||
462 | PeerTubeSocket.Instance.sendVideoLiveNewState(fullVideo) | |
463 | ||
464 | await federateVideoIfNeeded(fullVideo, false) | |
465 | } catch (err) { | |
0c9668f7 | 466 | logger.error('Cannot save/federate new video state of live streaming of video %s.', videoUUID, { err, ...lTags(videoUUID) }) |
8ebf2a5d C |
467 | } |
468 | } | |
469 | ||
8ebf2a5d | 470 | private async handleBrokenLives () { |
0c9668f7 C |
471 | await RunnerJobModel.cancelAllJobs({ type: 'live-rtmp-hls-transcoding' }) |
472 | ||
8ebf2a5d C |
473 | const videoUUIDs = await VideoModel.listPublishedLiveUUIDs() |
474 | ||
475 | for (const uuid of videoUUIDs) { | |
0c9668f7 | 476 | await this.onAfterMuxingCleanup({ videoUUID: uuid, cleanupNow: true }) |
8ebf2a5d C |
477 | } |
478 | } | |
479 | ||
4ec52d04 C |
480 | private async findReplayDirectory (video: MVideo) { |
481 | const directory = getLiveReplayBaseDirectory(video) | |
482 | const files = await readdir(directory) | |
483 | ||
484 | if (files.length === 0) return undefined | |
485 | ||
486 | return join(directory, files.sort().reverse()[0]) | |
487 | } | |
488 | ||
a32bf8cd | 489 | private buildAllResolutionsToTranscode (originResolution: number, hasAudio: boolean) { |
5e2afe42 | 490 | const includeInput = CONFIG.LIVE.TRANSCODING.ALWAYS_TRANSCODE_ORIGINAL_RESOLUTION |
84cae54e | 491 | |
8ebf2a5d | 492 | const resolutionsEnabled = CONFIG.LIVE.TRANSCODING.ENABLED |
a32bf8cd | 493 | ? computeResolutionsToTranscode({ input: originResolution, type: 'live', includeInput, strictLower: false, hasAudio }) |
8ebf2a5d C |
494 | : [] |
495 | ||
84cae54e C |
496 | if (resolutionsEnabled.length === 0) { |
497 | return [ originResolution ] | |
498 | } | |
499 | ||
500 | return resolutionsEnabled | |
8ebf2a5d C |
501 | } |
502 | ||
05a60d85 W |
503 | private async saveStartingSession (videoLive: MVideoLiveVideoWithSetting) { |
504 | const replaySettings = videoLive.saveReplay | |
505 | ? new VideoLiveReplaySettingModel({ | |
506 | privacy: videoLive.ReplaySetting.privacy | |
507 | }) | |
508 | : null | |
26e3e98f | 509 | |
05a60d85 W |
510 | return sequelizeTypescript.transaction(async t => { |
511 | if (videoLive.saveReplay) { | |
512 | await replaySettings.save({ transaction: t }) | |
513 | } | |
514 | ||
515 | return VideoLiveSessionModel.create({ | |
516 | startDate: new Date(), | |
517 | liveVideoId: videoLive.videoId, | |
518 | saveReplay: videoLive.saveReplay, | |
519 | replaySettingId: videoLive.saveReplay ? replaySettings.id : null, | |
520 | endingProcessed: false | |
521 | }, { transaction: t }) | |
522 | }) | |
26e3e98f C |
523 | } |
524 | ||
0c9668f7 C |
525 | private async saveEndingSession (videoUUID: string, error: LiveVideoError | null) { |
526 | const liveSession = await VideoLiveSessionModel.findCurrentSessionOf(videoUUID) | |
0755cb89 C |
527 | if (!liveSession) return |
528 | ||
26e3e98f C |
529 | liveSession.endDate = new Date() |
530 | liveSession.error = error | |
531 | ||
532 | return liveSession.save() | |
533 | } | |
534 | ||
8ebf2a5d C |
535 | static get Instance () { |
536 | return this.instance || (this.instance = new this()) | |
537 | } | |
538 | } | |
539 | ||
540 | // --------------------------------------------------------------------------- | |
541 | ||
542 | export { | |
543 | LiveManager | |
544 | } |