1 files changed, 3 insertions, 12 deletions
diff --git a/components/chatmessage.hpp b/components/chatmessage.hpp
index 3c3e29e..e7202a5 100644
--- a/components/chatmessage.hpp
+++ b/components/chatmessage.hpp
@@ -40,8 +40,7 @@ public:
 
     virtual void ShowMenu(const GdkEvent *event);
     void AddMenuItem(Gtk::MenuItem *item);
-    virtual void MarkAsDeleted() = 0;
-    virtual void MarkAsEdited() = 0;
+    virtual void Update() = 0;
 
 protected:
     void AttachMenuHandler(Gtk::Widget *widget);
@@ -74,17 +73,13 @@ public:
 
     void EditContent(std::string content);
 
-    virtual void MarkAsDeleted();
-    virtual void MarkAsEdited();
+    virtual void Update();
 
 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;
@@ -96,16 +91,12 @@ class ChatMessageEmbedItem
 public:
     ChatMessageEmbedItem(const Message *data);
 
-    virtual void MarkAsDeleted();
-    virtual void MarkAsEdited();
+    virtual void Update();
 
 protected:
     void DoLayout();
     void UpdateAttributes();
 
-    bool m_was_deleted = false;
-    bool m_was_edited = false;
-
     EmbedData m_embed;
     Gtk::Box *m_main;
     Gtk::Label *m_attrib_label = nullptr;