import pygame
from player import Player
from enemy import Enemy 
from settings import SCREEN_HEIGHT, SCREEN_WIDTH

from pygame.locals import (
    K_UP,
    K_DOWN,
    K_LEFT,
    K_RIGHT,
    K_ESCAPE,
    KEYDOWN,
    QUIT,
)

pygame.init()
ADDENEMY = pygame.USEREVENT + 1
pygame.time.set_timer(ADDENEMY, 1000)


screen = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))


running = True

player = Player()

enemy = Enemy()

enemies = pygame.sprite.Group()
all_sprites = pygame.sprite.Group()

all_sprites.add(player)
all_sprites.add(enemy)

enemies.add(enemy)


while running:
    for event in pygame.event.get():
        if event.type == KEYDOWN:
            if event.key == K_ESCAPE:
                running = False

        elif event.type == QUIT:
            running = False

        elif event.type == ADDENEMY:
            new_enemy = Enemy()
            enemies.add(new_enemy)
            all_sprites.add(new_enemy)

    screen.fill((255, 255, 255))
    
    pressed_keys = pygame.key.get_pressed()

    player.update(pressed_keys)

    enemies.update()

    for entity in all_sprites:
        screen.blit(entity.surf, entity.rect)

    if pygame.sprite.spritecollideany(player, enemies):

        player.kill()
        running=False                           

    pygame.display.flip()

pygame.quit()