"""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, TextButton
from mamba.widgets.imagebutton import ImageButtonWidget
#from mamba.widgets.messagebox import MessageBox
from mamba.widgets.entrybox import EntryBox
from mamba.level import Level, TILE_MAP, THING_MAP
from mamba.data import check_level_exists
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)
        self.mode = 'Tile'

    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 = 5

        filename = TextButton(
                (button_left, button_height),
                'File: %s' % self.level.level_name, color='white')
        filename.add_callback('clicked', self.edit_filename)
        self.container.add(filename)
        button_height += filename.rect.height + button_padding

        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')
        self.edit_widget.set_tool('.')
        button_height += floor_button.surface.get_height() + button_padding
        if self.mode == 'Tile':
            tile_map = TILE_MAP
            change_mode_text = 'Switch to Things'
        elif self.mode == 'Thing':
            tile_map = THING_MAP
            change_mode_text = 'Switch to Tiles'
        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
        button_height += 10

        mode_button = TextButton((button_left, button_height),
                change_mode_text)
        mode_button.add_callback('clicked', self.change_toolbar)
        self.container.add(mode_button)
        button_height += mode_button.surface.get_height() + button_padding
        button_height += 10

        save_button = TextButton((button_left, button_height), "Save Level")
        save_button.add_callback('clicked', self.save)
        self.container.add(save_button)

    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()

    def save(self, ev, widget):
        self.level.save_level()

    def change_toolbar(self, ev, widget):
        if self.mode == 'Tile':
            self.mode = 'Thing'
        elif self.mode == 'Thing':
            self.mode = 'Tile'
        self.clear_toolbar()
        self.setup_toolbar()

    def clear_toolbar(self):
        """Remove every non-edit widget from the container"""
        for widget in self.container.children[:]:
            if widget is not self.edit_widget:
                self.container.remove(widget)

    def edit_filename(self, ev, widget):
        editbox = EntryBox((200, 200), 'Enter level filename',
                self.level.level_name, self.check_file)
        self.container.add(editbox)
        editbox.grab_focus()

    def check_file(self, new_name):
        if new_name == self.level.level_name:
            return True  # No-op change
        if check_level_exists(new_name):
            # TODO: Popup warning
            return False
        self.level.level_name = new_name
        self.clear_toolbar()
        self.setup_toolbar()