import {Stream, ProviderContext} from '../types';

export const tokyoGetStream = async function ({
  link,
  providerContext,
}: {
  link: string;
  providerContext: ProviderContext;
}): Promise<Stream[]> {
  try {
    const {cheerio} = providerContext;
    const url = link;
    const res = await fetch(url);
    const data = await res.text();
    const $ = cheerio.load(data);
    const streamLinks: Stream[] = [];
    $('.c_h1,.c_h2').map((i, element) => {
      $(element).find('span').remove();
      const title = $(element).find('a').text() || '';
      const link = $(element).find('a').attr('href') || '';
      if (title && link.includes('media')) {
        streamLinks.push({
          server: title,
          link,
          type: link.split('.').pop() || 'mkv',
        });
      }
    });
    return streamLinks;
  } catch (err) {
    return [];
  }
};