import pygame

class Bullet:
    def __init__(self, game, screen, x, y, x_speed, y_speed):
        self._screen = screen
        self.game = game

        self._pos = [x,y]

        self._x_speed = x_speed
        self._y_speed = y_speed

        self._color = (0,0,0)
        self._size = 5

    def update(self):
        self.move(self._x_speed, self._y_speed)

        if not (0 <= self._pos[0] <= self._screen.get_width()) or not (0 <= self._pos[1] <= self._screen.get_height()):
            self.game.remove_object(self)

    def move(self, x, y):
        self._pos[0] += x
        self._pos[1] += y

    def draw(self):
        pygame.draw.rect(self._screen, self._color, pygame.rect.Rect(self._pos, (self._size, self._size)))
        
    def get_pos(self):
        return self._pos