from playwright.sync_api import sync_playwright
import os

OUT = os.path.abspath("shots")
os.makedirs(OUT, exist_ok=True)

with sync_playwright() as p:
    browser = p.chromium.launch(headless=True)

    # desktop
    ctx = browser.new_context(viewport={"width": 1440, "height": 820}, device_scale_factor=2)
    page = ctx.new_page()
    page.goto("http://localhost:4321/", wait_until="networkidle", timeout=60000)
    page.wait_for_timeout(2600)  # entrance laten settelen
    page.screenshot(path=os.path.join(OUT, "hero-desktop.png"))
    print("OK hero-desktop.png")

    # mobile
    m = browser.new_context(viewport={"width": 390, "height": 844}, device_scale_factor=2)
    mp = m.new_page()
    mp.goto("http://localhost:4321/", wait_until="networkidle", timeout=60000)
    mp.wait_for_timeout(2600)
    mp.screenshot(path=os.path.join(OUT, "hero-mobile.png"), full_page=False)
    print("OK hero-mobile.png")

    browser.close()