Creepso/webstreamr-github
1
0
Fork 0
Code Issues 27 Pull requests 7 Projects Releases 240 Packages Wiki Activity Actions

refactor: introduce guessSizeFromMp4

Browse source
This commit is contained in:
WebStreamr 2025-08-28 19:01:14 +00:00
parent cefc80fa55
commit 4a1a3b19aa
No known key found for this signature in database
2 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/extractor/DoodStream.ts
View file

@ -2,6 +2,7 @@ import * as cheerio from 'cheerio';
import randomstring from 'randomstring';
import { NotFoundError } from '../error';
import { Context, CountryCode, Format, UrlResult } from '../types';
import { guessSizeFromMp4 } from '../utils/size';
import { Extractor } from './Extractor';
export class DoodStream extends Extractor {
@ -41,10 +42,7 @@ export class DoodStream extends Extractor {
mp4Url = new URL(baseUrl);
} else {
mp4Url = new URL(`${baseUrl}${randomstring.generate(10)}?token=${token}&expiry=${Date.now()}`);
const mp4Head = await this.fetcher.head(ctx, mp4Url, { headers: { Referer: url.origin } });
if (mp4Head['content-length']) {
bytes = parseInt(mp4Head['content-length'] as string);
}
bytes = await guessSizeFromMp4(ctx, this.fetcher, mp4Url, { headers: { Referer: url.origin } });
}
return [

12
src/utils/size.ts Normal file
View file

@ -0,0 +1,12 @@
import { Context } from '../types';
import { CustomRequestInit, Fetcher } from './Fetcher';
export const guessSizeFromMp4 = async (ctx: Context, fetcher: Fetcher, url: URL, init?: CustomRequestInit): Promise<number | undefined> => {
const mp4Head = await fetcher.head(ctx, url, init);
if (mp4Head['content-length']) {
return parseInt(mp4Head['content-length'] as string);
}
return undefined;
};