import pygame
from pygame import *
import random

class Bullet:
    def __init__(self,x,y,radius,color,facing):
        self.x = x
        self.y = y
        self.radius = radius
        self.color = color
        self.facing = facing
        self.vel = 8 * facing

    def draw(self,win):
        pygame.draw.circle(win,self.color,(self.x,self.y),self.radius)




class Player:
    def __init__(self, x, y, w, h,speed):
        self.x = x
        self.y = y
        self.w = w
        self.h = h
        self.speed = speed
        self.isJump = False
        self.jumpCount = 10
        self.bullets = []
        self.lastMove ="R"
        self.facing = 1



    def draw_player(self):
        pygame.draw.rect(win,(255,255,255), (self.x, self.y, self.h, self.w))

    def shoot(self):
        for bullet in self.bullets:
            if bullet.x < win_w and bullet.x > 0:
                bullet.x += bullet.vel
            else:
                self.bullets.pop(self.bullets.index(bullet))


    def move_player1(self):
        keys = pygame.key.get_pressed()

        if keys[pygame.K_SPACE]:
            if self.lastMove == "R":
                self.facing = 1
            else:
                self.facing = -1

            if len(self.bullets) < 1:
                self.bullets.append(Bullet(round(self.x +self.w // 2),round(self.y +self.h // 2),5,(255,0,0),self.facing))

        if keys[pygame.K_a] and self.x > 5:
            self.x -= self.speed
            self.lastMove = "L"
        if keys[pygame.K_d] and self.x < win_w - self.w - 5:
            self.x += self.speed
            self.lastMove = "R"
        if not(self.isJump):
            if keys[pygame.K_w] and  self.y > 5 :
                self.isJump = True
            if keys[pygame.K_s] and self.y < win_h - self.h - 5:
                self.y += self.speed
        else:
            if self.jumpCount >= -10:
                if self.jumpCount < 0:
                    self.y += (self.jumpCount ** 2)/2
                else:
                    self.y -= (self.jumpCount ** 2)/2
                self.jumpCount -= 1
            else:
                self.isJump = False
                self.jumpCount = 10



    def move_player2(self):
        keys = pygame.key.get_pressed()
        if keys[pygame.K_z]:
            if self.lastMove == "R":
                self.facing = 1
            else:
                self.facing = -1

            if len(self.bullets) < 1:
                self.bullets.append(Bullet(round(self.x +self.w // 2),round(self.y +self.h // 2),5,(255,0,0),self.facing))
        if keys[pygame.K_LEFT] and self.x > 5:
            self.x -= self.speed
            self.lastMove = "L"
        if keys[pygame.K_RIGHT] and self.x < win_w - self.w - 5:
            self.x += self.speed
            self.lastMove = "R"
        if not(self.isJump):
            if keys[pygame.K_UP] and  self.y > 5 :
                self.isJump = True
            if keys[pygame.K_DOWN] and self.y < win_h - self.h - 5:
                self.y += self.speed
        else:
            if self.jumpCount >= -10:
                if self.jumpCount < 0:
                    self.y += (self.jumpCount ** 2)/2
                else:
                    self.y -= (self.jumpCount ** 2)/2
                self.jumpCount -= 1
            else:
                self.isJump = False
                self.jumpCount = 10

pygame.init()

pygame.display.set_caption("Dun")

player1 = Player(50, 590, 50, 50, 5)
player2 = Player(450,590, 50, 50, 5)

win_w = 800
win_h = 640

win = pygame.display.set_mode((win_w,win_h))

start = 1
game = 2
death = 3
winner = 4
buff = 5
bos = 6
state = game



run = True
while run:
    pygame.time.delay(20)

    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()


    if state == game:

        player1.move_player1()
        player1.shoot()
        player2.move_player2()
        player2.shoot()

        win.fill((0,0,0))

        player1.draw_player()
        for bullet in player1.bullets:
            bullet.draw(win)
        player2.draw_player()
        for bullet in player2.bullets:
            bullet.draw(win)
        pygame.display.update()