from time import sleep
import pygame,sys
import pygame_menu
from pygame_menu import themes

pygame.init()
surface = pygame.display.set_mode((600,400))

WHITE=(255,255,255)
RED=(255,0,0)
BLUE=(0,0,255)
GREEN=(0,255,0)
BLACK=(0,0,0)

def set_difficulty(value, difficulty):
    print(value)
    print(difficulty)

def start_the_game():
    mainmenu._open(loading)
    pygame.time.set_timer(update_loading, 30)

def level_menu():
    mainmenu._open(level)

mainmenu = pygame_menu.Menu("Welcome",600, 400, theme=themes.THEME_SOLARIZED)
mainmenu.add.text_input("Name: ", default="username")
mainmenu.add.button("Play", start_the_game)
mainmenu.add.button("Levels", level_menu)
mainmenu.add.button("Quit", pygame_menu.events.EXIT)

level = pygame_menu.Menu("Select a Difficulty", 600, 400, theme=themes.THEME_BLUE)
level.add.selector("Difficulty :", [("Hard", 1), ("Easy", 2)], onchange=set_difficulty)

loading = pygame_menu.Menu("Loading the Game...", 600, 400, theme=themes.THEME_DARK)
loading.add.progress_bar("Progress", progressbar_id = "1", default=0, width = 200)

arrow = pygame_menu.widgets.LeftArrowSelection(arrow_size = (10, 15))

update_loading = pygame.USEREVENT + 0

pygame.display.set_caption("Task0")


while True: 
    events = pygame.event.get()
    for event in events:
        if event.type == update_loading:
            progress = loading.get_widget("1")
            progress.set_value(progress.get_value() + 1)
            if progress.get_value() == 100:
                pygame.time.set_timer(update_loading, 0)

        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()
        
    if mainmenu.is_enabled():
        mainmenu.update(events)
        mainmenu.draw(surface)
        if(mainmenu.get_current().get_selected_widget()):
            arrow.draw(surface, mainmenu.get_current().get_selected_widget())
    
    pygame.display.update()