]>
Commit | Line | Data |
---|---|---|
1 | import cors from 'cors' | |
2 | import express from 'express' | |
3 | import { PassThrough, pipeline } from 'stream' | |
4 | import { logger } from '@server/helpers/logger' | |
5 | import { StreamReplacer } from '@server/helpers/stream-replacer' | |
6 | import { OBJECT_STORAGE_PROXY_PATHS } from '@server/initializers/constants' | |
7 | import { injectQueryToPlaylistUrls } from '@server/lib/hls' | |
8 | import { getHLSFileReadStream, getWebTorrentFileReadStream } from '@server/lib/object-storage' | |
9 | import { | |
10 | asyncMiddleware, | |
11 | ensureCanAccessPrivateVideoHLSFiles, | |
12 | ensureCanAccessVideoPrivateWebTorrentFiles, | |
13 | ensurePrivateObjectStorageProxyIsEnabled, | |
14 | optionalAuthenticate | |
15 | } from '@server/middlewares' | |
16 | import { HttpStatusCode } from '@shared/models' | |
17 | import { buildReinjectVideoFileTokenQuery, doReinjectVideoFileToken } from './shared/m3u8-playlist' | |
18 | import { GetObjectCommandOutput } from '@aws-sdk/client-s3' | |
19 | ||
20 | const objectStorageProxyRouter = express.Router() | |
21 | ||
22 | objectStorageProxyRouter.use(cors()) | |
23 | ||
24 | objectStorageProxyRouter.get(OBJECT_STORAGE_PROXY_PATHS.PRIVATE_WEBSEED + ':filename', | |
25 | ensurePrivateObjectStorageProxyIsEnabled, | |
26 | optionalAuthenticate, | |
27 | asyncMiddleware(ensureCanAccessVideoPrivateWebTorrentFiles), | |
28 | asyncMiddleware(proxifyWebTorrent) | |
29 | ) | |
30 | ||
31 | objectStorageProxyRouter.get(OBJECT_STORAGE_PROXY_PATHS.STREAMING_PLAYLISTS.PRIVATE_HLS + ':videoUUID/:filename', | |
32 | ensurePrivateObjectStorageProxyIsEnabled, | |
33 | optionalAuthenticate, | |
34 | asyncMiddleware(ensureCanAccessPrivateVideoHLSFiles), | |
35 | asyncMiddleware(proxifyHLS) | |
36 | ) | |
37 | ||
38 | // --------------------------------------------------------------------------- | |
39 | ||
40 | export { | |
41 | objectStorageProxyRouter | |
42 | } | |
43 | ||
44 | async function proxifyWebTorrent (req: express.Request, res: express.Response) { | |
45 | const filename = req.params.filename | |
46 | ||
47 | logger.debug('Proxifying WebTorrent file %s from object storage.', filename) | |
48 | ||
49 | try { | |
50 | const { response: s3Response, stream } = await getWebTorrentFileReadStream({ | |
51 | filename, | |
52 | rangeHeader: req.header('range') | |
53 | }) | |
54 | ||
55 | setS3Headers(res, s3Response) | |
56 | ||
57 | return stream.pipe(res) | |
58 | } catch (err) { | |
59 | return handleObjectStorageFailure(res, err) | |
60 | } | |
61 | } | |
62 | ||
63 | async function proxifyHLS (req: express.Request, res: express.Response) { | |
64 | const playlist = res.locals.videoStreamingPlaylist | |
65 | const video = res.locals.onlyVideo | |
66 | const filename = req.params.filename | |
67 | ||
68 | logger.debug('Proxifying HLS file %s from object storage.', filename) | |
69 | ||
70 | try { | |
71 | const { response: s3Response, stream } = await getHLSFileReadStream({ | |
72 | playlist: playlist.withVideo(video), | |
73 | filename, | |
74 | rangeHeader: req.header('range') | |
75 | }) | |
76 | ||
77 | setS3Headers(res, s3Response) | |
78 | ||
79 | const streamReplacer = filename.endsWith('.m3u8') && doReinjectVideoFileToken(req) | |
80 | ? new StreamReplacer(line => injectQueryToPlaylistUrls(line, buildReinjectVideoFileTokenQuery(req))) | |
81 | : new PassThrough() | |
82 | ||
83 | return pipeline( | |
84 | stream, | |
85 | streamReplacer, | |
86 | res, | |
87 | err => { | |
88 | if (!err) return | |
89 | ||
90 | handleObjectStorageFailure(res, err) | |
91 | } | |
92 | ) | |
93 | } catch (err) { | |
94 | return handleObjectStorageFailure(res, err) | |
95 | } | |
96 | } | |
97 | ||
98 | function handleObjectStorageFailure (res: express.Response, err: Error) { | |
99 | if (err.name === 'NoSuchKey') { | |
100 | logger.debug('Could not find key in object storage to proxify private HLS video file.', { err }) | |
101 | return res.sendStatus(HttpStatusCode.NOT_FOUND_404) | |
102 | } | |
103 | ||
104 | return res.fail({ | |
105 | status: HttpStatusCode.INTERNAL_SERVER_ERROR_500, | |
106 | message: err.message, | |
107 | type: err.name | |
108 | }) | |
109 | } | |
110 | ||
111 | function setS3Headers (res: express.Response, s3Response: GetObjectCommandOutput) { | |
112 | if (s3Response.$metadata.httpStatusCode === HttpStatusCode.PARTIAL_CONTENT_206) { | |
113 | res.status(HttpStatusCode.PARTIAL_CONTENT_206) | |
114 | } | |
115 | } |