#!/usr/bin/env python3
from pathlib import Path

from openpilot.system.hardware import HARDWARE

BIG_UI_DEVICE_TYPES = ("tici", "tizi")
SMALL_UI_DEVICE_TYPE = "mici"


def _normalize_device_type(raw: str) -> str:
  device_type = raw.replace("\x00", "").strip().lower()
  if "comma " in device_type:
    device_type = device_type.rsplit("comma ", 1)[-1].strip()
  return device_type


def _device_tree_device_type() -> str | None:
  model_path = Path("/sys/firmware/devicetree/base/model")
  if not model_path.is_file():
    return None

  try:
    return _normalize_device_type(model_path.read_text(encoding="utf-8", errors="ignore"))
  except Exception:
    return None


def _framebuffer_size() -> tuple[int, int] | None:
  fb_path = Path("/sys/class/graphics/fb0/virtual_size")
  if not fb_path.is_file():
    return None

  try:
    raw = fb_path.read_text().strip().replace(" ", "")
    width_s, height_s = raw.split(",", 1)
    return int(width_s), int(height_s)
  except Exception:
    return None


def _reported_device_type() -> str | None:
  try:
    return _normalize_device_type(HARDWARE.get_device_type())
  except Exception:
    return None


def _ui_device_type() -> str:
  # AGNOS' baked-in updater can run before the StarPilot wrapper gets a chance to
  # route mici devices through the small UI. Trust the device tree first and only
  # allow known large-screen devices to select the tici updater.
  device_tree_type = _device_tree_device_type()
  if device_tree_type in BIG_UI_DEVICE_TYPES:
    return device_tree_type
  if device_tree_type:
    return SMALL_UI_DEVICE_TYPE

  fb_size = _framebuffer_size()
  if fb_size is not None and max(fb_size) < 1000:
    return SMALL_UI_DEVICE_TYPE

  reported_type = _reported_device_type()
  if reported_type in BIG_UI_DEVICE_TYPES:
    return reported_type
  return SMALL_UI_DEVICE_TYPE


def _patch_hardware_device_type(device_type: str) -> None:
  HARDWARE.get_device_type = lambda: device_type
  try:
    import openpilot.system.hardware.tici.hardware as tici_hardware
    tici_hardware.get_device_type = lambda: device_type
  except Exception:
    pass


def main():
  device_type = _ui_device_type()

  # The updater imports application sizing during module import, so patch the
  # hardware probe before importing either UI implementation.
  _patch_hardware_device_type(device_type)

  if device_type in BIG_UI_DEVICE_TYPES:
    import openpilot.system.ui.tici_updater as updater_impl
  else:
    import openpilot.system.ui.mici_updater as updater_impl

  updater_impl.main()


if __name__ == "__main__":
  main()