add functionality to '@rustbot note' to allow it to edit the issue body accordingly

parser/src/command/note.rs

@@ -4,7 +4,7 @@ use std::fmt;
 
 #[derive(PartialEq, Eq, Debug)]
 pub enum NoteCommand {
-    Summary { title: String, summary: String },
+    Summary { title: String },
 }
 
 #[derive(PartialEq, Eq, Debug)]
@@ -15,14 +15,13 @@ pub enum ParseError {
 impl std::error::Error for ParseError {}
 impl fmt::Display for ParseError {
     fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
-        match self  {
+        match self {
             ParseError::MissingTitle => write!(f, "missing required summary title"),
             ParseError::MissingBody => write!(f, "missing summary notes"),
         }
     }
 }
 
-
 impl NoteCommand {
     pub fn parse<'a>(input: &mut Tokenizer<'a>) -> Result<Option<Self>, Error<'a>> {
         let mut toks = input.clone();
@@ -31,7 +30,6 @@ impl NoteCommand {
             if let Some(Token::Word(title)) = toks.next_token()? {
                 Ok(Some(NoteCommand::Summary {
                     title: title.to_string(),
-                    summary: String::from("body"),
                 }))
             } else {
                 Err(toks.error(ParseError::MissingTitle))

src/handlers/note.rs

@@ -11,44 +11,68 @@
 //! If this is the first summary entry, rustbot will amend the original post (the top-level comment) to add a "Notes" section:
 //!
 //! ```md
-//! <!-- rustbot summary start -->
+//! <!-- TRIAGEBOT_SUMMARY_START -->
 //!
-//! ### Notes
+//! ### Summary Notes
 //!
 //! - ["summary-title" by @username](link-to-comment)
 //!
-//! <!-- rustbot summary end -->
+//! <!-- TRIAGEBOT_SUMMARY_END -->
 //! ```
 //!
 //! If this is *not* the first summary entry, rustbot will simply append the new entry to the existing notes section:
 //!
 //! ```md
-//! <!-- rustbot summary start -->
+//! <!-- TRIAGEBOT_SUMMARY_START -->
 //!
-//! ### Notes
+//! ### Summary Notes
 //!
 //! - ["first-note" by @username](link-to-comment)
 //! - ["second-note" by @username](link-to-comment)
 //! - ["summary-title" by @username](link-to-comment)
 //!
-//! <!-- rustbot summary end -->
+//! <!-- TRIAGEBOT_SUMMARY_END -->
 //! ```
 //!
 
-use crate::{
-    config::NoteConfig,
-    github::{self, Event, Selection},
-    handlers::Context,
-    interactions::EditIssueBody,
-};
-use anyhow::Context as _;
+use crate::{config::NoteConfig, github::Event, handlers::Context, interactions::EditIssueBody};
 use parser::command::note::NoteCommand;
 use tracing as log;
 
 #[derive(Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize)]
+struct NoteDataEntry {
+    title: String,
+    comment_url: String,
+    author: String,
+}
+
+impl NoteDataEntry {
+    pub fn to_markdown(&self) -> String {
+        format!(
+            "\n- [\"{title}\" by @{author}]({comment_url})",
+            title = self.title.to_owned(),
+            author = self.author.to_owned(),
+            comment_url = self.comment_url.to_owned()
+        )
+    }
+}
+
+#[derive(Debug, PartialEq, Eq, serde::Serialize, serde::Deserialize, Default)]
 struct NoteData {
-    title: Option<String>,
-    summary: Option<String>,
+    entries: Vec<NoteDataEntry>,
+}
+
+impl NoteData {
+    pub fn to_markdown(&self) -> String {
+        if self.entries.is_empty() {
+            return String::new();
+        }
+        let mut text = String::from("\n### Summary Notes\n");
+        for entry in &self.entries {
+            text.push_str(&entry.to_markdown());
+        }
+        text
+    }
 }
 
 pub(super) async fn handle_command(
@@ -57,17 +81,27 @@ pub(super) async fn handle_command(
     event: &Event,
     cmd: NoteCommand,
 ) -> anyhow::Result<()> {
-    log::debug!("Handling Note Command");
+    let issue = event.issue().unwrap();
+    let e = EditIssueBody::new(&issue, "SUMMARY");
 
-    // TODO: edit the original post
+    let mut current: NoteData = e.current_data().unwrap_or_default();
 
-    let issue = event.issue().unwrap();
-    
-    log::debug!("Issue: {:?}", issue);
+    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,
+    };
+
+    log::debug!("New Note Entry: {:#?}", new_entry);
+    current.entries.push(new_entry);
+    let new_markdown = current.to_markdown();
+    log::debug!("New MD: {:#?}", new_markdown);
+
+    e.apply(&ctx.github, new_markdown, current).await?;
 
-    if issue.is_pr() {
-        let NoteCommand::Summary { title, summary } = &cmd;
-        log::debug!("Note: {}, {}", title, summary);
-    }
     Ok(())
 }