from abc import ABC, abstractmethod
import uuid


class Unit(ABC):
    """
    Abstract base class for all game units.

    Attributes
    ----------
    id : UUID
        Unique identifier for the unit.
    game : Game
        Reference to the main game object.
    position : tuple
        The current position of the unit (x, y).
    position_before : tuple
        The position of the unit before the last update.
    age : int
        The age of the unit in game ticks.
    speed : float
        Movement speed of the unit.
    partial_move : float
        Partial movement progress for smooth animation.
    bbox : tuple
        Bounding box for collision detection (x1, y1, x2, y2).
    stop : int
        Number of ticks to remain stationary.

    Methods
    -------
    move()
        Update unit position and state (must be implemented by subclasses).
    draw()
        Render the unit on screen (must be implemented by subclasses).
    collisions()
        Handle collisions with other units (optional override).
    die()
        Remove unit from game and handle cleanup.
    """
    def __init__(self, game, position=(0, 0), id=None):
        """Initialize a unit with game reference and position."""
        self.id = id if id else uuid.uuid4()
        self.game = game
        self.position = position
        self.position_before = position
        self.age = 0
        self.speed = 1.0
        self.partial_move = 0
        self.bbox = (0, 0, 0, 0)
        self.stop = 0

    @abstractmethod
    def move(self):
        """Update unit position and state. Must be implemented by subclasses."""
        pass

    @abstractmethod
    def draw(self):
        """Render the unit on screen. Must be implemented by subclasses."""
        pass

    def collisions(self):
        """Handle collisions with other units. Default implementation does nothing."""
        pass

    def die(self):
        """Remove unit from game and handle basic cleanup."""
        if self.id in self.game.units:
            self.game.units.pop(self.id)