From 4b7cdbd129cedd4029c2b99c85146ad5e07e9732 Mon Sep 17 00:00:00 2001
From: ouwou <26526779+ouwou@users.noreply.github.com>
Date: Sun, 20 Sep 2020 01:12:54 -0400
Subject: improve MESSAGE_UPDATE handling (hopefully)

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

(limited to 'components/chatmessage.hpp')

diff --git a/components/chatmessage.hpp b/components/chatmessage.hpp
index 9365dd5..94428b3 100644
--- a/components/chatmessage.hpp
+++ b/components/chatmessage.hpp
@@ -19,8 +19,10 @@ public:
 
     ChatMessageContainer(const Message *data);
     void AddNewContent(Gtk::Widget *widget, bool prepend = false);
+    void AddNewContentAtIndex(Gtk::Widget *widget, int index);
     void SetAvatarFromPixbuf(Glib::RefPtr<Gdk::Pixbuf> pixbuf);
     void Update();
+    int RemoveItem(Gtk::Widget *widget);
 
 protected:
     Gtk::Box *m_main_box;
@@ -43,7 +45,12 @@ public:
     void AddMenuItem(Gtk::MenuItem *item);
     virtual void Update() = 0;
 
+    void SetContainer(ChatMessageContainer *container);
+    ChatMessageContainer *GetContainer() const;
+
 protected:
+    ChatMessageContainer *m_container = nullptr;
+
     void AttachMenuHandler(Gtk::Widget *widget);
     void on_menu_copy_id();
     void on_menu_message_delete();
-- 
cgit v1.2.3