# Author - Kristiāns Francis Cagulis
# Date - 11.03.2022
# Title - Space invaders

import pygame
from random import choice
from os.path import abspath, dirname, join

WIDTH, HEIGHT = 800, 800

WINDOW = pygame.display.set_mode((WIDTH, HEIGHT))
pygame.display.set_caption("Space Invaders")
pygame.font.init()

# paths
BASE_PATH = abspath(dirname(__file__))
SPRITE_PATH = join(BASE_PATH, "assets", "sprites")
ENEMY_PATH = join(SPRITE_PATH, "enemies")
FONT = join(BASE_PATH, "fonts", "space_invaders.ttf")

# load sprites
SPACESHIP = pygame.image.load(join(SPRITE_PATH, "playership.png"))  # player
PLAYER_MISSILE = pygame.image.load(
    join(SPRITE_PATH, "missiles", "playermissile.png"))  # player missile

# enemies
ENEMY_1 = pygame.transform.scale(pygame.image.load(
    join(ENEMY_PATH, "enemy_magenta.png")), (40, 35))
ENEMY_2 = pygame.transform.scale(pygame.image.load(
    join(ENEMY_PATH, "enemy_cyan.png")), (40, 35))
ENEMY_3 = pygame.transform.scale(pygame.image.load(
    join(ENEMY_PATH, "enemy_lime.png")), (40, 35))
ENEMY_MISSILE = pygame.image.load(
    join(SPRITE_PATH, "missiles", "enemymissile.png"))  # enemy missile

pygame.display.set_icon(ENEMY_3)
# background
BACKGROUND = pygame.transform.scale(pygame.image.load(
    join(BASE_PATH, "assets", "background.jpg")), (WIDTH, HEIGHT))

# colors (R, G, B)
BLUE = (16, 16, 69)
WHITE = (255, 255, 255)
RED = (188, 2, 5)


class Missile:

    def __init__(self, x: int, y: int, img) -> None:
        self.x = x
        self.y = y
        self.img = img
        self.mask = pygame.mask.from_surface(self.img)

    def draw(self, window) -> None:
        window.blit(self.img, (self.x, self.y))

    def move(self, vel: int) -> None:
        self.y += vel

    def off_screen(self, height: int) -> bool:
        return not (self.y <= height and self.y >= 0)

    def collision(self, obj) -> bool:
        return collide(self, obj)


class Ship:
    COOLDOWN = 30  # cooldown before shooting next missile

    def __init__(self, x: int, y: int, lives: int = 3) -> None:
        self.x = x
        self.y = y
        self.lives = lives
        self.ship_img = None
        self.missile_img = None
        self.missiles = []
        self.cooldown_counter = 0

    def draw(self, window) -> None:
        window.blit(self.ship_img, (self.x, self.y))
        for missile in self.missiles:
            missile.draw(WINDOW)

    def move_missiles(self, vel: int, obj) -> None:
        self.cooldown()
        for missile in self.missiles:
            missile.move(vel)
            if missile.off_screen(HEIGHT):
                self.missiles.remove(missile)
            elif missile.collision(obj):
                obj.lives -= 1
                self.missiles.remove(missile)

    def cooldown(self) -> None:
        if self.cooldown_counter >= self.COOLDOWN:
            self.cooldown_counter = 0
        elif self.cooldown_counter > 0:
            self.cooldown_counter += 1

    def shoot(self) -> None:
        if self.cooldown_counter == 0:
            missile = Missile(self.x + self.get_width() / 2 - 5 / 2,
                              self.y, self.missile_img)  # missile spawn with offset
            self.missiles.append(missile)
            self.cooldown_counter = 1

    def get_width(self) -> int:
        return self.ship_img.get_width()

    def get_height(self) -> int:
        return self.ship_img.get_height()


class Player(Ship):

    def __init__(self, x: int, y: int, lives: int = 3) -> None:
        super().__init__(x, y, lives)
        self.ship_img = SPACESHIP
        self.missile_img = PLAYER_MISSILE
        self.mask = pygame.mask.from_surface(self.ship_img)
        self.max_lives = lives
        self.score = 0

    def move_missiles(self, vel: int, objs: list) -> None:
        self.cooldown()
        for missile in self.missiles:
            missile.move(vel)
            if missile.off_screen(HEIGHT):
                self.missiles.remove(missile)
                self.score -= 10
            else:
                for obj in objs:
                    if missile.collision(obj):
                        objs.remove(obj)

                        # different scores for different colored enemies
                        if obj.color == "lime":
                            self.score += 10
                        elif obj.color == "cyan":
                            self.score += 20
                        elif obj.color == "magenta":
                            self.score += 30
                        if missile in self.missiles:
                            self.missiles.remove(missile)


class Enemy(Ship):
    COLOR_MAP = {
        "magenta": ENEMY_1,
        "cyan": ENEMY_2,
        "lime": ENEMY_3,
    }

    def __init__(self, x: int, y: int, color: str) -> None:
        super().__init__(x, y)
        self.missile_img = ENEMY_MISSILE
        self.ship_img = self.COLOR_MAP[color]
        self.mask = pygame.mask.from_surface(self.ship_img)
        self.color = color

    def move(self, vel_x: int, vel_y: int = 0) -> None:
        self.x += vel_x
        self.y += vel_y


def main() -> None:
    FPS = 60
    run = True
    lost = False
    lost_count = 0
    level = 0

    enemies = []
    player_vel = 7
    player_missile_vel = 15
    enemy_x_vel = 2
    enemy_y_vel = 2
    vel_x = enemy_x_vel
    enemy_missile_vel = 6
    last_enemy_shot = 0

    player = Player(WIDTH / 2, 650)

    clock = pygame.time.Clock()

    def redraw_window() -> None:
        WINDOW.blit(BACKGROUND, (0, 0))
        # draw text
        lives_label = set_font(40).render(f"Lives: {player.lives}", 1, WHITE)
        score_label = set_font(40).render(f"Score {player.score}", 1, WHITE)
        level_label = set_font(40).render(f"Level: {level}", 1, WHITE)
        WINDOW.blit(lives_label, (10, 10))
        WINDOW.blit(score_label, (10, lives_label.get_height() + 10))
        WINDOW.blit(level_label, (WIDTH - level_label.get_width() - 10, 10))

        for enemy in enemies:
            enemy.draw(WINDOW)

        player.draw(WINDOW)

        if lost:
            lost_label = set_font(60).render("You lost!", 1, RED)
            WINDOW.blit(lost_label, (WIDTH / 2 -
                        lost_label.get_width() / 2, 350))

        pygame.display.update()

    while run:
        clock.tick(FPS)
        redraw_window()

        if player.lives <= 0:
            lost = True
            lost_count += 1

        # stop game
        if lost:
            if lost_count > FPS * 3:  # wait for 3 sec
                run = False
            else:
                continue

        # spawn enemies
        if len(enemies) == 0:
            level += 1
            margin = 75
            for x in range(margin, WIDTH - margin, margin):
                for y in range(margin, int(HEIGHT / 2), margin):
                    if y == margin:  # top row
                        color = "magenta"
                    elif y == 2 * margin:  # rows 2-3
                        color = "cyan"
                    elif y == 4 * margin:  # rows 4-5
                        color = "lime"
                    enemy = Enemy(x, y, color)
                    enemies.append(enemy)

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

        keys = pygame.key.get_pressed()

        # move left
        if (keys[pygame.K_a] or keys[pygame.K_LEFT]) and (player.x - player_vel > 0):
            player.x -= player_vel
        # move right
        if (keys[pygame.K_d] or keys[pygame.K_RIGHT]) and (player.x + player_vel + player.get_width() < WIDTH):
            player.x += player_vel
        # shoot
        if keys[pygame.K_SPACE]:
            player.shoot()

        # enemies action
        for enemy in enemies[:]:
            if enemy.x >= WIDTH - enemy.get_width():
                move(0, enemy_y_vel, enemies)
                vel_x = -enemy_x_vel

            elif enemy.x <= 0:
                move(0, enemy_y_vel, enemies)
                vel_x = enemy_x_vel

            enemy.move_missiles(enemy_missile_vel, player)

            if collide(enemy, player) or (enemy.y + enemy.get_height() > HEIGHT):
                player.score -= 10
                player.lives -= 1
                enemies.remove(enemy)

        move(vel_x, 0, enemies)

        if pygame.time.get_ticks() - last_enemy_shot > 2000:
            choice(enemies).shoot()
            last_enemy_shot = pygame.time.get_ticks()

        player.move_missiles(-player_missile_vel, enemies)


# lambda functions
def set_font(size, font=FONT): return pygame.font.Font(
    font, size)  # sets font size


def collide(obj1, obj2): return obj1.mask.overlap(obj2.mask, (obj2.x -
                                                              obj1.x, obj2.y - obj1.y)) != None  # checks if 2 objs collide/overlap


def move(vel_x: int, vel_y: int, enemies: list) -> None:
    for enemy in enemies:
        enemy.move(vel_x, vel_y)


def main_menu() -> None:
    while True:
        WINDOW.blit(BACKGROUND, (0, 0))
        title_label = set_font(50).render(
            "Press any key to start...", 1, WHITE)
        WINDOW.blit(title_label, (WIDTH / 2 - title_label.get_width() /
                    2, HEIGHT / 2 - title_label.get_height() / 2))
        pygame.display.update()
        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                quit()
            if event.type == pygame.KEYDOWN:
                main()


if __name__ == "__main__":
    main_menu()