"""HTTP REST client for the i.MX 8M Mini target.

Talks to the Flask server in `server/app.py`. The target must run that server
with appropriate privileges to access /sys/kernel/debug, memtool, and i2c-2.
"""

from __future__ import annotations

import logging

import requests

log = logging.getLogger(__name__)


class TargetController:
    def __init__(self, ip: str, port: int, timeout_s: float = 10.0) -> None:
        self.base_url = f"http://{ip}:{port}"
        self.timeout = timeout_s
        probe = requests.get(f"{self.base_url}/registers", timeout=timeout_s)
        probe.raise_for_status()
        log.info("Target reachable at %s", self.base_url)

    def _get(self, path: str) -> dict:
        r = requests.get(f"{self.base_url}{path}", timeout=self.timeout)
        r.raise_for_status()
        return r.json()

    def _put(self, path: str, payload: dict) -> dict:
        r = requests.put(f"{self.base_url}{path}", json=payload, timeout=self.timeout)
        r.raise_for_status()
        return r.json()

    def get_dsim_registers(self) -> dict:
        return self._get("/registers")

    def get_sn65_registers(self) -> dict:
        return self._get("/sn65_registers")

    def get_sn65_settling(self) -> dict:
        return self._get("/sn65_settling")

    def display_on(self) -> dict:
        return self._put("/display", {"state": "on"})

    def display_off(self) -> dict:
        return self._put("/display", {"state": "off"})

    def video_start(self, mode: str = "static-pink") -> dict:
        return self._put("/video", {"action": "start", "mode": mode})

    def video_stop(self) -> dict:
        return self._put("/video", {"action": "stop"})