Mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/wukko/cobalt.git synced 2024-11-14 04:10:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

stream: extract headers to shared file

Browse source
This commit is contained in:
dumbmoron 2024-04-27 10:59:27 +00:00
parent ec746f57a7
commit 49eaa7d4ed
No known key found for this signature in database
2 changed files with 23 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
src/modules/stream/shared.js Normal file
View file

@ -0,0 +1,21 @@
import { genericUserAgent } from "../config.js";
const defaultHeaders = {
'user-agent': genericUserAgent
}
const serviceHeaders = {
bilibili: {
referer: 'https://www.bilibili.com/'
},
youtube: {
accept: '*/*',
origin: 'https://www.youtube.com',
referer: 'https://www.youtube.com',
DNT: '?1'
}
}
export function getHeaders(service) {
return { ...defaultHeaders, ...serviceHeaders[service] }
}

21
src/modules/stream/types.js
View file

@ -5,26 +5,9 @@ import { create as contentDisposition } from "content-disposition-header";
import { metadataManager } from "../sub/utils.js";
import { destroyInternalStream } from "./manage.js";
import { ffmpegArgs, genericUserAgent } from "../config.js";
import { ffmpegArgs } from "../config.js";
import { getHeaders } from "./shared.js";
const defaultHeaders = {
'user-agent': genericUserAgent
}
const serviceHeaders = {
bilibili: {
referer: 'https://www.bilibili.com/'
},
youtube: {
accept: '*/*',
origin: 'https://www.youtube.com',
referer: 'https://www.youtube.com',
DNT: '?1'
}
}
function getHeaders(service) {
return { ...defaultHeaders, ...serviceHeaders[service] }
}
function toRawHeaders(headers) {
return Object.entries(headers)
.map(([key, value]) => `${key}: ${value}\r\n`)