summaryrefslogtreecommitdiff
path: root/lib/domain/libimaghabit/src/store.rs
diff options
context:
space:
mode:
Diffstat (limited to 'lib/domain/libimaghabit/src/store.rs')
-rw-r--r--lib/domain/libimaghabit/src/store.rs63
1 files changed, 63 insertions, 0 deletions
diff --git a/lib/domain/libimaghabit/src/store.rs b/lib/domain/libimaghabit/src/store.rs
new file mode 100644
index 0000000..5d30c03
--- /dev/null
+++ b/lib/domain/libimaghabit/src/store.rs
@@ -0,0 +1,63 @@
+//
+// imag - the personal information management suite for the commandline
+// Copyright (C) 2015, 2016 Matthias Beyer <mail@beyermatthias.de> and contributors
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; version
+// 2.1 of the License.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+//
+
+use chrono::NaiveDate;
+
+use error::Result;
+use error::HabitError as HE;
+use habit::builder::HabitBuilder;
+use iter::HabitTemplateStoreIdIterator;
+use iter::HabitInstanceStoreIdIterator;
+
+use libimagstore::store::Store;
+use libimagstore::store::FileLockEntry;
+use libimagstore::storeid::StoreIdIterator;
+
+/// Extension trait for libimagstore::store::Store which is basically our Habit-Store
+pub trait HabitStore {
+
+ /// Create a new habit
+ fn create_habit(&self) -> HabitBuilder {
+ HabitBuilder::default()
+ }
+
+ /// Get an iterator over all habits
+ fn all_habit_templates(&self) -> Result<HabitTemplateStoreIdIterator>;
+
+ /// Get instances
+ fn all_habit_instances(&self) -> Result<HabitInstanceStoreIdIterator>;
+
+ // /// Get instances of a certain date
+ // fn all_habit_instances_on(&self, date: &NaiveDate) -> Result<HabitInstanceStoreIdIterator>;
+
+ // /// Get instances between two dates
+ // fn all_habit_instances_between(&self, start: &NaiveDate, end: &NaiveDate) -> Result<HabitInstanceStoreIdIterator>;
+}
+
+impl HabitStore for Store {
+ /// Get an iterator over all habits
+ fn all_habit_templates(&self) -> Result<HabitTemplateStoreIdIterator> {
+ self.entries().map(HabitTemplateStoreIdIterator::from).map_err(From::from)
+ }
+
+ fn all_habit_instances(&self) -> Result<HabitInstanceStoreIdIterator> {
+ self.entries().map(HabitInstanceStoreIdIterator::from).map_err(From::from)
+ }
+}
+