]>
Commit | Line | Data |
---|---|---|
c55e3d72 C |
1 | /* eslint-disable @typescript-eslint/no-unused-expressions,@typescript-eslint/require-await */ |
2 | ||
3 | import { expect } from 'chai' | |
4 | import { pathExists, readdir } from 'fs-extra' | |
5 | import { join } from 'path' | |
cfd57d2c C |
6 | import { wait } from '@shared/core-utils' |
7 | import { LiveVideo, VideoStreamingPlaylistType } from '@shared/models' | |
8 | import { ObjectStorageCommand, PeerTubeServer } from '@shared/server-commands' | |
9 | import { checkLiveSegmentHash, checkResolutionsInMasterPlaylist } from './streaming-playlists' | |
c55e3d72 | 10 | |
4ec52d04 | 11 | async function checkLiveCleanup (server: PeerTubeServer, videoUUID: string, savedResolutions: number[] = []) { |
c55e3d72 C |
12 | const basePath = server.servers.buildDirectory('streaming-playlists') |
13 | const hlsPath = join(basePath, 'hls', videoUUID) | |
14 | ||
4ec52d04 | 15 | if (savedResolutions.length === 0) { |
cfd57d2c C |
16 | return checkUnsavedLiveCleanup(server, videoUUID, hlsPath) |
17 | } | |
18 | ||
19 | return checkSavedLiveCleanup(hlsPath, savedResolutions) | |
20 | } | |
21 | ||
22 | // --------------------------------------------------------------------------- | |
5333788c | 23 | |
cfd57d2c C |
24 | async function testVideoResolutions (options: { |
25 | originServer: PeerTubeServer | |
26 | servers: PeerTubeServer[] | |
27 | liveVideoId: string | |
28 | resolutions: number[] | |
29 | transcoded: boolean | |
30 | objectStorage: boolean | |
31 | }) { | |
32 | const { originServer, servers, liveVideoId, resolutions, transcoded, objectStorage } = options | |
5333788c | 33 | |
cfd57d2c C |
34 | for (const server of servers) { |
35 | const { data } = await server.videos.list() | |
36 | expect(data.find(v => v.uuid === liveVideoId)).to.exist | |
5333788c | 37 | |
cfd57d2c C |
38 | const video = await server.videos.get({ id: liveVideoId }) |
39 | expect(video.streamingPlaylists).to.have.lengthOf(1) | |
5333788c | 40 | |
cfd57d2c C |
41 | const hlsPlaylist = video.streamingPlaylists.find(s => s.type === VideoStreamingPlaylistType.HLS) |
42 | expect(hlsPlaylist).to.exist | |
43 | expect(hlsPlaylist.files).to.have.lengthOf(0) // Only fragmented mp4 files are displayed | |
44 | ||
45 | await checkResolutionsInMasterPlaylist({ server, playlistUrl: hlsPlaylist.playlistUrl, resolutions, transcoded }) | |
46 | ||
47 | if (objectStorage) { | |
48 | expect(hlsPlaylist.playlistUrl).to.contain(ObjectStorageCommand.getPlaylistBaseUrl()) | |
5333788c | 49 | } |
c55e3d72 | 50 | |
cfd57d2c C |
51 | for (let i = 0; i < resolutions.length; i++) { |
52 | const segmentNum = 3 | |
53 | const segmentName = `${i}-00000${segmentNum}.ts` | |
54 | await originServer.live.waitUntilSegmentGeneration({ videoUUID: video.uuid, playlistNumber: i, segment: segmentNum }) | |
55 | ||
56 | const baseUrl = objectStorage | |
57 | ? ObjectStorageCommand.getPlaylistBaseUrl() + 'hls' | |
58 | : originServer.url + '/static/streaming-playlists/hls' | |
59 | ||
60 | if (objectStorage) { | |
61 | // Playlist file upload | |
62 | await wait(500) | |
63 | ||
64 | expect(hlsPlaylist.segmentsSha256Url).to.contain(ObjectStorageCommand.getPlaylistBaseUrl()) | |
65 | } | |
66 | ||
67 | const subPlaylist = await originServer.streamingPlaylists.get({ url: `${baseUrl}/${video.uuid}/${i}.m3u8` }) | |
68 | ||
69 | expect(subPlaylist).to.contain(segmentName) | |
70 | ||
71 | await checkLiveSegmentHash({ | |
72 | server, | |
73 | baseUrlSegment: baseUrl, | |
74 | videoUUID: video.uuid, | |
75 | segmentName, | |
76 | hlsPlaylist | |
77 | }) | |
78 | } | |
c55e3d72 | 79 | } |
cfd57d2c C |
80 | } |
81 | ||
82 | // --------------------------------------------------------------------------- | |
83 | ||
84 | export { | |
85 | checkLiveCleanup, | |
86 | testVideoResolutions | |
87 | } | |
c55e3d72 | 88 | |
cfd57d2c C |
89 | // --------------------------------------------------------------------------- |
90 | ||
91 | async function checkSavedLiveCleanup (hlsPath: string, savedResolutions: number[] = []) { | |
c55e3d72 C |
92 | const files = await readdir(hlsPath) |
93 | ||
94 | // fragmented file and playlist per resolution + master playlist + segments sha256 json file | |
4ec52d04 | 95 | expect(files).to.have.lengthOf(savedResolutions.length * 2 + 2) |
c55e3d72 | 96 | |
4ec52d04 | 97 | for (const resolution of savedResolutions) { |
c55e3d72 C |
98 | const fragmentedFile = files.find(f => f.endsWith(`-${resolution}-fragmented.mp4`)) |
99 | expect(fragmentedFile).to.exist | |
100 | ||
101 | const playlistFile = files.find(f => f.endsWith(`${resolution}.m3u8`)) | |
102 | expect(playlistFile).to.exist | |
103 | } | |
104 | ||
105 | const masterPlaylistFile = files.find(f => f.endsWith('-master.m3u8')) | |
106 | expect(masterPlaylistFile).to.exist | |
107 | ||
108 | const shaFile = files.find(f => f.endsWith('-segments-sha256.json')) | |
109 | expect(shaFile).to.exist | |
110 | } | |
111 | ||
cfd57d2c C |
112 | async function checkUnsavedLiveCleanup (server: PeerTubeServer, videoUUID: string, hlsPath: string) { |
113 | let live: LiveVideo | |
114 | ||
115 | try { | |
116 | live = await server.live.get({ videoId: videoUUID }) | |
117 | } catch {} | |
118 | ||
119 | if (live?.permanentLive) { | |
120 | expect(await pathExists(hlsPath)).to.be.true | |
121 | ||
122 | const hlsFiles = await readdir(hlsPath) | |
123 | expect(hlsFiles).to.have.lengthOf(1) // Only replays directory | |
124 | ||
125 | const replayDir = join(hlsPath, 'replay') | |
126 | expect(await pathExists(replayDir)).to.be.true | |
127 | ||
128 | const replayFiles = await readdir(join(hlsPath, 'replay')) | |
129 | expect(replayFiles).to.have.lengthOf(0) | |
130 | ||
131 | return | |
132 | } | |
133 | ||
134 | expect(await pathExists(hlsPath)).to.be.false | |
c55e3d72 | 135 | } |