Use builder pattern in char section

1 files changed, 3 insertions, 7 deletions

diff --git a/src/modules/character.rs b/src/modules/character.rs
index 98b9b254e..9ebc8fa61 100644
--- a/src/modules/character.rs
+++ b/src/modules/character.rs
@@ -17,17 +17,13 @@ pub fn segment(args: &ArgMatches) -> Segment {
 
     let segment = Segment::new("char");
 
-    let color = if args.value_of("status_code").unwrap() == "0" {
+    if args.value_of("status_code").unwrap() == "0" {
         segment.set_style(COLOR_SUCCESS);
     } else {
-        COLOR_FAILURE
+        segment.set_style(COLOR_FAILURE);
     };
 
-    Segment {
-        value: String::from(PROMPT_CHAR),
-        style: Style::from(color),
-        ..Default::default()
-    }
+    segment
 }
 
 #[cfg(test)]