2 files changed, 6 insertions, 0 deletions
diff --git a/windows/mainwindow.cpp b/windows/mainwindow.cpp
index 24f39f3..6f0bfaa 100644
--- a/windows/mainwindow.cpp
+++ b/windows/mainwindow.cpp
@@ -115,6 +115,11 @@ void MainWindow::UpdateChatMessageDeleted(Snowflake id, Snowflake channel_id) {
         m_chat.DeleteMessage(id);
 }
 
+void MainWindow::UpdateChatMessageEditContent(Snowflake id, Snowflake channel_id) {
+    if (channel_id == GetChatActiveChannel())
+        m_chat.UpdateMessageContent(id);
+}
+
 void MainWindow::UpdateChatPrependHistory(const std::vector<MessageData> &msgs) {
     m_chat.AddNewHistory(msgs);
 }
diff --git a/windows/mainwindow.hpp b/windows/mainwindow.hpp
index 10d897c..43dd540 100644
--- a/windows/mainwindow.hpp
+++ b/windows/mainwindow.hpp
@@ -17,6 +17,7 @@ public:
     Snowflake GetChatActiveChannel() const;
     void UpdateChatNewMessage(Snowflake id);
     void UpdateChatMessageDeleted(Snowflake id, Snowflake channel_id);
+    void UpdateChatMessageEditContent(Snowflake id, Snowflake channel_id);
     void UpdateChatPrependHistory(const std::vector<MessageData> &msgs);
 
 protected: