1 files changed, 15 insertions, 0 deletions
diff --git a/public/make_move.js b/public/make_move.js
index e00b999..6233ed0 100644
--- a/public/make_move.js
+++ b/public/make_move.js
@@ -17,6 +17,21 @@ function make_move(board, prev_y_x, new_y_x) {
     }
 
     for (let i = 0; i < prev_y_x.length; i++) {
+        let color = board[prev_y_x[i][0]][prev_y_x[i][1]] > 0
+
+        if(Math.abs(board[prev_y_x[i][0]][prev_y_x[i][1]]) == 1){
+            if(color) castle_white_short = castle_white_long = false
+            else castle_black_short = castle_black_long = false
+        } else if(Math.abs(board[prev_y_x[i][0]][prev_y_x[i][1]]) == 3){
+            if(prev_y_x[i][1] == 0){
+                if(color) castle_white_long = false
+                else castle_black_long = false
+            } else if (prev_y_x[i][1] == 7){
+                if(color) castle_white_short = false
+                else castle_black_short = false
+            }
+        }
+
         board[prev_y_x[i][0]][prev_y_x[i][1]] = 0
     }
 }