display view mode in statusline

1 files changed, 14 insertions, 1 deletions

diff --git a/src/habit/prelude.rs b/src/habit/prelude.rs
index b8b2bb2..19b00a7 100644
--- a/src/habit/prelude.rs
+++ b/src/habit/prelude.rs
@@ -1,4 +1,6 @@
 use serde::{Deserialize, Serialize};
+use std::default;
+use std::fmt;
 
 pub enum TrackEvent {
     Increment,
@@ -13,12 +15,23 @@ pub enum ViewMode {
     Year,
 }
 
-impl std::default::Default for ViewMode {
+impl default::Default for ViewMode {
     fn default() -> Self {
         ViewMode::Day
     }
 }
 
+impl fmt::Display for ViewMode {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        match self {
+            ViewMode::Day => write!(f, "DAY"),
+            ViewMode::Week => write!(f, "WEEK"),
+            ViewMode::Month => write!(f, "MONTH"),
+            ViewMode::Year => write!(f, "YEAR"),
+        }
+    }
+}
+
 pub fn default_auto() -> bool {
     false
 }