mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/imputnet/cobalt.git synced 2025-07-18 03:08:30 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: camelcase for abbreviations, makeInternalSegments

Browse Source
This commit is contained in:
mikhail 2024-05-28 19:50:04 +05:00
parent 70b63495f2
commit d40b1d1e48
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/modules/stream/internal.js
View File

@ -75,36 +75,36 @@ async function handleYoutubeStream(streamInfo, res) {
}
}
function transformHLSMediaPlaylist(streamInfo, hlsPlaylist) {
function generateInternalStreamsOfSegments(segment) {
function transformHlsMediaPlaylist(streamInfo, hlsPlaylist) {
const makeInternalSegments = (segment) => {
const fullUri = new URL(segment.uri, streamInfo.url).toString();
segment.uri = createInternalStream(fullUri, streamInfo);
return segment;
}
hlsPlaylist.segments =
hlsPlaylist.segments.map(generateInternalStreamsOfSegments);
hlsPlaylist.prefetchSegments =
hlsPlaylist.prefetchSegments.map(generateInternalStreamsOfSegments);
hlsPlaylist.segments = hlsPlaylist.segments.map(makeInternalSegments);
hlsPlaylist.prefetchSegments = hlsPlaylist.prefetchSegments.map(makeInternalSegments);
return hlsPlaylist;
}
async function handleHLSPlaylist(streamInfo, req, res) {
async function handleHlsPlaylist(streamInfo, req, res) {
let hlsPlaylist = await req.body.text();
hlsPlaylist = HLS.parse(hlsPlaylist);
// NOTE no processing module is passing the master playlist
assert(!hlsPlaylist.isMasterPlaylist);
hlsPlaylist = transformHLSMediaPlaylist(streamInfo, hlsPlaylist);
hlsPlaylist = transformHlsMediaPlaylist(streamInfo, hlsPlaylist);
hlsPlaylist = HLS.stringify(hlsPlaylist);
res.write(hlsPlaylist);
res.end();
}
const HLS_MIME_TYPES = ["application/vnd.apple.mpegurl", "audio/mpegurl", "application/x-mpegURL"];
export async function internalStream(streamInfo, res) {
if (streamInfo.service === 'youtube') {
return handleYoutubeStream(streamInfo, res);
@ -129,8 +129,8 @@ export async function internalStream(streamInfo, res) {
if (req.statusCode < 200 || req.statusCode > 299)
return res.end();
if (["application/vnd.apple.mpegurl", "audio/mpegurl"].includes(req.headers['content-type'])) {
await handleHLSPlaylist(streamInfo, req, res);
if (HLS_MIME_TYPES.includes(req.headers['content-type'])) {
await handleHlsPlaylist(streamInfo, req, res);
} else {
req.body.pipe(res);
req.body.on('error', () => res.end());