import pygame
import pygame_widgets
from pygame_widgets.slider import Slider
from models.match import Match
from models.player import Player
from models.team import Team
from controllers.map_coord_controller import MapCoordController
from utils.json_object import JSONObject

class Renderer:
    def __init__(self, match: Match, screen):
        self.screen = screen
        self.match = match
        self.font = pygame.font.Font(None, 36)
        self.small_font = pygame.font.Font(None, 15)

        try:
            self.json_object = JSONObject(f"maps/{match.map_name}.json")
        except FileNotFoundError:
            raise NotImplementedError(f"Map {match.map_name} not implemented.")
        
        self.top_left_x = self.json_object.get("pos_x")
        self.top_left_y = self.json_object.get("pos_y")
        self.scale = self.json_object.get("scale")
        self.rotation = self.json_object.get("rotate")
        self.image_path = self.json_object.get("material")

        self.map_image = pygame.image.load(self.image_path)
        self.image_width = self.map_image.get_width()
        self.image_height = self.map_image.get_height()

        self.bottom_right_x = self.top_left_x + (self.image_width * self.scale)
        self.bottom_right_y = self.top_left_y - (self.image_height * self.scale)
        
    
        self.map_coord_controller = MapCoordController(self.screen.get_width(), self.screen.get_height(), 
                                                       self.top_left_x, self.bottom_right_x, self.top_left_y, self.bottom_right_y)
    def render_players(self):
        """Draws everything on screen."""
        self.map_coord_controller.update_screen_size(self.screen.get_width(), self.screen.get_height())

        for team in self.match.get_teams():
            for player in team.players:
                if player.dead:
                    continue
                mapped_x, mapped_y = self.map_coord_controller.map_to_screen(player.x, player.y)
                pygame.draw.circle(self.screen, team.colour, (mapped_x, mapped_y), 5)

                # Draw player name
                text = self.small_font.render(player.name, True, (255, 255, 255))
                self.screen.blit(text, (mapped_x + 10, mapped_y))

        pygame.display.flip()  # Update display

    def render_text(self):
        # Draw current tick
        text = self.font.render(f"Tick: {self.match.tick}/{self.match.max_tick}", True, (255, 255, 255))
        self.screen.blit(text, (10, 10))

        # Draw team scores
        text = self.font.render(f"Score: {self.match.team_1.score} - {self.match.team_2.score}", True, (255, 255, 255))
        self.screen.blit(text, (10, 40))

    def render_map(self):
        # Scale and rotate map image
        self.map_image = pygame.transform.scale(self.map_image, (self.screen.get_width(), self.screen.get_height()))

        # Draw map image
        self.screen.blit(self.map_image, (0, 0))

    '''
        def render_slider(self):
            # Create slider
            slider = Slider(self.screen, 100, 100, 400, 20, min=1, max=self.match.max_tick, step=1, initial=self.match.tick, 
                                            handleRadius=10, handleColour=(255, 255, 255), handleOutline=(255, 255, 255), 
                                            sliderColour=(100, 100, 100), sliderMarkColour=(150, 150, 150), 
                                            font=pygame.font.Font(None, 36), textColour=(255, 255, 255), 
                                            valueColour=(255, 255, 255), valueOutlineColour=(255, 255, 255), 
                                            onSlide=self.on_slide, sliderMark=0)
    '''

    def render(self):
        self.screen.fill((30, 30, 30))  # Clear screen
        self.render_map()
        self.render_text()
        self.render_players()