import { VideoExtractor, type IVideo, type ExtractorContext } from '../models';
import { defaultExtractorContext } from '../utils/extension-utils';

export class VidMoly extends VideoExtractor {
  protected override serverName = 'vidmoly';
  protected override sources: IVideo[] = [];

  /** @param ctx extractor context; requests go through its axios (default: the library's) */
  constructor(private readonly ctx: ExtractorContext = defaultExtractorContext) {
    super();
  }

  override extract = async (videoUrl: URL): Promise<IVideo[]> => {
    try {
      const { data } = await this.ctx.axios.get(videoUrl.href);

      const links = data.match(/file:\s*"([^"]+)"/);

      const m3u8Content = await this.ctx.axios.get(links[1], {
        headers: {
          Referer: videoUrl.href,
        },
      });

      this.sources.push({
        quality: 'auto',
        url: links[1],
        isM3U8: links[1].includes('.m3u8'),
      });

      if (m3u8Content.data.includes('EXTM3U')) {
        const videoList = m3u8Content.data.split('#EXT-X-STREAM-INF:');
        for (const video of videoList ?? []) {
          if (!video.includes('m3u8')) continue;

          const url = video.split('\n')[1];
          const quality = video.split('RESOLUTION=')[1].split(',')[0].split('x')[1];

          this.sources.push({
            url: url,
            quality: `${quality}`,
            isM3U8: url.includes('.m3u8'),
          });
        }
      }

      return this.sources;
    } catch (err) {
      throw new Error((err as Error).message);
    }
  };
}