import pygame
from random import randint
from time import time
from bullets import *
#from particles import * # !!!
import time
import sys

############
windowX = 832
windowY = 832
############ imgs for player-enemy(self,bullet)
bullet_image = pygame.image.load('pics/second_bullet.png')
enemy_bul_img = pygame.image.load('pics/bullet.png')
enemy_image = pygame.image.load('pics/32x64.png')
player_image = pygame.image.load('pics/32x64.png')
############

class Player:
    def __init__(self):
        self.bul_cooldown = 50
        self.tp_cooldown = 100
        self.x = 300
        self.y = 300
        self.vel = 10
        self.height = 64
        self.width = 32

    def update(self):
        keys = pygame.key.get_pressed()
        if keys[pygame.K_LEFT] and self.x - self.vel > 5 :
        	self.x -= self.vel

        if keys[pygame.K_RIGHT] and self.x + self.vel < windowX- self.width - 5 :
        	self.x += self.vel

        if keys[pygame.K_UP] and self.y + self.height - self.vel > self.height + 5 :
        	self.y -= self.vel

        if keys[pygame.K_DOWN] and self.y + self.height + self.vel < windowY-15 :
        	self.y += self.vel

    def draw(self,window):
        pygame.draw.rect(window,(0,0,0),(self.x,self.y,self.width,self.height))
        window.blit(player_image,(self.x, self.y))
        self.player_rect = pygame.Rect(self.x, self.y, self.width, self.height)


    def shoot(self,window):
        # if self.bul_cooldown >= 1: #With cooldown the game looks not so spicy.
        #     self.bul_cooldown -= 2 #Maybe it will be the matter of upgrades ?

        keys = pygame.key.get_pressed()
        if keys[pygame.K_z]: # and self.bul_cooldown == 0
            new_bullet = Bullet(self.x + self.width/2, self.y + 10)
            bullets_on_screen.append(new_bullet)
            #self.bul_cooldown = 20

        if int(len(bullets_on_screen)) > 0:
            for bullet in bullets_on_screen:
                bullet.draw(window,(0,0,0),bullet_image)
                bullet.move()
                if bullet.bullet_rect[1] <= -20:
                    bullets_on_screen.remove(bullet)

    def teleportation(self):
        self.vel = 10
        if self.tp_cooldown >= 1:
            self.tp_cooldown -= 2
            print('teleportation cooldown : '+ str(self.tp_cooldown))

        keys = pygame.key.get_pressed()
        if keys[pygame.K_d] and keys[pygame.K_UP] and self.tp_cooldown == 0: #forward-dash
            self.y -= self.height*2
            self.tp_cooldown = 100

        if keys[pygame.K_d] and keys[pygame.K_LEFT] and self.tp_cooldown == 0: #left-dash
            self.x -= self.height*2
            self.tp_cooldown = 100

        if keys[pygame.K_d] and keys[pygame.K_RIGHT] and self.tp_cooldown == 0: #right-dash
            self.x += self.height*2
            self.tp_cooldown = 100

        if keys[pygame.K_d] and keys[pygame.K_DOWN] and self.tp_cooldown == 0: #backward-dash
            self.y += self.height*2
            self.tp_cooldown = 100

    def out_of_area(self):
        if self.x + self.width > windowX or self.x < 0 or self.y+self.height > windowY or self.y < 0:
            return True
        else:
            return False
    def collision(self,window):
        for bullet in enemy_bul_on_screen:
            if self.player_rect.colliderect(bullet.bullet_rect):
                # new_particle = Particle(bullet.x,bullet.y)
                # particles_on_screen.append(new_particle)
                # for particle in particles_on_screen:
                #     particle.draw(window)
                #     particle.update()
                self.x += randint(-100,100)
                self.y += randint(-100,100)



class Enemy:
    def __init__(self):
        self.enemy_x = 100
        self.enemy_y = 100
        self.en_width = 32
        self.en_height = 64
        self.vel = 3
    def draw(self,window):
        pygame.draw.rect(window,(0,0,0),(self.enemy_x, self.enemy_y, self.en_width, self.en_height))
        window.blit(enemy_image,(self.enemy_x, self.enemy_y))
        self.enemy_rect = pygame.Rect(self.enemy_x, self.enemy_y, self.en_width, self.en_height) #Rect object for collision
    def move(self,side):
        if str(side) == "right":
            self.enemy_x += self.vel
        elif str(side) == "left":
            self.enemy_x -= self.vel
        elif str(side) == "up":
            self.enemy_y -= self.vel
        elif str(side) == "down":
            self.enemy_y += self.vel
    def enemy_shoot(self,window):
        new_bullet = Bullet(self.enemy_x + self.en_width/2, self.enemy_y + 10)
        enemy_bul_on_screen.append(new_bullet)
        if int(len(enemy_bul_on_screen)) > 0:
            for bullet in enemy_bul_on_screen:
                bullet.draw(window,(0,0,0),enemy_bul_img)
                bullet.movedwn()
                if bullet.bullet_rect[1] >= windowY +20:
                    enemy_bul_on_screen.remove(bullet) # !!???
        print('Bullets (Enemy,Player): ' + str(len(bullets_on_screen + enemy_bul_on_screen)))

    def out_of_area(self):
        if self.enemy_x > windowX+1 or self.enemy_x < -1 or self.enemy_y > windowY+1 or self.enemy_y < -1:
            return True
        else:
            return False

    def collision(self):
        for bullet in bullets_on_screen:
            if self.enemy_rect.colliderect(bullet.bullet_rect):
                #Particle effect here
                self.enemy_x += randint(-60,60)
                self.enemy_y += randint(-60,60)