diff options
Diffstat (limited to 'src/modules/mod.rs')
-rw-r--r-- | src/modules/mod.rs | 11 |
1 files changed, 10 insertions, 1 deletions
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), } |