import json
import sys
import urllib.parse
import urllib.request
import os

FLICKR_API_KEY = os.getenv("FLICKR_API_KEY", "YOUR_API_KEY_HERE")
FLICKR_USER_ID = os.getenv("FLICKR_USER_ID", "YOUR_USER_ID_HERE")  # e.g., "12345678@N00"

ENDPOINT = "https://api.flickr.com/services/rest/"

def flickr_get(method: str, **params):
    q = {
        "method": method,
        "api_key": FLICKR_API_KEY,
        "format": "json",
        "nojsoncallback": "1",
        **params,
    }
    url = ENDPOINT + "?" + urllib.parse.urlencode(q)
    with urllib.request.urlopen(url, timeout=10) as resp:
        data = json.load(resp)
    return data

def require_ok(resp, what):
    if resp.get("stat") != "ok":
        code = resp.get("code")
        msg = resp.get("message")
        raise SystemExit(f"{what} failed (code {code}): {msg}")

def main():
    if not FLICKR_API_KEY or "YOUR_API_KEY_HERE" in FLICKR_API_KEY:
        raise SystemExit("Set FLICKR_API_KEY (env var) or put your key in the script.")

    # 1) Basic key test
    echo = flickr_get("flickr.test.echo", test="ping")
    require_ok(echo, "flickr.test.echo")
    print("✓ API key works (flickr.test.echo)")

    # 2) Check user_id is valid/visible
    if not FLICKR_USER_ID or "YOUR_USER_ID_HERE" in FLICKR_USER_ID:
        print("! Skipping user check: set FLICKR_USER_ID to test your user as well.")
    else:
        info = flickr_get("flickr.people.getInfo", user_id=FLICKR_USER_ID)
        require_ok(info, "flickr.people.getInfo")
        username = info["person"]["username"]["_content"]
        nsid = info["person"]["nsid"]
        print(f"✓ User OK: username='{username}', nsid='{nsid}'")

        # 3) Pull a single public photo (if any)
        photos = flickr_get("flickr.people.getPublicPhotos", user_id=FLICKR_USER_ID, per_page=1, page=1)
        require_ok(photos, "flickr.people.getPublicPhotos")
        total = int(photos["photos"]["total"])
        print(f"✓ Public photos query OK (total={total})")

if __name__ == "__main__":
    main()