use std::cmp::Ordering; use libimagstore::store::Store; use libimagstore::storeid::IntoStoreId; use libimagerror::trace::trace_error; use chrono::offset::local::Local; use chrono::Datelike; use itertools::Itertools; use chrono::naive::datetime::NaiveDateTime; use entry::Entry; use diaryid::DiaryId; use error::DiaryError as DE; use error::DiaryErrorKind as DEK; use result::Result; use iter::DiaryEntryIterator; use is_in_diary::IsInDiary; #[derive(Debug)] pub struct Diary<'a> { store: &'a Store, name: &'a str, } impl<'a> Diary<'a> { pub fn open(store: &'a Store, name: &'a str) -> Diary<'a> { Diary { store: store, name: name, } } // create or get a new entry for today pub fn new_entry_today(&self) -> Result { let dt = Local::now(); let ndt = dt.naive_local(); let id = DiaryId::new(String::from(self.name), ndt.year(), ndt.month(), ndt.day(), 0, 0); self.new_entry_by_id(id) } pub fn new_entry_by_id(&self, id: DiaryId) -> Result { self.retrieve(id.with_diary_name(String::from(self.name))) } pub fn retrieve(&self, id: DiaryId) -> Result { id.into_storeid() .and_then(|id| self.store.retrieve(id)) .map(|fle| Entry::new(fle)) .map_err(|e| DE::new(DEK::StoreWriteError, Some(Box::new(e)))) } // Get an iterator for iterating over all entries pub fn entries(&self) -> Result> { self.store .retrieve_for_module("diary") .map(|iter| DiaryEntryIterator::new(self.name, self.store, iter)) .map_err(|e| DE::new(DEK::StoreReadError, Some(Box::new(e)))) } pub fn delete_entry(&self, entry: Entry) -> Result<()> { if !entry.is_in_diary(self.name) { return Err(DE::new(DEK::EntryNotInDiary, None)); } let id = entry.get_location().clone(); drop(entry); self.store.delete(id) .map_err(|e| DE::new(DEK::StoreWriteError, Some(Box::new(e)))) } pub fn get_youngest_entry(&self) -> Option> { match self.entries() { Err(e) => Some(Err(e)), Ok(entries) => { entries.sorted_by(|a, b| { match (a, b) { (&Ok(ref a), &Ok(ref b)) => { let a : NaiveDateTime = a.diary_id().into(); let b : NaiveDateTime = b.diary_id().into(); a.cmp(&b) }, (&Ok(_), &Err(ref e)) => { trace_error(e); Ordering::Less }, (&Err(ref e), &Ok(_)) => { trace_error(e); Ordering::Greater }, (&Err(ref e1), &Err(ref e2)) => { trace_error(e1); trace_error(e2); Ordering::Equal }, } }).into_iter().next() } } } pub fn name(&self) -> &'a str { &self.name } }