Actually carries out !edit and !s commands

2 files changed, 11 insertions, 0 deletions

diff --git a/src/system/bot/client.rs b/src/system/bot/client.rs
index 61d7515..4d4f7bb 100644
--- a/src/system/bot/client.rs
+++ b/src/system/bot/client.rs
@@ -90,6 +90,13 @@ impl Client {
         return Ok(())
     }
 
+    pub async fn edit_message(&self, channel_id: ChannelId, message_id: MessageId, new_content: String) -> Result<FullMessage, TwiError> {
+        Ok(self.client.lock().await.update_message(channel_id, message_id)
+            .content(Some(new_content.as_str())).expect("Invalid message contents")
+            .await.expect("Could not update message")
+            .model().await.unwrap())
+    }
+
     pub async fn duplicate_message(&self, message: &TwiMessage, content: &str) -> Result<TwiMessage, MessageDuplicateError> {
         let client = self.client.lock().await;
 
diff --git a/src/system/bot/mod.rs b/src/system/bot/mod.rs
index 2f38075..f0a2e45 100644
--- a/src/system/bot/mod.rs
+++ b/src/system/bot/mod.rs
@@ -74,6 +74,10 @@ impl Bot {
         self.client.resend_message(message_id, channel_id).await;
     }
 
+    pub async fn edit_message(&self, channel_id: ChannelId, message_id: MessageId, new_content: String) -> Result<FullMessage, TwiError> {
+        self.client.edit_message(channel_id, message_id, new_content).await
+    }
+
     pub async fn delete_message(&self, channel_id: ChannelId, message_id: MessageId) -> Result<(), TwiError> {
         self.client.delete_message(channel_id, message_id).await
     }