import pygame
from enums import PillType

class Pill(pygame.sprite.Sprite):
    def __init__(self, x, y, pill_type: PillType):
        super().__init__()
        self.x = x
        self.y = y
        self.pill_type = pill_type
        self.eaten = False
        self.radius = 2 if pill_type == PillType.NORMAL else 6
        self.color = (255, 255, 255)
        self.image = pygame.Surface((self.radius * 2, self.radius * 2), pygame.SRCALPHA)
        pygame.draw.circle(self.image, self.color, (self.radius, self.radius), self.radius)
        self.rect = self.image.get_rect(center=(x, y))

    def check_eaten(self, pacman_rect, game_state):
        if not self.eaten and self.rect.colliderect(pacman_rect):
            self.eaten = True
            self.kill()
            return self.on_eaten(game_state)
        return 0

    def on_eaten(self, game_state):
        """Override this in subclasses to apply effects when eaten"""
        return 0

class NormalPill(Pill):
    def __init__(self, x, y):
        super().__init__(x, y, PillType.NORMAL)

    def on_eaten(self, game_state):
        game_state["score"] += 100
        return 100

class PowerPill(Pill):
    def __init__(self, x, y):
        super().__init__(x, y, PillType.POWER)

    def on_eaten(self, game_state):
        game_state["score"] += 500
        game_state["ghost_mode"] = "frightened"
        game_state["frightened_timer"] = pygame.time.get_ticks()  # Optional
        return 500