From 8f156616f0a59f07d5fdfbc19ed9fdb672da9056 Mon Sep 17 00:00:00 2001
From: Karma Riuk <riukkarma@gmail.com>
Date: Wed, 29 Jan 2025 12:08:03 +0100
Subject: [PATCH] pawns have their legal move list

---
 src/logic/board.py       | 10 ++++++++++
 src/logic/pieces/pawn.py | 32 ++++++++++++++++++++++++++++----
 2 files changed, 38 insertions(+), 4 deletions(-)

diff --git a/src/logic/board.py b/src/logic/board.py
index 451fbe0..7c79483 100644
--- a/src/logic/board.py
+++ b/src/logic/board.py
@@ -37,6 +37,16 @@ class Board:
             self._white[pos_w_piece] = piece(pos_w_piece, Piece.WHITE)
             self._black[pos_b_piece] = piece(pos_b_piece, Piece.BLACK)
 
+    def piece_at(self, x: int, y: int) -> Piece | None:
+        pos = Position(x, y)
+        white_piece = self._white.get(pos, None)
+        black_piece = self._black.get(pos, None)
+
+        assert white_piece == None or black_piece == None, f"There are two pieces at the same position {pos}, this shouldn't happen!"
+
+        if white_piece != None:
+            return white_piece
+        return black_piece
 
 def create_board():
     return Board()
diff --git a/src/logic/pieces/pawn.py b/src/logic/pieces/pawn.py
index 748e941..1be50b0 100644
--- a/src/logic/pieces/pawn.py
+++ b/src/logic/pieces/pawn.py
@@ -1,7 +1,31 @@
-from .piece import Piece
+from logic.position import Position
+from logic.pieces.piece import Piece
 
 class Pawn(Piece):
-    def __init__(self, pos) -> None:
-        super().__init__(pos)
-        self.already_moved = False
+    def legal_moves(self, board) -> list[Position]:
+        ret = []
 
+        # can we capture to the left?
+        if self.pos.x > 0 and (
+            (self.colour == self.WHITE and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y + 1)))
+            or
+            (self.colour == self.BLACK and (capturable_piece := board.piece_at(self.pos.x - 1, self.pos.y - 1)))
+        ):
+            if capturable_piece.colour != self.colour:
+                ret.append(capturable_piece.pos)
+
+        # can we capture to the right?
+        if self.pos.x < 7 and (
+            (self.colour == self.WHITE and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y + 1)))
+            or
+            (self.colour == self.BLACK and (capturable_piece := board.piece_at(self.pos.x + 1, self.pos.y - 1)))
+        ):
+            if capturable_piece.colour != self.colour:
+                ret.append(capturable_piece.pos)
+
+        for dy in range(1, 3 if self.pos.y == 1 else 2):
+            if self.pos.y + dy > 7 or board.piece_at(self.pos.x, self.pos.y + dy):
+                break
+            ret.append(Position(self.pos.x, self.pos.y + dy))
+
+        return ret