From db78ff212da7abed937b03b120b1cf1afba2f37f Mon Sep 17 00:00:00 2001
From: Loic Guegan <manzerbredes@mailbox.org>
Date: Tue, 10 Jan 2023 17:07:27 +0100
Subject: [PATCH] Add plus sign to SAN moves when player is in check

---
 src/ChessArbiter.cpp   |  5 +++++
 tests/chessarbiter.cpp | 11 +++++++++++
 2 files changed, 16 insertions(+)

diff --git a/src/ChessArbiter.cpp b/src/ChessArbiter.cpp
index 593ece2..fdda48d 100644
--- a/src/ChessArbiter.cpp
+++ b/src/ChessArbiter.cpp
@@ -178,6 +178,11 @@ bool ChessArbiter::Play(const std::string &move, char promote) {
       return (false);
     }
 
+    // Don't forget the plus sign on the SAN move
+    if(IsCheck(fen.player)){
+      SAN+="+";
+    }
+
     // Update position map (repetitions draw)
     if (positions.count(fen.board) == 0) {
       positions[fen.board] = 1;
diff --git a/tests/chessarbiter.cpp b/tests/chessarbiter.cpp
index a6ee016..931b545 100644
--- a/tests/chessarbiter.cpp
+++ b/tests/chessarbiter.cpp
@@ -513,4 +513,15 @@ TEST_CASE("Specific bugs found on a game", "[BugFixes]") {
   CHECK(a.ParseSAN("Ne2") == "g1e2");
   a.Play("g1e2");
   CHECK(a.GetFEN() == "r1bqk2r/pp1n1p2/3p4/1BpP2pp/1b2n2P/2N1P1B1/PP2NPP1/R2QK2R b KQkq - 4 12");
+
+  // Bug 6 SAN move that lead to check must have a + sign on the SAN moves
+  a.Setup("rnb1kbnr/pppp1ppp/3qP3/8/8/8/PPP1PPPP/RNB1KBNR b KQkq - 0 4");
+  a.Play("d6d1");
+  CHECK(a.IsCheck(false));
+  CHECK(!a.IsCheck(!false));
+  CHECK(a.GetSAN()=="Qd1+");
+  a.Setup("rnb1kbnr/pppp1ppp/3qP3/8/8/8/PPP1PPPP/RNB1KBNR b KQkq - 0 4");
+  a.Play("d6d3");
+  CHECK(!a.IsCheck(false));
+  CHECK(a.GetSAN()=="Qd3");
 }
\ No newline at end of file