116 lines
3.7 KiB
TypeScript
116 lines
3.7 KiB
TypeScript
import * as cheerio from 'cheerio';
|
|
import { ContentType } from 'stremio-addon-sdk';
|
|
import { Context, CountryCode } from '../types';
|
|
import { Fetcher, getTmdbId, getTmdbNameAndYear, Id, TmdbId } from '../utils';
|
|
import { Source, SourceResult } from './types';
|
|
|
|
export class Cuevana implements Source {
|
|
public readonly id = 'cuevana';
|
|
|
|
public readonly label = 'Cuevana';
|
|
|
|
public readonly contentTypes: ContentType[] = ['movie', 'series'];
|
|
|
|
public readonly countryCodes: CountryCode[] = [CountryCode.es, CountryCode.mx];
|
|
|
|
public readonly baseUrl = 'https://ww1.cuevana3.is';
|
|
|
|
private readonly fetcher: Fetcher;
|
|
|
|
public constructor(fetcher: Fetcher) {
|
|
this.fetcher = fetcher;
|
|
}
|
|
|
|
public async handle(ctx: Context, _type: string, id: Id): Promise<SourceResult[]> {
|
|
const tmdbId = await getTmdbId(ctx, this.fetcher, id);
|
|
|
|
const [name] = await getTmdbNameAndYear(ctx, this.fetcher, tmdbId, 'es');
|
|
|
|
const pageUrl = await this.fetchPageUrl(ctx, name);
|
|
if (!pageUrl) {
|
|
return [];
|
|
}
|
|
|
|
let title: string = name;
|
|
let html: string;
|
|
|
|
if (tmdbId.season) {
|
|
title += ` ${tmdbId.season}x${tmdbId.episode}`;
|
|
|
|
const episodeUrl = await this.fetchEpisodeUrl(ctx, pageUrl, tmdbId);
|
|
if (!episodeUrl) {
|
|
return [];
|
|
}
|
|
|
|
html = await this.fetcher.text(ctx, episodeUrl);
|
|
} else {
|
|
html = await this.fetcher.text(ctx, pageUrl);
|
|
}
|
|
|
|
const $ = cheerio.load(html);
|
|
|
|
const urlResults = $('.open_submenu')
|
|
.map((_i, el) => {
|
|
const elText = $(el).text();
|
|
if (!elText.includes('Español')) {
|
|
return [];
|
|
}
|
|
|
|
const isLatino = elText.includes('Latino');
|
|
|
|
if (isLatino && CountryCode.mx in ctx.config) {
|
|
return $('[data-tr], [data-video]', el)
|
|
.map((_i, el) => ({ countryCode: CountryCode.mx, title, url: new URL($(el).attr('data-tr') ?? $(el).attr('data-video') as string) }))
|
|
.toArray();
|
|
} else if (!isLatino && CountryCode.es in ctx.config) {
|
|
return $('[data-tr], [data-video]', el)
|
|
.map((_i, el) => ({ countryCode: CountryCode.es, title, url: new URL($(el).attr('data-tr') ?? $(el).attr('data-video') as string) }))
|
|
.toArray();
|
|
} else {
|
|
return [];
|
|
}
|
|
})
|
|
.toArray();
|
|
|
|
return Promise.all(
|
|
urlResults.map(async (urlResult) => {
|
|
if (!urlResult.url.host.includes('cuevana3')) {
|
|
return urlResult;
|
|
}
|
|
|
|
const html = await this.fetcher.text(ctx, urlResult.url, { headers: { Referer: pageUrl.origin } });
|
|
|
|
const urlMatcher = html.match(/url ?= ?'(.*)'/) as string[];
|
|
|
|
return { ...urlResult, url: new URL(urlMatcher[1] as string) };
|
|
}),
|
|
);
|
|
};
|
|
|
|
private async fetchPageUrl(ctx: Context, keyword: string): Promise<URL | undefined> {
|
|
const searchUrl = new URL(`/search/${encodeURIComponent(keyword)}/`, this.baseUrl);
|
|
const html = await this.fetcher.text(ctx, searchUrl, { headers: { Referer: searchUrl.origin } });
|
|
|
|
const $ = cheerio.load(html);
|
|
|
|
const urlPath = $('.TPost .Title')
|
|
.filter((_i, el) => $(el).text().trim() === keyword)
|
|
.closest('a')
|
|
.attr('href');
|
|
|
|
return urlPath !== undefined ? new URL(urlPath, searchUrl.origin) : urlPath;
|
|
};
|
|
|
|
private async fetchEpisodeUrl(ctx: Context, pageUrl: URL, tmdbId: TmdbId): Promise<URL | undefined> {
|
|
const html = await this.fetcher.text(ctx, pageUrl, { headers: { Referer: pageUrl.origin } });
|
|
|
|
const $ = cheerio.load(html);
|
|
|
|
const urlPath = $('.TPost .Year')
|
|
.filter((_i, el) => $(el).text().trim() === `${tmdbId.season}x${tmdbId.episode}`)
|
|
.closest('a')
|
|
.attr('href');
|
|
|
|
return urlPath !== undefined ? new URL(urlPath, pageUrl.origin) : urlPath;
|
|
}
|
|
}
|