Implemented actions

board_actions.py

@@ -5,14 +5,72 @@ import board
 
 
 @dataclass
-class MoveAction:
-    """Moving a card from one stack to another"""
-    card: board.Card
-    source_position: board.Position
+class GoalAction:
+    """Move card from field to goal"""
+    card: board.NumberCard
+    source_id: int
+    source_position: board.Position
+
+    def apply(self, action_board: board.Board) -> None:
+        """Do action"""
+        if self.source_position == board.Position.Field:
+            assert action_board.field[self.source_id][-1] == self.card
+            assert action_board.goal[self.card.suit] + 1 == self.card.number
+            action_board.field[self.source_id].pop()
+            action_board.goal[self.card.suit] += 1
+        elif self.source_position == board.Position.Bunker:
+            assert action_board.bunker[self.source_id] == self.card
+            assert action_board.goal[self.card.suit] + 1 == self.card.number
+            action_board.bunker[self.source_id] = None
+            action_board.goal[self.card.suit] += 1
+        else:
+            raise RuntimeError("Unknown position")
+
+    def undo(self, action_board: board.Board) -> None:
+        """Undo action"""
+        assert action_board.goal[self.card.suit] == self.card.number
+        if self.source_position == board.Position.Field:
+            action_board.field[self.source_id].append(self.card)
+        elif self.source_position == board.Position.Bunker:
+            assert action_board.bunker[self.source_id] is None
+            action_board.bunker[self.source_id] = self.card
+        else:
+            raise RuntimeError("Unknown position")
+        action_board.goal[self.card.suit] -= 1
+
+
+@dataclass
+class RestoreAction:
+    """Move card from bunker to field"""
+    card: board.Card
     source_id: int
-    destination_position: board.Position
     destination_id: int
 
+    def apply(self, action_board: board.Board) -> None:
+        """Do action"""
+        assert action_board.bunker[self.source_id] == self.card
+        action_board.bunker[self.source_id] = None
+
+
+
+@dataclass
+class StoreAction:
+    """Move card from field to bunker"""
+    card: board.Card
+    source_id: int
+    destination_id: int
+
+
+@dataclass
+class MoveAction:
+    """Moving a card from one field stack to another"""
+    card: board.Card
+    source_id: int
+    destination_id: int
+
+    def apply(self, action_board: board.Board) -> None:
+        """Do action"""
+
 
 @dataclass
 class DragonKillAction:
@@ -21,11 +79,57 @@ class DragonKillAction:
     source_stacks: List[Tuple[board.Position, int]]
     destination_bunker_id: int
 
+    def apply(self, action_board: board.Board) -> None:
+        """Do action"""
+        assert (action_board.bunker[self.destination_bunker_id] is None or
+                action_board.bunker[self.destination_bunker_id] == self.dragon)
+        assert len(self.source_stacks) == 4
+        for position, index in self.source_stacks:
+            if position == board.Position.Field:
+                assert action_board.field[index]
+                assert action_board.field[index][-1] == self.dragon
+                action_board.field[index].pop()
+            elif position == board.Position.Bunker:
+                assert action_board.bunker[index] == self.dragon
+                action_board.bunker[index] = None
+            else:
+                raise RuntimeError("Can only kill dragons in field and bunker")
+        action_board.bunker[self.destination_bunker_id] = board.KilledDragon(
+            self.dragon)
+
+    def undo(self, action_board: board.Board) -> None:
+        """Undo action"""
+        assert action_board.bunker[self.destination_bunker_id] == board.KilledDragon(
+            self.dragon)
+        assert len(self.source_stacks) == 4
+        for position, index in self.source_stacks:
+            if position == board.Position.Field:
+                action_board.field[index].append(self.dragon)
+            elif position == board.Position.Bunker:
+                action_board.bunker[index] = self.dragon
+            else:
+                raise RuntimeError("Can only kill dragons in field and bunker")
+        action_board.bunker[self.destination_bunker_id] = None
+
 
 @dataclass
 class HuaKillAction:
     """Remove the flower card"""
     source_field_id: int
 
+    def apply(self, action_board: board.Board) -> None:
+        """Do action"""
+        assert not action_board.flowerGone
+        assert action_board.field[self.source_field_id] == board.SpecialCard.Hua
+        action_board.field[self.source_field_id].pop()
+        action_board.flowerGone = True
 
-Action = Union[MoveAction, DragonKillAction, HuaKillAction]
+    def undo(self, action_board: board.Board) -> None:
+        """Undo action"""
+        assert action_board.flowerGone
+        action_board.field[self.source_field_id].append(board.SpecialCard.Hua)
+        action_board.flowerGone = False
+
+
+Action = Union[MoveAction, DragonKillAction,
+               HuaKillAction, StoreAction, RestoreAction, GoalAction]