From b5365f8a82c9ead4030feaa70727d7a50de8ec69 Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Wed, 27 Jul 2022 23:40:38 +0200
Subject: [PATCH] Add FEN en-passant target square parsing

---
 src/fen.rs | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/fen.rs b/src/fen.rs
index 4ee320d..4118b59 100644
--- a/src/fen.rs
+++ b/src/fen.rs
@@ -1,4 +1,4 @@
-use crate::board::Color;
+use crate::board::{Color, File, Rank, Square};
 
 /// A trait to mark items that can be converted from a FEN input.
 pub trait FromFen: Sized {
@@ -38,3 +38,20 @@ impl FromFen for Color {
         Ok(res)
     }
 }
+
+/// Convert an en-passant target square segment of a FEN string to an optional [Square].
+impl FromFen for Option<Square> {
+    type Err = FenError;
+
+    fn from_fen(s: &str) -> Result<Self, Self::Err> {
+        let res = match s.as_bytes() {
+            [b'-'] => None,
+            [file @ b'a'..=b'h', rank @ b'1'..=b'8'] => Some(Square::new(
+                File::from_index((file - b'a') as usize),
+                Rank::from_index((rank - b'1') as usize),
+            )),
+            _ => return Err(FenError::InvalidFen),
+        };
+        Ok(res)
+    }
+}