from typing import Any, Callable, Optional

import pygame
from utils import CONFIG

from .base import BaseScreen, SceenElement, TextScreen
from .base_button import BaseButton


class Button(BaseButton, BaseScreen, SceenElement, TextScreen):
    def __init__(
        self, text: str, action: Optional[Callable[[], Optional[Any]]]
    ) -> None:
        super().__init__(text, action)
        self._initialize_surface()
        self._initialize_font()
        self._set_default_background_color()
        self._set_default_text_color()

    def on_click(self) -> None:
        """Handle click event."""
        if self.action:
            self.action()

    def on_hover(self, event: pygame.Event) -> None:
        """Handle hover event."""
        if self.rect.collidepoint(event.pos):
            self._set_background_color(CONFIG.colors.bg_visual)
        else:
            self._set_default_background_color()

    def run(self) -> None:
        pass

    def update(self) -> None:
        """Update the button."""

    def draw(self, surface: pygame.Surface, pos: tuple[float, float]) -> None:
        """Draw the button on the button surface."""
        self._initialize_rect(pos)
        self._update_display_surface()
        self._draw_background()
        self._draw_text()
        self._draw_border()

    def _initialize_surface(self) -> None:
        """Initialize the button surface."""
        self.surface = pygame.Surface(CONFIG.window.button.size)
        self.display_surface = pygame.display.get_surface()

    def _initialize_rect(self, pos: tuple[float, float]) -> None:
        """Initialize the button rectangle."""
        self.rect = self.surface.get_rect(center=pos)

    def _draw_text(self) -> None:
        """Draw the text on the text surface."""
        x = self.surface.get_width() / 2
        y = self.surface.get_height() / 2
        self._display_text(self.text, (x, y))

    def _display_text(self, text: str, pos: tuple[float, float]) -> None:
        """
        Display a single text element on the button surface.

        Args:
            text: The text to be displayed.
            pos: The position (x, y) where the text should be displayed.
        """
        text_surface = self.font.render(text, True, self.text_color)
        text_rect = text_surface.get_rect(center=pos)
        self.surface.blit(text_surface, text_rect)

    def _initialize_font(self) -> None:
        """Initialize the font used to display the score."""
        self.font = pygame.font.Font(CONFIG.font.family, CONFIG.font.size)

    def _draw_background(self) -> None:
        """Fill the background of the button."""
        self.surface.fill(self.color)

    def _draw_border(self) -> None:
        """Draw the border of the button."""
        pygame.draw.rect(
            self.display_surface,
            CONFIG.colors.border_highlight,
            self.rect,
            CONFIG.game.line_width * 2,
            CONFIG.game.border_radius,
        )

    def _update_display_surface(self) -> None:
        """Update the display surface."""
        self.display_surface.blit(self.surface, self.rect)

    def _set_default_background_color(self) -> None:
        """Set the default background color of the button."""
        self._set_background_color(CONFIG.colors.bg_sidebar)

    def _set_default_text_color(self) -> None:
        """Set the default text color of the button."""
        self._set_text_color(CONFIG.colors.fg_float)

    def _set_text_color(self, color: str = CONFIG.colors.fg_float) -> None:
        """Set the text color of the button."""
        if self.action is None:
            self.text_color = CONFIG.colors.fg_dark
            return
        self.text_color = color

    def _set_background_color(self, color: str) -> None:
        """Set the background color of the button."""
        if self.action is None:
            self.color = CONFIG.colors.comment
            return
        self.color = color