"""Display a menu screen."""

import pygame

from nagslang.screens.base import Screen
from nagslang.events import QuitEvent, ScreenChange
from nagslang.widgets.text import TextWidget


class MenuScreen(Screen):

    def handle_event(self, ev):
        if ev.type == pygame.locals.KEYDOWN:
            if ev.key == pygame.locals.K_ESCAPE:
                QuitEvent.post()
            elif ev.key == pygame.locals.K_1:
                ScreenChange.post('level1', None)

    def render(self, surface):
        surface.fill(pygame.color.Color(255, 255, 255))
        TextWidget((10, 10), 'Menu').draw(surface)
        TextWidget((10, 30), 'Press 1 to start').draw(surface)