import pygame
import time
import random

pygame.init()

# Defining colors
white = (255, 255, 255)
black = (0, 0, 0)
red = (255, 0, 0)
green = (0, 255, 0)
blue = (0, 0, 255)

# Setting the "Surface" (IE game window)
display_width = 800
display_height = 600
gameDisplay = pygame.display.set_mode((display_width, display_height))
pygame.display.set_caption('Slither')

# Defining some game constants
font = pygame.font.SysFont(None, 25)
clock = pygame.time.Clock()
fps = 15


def snake(lead_x, lead_y, player_width, player_height):  # Snake rendering/logic(someday)
    pygame.draw.rect(gameDisplay, green, [lead_x, lead_y, player_width, player_height])


def message_to_screen(msg, color):  # Display text in center of screen
    screen_text = font.render(msg, True, color)
    gameDisplay.blit(screen_text, [display_width / 2, display_height / 2])


def game_loop():  # Actual main loop

    # Defining some game variables
    game_exit = False
    game_over = False
    player_height = 10
    player_width = 10
    apple_size = 10
    lead_x = display_width / 2
    lead_y = display_height / 2
    lead_x_change = 0
    lead_y_change = 0
    rand_apple_x = round(random.randrange(0, display_width - apple_size) / 10.0) * 10.0
    rand_apple_y = round(random.randrange(0, display_height - apple_size) / 10.0) * 10.0

    while not game_exit:  # Main Game loop

        while game_over:  # Game Over menu
            gameDisplay.fill(white)
            message_to_screen("Game over, press C to play again or Q to quit", red)
            pygame.display.update()
            for event in pygame.event.get():  # Event checking for game over
                if event.type == pygame.KEYDOWN:
                    if event.key == pygame.K_c:
                        game_loop()
                    elif event.key == pygame.K_q:
                        game_exit = True
                        game_over = False

        for event in pygame.event.get():  # Checking for events
            if event.type == pygame.QUIT:
                game_exit = True
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_LEFT:
                    lead_x_change = -player_width
                    lead_y_change = 0
                if event.key == pygame.K_RIGHT:
                    lead_x_change = player_width
                    lead_y_change = 0
                if event.key == pygame.K_UP:
                    lead_y_change = -player_height
                    lead_x_change = 0
                if event.key == pygame.K_DOWN:
                    lead_y_change = player_height
                    lead_x_change = 0

        if lead_x > display_width - player_width or lead_x < 0 or lead_y > display_height - player_height or lead_y < 0:  # Check if player is within bounds
            game_over = True

        # Moving Player
        lead_x += lead_x_change
        lead_y += lead_y_change

        # Drawing Game - - - -
        gameDisplay.fill(white)
        pygame.draw.rect(gameDisplay, red, [rand_apple_x, rand_apple_y, apple_size, apple_size])
        snake(lead_x, lead_y, player_width, player_height)
        pygame.display.update()  # Render all changes to graphics since last frame

        if lead_x == rand_apple_x and lead_y == rand_apple_y:  # Snake Apple collision detection
            print("om nom nom")
            rand_apple_x = round(random.randrange(0, display_width - apple_size) / 10.0) * 10.0
            rand_apple_y = round(random.randrange(0, display_height - apple_size) / 10.0) * 10.0

        clock.tick(fps)

    pygame.quit()
    quit()


game_loop()