"""Level menu."""

from pygame.locals import KEYDOWN

from mamba.constants import ESCAPE_KEYS
from mamba.engine import Habitat, NewHabitatEvent, HabitatSetupEvent
from mamba.gamestate import done_levels
from mamba.data import get_official_levels
from mamba.level import Level
from mamba.widgets.base import GridContainer
from mamba.widgets.levelbutton import LevelButton


OFFICIAL_LEVELS = []
LEVEL_CACHE = {}


class LevelMenu(Habitat):

    WIDTH = 5

    def __init__(self):
        super(LevelMenu, self).__init__()
        self.level_buttons = {}
        self.grid_container = GridContainer(width=self.WIDTH)
        self.container.add(self.grid_container)
        self.container.add_callback(KEYDOWN, self.keydown_event)

    def on_enter(self):
        super(LevelMenu, self).on_enter()
        self._setup_levels()

    def dispatch(self, ev):
        if HabitatSetupEvent.matches(ev):
            self._setup_levels(ev)
            return
        super(LevelMenu, self).dispatch(ev)

    def _setup_levels(self, ev=None):
        if ev is not None:
            self.setup_levels()
            # We now have widgets, let's give them focus
            self.container.adjust_focus(1)
        else:
            HabitatSetupEvent.post()

    def setup_level(self, i, name):
        level = self.get_level(name)
        button = LevelButton((20 + 120 * (i % self.WIDTH),
                              20 + 120 * (i // self.WIDTH)),
                             level)
        button.add_callback('clicked', self.level_selected, name)
        button.done = level.unique_name() in done_levels
        self.grid_container.add(button)
        self.level_buttons[name] = button

    def setup_levels(self):
        for i, name in enumerate(self.list_levels()):
            self.setup_level(i, name)

    def list_levels(self):
        raise NotImplementedError()

    def get_level(self, name):
        raise NotImplementedError()

    def level_selected(self, ev, widget, name):
        from mamba.habitats.level import LevelHabitat
        next_level = self.mk_next_level(name)
        NewHabitatEvent.post(
            LevelHabitat(widget.level, self.go_menu, next_level))
        return True

    def mk_next_level(self, name):
        return self.go_menu

    @classmethod
    def go_menu(cls):
        NewHabitatEvent.post(cls())

    def keydown_event(self, ev, widget):
        if ev.key in ESCAPE_KEYS:
            return self.return_to_mainmenu(ev, widget)

    def return_to_mainmenu(self, ev, widget):
        from mamba.habitats.mainmenu import MainMenu
        NewHabitatEvent.post(MainMenu())
        return True


class OfficialLevelMenu(LevelMenu):

    level_namespace = 'official'

    def mk_next_level(self, name):
        levels = self.list_levels()
        if name not in levels:
            return self.go_menu
        idx = levels.index(name) + 1
        if idx >= len(levels):
            return self.go_menu
        next_name = levels[idx]

        def next_level():
            from mamba.habitats.level import LevelHabitat
            next_level = self.mk_next_level(next_name)
            NewHabitatEvent.post(LevelHabitat(self.get_level(next_name),
                                              self.go_menu, next_level))

        print "next_level end"
        return next_level

    def list_levels(self):
        if not OFFICIAL_LEVELS:
            OFFICIAL_LEVELS.extend(get_official_levels())
        return OFFICIAL_LEVELS[:]

    def get_level(self, name):
        if name not in LEVEL_CACHE:
            LEVEL_CACHE[name] = Level(name, self.level_namespace)
        return LEVEL_CACHE[name]