import axios from 'axios';
import { VideoExtractor, type IVideo } from '../models';

class VizCloud extends VideoExtractor {
  protected override serverName = 'VizCloud';
  protected override sources: IVideo[] = [];

  private readonly host = 'https://vidstream.pro';
  private keys: {
    cipher: string;
    encrypt: string;
    main: string;
    operations: Map<string, string>;
    post: string[];
    pre: string[];
  } = {
    cipher: '',
    encrypt: '',
    main: '',
    operations: new Map<string, string>(),
    pre: [],
    post: [],
  };

  override extract = async (videoUrl: URL, vizCloudHelper: string, apiKey: string): Promise<IVideo[]> => {
    const vizID: Array<string> = videoUrl.href.split('/');
    let url;
    if (!vizID.length) {
      throw new Error('Video not found');
    } else {
      url = `${vizCloudHelper}/vizcloud?query=${encodeURIComponent(vizID.pop() ?? '')}&apikey=${apiKey}`;
    }

    const { data } = await axios.get(url);
    if (!data.data?.media) throw new Error('Video not found');

    this.sources = [
      ...this.sources,
      ...data.data.media.sources.map((source: any) => ({
        url: source.file,
        quality: 'auto',
        isM3U8: source.file?.includes('.m3u8'),
      })),
    ];

    const main = this.sources[this.sources.length - 1]!.url;
    const req = await axios({
      method: 'get',
      url: main,
      headers: { referer: 'https://9anime.to' },
    });
    const resolutions = req.data.match(/(RESOLUTION=)(.*)(\s*?)(\s*.*)/g);
    resolutions?.forEach((res: string) => {
      const index = main.lastIndexOf('/');
      const quality = res.split('\n')[0]?.split('x')[1]!.split(',')[0];
      const url = main.slice(0, index);
      this.sources.push({
        url: url + '/' + res.split('\n')[1],
        isM3U8: (url + res.split('\n')[1]).includes('.m3u8'),
        quality: quality + 'p',
      });
    });
    return this.sources;
  };
}

export default VizCloud;