#!/usr/bin/env python3
"""Cari sinyal 'GA4 sudah siap' di halaman studiosoft.id + uji apakah event kustom lolos setelah siap."""
import asyncio, json, sys, time, urllib.request, websockets

PORT = int(sys.argv[2]) if len(sys.argv) > 2 else 9481
TARGET = sys.argv[1] if len(sys.argv) > 1 else "https://studiosoft.id/"
NAMA = "hermes_siap_%d" % int(time.time())
hits = []


def tab():
    req = urllib.request.Request(f"http://127.0.0.1:{PORT}/json/new?about:blank", method="PUT")
    for _ in range(15):
        try:
            return json.load(urllib.request.urlopen(req, timeout=3))
        except Exception:
            time.sleep(0.5)
    raise RuntimeError("tab gagal")


async def main():
    t = tab()
    async with websockets.connect(t["webSocketDebuggerUrl"], max_size=2**24) as ws:
        pending, mid = {}, 0

        async def reader():
            async for raw in ws:
                m = json.loads(raw)
                if "id" in m and m["id"] in pending:
                    pending[m["id"]].set_result(m)
                elif m.get("method") == "Network.requestWillBeSent":
                    u = m["params"]["request"]["url"]
                    if "/collect" in u or "analytics" in u:
                        hits.append(u)

        rt = asyncio.create_task(reader())

        async def cmd(method, params=None):
            nonlocal mid
            mid += 1
            f = asyncio.get_event_loop().create_future()
            pending[mid] = f
            await ws.send(json.dumps({"id": mid, "method": method, "params": params or {}}))
            return await asyncio.wait_for(f, 30)

        async def ev(x):
            r = await cmd("Runtime.evaluate", {"expression": x, "returnByValue": True})
            return r.get("result", {}).get("result", {}).get("value")

        await cmd("Page.enable"); await cmd("Network.enable"); await cmd("Runtime.enable")
        await cmd("Page.navigate", {"url": TARGET})
        await asyncio.sleep(8)

        print("sebelum interaksi :", await ev("JSON.stringify({gtm: typeof window.google_tag_manager, keys:(window.google_tag_manager?Object.keys(window.google_tag_manager):[]), delayed: document.querySelectorAll('script[type=rocketlazyloadscript]').length, dl:(window.dataLayer||[]).length})"))

        # interaksi nyata supaya WP Rocket melepas skrip yang ditunda
        for jenis, extra in [("mouseMoved", {}), ("mouseWheel", {"deltaY": 600}), ("mousePressed", {"button": "left", "clickCount": 1}), ("mouseReleased", {"button": "left", "clickCount": 1})]:
            p = {"x": 400, "y": 300}; p.update(extra)
            await cmd("Input.dispatchMouseEvent", {"type": jenis, **p})
            await asyncio.sleep(1)
        await asyncio.sleep(8)

        print("setelah interaksi :", await ev("JSON.stringify({gtm: typeof window.google_tag_manager, keys:(window.google_tag_manager?Object.keys(window.google_tag_manager):[]), delayed: document.querySelectorAll('script[type=rocketlazyloadscript]').length, dl:(window.dataLayer||[]).length})"))

        hits.clear()
        print(f"kirim event kustom: {NAMA}")
        await ev(f"window.gtag('event','{NAMA}',{{sumber:'uji-siap'}}); 'ok'")
        await asyncio.sleep(10)
        print(f"beacon setelah GA4 siap: {len(hits)}")
        for h in hits[:6]:
            en = h.split("en=")[1].split("&")[0] if "en=" in h else "(page_view)"
            print(f"   {en}  ←  {h[:110]}")
        rt.cancel()
    print("NAMA=" + NAMA)


try:
    asyncio.run(main())
except Exception as e:
    print("ERR", type(e).__name__, str(e)[:200])