#!/usr/bin/env python3
import argparse
import fcntl
import os
import sys
from datetime import datetime
from pathlib import Path

from PIL import Image

sys.path.append(str(Path(__file__).parent / "lib"))
from crop import face_aware_crop
from homeassistant import HomeAssistantClient
from immich import ImmichClient, get_random_photo_from_album, get_random_photo_of_people
from overlay import format_age, format_location

# waveshare_epd is imported lazily after the lock — its epdconfig claims
# GPIO pins at import time, so two overlapping invocations would both crash
# on "GPIO busy" before reaching the flock below.

IMMICH_URL = os.environ.get("IMMICH_URL", "https://immich.schmelczer.dev")
IMMICH_API_KEY = os.environ.get("IMMICH_API_KEY", "6crxVS1JLTJxsfGlzVhN2kefdL4EP7HPkkoMk9L6ZOE")

HA_URL = os.environ.get("HA_URL", "https://homeassistant.schmelczer.dev")
HA_TOKEN = os.environ.get(
    "HA_TOKEN",
    "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJmZjk3OTNmOWMzOWU0YjdmYmRjYTc5YmJkMTUyODcyNSIsImlhdCI6MTc2OTIwMjg1NCwiZXhwIjoyMDg0NTYyODU0fQ.IiL_1vTrGMlOoPMksN6lAopE0aInlY_wRnL4Jc-CeBs",
)
HA_PRESENCE = {"Andras": "person.andras", "Ruby": "person.ruby"}


def main() -> None:
    parser = argparse.ArgumentParser(description="Display image on e-ink frame")
    parser.add_argument(
        "--people", default="Me,Ruby", help="Comma-separated names for Immich search"
    )
    parser.add_argument("--album", help="Fetch from album (overrides --people)")
    parser.add_argument(
        "-o",
        "--orientation",
        type=int,
        choices=[0, 90, 180, 270],
        default=0,
        help="Rotation in degrees",
    )
    parser.add_argument("--saturation", type=float, default=1.3)
    parser.add_argument("--contrast", type=float, default=1.05)
    parser.add_argument("--gamma", type=float, default=0.90)
    args = parser.parse_args()

    lock_fd = open("/tmp/frame.lock", "w")  # noqa: SIM115 — held for process lifetime
    try:
        fcntl.flock(lock_fd, fcntl.LOCK_EX | fcntl.LOCK_NB)
    except BlockingIOError:
        print("Another instance running, skipping")
        sys.exit(0)

    from waveshare_epd import epd7in3e

    now = datetime.now()
    print(f"Time: {now.strftime('%H:%M')}")
    if now.hour < 7:
        print("Night time, skipping")
        sys.exit(0)

    ha = HomeAssistantClient(HA_URL, HA_TOKEN)
    home = [name for name, eid in HA_PRESENCE.items() if ha.is_person_home(eid)]
    if not home:
        print("No one home, skipping")
        sys.exit(0)
    print(f"Home: {', '.join(home)}")

    client = ImmichClient(IMMICH_URL, IMMICH_API_KEY)
    if args.album:
        image_path, asset = get_random_photo_from_album(client, args.album, args.orientation)
        print(f"Album: {args.album}")
    else:
        names = [n.strip() for n in args.people.split(",")]
        image_path, asset = get_random_photo_of_people(client, names, args.orientation)
        print(f"People: {', '.join(names)}")

    left_text = format_age(asset)
    right_text = format_location(asset)
    if left_text or right_text:
        print(f"Overlay: {left_text or '-'} | {right_text or '-'}")

    try:
        epd = epd7in3e.EPD()
        try:
            epd.init()
            img = Image.open(image_path).convert("RGB")
            faces = client.get_asset_faces(asset["id"])
            print(f"Faces: {len(faces)}")
            target_w, target_h = (480, 800) if args.orientation in (90, 270) else (800, 480)
            img = face_aware_crop(img, target_w, target_h, faces)
            if args.orientation:
                img = img.rotate(args.orientation, expand=True)
            buf = epd.getbuffer(
                img,
                saturation=args.saturation,
                contrast=args.contrast,
                gamma=args.gamma,
                left_text=left_text,
                right_text=right_text,
                orientation=args.orientation,
            )
            epd.display(buf)
        finally:
            epd.sleep()
    finally:
        image_path.unlink(missing_ok=True)


if __name__ == "__main__":
    main()