"""
Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
"""
import math
from enum import IntEnum

import pyray as rl

from iqpilot.common.params import Params
from iqpilot.system.ui.lib.text_measure import measure_text_cached

class SourceState(IntEnum):
  HIDDEN = 0
  LOADING = 1
  ACTIVE = 2
  FAILED = 3
  CROSSED = 4


_GREEN = rl.Color(46, 204, 113, 255)
_ORANGE = rl.Color(255, 115, 0, 255)
_WHITE = rl.Color(255, 255, 255, 255)


def _emac_state(params: Params, engaged: bool) -> SourceState:
  if not params.get_bool("MacModelReachable"):
    return SourceState.HIDDEN
  if params.get_bool("MacModelActive"):
    return SourceState.ACTIVE
  if params.get_bool("MacModelFailed"):
    return SourceState.CROSSED if engaged else SourceState.FAILED
  return SourceState.LOADING


def _egpu_state(params: Params, engaged: bool) -> SourceState:
  if not params.get_bool("UsbGpuPresent"):
    return SourceState.HIDDEN
  if params.get_bool("UsbGpuActive"):
    return SourceState.ACTIVE
  if params.get_bool("UsbGpuFailed"):
    return SourceState.CROSSED if engaged else SourceState.FAILED
  return SourceState.LOADING


def resolve_source(params: Params, engaged: bool) -> tuple[str, SourceState]:
  if params.get_bool("IQEmacEnabled"):
    return "MAC", _emac_state(params, engaged)
  if params.get_bool("UsbGpuPresent") or params.get_bool("IQEgpuEnabled"):
    return "GPU", _egpu_state(params, engaged)
  return "", SourceState.HIDDEN


def draw_source_label(font: rl.Font, label: str, state: SourceState,
                      pos: rl.Vector2, font_size: int) -> None:
  if state == SourceState.HIDDEN or not label:
    return
  if state == SourceState.ACTIVE:
    color, opacity, strike = _GREEN, 1.0, False
  elif state == SourceState.FAILED:
    color, opacity, strike = _ORANGE, 1.0, False
  elif state == SourceState.CROSSED:
    color, opacity, strike = _WHITE, 0.65, True
  else:
    color, opacity, strike = _WHITE, 0.35 + 0.65 * (0.5 - 0.5 * math.cos(rl.get_time() * 6.0)), False

  col = rl.Color(color.r, color.g, color.b, int(255 * opacity))
  rl.draw_text_ex(font, label, pos, font_size, 0, col)
  if strike:
    size = measure_text_cached(font, label, font_size)
    y = int(pos.y + size.y / 2)
    rl.draw_line_ex(rl.Vector2(pos.x - 2, y), rl.Vector2(pos.x + size.x + 2, y), 4, col)