import pyray as rl
from openpilot.system.ui.widgets import Widget
from openpilot.system.ui.lib.application import gui_app

# ---------------------------------------------------------------------------
#  Constants extracted from the original Qt style
# ---------------------------------------------------------------------------
# TODO: this should be corrected, but Scroller relies on this being incorrect :/
WIDTH, HEIGHT = 112, 240


class SideButton(Widget):
  def __init__(self, btn_type: str):
    super().__init__()
    self.type = btn_type
    self.set_rect(rl.Rectangle(0, 0, WIDTH, HEIGHT))

    # load pre-rendered button images
    if btn_type not in ("check", "back"):
      btn_type = "back"
    btn_img_path = f"icons_mici/buttons/button_side_{btn_type}.png"
    btn_img_pressed_path = f"icons_mici/buttons/button_side_{btn_type}_pressed.png"
    self._txt_btn, self._txt_btn_back = gui_app.texture(btn_img_path, 100, 224), gui_app.texture(btn_img_pressed_path, 100, 224)

  def _render(self, _) -> bool:
    x = int(self._rect.x + 12)
    y = int(self._rect.y + (self._rect.height - self._txt_btn.height) / 2)
    rl.draw_texture(self._txt_btn if not self.is_pressed else self._txt_btn_back,
                    x, y, rl.WHITE)

    return False