Merge pull request #451 from matthiasbeyer/imag-notes/get

Imag notes/get

imag-notes/src/main.rs

@@ -69,14 +69,19 @@ fn edit(rt: &Runtime) {
 }
 
 fn edit_entry(rt: &Runtime, name: String) -> bool {
-    let note = Note::retrieve(rt.store(), name);
-    if note.is_err() {
-        trace_error(&note.unwrap_err());
-        warn!("Cannot edit nonexistent Note");
-        return false
-    }
+    let mut note = match Note::get(rt.store(), name) {
+        Ok(Some(note)) => note,
+        Ok(None) => {
+            warn!("Cannot edit nonexistent Note");
+            return false
+        },
+        Err(e) => {
+            trace_error(&e);
+            warn!("Cannot edit nonexistent Note");
+            return false
+        },
+    };
 
-    let mut note = note.unwrap();
     if let Err(e) = note.edit_content(rt) {
         trace_error(&e);
         warn!("Editing failed");

libimagnotes/src/note.rs

@@ -102,6 +102,16 @@ impl<'a> Note<'a> {
             .map(|entry| Note { entry: entry })
     }
 
+    pub fn get(store: &Store, name: String) -> Result<Option<Note>> {
+        use libimagerror::into::IntoError;
+
+        match store.get(ModuleEntryPath::new(name).into_storeid()) {
+            Ok(Some(entry)) => Ok(Some(Note { entry: entry })),
+            Ok(None) => Ok(None),
+            Err(e) => Err(NEK::StoreWriteError.into_error_with_cause(Box::new(e))),
+        }
+    }
+
     pub fn all_notes(store: &Store) -> Result<NoteIterator> {
         store.retrieve_for_module("notes")
             .map(|iter| NoteIterator::new(store, iter))