From 99d2a835b4f032de78d9cdbe27521363a25204bd Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Wed, 15 Jul 2020 15:30:52 +0530
Subject: display view mode in statusline

---
 src/habit/prelude.rs | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

(limited to 'src/habit')

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
 }
-- 
cgit v1.2.3