import pygame
import random
from planet import generate_planet_surface
from space import draw_star_map

pygame.init()
WIDTH, HEIGHT = 800, 600
screen = pygame.display.set_mode((WIDTH, HEIGHT))
clock = pygame.time.Clock()

mode = "space"  # or 'planet'
planet_seed = None

running = True
while running:
    screen.fill((0, 0, 0))
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

        # Press SPACE to land on a planet
        if event.type == pygame.KEYDOWN:
            if event.key == pygame.K_SPACE and mode == "space":
                planet_seed = random.randint(0, 999999)
                mode = "planet"
            elif event.key == pygame.K_BACKSPACE and mode == "planet":
                mode = "space"

    if mode == "space":
        draw_star_map(screen)
    elif mode == "planet":
        generate_planet_surface(screen, planet_seed)

    pygame.display.flip()
    clock.tick(60)

pygame.quit()