import pygame
from bullet import Bullet

class Player:
    def __init__(self, screen):
        self.health = 200
        self.pos = [500, 250]
        self.size = 20
        self.speed = 5

        self.color = (0,0,0)

        self.screen = screen

        self.bullets = []

        
    def shoot(self):
        m_x, m_y = pygame.mouse.get_pos()

        dx = m_x - self.pos[0]
        dy = m_y - self.pos[1]

        length = (dx**2+dy**2)**(0.5)

        self.bullets.append(Bullet(
            self.screen,
            self.pos[0]+(self.size/2),
            self.pos[1]+(self.size/2),
            (dx/length)*10,
            (dy/length)*10
        ))
    
    def move(self, x,y):
        self.pos[0] += x*self.speed
        self.pos[1] += y*self.speed

        if self.pos[0] < 0:
            self.pos[0] = 0
        elif self.pos[0]+self.size > self.screen.get_width():
            self.pos[0] = self.screen.get_width()-self.size

        if self.pos[1] < 0:
            self.pos[1] = 0
        elif self.pos[1]+self.size > self.screen.get_height():
            self.pos[1] = self.screen.get_height()-self.size

    def draw(self):
        for bullet in self.bullets:
            bullet.draw()
        pygame.draw.rect(self.screen, self.color, pygame.rect.Rect(self.pos, (self.size, self.size)))


    def update(self):
        for bullet in self.bullets:
            bullet.update()
            
        pressed = pygame.key.get_pressed()

        y = pressed[pygame.K_s] - pressed[pygame.K_w]
        x = pressed[pygame.K_d] - pressed[pygame.K_a]
        
        self.move(x,y)