"""Main menu."""

from pygame.constants import K_RETURN
from pygame.locals import MOUSEMOTION, KEYDOWN, QUIT
import pygame.event

from mamba.constants import ESCAPE_KEYS
from mamba.engine import Habitat
from mamba.widgets.text import TextButton


class MainMenu(Habitat):
    def __init__(self):
        super(MainMenu, self).__init__()
        hello = TextButton((100, 100), "Hello", color='white')
        self.container.add(hello)
        quit = TextButton((100, 200), "Quit", color='white')
        quit.add_callback(KEYDOWN, self.quit_keydown_event)
        self.container.add(quit)
        self.container.add_callback(KEYDOWN, self.keydown_event)

    def keydown_event(self, ev, widget):
        if ev.key in ESCAPE_KEYS:
            pygame.event.post(pygame.event.Event(QUIT))
            return True

    def quit_keydown_event(self, ev, widget):
        if ev.key == K_RETURN:
            pygame.event.post(pygame.event.Event(QUIT))
            return True