mirror of https://github.com/Chocobozzz/PeerTube
98 lines
2.9 KiB
TypeScript
98 lines
2.9 KiB
TypeScript
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)
|
|
}
|
|
}
|