import pygame
import settings
from settings import MyColors

class Endscreen:
    def __init__(self, screen, score, time_passed=0):
        self.screen = screen
        self.score = score

        self.background = pygame.image.load(settings.BACKGROUND_IMG_COMP).convert_alpha()
        icon_controller_retry = pygame.image.load(settings.icons.controller.viereck).convert_alpha()
        self.icon_controller_retry = pygame.transform.scale(icon_controller_retry, (50, 48))
        icon_controller_menu = pygame.image.load(settings.icons.controller.dreieck).convert_alpha()
        self.icon_controller_menu = pygame.transform.scale(icon_controller_menu, (50, 48))

        self.font = pygame.font.Font(settings.fonts.PressStart2P, 60)
        self.small_font = pygame.font.Font(settings.fonts.PressStart2P, 24)

        self.retry_button = pygame.Rect(390, 400, 200, 50)
        self.menu_button = pygame.Rect(690, 400, 200, 50)

        self.time_passed = time_passed

    def draw(self):
        self.screen.blit(self.background, (0, 0))

        # Score-Text
        score_text = self.font.render(f"Dein Score: {self.score}", True, (255, 255, 255))  # z.B. Gold
        self.screen.blit(score_text, (settings.SCREEN_WIDTH // 2 - score_text.get_width() // 2, 200))

        time_text = self.small_font.render(f"Deine Zeit: {self.time_passed:.2f} Sekunden", True, (255, 255, 255))
        self.screen.blit(time_text, (settings.SCREEN_WIDTH // 2 - time_text.get_width() // 2, 300))


        pygame.draw.rect(self.screen, MyColors.yellow, self.retry_button)
        pygame.draw.rect(self.screen, MyColors.yellow, self.menu_button)

        nochmal_text = self.small_font.render("Nochmal", True, MyColors.brown)
        menu_text = self.small_font.render("Menü", True, MyColors.brown)

        self.screen.blit(nochmal_text, (self.retry_button.x + 15, self.retry_button.y + 10))
        self.screen.blit(menu_text, (self.menu_button.x + 50, self.menu_button.y + 10))

        self.screen.blit(self.icon_controller_retry, (self.retry_button.x + 70, self.retry_button.y + 40))
        self.screen.blit(self.icon_controller_menu, ((self.menu_button.x + 70), self.menu_button.y + 40))

        return self.retry_button, self.menu_button
    
 
    def handle_event(self, event):
        if event.type == pygame.MOUSEBUTTONDOWN:
            print("MOUSEBUTTONDOWN") if settings.DEBUG else None
            if self.retry_button.collidepoint(event.pos):
                return "retry"
            elif self.menu_button.collidepoint(event.pos):
                return "menu"
        if event.type == pygame.MOUSEBUTTONUP:
            if self.retry_button.collidepoint(event.pos):
                return "retry"
            elif self.menu_button.collidepoint(event.pos):
                return "menu"
        if event.type == pygame.JOYBUTTONDOWN:
            if event.button == 3:
                return "menu"
            elif event.button == 2:
                return "retry"
        return None