track edits and deletions

Author: tinaun

src/api.rs

@@ -2,16 +2,36 @@ use crate::{
     commands::{Args, Result},
     db::DB,
     schema::roles,
+    CommandHistory,
 };
 use diesel::prelude::*;
 use serenity::{model::prelude::*, utils::parse_username};
 
 /// Send a reply to the channel the message was received on.  
 pub(crate) fn send_reply(args: &Args, message: &str) -> Result<()> {
-    args.msg.channel_id.say(&args.cx, message)?;
+    if let Some(response_id) = response_exists(args) {
+        info!("editing message: {:?}", response_id);
+        args.msg
+            .channel_id
+            .edit_message(&args.cx, response_id, |msg| msg.content(message))?;
+    } else {
+        let command_id = args.msg.id;
+        let response = args.msg.channel_id.say(&args.cx, message)?;
+
+        let mut data = args.cx.data.write();
+        let history = data.get_mut::<CommandHistory>().unwrap();
+        history.insert(command_id, response.id);
+    }
+
     Ok(())
 }
 
+fn response_exists(args: &Args) -> Option<MessageId> {
+    let data = args.cx.data.read();
+    let history = data.get::<CommandHistory>().unwrap();
+    history.get(&args.msg.id).cloned()
+}
+
 /// Determine if a member sending a message has the `Role`.  
 pub(crate) fn has_role(args: &Args, role: &RoleId) -> Result<bool> {
     Ok(args

src/main.rs

@@ -24,6 +24,7 @@ use commands::{Args, Commands, GuardFn, Result};
 use diesel::prelude::*;
 use envy;
 use indexmap::IndexMap;
+use std::collections::HashMap;
 use serde::Deserialize;
 use serenity::{model::prelude::*, prelude::*};
 
@@ -221,6 +222,11 @@ fn main() {
     }
 }
 
+struct CommandHistory {}
+impl TypeMapKey for CommandHistory {
+    type Value = HashMap<MessageId, MessageId>;
+}
+
 struct Events {
     cmds: Commands,
 }
@@ -230,11 +236,34 @@ impl RawEventHandler for Events {
         match event {
             Event::Ready(ev) => {
                 info!("{} connected to discord", ev.ready.user.name);
+                let mut data = cx.data.write();
+                data.insert::<CommandHistory>(HashMap::new());
+                drop(data);
+
                 ban::start_unban_thread(cx);
             }
             Event::MessageCreate(ev) => {
                 self.cmds.execute(cx, &ev.message);
             }
+            Event::MessageUpdate(ev) => {
+                use serenity::utils::CustomMessage;
+                let mut msg = CustomMessage::new();
+        
+                msg.id(ev.id)
+                    .channel_id(ev.channel_id)
+                    .content(ev.content.unwrap_or_else(|| String::new()));
+        
+                let msg = msg.build();
+                self.cmds.execute(cx, msg);
+            }
+            Event::MessageDelete(ev) => {
+                let mut data = cx.data.write();
+                let history = data.get_mut::<CommandHistory>().unwrap();
+                if let Some(response_id) = history.remove(&ev.message_id) {
+                    info!("deleting message: {:?}", response_id);
+                    let _ = ev.channel_id.delete_message(&cx, response_id);
+                }
+            }
             Event::ReactionAdd(ev) => {
                 if let Err(e) = welcome::assign_talk_role(&cx, &ev) {
                     error!("{}", e);