"""Screenshot the System Settings > Floor Map tab of the dev UI.

Logs in via the API (same as tools/shot.py), seeds the token into
localStorage, opens /settings/system, clicks the Floor Map tab, and shots it.
"""

import os
import json
import urllib.request

from playwright.sync_api import sync_playwright

UI = "http://localhost:5173"
API = "http://localhost:5001/api"
# creds from env so no real login lands in source; fall back to dev defaults
USERNAME = os.environ.get("SHOT_USERNAME", "270015376")
PASSWORD = os.environ.get("SHOT_PASSWORD", "changeme")
OUT = "/tmp/shopdb-shots"
os.makedirs(OUT, exist_ok=True)


def login():
    body = json.dumps({"username": USERNAME, "password": PASSWORD}).encode()
    request = urllib.request.Request(f"{API}/auth/login", data=body,
                                     headers={"Content-Type": "application/json"})
    data = json.load(urllib.request.urlopen(request))["data"]
    return data["access_token"], data["refresh_token"], data["user"]


def main():
    token, refresh, user = login()
    seed = f"""
      localStorage.setItem('token', {json.dumps(token)});
      localStorage.setItem('refreshToken', {json.dumps(refresh)});
      localStorage.setItem('user', {json.dumps(json.dumps(user))});
    """
    with sync_playwright() as p:
        browser = p.chromium.launch()
        context = browser.new_context(viewport={"width": 1400, "height": 1100})
        context.add_init_script(seed)
        page = context.new_page()
        page.goto(f"{UI}/settings/system", wait_until="networkidle", timeout=30000)
        page.wait_for_timeout(800)
        # Click the Floor Map tab button.
        page.get_by_text("Floor Map", exact=True).first.click()
        page.wait_for_timeout(600)
        page.screenshot(path=OUT, full_page=True)
        print(OUT)
        browser.close()


if __name__ == "__main__":
    main()
