#pgzero
import random

WIDTH = 600
HEIGHT = 450

TITLE = "Космическое путешествие"
FPS = 30

# Объекты и переменные
ship = Actor("ship", (300, 400))
space = Actor("space")
enemies = []
planets = [Actor("plan1", (random.randint(0, 600), -100)), Actor("plan2", (random.randint(0, 600), -100)), Actor("plan3", (random.randint(0, 600), -100))]
meteors = []
bullets = []
mode = 'menu'
type1 = Actor('ship1', (100, 200))
type2 = Actor('ship2', (300, 200))
type3 = Actor('ship3', (500, 200))

# Заполнение списка врагов
for i in range(5):
    x = random.randint(0, 600)
    y = random.randint(-450, -50)
    enemy = Actor("enemy", (x, y))
    enemy.speed = random.randint(2, 8)
    enemies.append(enemy)

# Заполнение списка метеоритов
for i in range(5):
    x = random.randint(0, 600)
    y = random.randint(-450, -50)
    meteor = Actor("meteor", (x, y))
    meteor.speed = random.randint(2, 10)
    meteors.append(meteor)

# Отрисовка
def draw():
    # Режим игры
    if mode == 'game':
        space.draw()
        planets[0].draw()
        # Отрисовка метеоритов
        for i in range(len(meteors)):
            meteors[i].draw()
        ship.draw()
        # Отрисовка врагов
        for i in range(len(enemies)):
            enemies[i].draw()
        for bul in bullets:
            bul.draw()
    # Окно проигрыша
    elif mode == 'end':
        space.draw()
        screen.draw.text("GAME OVER!", center = (300, 200), color = "white", fontsize = 36)
    elif mode =='menu':
        space.draw()
        type1.draw()
        type2.draw()
        type3.draw()


# Управление
def on_mouse_move(pos):
    ship.pos = pos

def on_mouse_down(button, pos):
    global mode, ship, bullets
    if mode == 'menu' and button == mouse.LEFT:
        if type1.collidepoint(pos):
            ship.image = 'ship1'
            mode = 'game'
        if type2.collidepoint(pos):
            ship.image = 'ship2'
            mode = 'game'
        if type3.collidepoint(pos):
            ship.image = 'ship3'
            mode = 'game'
    else:
        bullet = Actor('missiles', pos)
        bullet.speed = 15
        bullets.append(bullet)



# Добавление в список нового врага
def new_enemy():
    x = random.randint(0, 400)
    y = -50
    enemy = Actor("enemy", (x, y))
    enemy.speed = random.randint(2, 8)
    enemies.append(enemy)

# Движение врагов
def enemy_ship():
    for i in range(len(enemies)):
        if enemies[i].y < 650:
            enemies[i].y = enemies[i].y + enemies[i].speed
        else:
            enemies.pop(i)
            new_enemy()

# Движение планет
def planet():
    if planets[0].y < 550:
            planets[0].y = planets[0].y + 1
    else:
        planets[0].y = -100
        planets[0].x = random.randint(0, 600)
        first = planets.pop(0)
        planets.append(first)

# Движение метеоритов
def meteorites():
    for i in range(len(meteors)):
        if meteors[i].y < 450:
            meteors[i].y = meteors[i].y + meteors[i].speed
        else:
            meteors[i].x = random.randint(0, 600)
            meteors[i].y = -20
            meteors[i].speed = random.randint(2, 10)

def bullets_fire():
    global bullets
    for bul in bullets:
        if bul.y > 0:
            bul.y -= bul.speed
        else:
            bullets.remove(bul)
            break
# Столкновения
def collisions():
    global mode
    for i in range(len(enemies)):
        if ship.colliderect(enemies[i]):
            mode = 'end'

def update(dt):
    if mode == 'game':
        enemy_ship()
        collisions()
        planet()
        meteorites()
        bullets_fire()