# tsk/ui/layout.py
import pyray as rl

class Theme:
  """Defines the visual theme and constants for the TSK Manager."""

  # --- Colors ---
  nav_bg_color = rl.Color(20, 20, 20, 255)
  key_text_color = rl.Color(255, 255, 255, 255)
  button_color = rl.Color(51, 51, 51, 255)
  title_bg_color = rl.Color(40, 40, 40, 255)
  nav_text_color = rl.Color(255, 255, 255, 255)  # Added nav_text_color

  # --- Font Sizes ---
  title_font_size = 90
  nav_button_font_size = 80
  key_status_font_size = 65

  # --- Text ---
  nav_button_text_left = "< Tools\n< Menu"
  nav_button_text_right = "Reboot >\n   Menu >"
  nav_button_text_to_tools = "< Tools\n< Menu"

  # --- Layout ---
  title_prefix_x_offset_percent = 0.2
  title_x_offset = 180
  brighten_amount = 0.6
  status_update_interval = 1  # seconds

  # --- Menu Identifiers ---
  menu_tools = 1
  menu_reboot = 2

  menu_names = {
    menu_tools: "Tools Menu",
    menu_reboot: "Reboot Menu",
  }

  @staticmethod
  def brighten_color(color: rl.Color, amount: float) -> rl.Color:
    """Brightens a color by a given amount (0.0 to 1.0)."""
    r = int(min(color.r + (255 - color.r) * amount, 255))
    g = int(min(color.g + (255 - color.g) * amount, 255))
    b = int(min(color.b + (255 - color.b) * amount, 255))
    return rl.Color(r, g, b, color.a)


class Layout:
  """Provides layout calculation functions."""

  @staticmethod
  def calculate_button_dimensions(rect_height: int, header_height: int) -> int:
    """Calculates button height based on available screen space and desired spacing."""
    available_height = rect_height - header_height
    guide_button_height = 200
    num_spacers = 3  # Top, between buttons, and below guide button
    remaining_height = available_height - guide_button_height - (num_spacers * 80)
    button_height = remaining_height
    return int(button_height)

  @staticmethod
  def calculate_button_positions(rect: rl.Rectangle, header_height: int, num_buttons: int) -> tuple:
    """Calculates the starting positions for a row of buttons."""
    total_width = (num_buttons * 600) + ((num_buttons - 1) * 80)
    start_x = (rect.width - total_width) / 2 + rect.x
    start_y = rect.y + header_height + 80
    return start_x, start_y