From 7683f33bc870cd928e7d572fcdde65a1fd5eda0c Mon Sep 17 00:00:00 2001
From: Matan Kushner <hello@matchai.me>
Date: Thu, 4 Apr 2019 14:18:15 -0400
Subject: Add dir section with home truncation

---
 src/modules/mod.rs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'src/modules/mod.rs')

diff --git a/src/modules/mod.rs b/src/modules/mod.rs
index 869a82c2c..13b18fa13 100644
--- a/src/modules/mod.rs
+++ b/src/modules/mod.rs
@@ -1,4 +1,5 @@
 mod char;
+mod dir;
 
 use clap::ArgMatches;
 use ansi_term::Style;
@@ -12,11 +13,18 @@ pub struct Segment {
 
 impl Default for Segment {
     fn default() -> Segment {
+        let default_suffix = Some(Box::new(Segment {
+            style: Style::default(),
+            value: String::from(" "),
+            prefix: None,
+            suffix: None
+        }));
+
         Segment {
             style: Style::default(),
             value: String::from(""),
             prefix: None,
-            suffix: None
+            suffix: default_suffix
         }
     }
 }
@@ -24,6 +32,7 @@ impl Default for Segment {
 pub fn handle(module: &str, args: &ArgMatches) -> Segment {
     match module {
         "char" => char::segment(&args),
+        "dir" => dir::segment(&args),
 
         _ => panic!("Unknown module: {}", module),
     }
-- 
cgit v1.2.3