Add exmine mode

[naja.git] / naja / widgets / board.py

diff --git a/naja/widgets/board.py b/naja/widgets/board.py
index c5eef2ac50ae00784e120fc2a70fc180fa5a7733..aa608cef8673c8a4df1b19739b471bbf064c4d25 100644 (file)
--- a/naja/widgets/board.py
+++ b/naja/widgets/board.py
@@ -1,8 +1,11 @@
 """
 Widget that holds the game tiles.
 """
+import pygame.locals as pgl
+
+from naja.constants import BOARD_SIZE, TILE_SIZE, KEYS, MOVE, ACT
+from naja.events import finish_event
 
-from naja.constants import BOARD_SIZE, TILE_SIZE
 from naja.widgets.base import Widget
 from naja.widgets.tile import TileWidget
 
@@ -11,9 +14,11 @@ class BoardWidget(Widget):
     """
     Widget which holds all the tiles that make up the gameboard.
     """
-    def __init__(self, pos, state):
+    def __init__(self, pos, state, info):
         super(BoardWidget, self).__init__(pos, BOARD_SIZE)
-        # FIXME: Placeholder logic
+        self.info = info
+        self.state = state
+        self.pos = (2, 2)
         self._tiles = []
         for y in range(0, 5):
             for x in range(0, 5):
@@ -23,9 +28,42 @@ class BoardWidget(Widget):
 
     def prepare(self):
         for tile in self._tiles:
+            tile.set_highlight(self.pos)
             tile.prepare()
         self.size = BOARD_SIZE
+        if self.state.gameboard.player_mode in (ACT, MOVE):
+            self.pos = self.state.player.position
 
     def draw(self, surface):
         for tile in self._tiles:
             tile.draw(surface)
+
+    def change_pos(self, offset):
+        tpos = (offset[0] + self.pos[0], offset[1] + self.pos[1])
+        if tpos[0] < 0 or tpos[1] < 0 or tpos[0] > 4 or tpos[1] > 4:
+            return False
+        self.pos = tpos
+        self.info.set_position(self.pos)
+        return True
+
+    def handle_event(self, ev):
+        if self.state.gameboard.player_mode in (ACT, MOVE):
+            return super(BoardWidget, self).handle_event(ev)
+        if ev.type == pgl.KEYDOWN:
+            if ev.key in KEYS.UP:
+                if self.change_pos((0, -1)):
+                    return finish_event()
+            if ev.key in KEYS.DOWN:
+                if self.change_pos((0, +1)):
+                    return finish_event()
+            if ev.key in KEYS.LEFT:
+                if self.change_pos((-1, 0)):
+                    return finish_event()
+            if ev.key in KEYS.RIGHT:
+                if self.change_pos((+1, 0)):
+                    return finish_event()
+            if ev.key in KEYS.SWITCH:
+                self.state.gameboard.change_mode(MOVE)
+                self.info.set_position(self.state.player.position)
+                return finish_event()
+        return super(BoardWidget, self).handle_event(ev)