From 859ac5d3e49bc9a123df3f5a74b43d2281a3bed1 Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Wed, 15 Jul 2020 20:39:46 +0530
Subject: add utils for auto habits

auto trackable habits can be added, but not tracked as of this commit
---
 src/utils.rs | 28 +++++++++++++++++++---------
 1 file changed, 19 insertions(+), 9 deletions(-)

(limited to 'src/utils.rs')

diff --git a/src/utils.rs b/src/utils.rs
index 1d56377..e6ec6ac 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -34,13 +34,23 @@ pub fn load_configuration_file() -> AppConfig {
     };
 }
 
-pub fn data_file() -> PathBuf {
-    if let Some(proj_dirs) = ProjectDirs::from("rs", "nerdypepper", "dijo") {
-        let mut data_file = PathBuf::from(proj_dirs.data_dir());
-        fs::create_dir_all(&data_file);
-        data_file.push("habit_record.json");
-        return data_file;
-    } else {
-        panic!("Invalid home directory!")
-    };
+fn project_dirs() -> ProjectDirs {
+    ProjectDirs::from("rs", "nerdypepper", "dijo")
+        .unwrap_or_else(|| panic!("Invalid home directory!"))
+}
+
+pub fn habit_file() -> PathBuf {
+    let proj_dirs = project_dirs();
+    let mut data_file = PathBuf::from(proj_dirs.data_dir());
+    fs::create_dir_all(&data_file);
+    data_file.push("habit_record.json");
+    return data_file;
+}
+
+pub fn auto_habit_file() -> PathBuf {
+    let proj_dirs = project_dirs();
+    let mut data_file = PathBuf::from(proj_dirs.data_dir());
+    fs::create_dir_all(&data_file);
+    data_file.push("habit_record[auto].json");
+    return data_file;
 }
-- 
cgit v1.2.3