]>
Commit | Line | Data |
---|---|---|
1 | import ffmpeg, { FfmpegCommand } from 'fluent-ffmpeg' | |
2 | import { truncate } from 'lodash' | |
3 | import { buildAbsoluteFixturePath, wait } from '@shared/core-utils' | |
4 | import { VideoDetails, VideoInclude, VideoPrivacy } from '@shared/models' | |
5 | import { PeerTubeServer } from '../server/server' | |
6 | ||
7 | function sendRTMPStream (options: { | |
8 | rtmpBaseUrl: string | |
9 | streamKey: string | |
10 | fixtureName?: string // default video_short.mp4 | |
11 | copyCodecs?: boolean // default false | |
12 | }) { | |
13 | const { rtmpBaseUrl, streamKey, fixtureName = 'video_short.mp4', copyCodecs = false } = options | |
14 | ||
15 | const fixture = buildAbsoluteFixturePath(fixtureName) | |
16 | ||
17 | const command = ffmpeg(fixture) | |
18 | command.inputOption('-stream_loop -1') | |
19 | command.inputOption('-re') | |
20 | ||
21 | if (copyCodecs) { | |
22 | command.outputOption('-c copy') | |
23 | } else { | |
24 | command.outputOption('-c:v libx264') | |
25 | command.outputOption('-g 120') | |
26 | command.outputOption('-x264-params "no-scenecut=1"') | |
27 | command.outputOption('-r 60') | |
28 | } | |
29 | ||
30 | command.outputOption('-f flv') | |
31 | ||
32 | const rtmpUrl = rtmpBaseUrl + '/' + streamKey | |
33 | command.output(rtmpUrl) | |
34 | ||
35 | command.on('error', err => { | |
36 | if (err?.message?.includes('Exiting normally')) return | |
37 | ||
38 | if (process.env.DEBUG) console.error(err) | |
39 | }) | |
40 | ||
41 | if (process.env.DEBUG) { | |
42 | command.on('stderr', data => console.log(data)) | |
43 | command.on('stdout', data => console.log(data)) | |
44 | } | |
45 | ||
46 | command.run() | |
47 | ||
48 | return command | |
49 | } | |
50 | ||
51 | function waitFfmpegUntilError (command: FfmpegCommand, successAfterMS = 10000) { | |
52 | return new Promise<void>((res, rej) => { | |
53 | command.on('error', err => { | |
54 | return rej(err) | |
55 | }) | |
56 | ||
57 | setTimeout(() => { | |
58 | res() | |
59 | }, successAfterMS) | |
60 | }) | |
61 | } | |
62 | ||
63 | async function testFfmpegStreamError (command: FfmpegCommand, shouldHaveError: boolean) { | |
64 | let error: Error | |
65 | ||
66 | try { | |
67 | await waitFfmpegUntilError(command, 45000) | |
68 | } catch (err) { | |
69 | error = err | |
70 | } | |
71 | ||
72 | await stopFfmpeg(command) | |
73 | ||
74 | if (shouldHaveError && !error) throw new Error('Ffmpeg did not have an error') | |
75 | if (!shouldHaveError && error) throw error | |
76 | } | |
77 | ||
78 | async function stopFfmpeg (command: FfmpegCommand) { | |
79 | command.kill('SIGINT') | |
80 | ||
81 | await wait(500) | |
82 | } | |
83 | ||
84 | async function waitUntilLivePublishedOnAllServers (servers: PeerTubeServer[], videoId: string) { | |
85 | for (const server of servers) { | |
86 | await server.live.waitUntilPublished({ videoId }) | |
87 | } | |
88 | } | |
89 | ||
90 | async function waitUntilLiveWaitingOnAllServers (servers: PeerTubeServer[], videoId: string) { | |
91 | for (const server of servers) { | |
92 | await server.live.waitUntilWaiting({ videoId }) | |
93 | } | |
94 | } | |
95 | ||
96 | async function waitUntilLiveReplacedByReplayOnAllServers (servers: PeerTubeServer[], videoId: string) { | |
97 | for (const server of servers) { | |
98 | await server.live.waitUntilReplacedByReplay({ videoId }) | |
99 | } | |
100 | } | |
101 | ||
102 | async function findExternalSavedVideo (server: PeerTubeServer, liveDetails: VideoDetails) { | |
103 | const include = VideoInclude.BLACKLISTED | |
104 | const privacyOneOf = [ VideoPrivacy.INTERNAL, VideoPrivacy.PRIVATE, VideoPrivacy.PUBLIC, VideoPrivacy.UNLISTED ] | |
105 | ||
106 | const { data } = await server.videos.list({ token: server.accessToken, sort: '-publishedAt', include, privacyOneOf }) | |
107 | ||
108 | const videoNameSuffix = ` - ${new Date(liveDetails.publishedAt).toLocaleString()}` | |
109 | const truncatedVideoName = truncate(liveDetails.name, { | |
110 | length: 120 - videoNameSuffix.length | |
111 | }) | |
112 | const toFind = truncatedVideoName + videoNameSuffix | |
113 | ||
114 | return data.find(v => v.name === toFind) | |
115 | } | |
116 | ||
117 | export { | |
118 | sendRTMPStream, | |
119 | waitFfmpegUntilError, | |
120 | testFfmpegStreamError, | |
121 | stopFfmpeg, | |
122 | ||
123 | waitUntilLivePublishedOnAllServers, | |
124 | waitUntilLiveReplacedByReplayOnAllServers, | |
125 | waitUntilLiveWaitingOnAllServers, | |
126 | ||
127 | findExternalSavedVideo | |
128 | } |