1 files changed, 12 insertions, 0 deletions
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;