"""Habitat for editing levels."""

import pygame.display
from pygame.locals import SWSURFACE, KEYDOWN

from mamba.engine import Habitat, NewHabitatEvent
from mamba.widgets.level import EditLevelWidget
from mamba.widgets.text import TextWidget
from mamba.widgets.imagebutton import ImageButtonWidget
from mamba.level import Level, TILE_MAP
from mamba.constants import SCREEN, EDIT_SCREEN, NAME, ESCAPE_KEYS


class EditorHabitat(Habitat):
    def __init__(self, level_name):
        super(EditorHabitat, self).__init__(EDIT_SCREEN)
        self.level = Level(level_name)
        self.edit_widget = EditLevelWidget(self.level)
        self.container.add(self.edit_widget)
        self.container.add_callback(KEYDOWN, self.keydown_event)

    def on_enter(self):
        # We need to juggle the display to the correct size
        # This is a horrible hack
        pygame.display.quit()
        pygame.display.init()
        pygame.display.set_mode(EDIT_SCREEN, SWSURFACE)
        pygame.display.set_caption('%s Level editor' % NAME)
        super(EditorHabitat, self).on_enter()
        self.setup_toolbar()

    def on_exit(self):
        # We need to juggle the display to the correct size
        # This is a horrible hack
        super(EditorHabitat, self).on_exit()
        pygame.display.quit()
        pygame.display.init()
        pygame.display.set_mode(SCREEN, SWSURFACE)
        pygame.display.set_caption(NAME)

    def keydown_event(self, ev, widget):
        if ev.key in ESCAPE_KEYS:
            from mamba.habitats.mainmenu import MainMenu
            NewHabitatEvent.post(MainMenu())

    def setup_toolbar(self):
        """Draw the editor toolbar"""
        button_height = 20
        button_left = 820
        button_padding = 10

        levelname = TextWidget(
                (button_left, button_height),
                'Level: %s' % self.level.name, color='white')
        self.container.add(levelname)
        button_height += levelname.surface.get_height() + button_padding

        tilesetname = TextWidget(
                (button_left, button_height),
                'Tileset: %s' % self.level.tileset.name, color='white')
        self.container.add(tilesetname)
        button_height += tilesetname.surface.get_height() + button_padding
        # TODO: Add Image widget for the current tool
        self.current_tool = TextWidget((button_left, button_height),
                'Tool: Floor', color='white')
        self.container.add(self.current_tool)
        button_height += self.current_tool.surface.get_height()
        button_height += button_padding
        floor_button = ImageButtonWidget(
                (button_left, button_height), self.level.tileset.floor,
                'Floor', color='white')
        self.container.add(floor_button)
        floor_button.add_callback('clicked', self.change_tool,
                '.', 'Floor')
        button_height += floor_button.surface.get_height() + button_padding
        for tile_char in TILE_MAP:
            try:
                tile = self.level.tileset[tile_char]
            except pygame.error:
                # Ignore stuff we can't load for now
                continue
            if tile is None:
                continue
            if tile.name:
                text = tile.name
            else:
                text = 'Tile'
            tile_button = ImageButtonWidget(
                    (button_left, button_height), tile.image,
                    text, color='white')
            tile_button.add_callback('clicked', self.change_tool,
                    tile_char, text)
            self.container.add(tile_button)
            button_height += \
                    tile_button.surface.get_height() + button_padding

    def change_tool(self, ev, widget, new_tool, text):
        self.edit_widget.set_tool(new_tool)
        self.current_tool.text = 'Tool: %s' % text
        self.current_tool.prepare()