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'
11 ensureCanAccessPrivateVideoHLSFiles,
12 ensureCanAccessVideoPrivateWebTorrentFiles,
13 ensurePrivateObjectStorageProxyIsEnabled,
15 } from '@server/middlewares'
16 import { HttpStatusCode } from '@shared/models'
17 import { buildReinjectVideoFileTokenQuery, doReinjectVideoFileToken } from './shared/m3u8-playlist'
19 const objectStorageProxyRouter = express.Router()
21 objectStorageProxyRouter.use(cors())
23 objectStorageProxyRouter.get(OBJECT_STORAGE_PROXY_PATHS.PRIVATE_WEBSEED + ':filename',
24 ensurePrivateObjectStorageProxyIsEnabled,
26 asyncMiddleware(ensureCanAccessVideoPrivateWebTorrentFiles),
27 asyncMiddleware(proxifyWebTorrent)
30 objectStorageProxyRouter.get(OBJECT_STORAGE_PROXY_PATHS.STREAMING_PLAYLISTS.PRIVATE_HLS + ':videoUUID/:filename',
31 ensurePrivateObjectStorageProxyIsEnabled,
33 asyncMiddleware(ensureCanAccessPrivateVideoHLSFiles),
34 asyncMiddleware(proxifyHLS)
37 // ---------------------------------------------------------------------------
40 objectStorageProxyRouter
43 async function proxifyWebTorrent (req: express.Request, res: express.Response) {
44 const filename = req.params.filename
46 logger.debug('Proxifying WebTorrent file %s from object storage.', filename)
49 const stream = await getWebTorrentFileReadStream({
51 rangeHeader: req.header('range')
54 return stream.pipe(res)
56 return handleObjectStorageFailure(res, err)
60 async function proxifyHLS (req: express.Request, res: express.Response) {
61 const playlist = res.locals.videoStreamingPlaylist
62 const video = res.locals.onlyVideo
63 const filename = req.params.filename
65 logger.debug('Proxifying HLS file %s from object storage.', filename)
68 const stream = await getHLSFileReadStream({
69 playlist: playlist.withVideo(video),
71 rangeHeader: req.header('range')
74 const streamReplacer = filename.endsWith('.m3u8') && doReinjectVideoFileToken(req)
75 ? new StreamReplacer(line => injectQueryToPlaylistUrls(line, buildReinjectVideoFileTokenQuery(req)))
85 handleObjectStorageFailure(res, err)
89 return handleObjectStorageFailure(res, err)
93 function handleObjectStorageFailure (res: express.Response, err: Error) {
94 if (err.name === 'NoSuchKey') {
95 logger.debug('Could not find key in object storage to proxify private HLS video file.', { err })
96 return res.sendStatus(HttpStatusCode.NOT_FOUND_404)
100 status: HttpStatusCode.INTERNAL_SERVER_ERROR_500,
101 message: err.message,