X-Git-Url: http://git.scottworley.com/pluta-lesnura/blobdiff_plain/aa0622ab28e4bf320db9a2ff747e5004923e2570..2d3998b9d91b9016dd41fa871a2e2d9e2c15180a:/src/lib.rs

diff --git a/src/lib.rs b/src/lib.rs
index 09bd5fb..444b3fe 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -181,6 +181,17 @@ impl Hand {
     fn random(&self) -> Option<&Card> {
         self.cards.choose(&mut rand::thread_rng())
     }
+    /// Make a new Hand that contains only cards of the requested suit
+    fn filter_by_suit(&self, suit: Suit) -> Self {
+        Self {
+            cards: self
+                .cards
+                .iter()
+                .filter(|c| c.suit().expect("I shouldn't have jokers in my hand") == suit)
+                .copied()
+                .collect(),
+        }
+    }
 }
 
 #[derive(Copy, Clone)]
@@ -428,6 +439,22 @@ pub fn random_player(draw_chance: f64) -> impl FnMut(&Game) -> Play {
     }
 }
 
+/// When available, make plays that grant momentum.  Otherwise, play randomly.
+pub fn momentum_player(draw_chance: f64) -> impl FnMut(&Game) -> Play {
+    let mut fallback = random_player(draw_chance);
+    move |game: &Game| -> Play {
+        match (&game.phase, game.discard.top().and_then(Card::suit)) {
+            (Phase::Play, Some(suit)) => {
+                match game.current_player_hand().filter_by_suit(suit).random() {
+                    Some(card) => Play::Play(*card),
+                    _ => fallback(game),
+                }
+            }
+            _ => fallback(game),
+        }
+    }
+}
+
 /// # Errors
 ///
 /// Will return `Err` on invalid plays, like trying to draw during Play phase,