Add max bomb placement restriction

Signed-off-by: pacien <pacien.trangirard@pacien.net>

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main/java/fr/umlv/java/wallj/block/RobotBlock.java b/src/main/java/fr/umlv/java/wallj/block/RobotBlock.java
index da7f3cf..e4e0aad 100644
--- a/src/main/java/fr/umlv/java/wallj/block/RobotBlock.java
+++ b/src/main/java/fr/umlv/java/wallj/block/RobotBlock.java
@@ -23,10 +23,12 @@ import java.util.List;
  */
 public class RobotBlock extends Block {
   private static final float SPEED = 10f; // px/ms
+  private static final int MAX_BOMB_PLACEMENTS = 3;
 
   private Vec2 pos;
   private PathFinder pathFinder;
   private Deque<TileVec2> path = new LinkedList<>();
+  private int droppedBombCount = 0;
 
   RobotBlock(Vec2 pos) {
     super(BlockType.ROBOT);
@@ -62,10 +64,17 @@ public class RobotBlock extends Block {
     return Events.findFirst(events, BombSetupOrder.class)
            .map(event -> isOnBomb(stage) ?
                          Collections.<Event>singletonList(new BombTimerIncrEvent(getTile())) :
-                         Collections.<Event>singletonList(new BlockCreateEvent(BlockType.BOMB, getTile())))
+                         dropBomb(event))
            .orElse(Collections.emptyList());
   }
 
+  private List<Event> dropBomb(BombSetupOrder order) {
+    if (droppedBombCount >= MAX_BOMB_PLACEMENTS) return Collections.emptyList();
+
+    droppedBombCount++;
+    return Collections.singletonList(new BlockCreateEvent(BlockType.BOMB, getTile()));
+  }
+
   private void updatePath(Board board, TileVec2 target) {
     if (!board.inside(target) || !board.getBlockTypeAt(target).isTraversable()) return;
     if (pathFinder == null) pathFinder = new PathFinder(board);