1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
import express from 'express'
import { PassThrough, pipeline } from 'stream'
import { GetObjectCommandOutput } from '@aws-sdk/client-s3'
import { buildReinjectVideoFileTokenQuery } from '@server/controllers/shared/m3u8-playlist'
import { logger } from '@server/helpers/logger'
import { StreamReplacer } from '@server/helpers/stream-replacer'
import { MStreamingPlaylist, MVideo } from '@server/types/models'
import { HttpStatusCode } from '@shared/models'
import { injectQueryToPlaylistUrls } from '../hls'
import { getHLSFileReadStream, getWebTorrentFileReadStream } from './videos'
export async function proxifyWebTorrentFile (options: {
req: express.Request
res: express.Response
filename: string
}) {
const { req, res, filename } = options
logger.debug('Proxifying WebTorrent file %s from object storage.', filename)
try {
const { response: s3Response, stream } = await getWebTorrentFileReadStream({
filename,
rangeHeader: req.header('range')
})
setS3Headers(res, s3Response)
return stream.pipe(res)
} catch (err) {
return handleObjectStorageFailure(res, err)
}
}
export async function proxifyHLS (options: {
req: express.Request
res: express.Response
playlist: MStreamingPlaylist
video: MVideo
filename: string
reinjectVideoFileToken: boolean
}) {
const { req, res, playlist, video, filename, reinjectVideoFileToken } = options
logger.debug('Proxifying HLS file %s from object storage.', filename)
try {
const { response: s3Response, stream } = await getHLSFileReadStream({
playlist: playlist.withVideo(video),
filename,
rangeHeader: req.header('range')
})
setS3Headers(res, s3Response)
const streamReplacer = reinjectVideoFileToken
? new StreamReplacer(line => injectQueryToPlaylistUrls(line, buildReinjectVideoFileTokenQuery(req, filename.endsWith('master.m3u8'))))
: new PassThrough()
return pipeline(
stream,
streamReplacer,
res,
err => {
if (!err) return
handleObjectStorageFailure(res, err)
}
)
} catch (err) {
return handleObjectStorageFailure(res, err)
}
}
// ---------------------------------------------------------------------------
// Private
// ---------------------------------------------------------------------------
function handleObjectStorageFailure (res: express.Response, err: Error) {
if (err.name === 'NoSuchKey') {
logger.debug('Could not find key in object storage to proxify private HLS video file.', { err })
return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
}
return res.fail({
status: HttpStatusCode.INTERNAL_SERVER_ERROR_500,
message: err.message,
type: err.name
})
}
function setS3Headers (res: express.Response, s3Response: GetObjectCommandOutput) {
if (s3Response.$metadata.httpStatusCode === HttpStatusCode.PARTIAL_CONTENT_206) {
res.setHeader('Content-Range', s3Response.ContentRange)
res.status(HttpStatusCode.PARTIAL_CONTENT_206)
}
}
|
