89 lines
3.1 KiB
TypeScript
89 lines
3.1 KiB
TypeScript
import * as cheerio from 'cheerio';
|
|
import slugify from 'slugify';
|
|
import { BlockedError, NotFoundError, TooManyRequestsError } from '../error';
|
|
import { Context, Format, Meta, NonEmptyArray, UrlResult } from '../types';
|
|
import { Fetcher, guessHeightFromPlaylist } from '../utils';
|
|
import { Extractor } from './Extractor';
|
|
|
|
export class VidSrc extends Extractor {
|
|
public readonly id = 'vidsrc';
|
|
|
|
public readonly label = 'VidSrc';
|
|
|
|
public override readonly ttl: number = 10800000; // 3h
|
|
|
|
private readonly tlds: NonEmptyArray<string>;
|
|
|
|
public constructor(fetcher: Fetcher, tlds: NonEmptyArray<string>) {
|
|
super(fetcher);
|
|
|
|
this.tlds = tlds;
|
|
}
|
|
|
|
public supports(_ctx: Context, url: URL): boolean {
|
|
return null !== url.host.match(/vidsrc/);
|
|
}
|
|
|
|
protected async extractInternal(ctx: Context, url: URL, meta: Meta): Promise<UrlResult[]> {
|
|
return this.extractUsingRandomTld(ctx, url, meta, [...this.tlds]);
|
|
};
|
|
|
|
private async extractUsingRandomTld(ctx: Context, url: URL, meta: Meta, tlds: string[]): Promise<UrlResult[]> {
|
|
const tldIndex = Math.floor(Math.random() * tlds.length);
|
|
const [tld] = tlds.splice(tldIndex, 1) as [string];
|
|
|
|
const newUrl = new URL(url);
|
|
const hostnameParts = newUrl.hostname.split('.');
|
|
hostnameParts[hostnameParts.length - 1] = tld;
|
|
newUrl.hostname = hostnameParts.join('.');
|
|
|
|
let html: string;
|
|
try {
|
|
html = await this.fetcher.text(ctx, newUrl, { queueLimit: 1 });
|
|
} catch (error) {
|
|
if (tlds.length && (error instanceof TooManyRequestsError || error instanceof BlockedError)) {
|
|
return this.extractUsingRandomTld(ctx, url, meta, tlds);
|
|
}
|
|
|
|
throw error;
|
|
}
|
|
|
|
const $ = cheerio.load(html);
|
|
|
|
const iframeUrl = new URL(($('#player_iframe').attr('src') as string).replace(/^\/\//, 'https://'));
|
|
const title = $('title').text().trim();
|
|
|
|
return Promise.all(
|
|
$('.server')
|
|
.map((_i, el) => ({ serverName: $(el).text(), dataHash: $(el).data('hash') }))
|
|
.toArray()
|
|
.filter(({ serverName }) => serverName === 'CloudStream Pro')
|
|
.map(async ({ serverName, dataHash }) => {
|
|
const rcpUrl = new URL(`/rcp/${dataHash}`, iframeUrl.origin);
|
|
const iframeHtml = await this.fetcher.text(ctx, rcpUrl, { headers: { Referer: newUrl.origin } });
|
|
const srcMatch = iframeHtml.match(`src:\\s?'(.*)'`) as string[];
|
|
|
|
const playerHtml = await this.fetcher.text(ctx, new URL(srcMatch[1] as string, iframeUrl.origin), { headers: { Referer: rcpUrl.href } });
|
|
const fileMatch = playerHtml.match(`file:\\s?'(.*)'`);
|
|
if (!fileMatch) {
|
|
throw new NotFoundError();
|
|
}
|
|
|
|
const m3u8Url = new URL(fileMatch[1] as string);
|
|
|
|
return {
|
|
url: m3u8Url,
|
|
format: Format.hls,
|
|
label: serverName,
|
|
sourceId: `${this.id}_${slugify(serverName)}_${meta.countryCodes?.join('_')}`,
|
|
ttl: this.ttl,
|
|
meta: {
|
|
...meta,
|
|
height: await guessHeightFromPlaylist(ctx, this.fetcher, m3u8Url, { headers: { Referer: iframeUrl.href } }),
|
|
title,
|
|
},
|
|
};
|
|
}),
|
|
);
|
|
}
|
|
}
|