from random import randint

from lifxlan import LifxLAN
from litestar import Litestar, get, post

lifxlan = LifxLAN()


@get("/")
async def get_index() -> dict[str, str]:
    try:
        # colors = lifxlan.get_color_all_lights()
        lifxlan.set_color_all_lights(
            [randint(0, 65535), randint(0, 65535), randint(0, 65535), randint(0, 9000)],
            rapid=True,
        )
    except Exception as e:
        return {"status": "error", "message": str(e)}

    # return {"status": str(colors)}
    return {"status": "ok"}


@post("/")
async def post_index() -> dict[str, str]:
    try:
        lifxlan.set_color_all_lights(
            [randint(0, 65535), randint(0, 65535), randint(0, 65535), randint(0, 9000)],
            rapid=True,
        )
    except Exception as e:
        return {"status": "error", "message": str(e)}

    return {"status": "ok"}


app = Litestar(
    [
        get_index,
        post_index,
    ]
)