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

export const getStream = async function ({
  link: url,
  type,
  providerContext,
}: {
  link: string;
  type: string;
  providerContext: ProviderContext;
}): Promise<Stream[]> {
  const { axios, cheerio } = providerContext;
  const headers = {
    "sec-ch-ua":
      '"Not_A Brand";v="8", "Chromium";v="120", "Microsoft Edge";v="120"',
    "sec-ch-ua-mobile": "?0",
    "sec-ch-ua-platform": '"Windows"',
    "Sec-Fetch-Site": "none",
    "Sec-Fetch-User": "?1",
    "Upgrade-Insecure-Requests": "1",
    "User-Agent":
      "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36 Edg/120.0.0.0",
  };

  try {
    if (type === "movie") {
      const linkRes = await axios.get(url, { headers });
      const linkData = linkRes.data;
      const $ = cheerio.load(linkData);
      url = $('strong:contains("INSTANT")').parent().attr("href") || url;
    }

    // fastilinks
    if (url.includes("fastilinks")) {
      const fastilinksRes = await axios.get(url, { headers });
      const fastilinksData = fastilinksRes.data;
      const $$ = cheerio.load(fastilinksData);
      const fastilinksKey = $$(
        'input[name="_csrf_token_645a83a41868941e4692aa31e7235f2"]'
      ).attr("value");
      console.log("fastilinksKey", fastilinksKey);
      const fastilinksFormData = new FormData();
      fastilinksFormData.append(
        "_csrf_token_645a83a41868941e4692aa31e7235f2",
        fastilinksKey || ""
      );
      const fastilinksRes2 = await fetch(url, {
        method: "POST",
        headers: headers,
        body: fastilinksFormData,
      });
      const fastilinksHtml = await fastilinksRes2.text();
      // console.log('fastilinksHtml', fastilinksHtml);
      const $$$ = cheerio.load(fastilinksHtml);
      const fastilinksLink =
        $$$('a:contains("mediafire")').attr("href") ||
        $$$('a:contains("photolinx")').attr("href");
      console.log("fastilinksLink", fastilinksLink);
      url = fastilinksLink || url;
    }
    console.log("world4uGetStream", type, url);

    if (url.includes("photolinx")) {
      console.log("photolinx", url);
      const photolinxBaseUrl = url.split("/").slice(0, 3).join("/");
      console.log("photolinxBaseUrl", photolinxBaseUrl);
      // const photolinxBaseUrl = url.split('/').slice(0, 3).join('/');
      const photolinxRes = await fetch(
        "https://photolinx.space/download/SzbPKzt6YMO",
        {
          headers: {
            accept:
              "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
            "accept-language": "en-US,en;q=0.9,en-IN;q=0.8",
            "cache-control": "no-cache",
            pragma: "no-cache",
            priority: "u=0, i",
            "sec-ch-ua":
              '"Not;A=Brand";v="99", "Microsoft Edge";v="139", "Chromium";v="139"',
            "sec-ch-ua-mobile": "?0",
            "sec-ch-ua-platform": '"Windows"',
            "sec-fetch-dest": "document",
            "sec-fetch-mode": "navigate",
            "sec-fetch-site": "none",
            "sec-fetch-user": "?1",
            "upgrade-insecure-requests": "1",
            cookie:
              "PHPSESSID=f2211def7938d7228daaa37ffeabcfe0; ext_name=ojplmecpdpgccookcobabopnaifgidhf",
          },
          body: null,
          method: "GET",
        }
      );
      const photolinxData = await photolinxRes.text();
      const $$$ = cheerio.load(photolinxData);
      const access_token = $$$("#generate_url").attr("data-token");
      const uid = $$$("#generate_url").attr("data-uid");
      const body = {
        type: "DOWNLOAD_GENERATE",
        payload: {
          access_token,
          uid,
        },
      };
      console.log("photolinxData", JSON.stringify(body));

      const photolinxRes2 = await fetch(`${photolinxBaseUrl}/action`, {
        headers: {
          accept: "application/json, text/plain, */*",
          "accept-language": "en-US,en;q=0.9,en-IN;q=0.8",
          "cache-control": "no-cache",
          "content-type": "application/json; charset=UTF-8",
          pragma: "no-cache",
          priority: "u=1, i",
          "sec-ch-ua":
            '"Not;A=Brand";v="99", "Microsoft Edge";v="139", "Chromium";v="139"',
          "sec-ch-ua-mobile": "?0",
          "sec-ch-ua-platform": '"Windows"',
          "sec-fetch-dest": "empty",
          "sec-fetch-mode": "cors",
          "sec-fetch-site": "same-origin",
          "x-requested-with": "xmlhttprequest",
          cookie:
            "PHPSESSID=f2211def7938d7228daaa37ffeabcfe0; ext_name=ojplmecpdpgccookcobabopnaifgidhf",
          Referer: url,
        },
        body: JSON.stringify(body),
        method: "POST",
      });
      const photolinxData2 = await photolinxRes2.json();
      console.log("photolinxData2", photolinxData2);
      const dwUrl = photolinxData2?.download_url;
      if (dwUrl) {
        const streamLinks = [
          {
            server: "Photolinx",
            link: dwUrl,
            type: "mkv",
          },
        ];
        return streamLinks;
      }
    }

    const res = await axios.get(url, { headers });
    const html = res.data;
    const streamLinks: Stream[] = [];
    let data = { download: "" };
    try {
      const key =
        html.match(/formData\.append\('key',\s*'(\d+)'\);/)?.[1] || "";
      console.log("key", key, "url", url);
      const formData = new FormData();
      formData.append("key", key);
      const streamRes = await fetch(url, {
        method: "POST",
        headers: headers,
        body: formData,
      });
      data = await streamRes.json();
    } catch (err) {
      console.log(
        "error in world4uGetStream",
        err instanceof Error ? err.message : err
      );
    }

    // console.log('streamRes', streamRes);
    let $ = cheerio.load(html);
    // console.log('data', html);
    const mediafireUrl =
      $('h1:contains("Download")').find("a").attr("href") ||
      $(".input.popsok").attr("href");
    console.log("mediafireUrl", mediafireUrl);
    if (mediafireUrl) {
      const directUrl = await fetch(mediafireUrl, {
        headers: {
          Referer: url,
        },
      });
      const urlContentType = directUrl.headers.get("content-type");
      console.log("mfcontentType", urlContentType);
      if (urlContentType && urlContentType.includes("video")) {
        streamLinks.push({
          server: "Mediafire",
          link: mediafireUrl,
          type: "mkv",
        });
        return streamLinks;
      } else {
        const repairRes = await fetch(mediafireUrl, {
          headers: {
            Referer: url,
          },
        });
        const repairHtml = await repairRes.text();

        // Regex to match the window.location.href assignment in the script content
        const base64Link = cheerio
          .load(repairHtml)(".input.popsok")
          .attr("data-scrambled-url");
        console.log("base64Link", base64Link);
        const href = base64Link ? atob(base64Link) : null;
        console.log("href", href);

        // If a match is found, return the URL; otherwise return null
        let downloadLInk = href?.startsWith("https://") ? href : null;
        console.log("downloadLInk", downloadLInk);

        if (downloadLInk) {
          streamLinks.push({
            server: "Mediafire",
            link: downloadLInk,
            type: "mkv",
          });
        }
        return streamLinks;
      }
    }

    const requireRepairRes = await fetch(data.download);
    const contentType = requireRepairRes.headers.get("content-type");
    console.log("contentType", contentType);
    if (contentType && contentType.includes("video")) {
      streamLinks.push({
        server: "Mediafire",
        link: data.download,
        type: "mkv",
      });
      return streamLinks;
    } else {
      const repairRes = await fetch(data.download, {
        headers: {
          Referer: url,
        },
      });
      const repairHtml = await repairRes.text();
      const $ = cheerio.load(repairHtml);
      const repairLink = $("#continue-btn").attr("href");
      console.log("repairLink", "https://www.mediafire.com" + repairLink);
      const repairRequireRepairRes = await fetch(
        "https://www.mediafire.com" + repairLink
      );
      const $$ = cheerio.load(await repairRequireRepairRes.text());
      const repairDownloadLink = $$(".input.popsok").attr("href");
      console.log("repairDownloadLink", repairDownloadLink);
      if (repairDownloadLink) {
        streamLinks.push({
          server: "Mediafire",
          link: repairDownloadLink,
          type: "mkv",
        });
      }
    }

    return streamLinks;
  } catch (err) {
    console.log(err instanceof Error ? err.message : err);
    return [];
  }
};