From a6260deefd468fc1859cb6fa2536536f1adb3abd Mon Sep 17 00:00:00 2001
From: Danila Fedorin <danila.fedorin@gmail.com>
Date: Sat, 26 May 2018 00:19:14 -0700
Subject: [PATCH] Add removing pieces.

---
 src/Go/Game.cr | 55 ++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 53 insertions(+), 2 deletions(-)

diff --git a/src/Go/Game.cr b/src/Go/Game.cr
index f388ac6..2d0018a 100644
--- a/src/Go/Game.cr
+++ b/src/Go/Game.cr
@@ -56,9 +56,60 @@ module Go
             end
         end
 
+        private def count_neighbors(x, y, color, visited)
+            if visited.includes?({x, y}) || (x < 0 || x >= @size.value || y < 0 || y >= @size.value)
+                return 0
+            else
+                visited.push({x, y})
+                case @board[{x, y}]?
+                when color
+                    return count_neighbors(x - 1, y, color, visited) + 
+                        count_neighbors(x + 1, y, color, visited) +
+                        count_neighbors(x, y - 1, color, visited) +
+                        count_neighbors(x, y + 1, color, visited)
+                when nil
+                    return 1
+                else
+                    return 0
+                end
+            end
+            return 0
+        end
+
+        private def remove_color(x, y, color)
+            if !(x < 0 || x >= @size.value || y < 0 || y >= @size.value) && @board[{x, y}]? == color
+                @board.delete({x, y})
+                remove_color(x - 1, y, color)
+                remove_color(x + 1, y, color)
+                remove_color(x, y - 1, color)
+                remove_color(x, y + 1, color)
+            end
+        end
+
+        private def try_remove_branch(x, y, color)
+            if @board[{x, y}]? == color
+                neighbor_count = count_neighbors(x, y, color, [] of Tuple(Int8, Int8))
+                if neighbor_count == 0
+                    remove_color(x, y, color)
+                end
+            end
+        end
+
+        def invert(color)
+            color == Color::Black ? Color::White : Color::Black
+        end
+
         def update(x, y, color)
-            @board[{x, y}] = color
-            @turn = @turn == Color::Black ? Color::White : Color::Black
+            if @turn == color
+                @board[{x, y}] = color
+                new_color = invert(color)
+                try_remove_branch(x - 1, y, new_color)
+                try_remove_branch(x + 1, y, new_color)
+                try_remove_branch(x, y - 1, new_color)
+                try_remove_branch(x, y + 1, new_color)
+                try_remove_branch(x, y, color)
+                @turn = new_color
+            end
         end
 
         private def color_char(color)