import time
import cloudscraper
from bs4 import BeautifulSoup
from urllib.parse import urlparse

def bypass_shrinkme(url: str) -> str | None:
    """
    Mencoba bypass link shrinkme.click / shrinkme.io style.
    """
    # Bersihkan trailing slash
    url = url.rstrip("/")
    code = url.split("/")[-1]

    # Beberapa domain mirror yang sering dipakai
    domains = [
        "https://en.shrinke.me",
        "https://shrinkme.click",
        "https://shrinkme.io",
    ]

    scraper = cloudscraper.create_scraper(
        browser={"browser": "chrome", "platform": "windows", "mobile": False},
        allow_brotli=False
    )

    headers_base = {
        "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",
        "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8",
        "Accept-Language": "en-US,en;q=0.5",
    }

    for domain in domains:
        try:
            final_url = f"{domain}/{code}"
            print(f"[*] Mencoba: {final_url}")

            # Referer yang sering dipakai agar lolos
            headers = headers_base.copy()
            headers["Referer"] = "https://mrproblogger.com/"

            resp = scraper.get(final_url, headers=headers, timeout=20)
            if resp.status_code != 200:
                print(f"    Status {resp.status_code}, skip...")
                continue

            soup = BeautifulSoup(resp.text, "html.parser")
            inputs = soup.find_all("input")
            if not inputs:
                print("    Tidak ketemu form input")
                continue

            data = {inp.get("name"): inp.get("value") for inp in inputs if inp.get("name")}

            # Tunggu seperti timer asli (biasanya 10-15 detik)
            print("    Menunggu timer...")
            time.sleep(12)

            # POST ke endpoint go
            post_headers = headers.copy()
            post_headers["X-Requested-With"] = "XMLHttpRequest"
            post_headers["Content-Type"] = "application/x-www-form-urlencoded"

            go_url = f"{domain}/links/go"
            r = scraper.post(go_url, data=data, headers=post_headers, timeout=20)

            try:
                result = r.json()
                if "url" in result and result["url"]:
                    return result["url"]
            except Exception:
                # Kadang langsung redirect atau text
                if "http" in r.text:
                    return r.text.strip()

        except Exception as e:
            print(f"    Error: {e}")
            continue

    return None


if __name__ == "__main__":
    target = "https://shrinkme.click/blckrose"
    print(f"Target: {target}\n")

    final = bypass_shrinkme(target)
    if final:
        print(f"\n✅ Berhasil!\nFinal URL: {final}")
    else:
        print("\n❌ Gagal. Coba ganti domain/referer atau gunakan userscript di browser.")