From 44b7989f50d458a8d60024f29e225e40a106d927 Mon Sep 17 00:00:00 2001
From: ouwou <26526779+ouwou@users.noreply.github.com>
Date: Sun, 30 Aug 2020 20:24:02 -0400
Subject: add MESSAGE_UPDATE

---
 components/chatmessage.hpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'components/chatmessage.hpp')

diff --git a/components/chatmessage.hpp b/components/chatmessage.hpp
index dc967af..82f4504 100644
--- a/components/chatmessage.hpp
+++ b/components/chatmessage.hpp
@@ -37,6 +37,7 @@ public:
     virtual void ShowMenu(const GdkEvent *event);
     void AddMenuItem(Gtk::MenuItem *item);
     virtual void MarkAsDeleted() = 0;
+    virtual void MarkAsEdited() = 0;
 
 protected:
     void AttachMenuHandler(Gtk::Widget *widget);
@@ -55,9 +56,20 @@ class ChatMessageTextItem
     , public ChatMessageItem {
 public:
     ChatMessageTextItem(const MessageData *data);
+
+    void EditContent(std::string content);
+
     virtual void MarkAsDeleted();
+    virtual void MarkAsEdited();
 
 protected:
+    void UpdateAttributes();
+
+    std::string m_content;
+
+    bool m_was_deleted = false;
+    bool m_was_edited = false;
+
     void on_menu_copy_content();
     Gtk::MenuItem *m_menu_copy_content;
     Gtk::MenuItem *m_menu_delete_message;
-- 
cgit v1.2.3