from states.game_state import GameState
from controllers.player_controller import PlayerController
from render.map_renderer import MapRenderer
from render.gui_renderer import GUIRenderer
from render.player_renderer import PlayerRenderer
from render.info_renderer import InfoRenderer
from render.control_renderer import ControlRenderer
from utils.map_coord_converter import MapCoordConverter
from controllers.gui_controller import GUIController
from controllers.info_controller import InfoController
from controllers.control_controller import ControlController
import pygame

class Game(GameState):
    def __init__(self, switch_state_callback, context):
        super().__init__(switch_state_callback, context)

        match_data_path = f"assets/maps/config/{self.match.map_name}.json"
        match_image_path = f"assets/maps/overview/{self.match.map_name}.png"

        # Screen Areas
        self.info_box = pygame.Surface((350, self.screen.get_height()), pygame.SRCALPHA)
        self.info_box_top_left = (0, 0)

        self.game_box = pygame.Surface((650, 650), pygame.SRCALPHA)
        self.game_box_top_left = (350, 0)

        self.control_box = pygame.Surface((650, 120), pygame.SRCALPHA)
        self.control_box_top_left = (350, 650)

        # Helper Classes
        self.map_coord_controller = MapCoordConverter(self.game_box.get_width(), self.game_box.get_height(), match_data_path, match_image_path)

        # Renderers
        self.map_renderer = MapRenderer(self.game_box, match_data_path, match_image_path)
        self.player_renderer = PlayerRenderer(self.game_box, self.match, self.map_coord_controller, self.options, self.styling)
        self.gui_render = GUIRenderer(self.screen, self.match)
        self.info_render = InfoRenderer(self.info_box, self.styling, self.match)
        self.control_render = ControlRenderer(self.control_box, self.match)

        # Controllers
        self.player_controller = PlayerController(self.player_renderer, self.match, self.game_box_top_left)
        self.gui_controller = GUIController(self.gui_render, self.switch_state)
        self.info_controller = InfoController(self.info_render, self.player_controller)
        self.control_controller = ControlController(self.control_render, self.control_box_top_left)

    def handle_events(self, events):
        """Handles user inputs."""
        for event in events:
            if event.type == pygame.KEYDOWN and event.key == pygame.K_ESCAPE:
                self.switch_state("menu")
            self.player_controller.update(event)
            self.gui_controller.update(event)
            self.info_controller.update(event)
            self.control_controller.update(event)

    def update(self):
        """Updates game objects."""
        self.match.next_tick()

    def draw(self):
        """Draws everything on screen."""
        self.screen.fill(self.styling["background_colour"])
        self.game_box.fill(self.styling["background_colour"])
        self.info_box.fill(self.styling["foreground_colour"])
        self.control_box.fill(self.styling["background_colour"])

        self.map_renderer.render()
        self.player_renderer.render()
        self.gui_render.render()
        self.info_render.render()
        self.control_render.render()
        self.screen.blit(self.info_box, self.info_box_top_left)
        self.screen.blit(self.control_box, self.control_box_top_left)
        self.screen.blit(self.game_box, self.game_box_top_left)