1 files changed, 24 insertions, 5 deletions

diff --git a/src/client/games/slots.rs b/src/client/games/slots.rs
index 39c43b2..c367fcb 100644
--- a/src/client/games/slots.rs
+++ b/src/client/games/slots.rs
@@ -13,22 +13,40 @@ impl SlotsGame {
         Self
     }
 
-    fn cell(chars: &[char], row: usize, col: usize) -> char {
-        if chars.len() == REELS * ROWS {
+    fn cell(chars: &[char], row: usize, col: usize, winning: &[usize]) -> String {
+        let symbol = if chars.len() == REELS * ROWS {
             chars.get(col * ROWS + row).copied().unwrap_or('?')
         } else if chars.len() == 3 && row == 0 {
             // Legacy 3-symbol server: degrade to the first row.
             chars.get(col).copied().unwrap_or('?')
         } else {
             '?'
+        };
+
+        if winning.contains(&(col * ROWS + row)) {
+            format!("{{ {} }}", symbol)
+        } else {
+            format!("[ {} ]", symbol)
         }
     }
 
-    fn render_grid(&self, reels: &str) -> [String; ROWS] {
+    fn highlighted_cells(state: &GameState) -> Vec<usize> {
+        state
+            .data
+            .get("win_cells")
+            .map(|raw| {
+                raw.split(',')
+                    .filter_map(|value| value.trim().parse::<usize>().ok())
+                    .collect()
+            })
+            .unwrap_or_default()
+    }
+
+    fn render_grid(&self, reels: &str, winning: &[usize]) -> [String; ROWS] {
         let chars: Vec<char> = reels.chars().collect();
         std::array::from_fn(|row| {
             (0..REELS)
-                .map(|col| format!("[ {} ]", Self::cell(&chars, row, col)))
+                .map(|col| Self::cell(&chars, row, col, winning))
                 .collect::<Vec<_>>()
                 .join(" ")
         })
@@ -84,7 +102,8 @@ impl Game for SlotsGame {
             .get("balance")
             .map(String::as_str)
             .unwrap_or("?");
-        let grid = self.render_grid(&reels);
+        let winning = Self::highlighted_cells(state);
+        let grid = self.render_grid(&reels, &winning);
 
         format!(
             "=== SLOTS ===\r\nBet: {} PLN   Lines: {}   Balance: {} PLN\r\n{}\r\n{}\r\n{}\r\n\r\n{}\r\n{}\r\n[+/-] Bet   [SPACE] Spin   [Q] Quit",