yetangitu/Peertube
1
0
Fork 0
mirror of https://github.com/Chocobozzz/PeerTube.git synced 2025-10-04 10:19:35 +02:00
Code Issues Releases Wiki Activity

Support reinjecting token in private m3u8 playlist

Browse source
This commit is contained in:
Chocobozzz 2022-12-02 14:47:21 +01:00
parent 04509c4325
commit 71e3e879c0
No known key found for this signature in database
GPG key ID: 583A612D890159BE
22 changed files with 391 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

20
server/controllers/object-storage-proxy.ts
View file

@ -1,7 +1,10 @@
import cors from 'cors'
import express from 'express'
import { PassThrough, pipeline } from 'stream'
import { logger } from '@server/helpers/logger'
import { StreamReplacer } from '@server/helpers/stream-replacer'
import { OBJECT_STORAGE_PROXY_PATHS } from '@server/initializers/constants'
import { injectQueryToPlaylistUrls } from '@server/lib/hls'
import { getHLSFileReadStream, getWebTorrentFileReadStream } from '@server/lib/object-storage'
import {
asyncMiddleware,
@ -11,6 +14,7 @@ import {
optionalAuthenticate
} from '@server/middlewares'
import { HttpStatusCode } from '@shared/models'
import { buildReinjectVideoFileTokenQuery, doReinjectVideoFileToken } from './shared/m3u8-playlist'
const objectStorageProxyRouter = express.Router()
@ -67,7 +71,20 @@ async function proxifyHLS (req: express.Request, res: express.Response) {
rangeHeader: req.header('range')
})
return stream.pipe(res)
const streamReplacer = filename.endsWith('.m3u8') && doReinjectVideoFileToken(req)
? new StreamReplacer(line => injectQueryToPlaylistUrls(line, buildReinjectVideoFileTokenQuery(req)))
: new PassThrough()
return pipeline(
stream,
streamReplacer,
res,
err => {
if (!err) return
handleObjectStorageFailure(res, err)
}
)
} catch (err) {
return handleObjectStorageFailure(res, err)
}
@ -75,6 +92,7 @@ async function proxifyHLS (req: express.Request, res: express.Response) {
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)
}