from fastapi import FastAPI, Path, HTTPException, status
from projectorpi.cli import ProjectorSerial, ExtronSerial
from pydantic import BaseModel
from time import sleep

app = FastAPI()

extron = ExtronSerial()
projector = ProjectorSerial()


class Response(BaseModel):
    status: str
    error: str | None = None


@app.post("/sleep", response_model=Response)
async def sleep_post() -> Response:
    extron.sleep()
    projector.power_off()

    return Response(status="ok")


@app.post("/select/{input_id}", response_model=Response)
async def select(
    input_id: int = Path(title="The input to select", ge=1, le=4),
) -> Response:
    if extron.is_sleeping():
        extron.wake()
        sleep(1)
    extron.change_input(input_id)

    if not projector.is_on():
        projector.power_on()

    return Response(status="ok")


@app.post("/control/{direction}", response_model=Response)
async def control(direction: str) -> Response:
    directions = {
        "vol_up": extron.volume_up,
        "vol_down": extron.volume_down,
        "up": extron.menu_up,
        "right": extron.menu_right,
        "down": extron.menu_down,
        "left": extron.menu_left,
        "enter": extron.menu_enter,
        "menu": extron.menu_toggle,
    }

    runnable = directions.get(direction)

    if not runnable:
        raise HTTPException(status.HTTP_400_BAD_REQUEST, "invalid control")

    runnable()

    return Response(status="ok")