From 04dabe409155cb2119044223c7a8e9820888a117 Mon Sep 17 00:00:00 2001
From: Wuzzy <Wuzzy@disroot.org>
Date: Fri, 14 Jul 2023 16:12:52 +0200
Subject: [PATCH] Chess: Add castling to possible moves detection

---
 src/chess.lua | 33 ++++++++++++++++++---------------
 1 file changed, 18 insertions(+), 15 deletions(-)

diff --git a/src/chess.lua b/src/chess.lua
index 89dfc4c..a8654fd 100644
--- a/src/chess.lua
+++ b/src/chess.lua
@@ -589,23 +589,26 @@ local function get_theoretical_moves_from(meta, board, from_idx)
 
 		-- KING
 		elseif piece == "king" then
-			local dx = from_x - to_x
-			local dy = from_y - to_y
-
-			if dx < 0 then
-				dx = -dx
-			end
-
-			if dy < 0 then
-				dy = -dy
-			end
-
-			if dx > 1 or dy > 1 then
-				moves[to_idx] = nil
-			end
-
 			if attacked(color, xy_to_index(to_x, to_y), board) then
 				moves[to_idx] = nil
+			else
+				local dx = from_x - to_x
+				local dy = from_y - to_y
+
+				if dx < 0 then
+					dx = -dx
+				end
+
+				if dy < 0 then
+					dy = -dy
+				end
+
+				if dx > 1 or dy > 1 then
+					local cc = can_castle(meta, board, "board", from_idx, to_idx)
+					if not cc then
+						moves[to_idx] = nil
+					end
+				end
 			end
 		end
 	end