Update valid.ts

This commit is contained in:
Jorrin
2024-04-01 00:13:04 +02:00
parent 84105c3ac2
commit 77540229a0

View File

@@ -22,26 +22,26 @@ export async function validatePlayableStream(
stream: Stream, stream: Stream,
ops: ProviderRunnerOptions | IndividualEmbedRunnerOptions, ops: ProviderRunnerOptions | IndividualEmbedRunnerOptions,
): Promise<Stream | null> { ): Promise<Stream | null> {
const fetcher = stream.flags.length === 1 && stream.flags.includes('cors-allowed') ? ops.fetcher : ops.proxiedFetcher;
if (stream.type === 'hls') { if (stream.type === 'hls') {
const headResult = await fetcher.full(stream.playlist, { const result = await ops.proxiedFetcher.full(stream.playlist, {
method: 'HEAD', method: 'GET',
headers: { headers: {
...stream.preferredHeaders, ...stream.preferredHeaders,
...stream.headers, ...stream.headers,
}, },
}); });
if (headResult.statusCode !== 200) return null; if (result.statusCode !== 200) return null;
return stream; return stream;
} }
if (stream.type === 'file') { if (stream.type === 'file') {
const validQualitiesResults = await Promise.all( const validQualitiesResults = await Promise.all(
Object.values(stream.qualities).map((quality) => Object.values(stream.qualities).map((quality) =>
fetcher.full(quality.url, { ops.proxiedFetcher.full(quality.url, {
method: 'HEAD', method: 'GET',
headers: { headers: {
...stream.preferredHeaders, ...stream.preferredHeaders,
...stream.headers, ...stream.headers,
Range: 'bytes=0-1',
}, },
}), }),
), ),
@@ -49,7 +49,7 @@ export async function validatePlayableStream(
// remove invalid qualities from the stream // remove invalid qualities from the stream
const validQualities = stream.qualities; const validQualities = stream.qualities;
Object.keys(stream.qualities).forEach((quality, index) => { Object.keys(stream.qualities).forEach((quality, index) => {
if (validQualitiesResults[index].statusCode !== 200) { if (validQualitiesResults[index].statusCode < 200 || validQualitiesResults[index].statusCode >= 400) {
delete validQualities[quality as keyof typeof stream.qualities]; delete validQualities[quality as keyof typeof stream.qualities];
} }
}); });