Change display to print robot health instead of a square

robot-rumble · Oct 26, 2020 · 3d869ac · 3d869ac
1 parent d5ee08c
commit 3d869ac
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/display.rs b/src/display.rs
@@ -10,9 +10,12 @@ pub fn display_state(state: &StateForOutput) {
                     let obj = state.objs.get(&id).unwrap();
                     match &obj.1 {
                         ObjDetails::Terrain(_) => "■".white(),
-                        ObjDetails::Unit(unit) => match unit.team {
-                            Team::Red => "r".white().on_red(),
-                            Team::Blue => "b".white().on_blue(),
+                        ObjDetails::Unit(unit) => {
+                            let health = unit.health.to_string();
+                            match unit.team {
+                                Team::Red => health.red(),
+                                Team::Blue => health.blue()
+                            }
                         },
                     }
                 }