add support for 'note remove TITLE'

parser/src/command/note.rs

@@ -5,6 +5,7 @@ use std::fmt;
 #[derive(PartialEq, Eq, Debug)]
 pub enum NoteCommand {
     Summary { title: String },
+    Remove { title: String },
 }
 
 #[derive(PartialEq, Eq, Debug)]
@@ -25,17 +26,27 @@ impl NoteCommand {
         let mut toks = input.clone();
         if let Some(Token::Word("note")) = toks.peek_token()? {
             toks.next_token()?;
-            let title = match toks.next_token()? {
-                Some(Token::Word(title)) => Some(title),
-                Some(Token::Quote(multi_word_title)) => Some(multi_word_title),
-                _ => None,
-            };
-            if let Some(title) = title {
-                Ok(Some(NoteCommand::Summary {
-                    title: title.to_string(),
-                }))
-            } else {
-                Err(toks.error(ParseError::MissingTitle))
+            let mut remove = false;
+            loop {
+                match toks.next_token()? {
+                    Some(Token::Word(title)) if title == "remove" => {
+                        remove = true;
+                        continue;
+                    }
+                    Some(Token::Word(title)) | Some(Token::Quote(title)) => {
+                        let command = if remove {
+                            NoteCommand::Remove {
+                                title: title.to_string(),
+                            }
+                        } else {
+                            NoteCommand::Summary {
+                                title: title.to_string(),
+                            }
+                        };
+                        break Ok(Some(command));
+                    }
+                    _ => break Err(toks.error(ParseError::MissingTitle)),
+                };
             }
         } else {
             Ok(None)

src/handlers/note.rs

@@ -62,6 +62,15 @@ struct NoteData {
 }
 
 impl NoteData {
+    pub fn remove(&mut self, title: &str) -> () {
+        let idx = self.entries.iter().position(|x| x.title == title).unwrap();
+        log::debug!(
+            "Removing element {:#?} from index {}",
+            self.entries[idx],
+            idx
+        );
+        self.entries.remove(idx);
+    }
     pub fn to_markdown(&self) -> String {
         if self.entries.is_empty() {
             return String::new();
@@ -88,16 +97,23 @@ pub(super) async fn handle_command(
 
     let comment_url = String::from(event.html_url().unwrap());
     let author = event.user().login.to_owned();
-    let NoteCommand::Summary { title } = &cmd;
 
-    let new_entry = NoteDataEntry {
-        title: title.to_owned(),
-        comment_url,
-        author,
-    };
+    match &cmd {
+        NoteCommand::Summary { title } => {
+            let new_entry = NoteDataEntry {
+                title: title.to_owned(),
+                comment_url,
+                author,
+            };
+
+            log::debug!("New Note Entry: {:#?}", new_entry);
+            current.entries.push(new_entry);
+        }
+        NoteCommand::Remove { title } => {
+            current.remove(title);
+        }
+    }
 
-    log::debug!("New Note Entry: {:#?}", new_entry);
-    current.entries.push(new_entry);
     let new_markdown = current.to_markdown();
     log::debug!("New MD: {:#?}", new_markdown);