import asyncio
import httpx
from playwright.async_api import async_playwright, TimeoutError as PlaywrightTimeoutError

SCREENSHOT_PATH = "/tmp/screenshot.png"
TELEGRAM_BOT_TOKEN = "8865512961:AAG4VVJk7S58oAF5Ps00xgwfoPjXobbcm1E"
TELEGRAM_CHAT_ID = "-1004385538803"
TELEGRAM_API_URL = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendPhoto"
TARGET_URL = "https://www.google.com"


async def capture_screenshot() -> bool:
    try:
        async with async_playwright() as p:
            browser = await p.chromium.launch(headless=True)
            page = await browser.new_page()
            await page.goto(TARGET_URL, wait_until="networkidle", timeout=30000)
            await page.screenshot(path=SCREENSHOT_PATH, full_page=True)
            await browser.close()
            print(f"[+] Screenshot saved to {SCREENSHOT_PATH}")
            return True
    except PlaywrightTimeoutError:
        print(f"[-] Timeout loading {TARGET_URL}")
        return False
    except Exception as e:
        print(f"[-] Error: {e}")
        return False


def send_screenshot() -> bool:
    try:
        with open(SCREENSHOT_PATH, "rb") as photo:
            with httpx.Client(timeout=30.0) as client:
                resp = client.post(
                    TELEGRAM_API_URL,
                    data={"chat_id": TELEGRAM_CHAT_ID, "caption": f"Screenshot of {TARGET_URL}"},
                    files={"photo": ("screenshot.png", photo, "image/png")},
                )
        if resp.status_code == 200 and resp.json().get("ok"):
            print("[+] Screenshot sent to Telegram!")
            return True
        else:
            print(f"[-] Telegram error: {resp.text}")
            return False
    except Exception as e:
        print(f"[-] Send failed: {e}")
        return False


async def main():
    if await capture_screenshot():
        send_screenshot()


if __name__ == "__main__":
    asyncio.run(main())