1 files changed, 22 insertions, 0 deletions
diff --git a/src/model/time.rs b/src/model/time.rs
new file mode 100644
index 0000000..10cf6d3
--- /dev/null
+++ b/src/model/time.rs
@@ -0,0 +1,22 @@
+use chrono::{NaiveTime, Timelike};
+
+pub fn pprint(t: NaiveTime) -> String {
+    if t.minute() == 0 {
+        format!("{}h", t.hour())
+    } else {
+        format!("{}h{}", t.hour(), t.minute())
+    }
+}
+
+pub fn parse(t: &str) -> Option<NaiveTime> {
+    match t.split('h').collect::<Vec<&str>>()[..] {
+        [hours, minutes] => {
+            if minutes.trim().is_empty() {
+                NaiveTime::from_hms_opt(hours.parse().ok()?, 0, 0)
+            } else {
+                NaiveTime::from_hms_opt(hours.parse().ok()?, minutes.parse().ok()?, 0)
+            }
+        }
+        _ => None,
+    }
+}