pull out chat list into a separate component

1 files changed, 44 insertions, 223 deletions

diff --git a/components/chatwindow.cpp b/components/chatwindow.cpp
index 55ff06d..f69efad 100644
--- a/components/chatwindow.cpp
+++ b/components/chatwindow.cpp
@@ -4,15 +4,13 @@
 #include "chatinputindicator.hpp"
 #include "ratelimitindicator.hpp"
 #include "chatinput.hpp"
-
-constexpr static uint64_t SnowflakeSplitDifference = 600;
+#include "chatlist.hpp"
 
 ChatWindow::ChatWindow() {
     Abaddon::Get().GetDiscordClient().signal_message_send_fail().connect(sigc::mem_fun(*this, &ChatWindow::OnMessageSendFail));
 
     m_main = Gtk::manage(new Gtk::Box(Gtk::ORIENTATION_VERTICAL));
-    m_list = Gtk::manage(new Gtk::ListBox);
-    m_scroll = Gtk::manage(new Gtk::ScrolledWindow);
+    m_chat = Gtk::manage(new ChatList);
     m_input = Gtk::manage(new ChatInput);
     m_input_indicator = Gtk::manage(new ChatInputIndicator);
     m_rate_limit_indicator = Gtk::manage(new RateLimitIndicator);
@@ -29,34 +27,10 @@ ChatWindow::ChatWindow() {
     m_input_indicator->show();
 
     m_main->get_style_context()->add_class("messages");
-    m_list->get_style_context()->add_class("messages");
 
     m_main->set_hexpand(true);
     m_main->set_vexpand(true);
 
-    m_scroll->signal_edge_reached().connect(sigc::mem_fun(*this, &ChatWindow::OnScrollEdgeOvershot));
-
-    auto v = m_scroll->get_vadjustment();
-    v->signal_value_changed().connect([this, v] {
-        m_should_scroll_to_bottom = v->get_upper() - v->get_page_size() <= v->get_value();
-    });
-
-    m_scroll->set_can_focus(false);
-    m_scroll->set_policy(Gtk::POLICY_AUTOMATIC, Gtk::POLICY_ALWAYS);
-    m_scroll->show();
-
-    m_list->signal_size_allocate().connect([this](Gtk::Allocation &) {
-        if (m_should_scroll_to_bottom)
-            ScrollToBottom();
-    });
-
-    m_list->set_selection_mode(Gtk::SELECTION_NONE);
-    m_list->set_hexpand(true);
-    m_list->set_vexpand(true);
-    m_list->set_focus_hadjustment(m_scroll->get_hadjustment());
-    m_list->set_focus_vadjustment(m_scroll->get_vadjustment());
-    m_list->show();
-
     m_input->signal_submit().connect(sigc::mem_fun(*this, &ChatWindow::OnInputSubmit));
     m_input->signal_escape().connect([this]() {
         if (m_is_replying)
@@ -71,41 +45,49 @@ ChatWindow::ChatWindow() {
     });
 
     m_completer.SetGetRecentAuthors([this]() -> auto {
-        const auto &discord = Abaddon::Get().GetDiscordClient();
-        std::vector<Snowflake> ret;
-
-        std::map<Snowflake, Gtk::Widget *> ordered(m_id_to_widget.begin(), m_id_to_widget.end());
-
-        for (auto it = ordered.crbegin(); it != ordered.crend(); it++) {
-            const auto *widget = dynamic_cast<ChatMessageItemContainer *>(it->second);
-            if (widget == nullptr) continue;
-            const auto msg = discord.GetMessage(widget->ID);
-            if (!msg.has_value()) continue;
-            if (std::find(ret.begin(), ret.end(), msg->Author.ID) == ret.end())
-                ret.push_back(msg->Author.ID);
-        }
-
-        const auto chan = discord.GetChannel(m_active_channel);
-        if (chan->GuildID.has_value()) {
-            const auto others = discord.GetUsersInGuild(*chan->GuildID);
-            for (const auto id : others)
-                if (std::find(ret.begin(), ret.end(), id) == ret.end())
-                    ret.push_back(id);
-        }
-
-        return ret;
+        return m_chat->GetRecentAuthors();
     });
 
     m_completer.show();
 
+    m_chat->signal_action_channel_click().connect([this](Snowflake id) {
+        m_signal_action_channel_click.emit(id);
+    });
+    m_chat->signal_action_chat_load_history().connect([this](Snowflake id) {
+        m_signal_action_chat_load_history.emit(id);
+    });
+    m_chat->signal_action_chat_submit().connect([this](const std::string &str, Snowflake channel_id, Snowflake referenced_id) {
+        m_signal_action_chat_submit.emit(str, channel_id, referenced_id);
+    });
+    m_chat->signal_action_insert_mention().connect([this](Snowflake id) {
+        // lowkey gross
+        m_signal_action_insert_mention.emit(id);
+    });
+    m_chat->signal_action_message_delete().connect([this](Snowflake channel_id, Snowflake message_id) {
+        m_signal_action_message_delete.emit(channel_id, message_id);
+    });
+    m_chat->signal_action_message_edit().connect([this](Snowflake channel_id, Snowflake message_id) {
+        m_signal_action_message_edit.emit(channel_id, message_id);
+    });
+    m_chat->signal_action_reaction_add().connect([this](Snowflake id, const Glib::ustring &param) {
+        m_signal_action_reaction_add.emit(id, param);
+    });
+    m_chat->signal_action_reaction_remove().connect([this](Snowflake id, const Glib::ustring &param) {
+        m_signal_action_reaction_remove.emit(id, param);
+    });
+    m_chat->signal_action_reply_to().connect([this](Snowflake id) {
+        StartReplying(id);
+    });
+    m_chat->show();
+
     m_meta->set_hexpand(true);
     m_meta->set_halign(Gtk::ALIGN_FILL);
     m_meta->show();
 
     m_meta->add(*m_input_indicator);
     m_meta->add(*m_rate_limit_indicator);
-    m_scroll->add(*m_list);
-    m_main->add(*m_scroll);
+    //m_scroll->add(*m_list);
+    m_main->add(*m_chat);
     m_main->add(m_completer);
     m_main->add(*m_input);
     m_main->add(*m_meta);
@@ -121,25 +103,12 @@ void ChatWindow::Clear() {
 }
 
 void ChatWindow::SetMessages(const std::set<Snowflake> &msgs) {
-    // empty the listbox
-    auto children = m_list->get_children();
-    auto it = children.begin();
-    while (it != children.end()) {
-        delete *it;
-        it++;
-    }
-
-    m_num_rows = 0;
-    m_num_messages = 0;
-    m_id_to_widget.clear();
-
-    for (const auto &id : msgs) {
-        ProcessNewMessage(id, false);
-    }
+    m_chat->SetMessages(msgs.begin(), msgs.end());
 }
 
 void ChatWindow::SetActiveChannel(Snowflake id) {
     m_active_channel = id;
+    m_chat->SetActiveChannel(id);
     m_input_indicator->SetActiveChannel(id);
     m_rate_limit_indicator->SetActiveChannel(id);
     if (m_is_replying)
@@ -147,33 +116,19 @@ void ChatWindow::SetActiveChannel(Snowflake id) {
 }
 
 void ChatWindow::AddNewMessage(Snowflake id) {
-    ProcessNewMessage(id, false);
+    m_chat->ProcessNewMessage(id, false);
 }
 
 void ChatWindow::DeleteMessage(Snowflake id) {
-    auto widget = m_id_to_widget.find(id);
-    if (widget == m_id_to_widget.end()) return;
-
-    auto *x = dynamic_cast<ChatMessageItemContainer *>(widget->second);
-    if (x != nullptr)
-        x->UpdateAttributes();
+    m_chat->DeleteMessage(id);
 }
 
 void ChatWindow::UpdateMessage(Snowflake id) {
-    auto widget = m_id_to_widget.find(id);
-    if (widget == m_id_to_widget.end()) return;
-
-    auto *x = dynamic_cast<ChatMessageItemContainer *>(widget->second);
-    if (x != nullptr) {
-        x->UpdateContent();
-        x->UpdateAttributes();
-    }
+    m_chat->RefetchMessage(id);
 }
 
 void ChatWindow::AddNewHistory(const std::vector<Snowflake> &id) {
-    std::set<Snowflake> ids(id.begin(), id.end());
-    for (auto it = ids.rbegin(); it != ids.rend(); it++)
-        ProcessNewMessage(*it, true);
+    m_chat->PrependMessages(id.begin(), id.end());
 }
 
 void ChatWindow::InsertChatInput(std::string text) {
@@ -181,15 +136,11 @@ void ChatWindow::InsertChatInput(std::string text) {
 }
 
 Snowflake ChatWindow::GetOldestListedMessage() {
-    return m_id_to_widget.begin()->first;
+    return m_chat->GetOldestListedMessage();
 }
 
 void ChatWindow::UpdateReactions(Snowflake id) {
-    auto it = m_id_to_widget.find(id);
-    if (it == m_id_to_widget.end()) return;
-    auto *widget = dynamic_cast<ChatMessageItemContainer *>(it->second);
-    if (widget == nullptr) return;
-    widget->UpdateReactions();
+    m_chat->UpdateMessageReactions(id);
 }
 
 Snowflake ChatWindow::GetActiveChannel() const {
@@ -221,122 +172,6 @@ bool ChatWindow::OnKeyPressEvent(GdkEventKey *e) {
     return false;
 }
 
-ChatMessageItemContainer *ChatWindow::CreateMessageComponent(Snowflake id) {
-    auto *container = ChatMessageItemContainer::FromMessage(id);
-    return container;
-}
-
-void ChatWindow::RemoveMessageAndHeader(Gtk::Widget *widget) {
-    ChatMessageHeader *header = dynamic_cast<ChatMessageHeader *>(widget->get_ancestor(Gtk::ListBoxRow::get_type()));
-    if (header != nullptr) {
-        if (header->GetChildContent().size() == 1) {
-            m_num_rows--;
-            delete header;
-        } else
-            delete widget;
-    } else
-        delete widget;
-    m_num_messages--;
-}
-
-constexpr static int MaxMessagesForCull = 50; // this has to be 50 cuz that magic number is used in a couple other places and i dont feel like replacing them
-void ChatWindow::ProcessNewMessage(Snowflake id, bool prepend) {
-    const auto &client = Abaddon::Get().GetDiscordClient();
-    if (!client.IsStarted()) return; // e.g. load channel and then dc
-    const auto data = client.GetMessage(id);
-    if (!data.has_value()) return;
-
-    if (!data->IsPending && data->Nonce.has_value() && data->Author.ID == client.GetUserData().ID) {
-        for (auto [id, widget] : m_id_to_widget) {
-            if (dynamic_cast<ChatMessageItemContainer *>(widget)->Nonce == *data->Nonce) {
-                RemoveMessageAndHeader(widget);
-                m_id_to_widget.erase(id);
-                break;
-            }
-        }
-    }
-
-    ChatMessageHeader *last_row = nullptr;
-    bool should_attach = false;
-    if (m_num_rows > 0) {
-        if (prepend)
-            last_row = dynamic_cast<ChatMessageHeader *>(m_list->get_row_at_index(0));
-        else
-            last_row = dynamic_cast<ChatMessageHeader *>(m_list->get_row_at_index(m_num_rows - 1));
-
-        if (last_row != nullptr) {
-            const uint64_t diff = std::max(id, last_row->NewestID) - std::min(id, last_row->NewestID);
-            if (last_row->UserID == data->Author.ID && (prepend || (diff < SnowflakeSplitDifference * Snowflake::SecondsInterval)))
-                should_attach = true;
-        }
-    }
-
-    m_num_messages++;
-
-    if (m_should_scroll_to_bottom && !prepend)
-        while (m_num_messages > MaxMessagesForCull) {
-            auto first_it = m_id_to_widget.begin();
-            RemoveMessageAndHeader(first_it->second);
-            m_id_to_widget.erase(first_it);
-        }
-
-    ChatMessageHeader *header;
-    if (should_attach) {
-        header = last_row;
-    } else {
-        const auto guild_id = *client.GetChannel(m_active_channel)->GuildID;
-        const auto user_id = data->Author.ID;
-        const auto user = client.GetUser(user_id);
-        if (!user.has_value()) return;
-
-        header = Gtk::manage(new ChatMessageHeader(&*data));
-        header->signal_action_insert_mention().connect([this, user_id]() {
-            m_signal_action_insert_mention.emit(user_id);
-        });
-
-        header->signal_action_open_user_menu().connect([this, user_id, guild_id](const GdkEvent *event) {
-            m_signal_action_open_user_menu.emit(event, user_id, guild_id);
-        });
-
-        m_num_rows++;
-    }
-
-    auto *content = CreateMessageComponent(id);
-    if (content != nullptr) {
-        header->AddContent(content, prepend);
-        m_id_to_widget[id] = content;
-
-        if (!data->IsPending) {
-            content->signal_action_delete().connect([this, id] {
-                m_signal_action_message_delete.emit(m_active_channel, id);
-            });
-            content->signal_action_edit().connect([this, id] {
-                m_signal_action_message_edit.emit(m_active_channel, id);
-            });
-            content->signal_action_reaction_add().connect([this, id](const Glib::ustring &param) {
-                m_signal_action_reaction_add.emit(id, param);
-            });
-            content->signal_action_reaction_remove().connect([this, id](const Glib::ustring &param) {
-                m_signal_action_reaction_remove.emit(id, param);
-            });
-            content->signal_action_channel_click().connect([this](const Snowflake &id) {
-                m_signal_action_channel_click.emit(id);
-            });
-            content->signal_action_reply_to().connect(sigc::mem_fun(*this, &ChatWindow::StartReplying));
-        }
-    }
-
-    header->set_margin_left(5);
-    header->show_all();
-
-    if (!should_attach) {
-        if (prepend)
-            m_list->prepend(*header);
-        else
-            m_list->add(*header);
-    }
-}
-
 void ChatWindow::StartReplying(Snowflake message_id) {
     const auto &discord = Abaddon::Get().GetDiscordClient();
     const auto message = *discord.GetMessage(message_id);
@@ -363,18 +198,8 @@ void ChatWindow::OnScrollEdgeOvershot(Gtk::PositionType pos) {
         m_signal_action_chat_load_history.emit(m_active_channel);
 }
 
-void ChatWindow::ScrollToBottom() {
-    auto x = m_scroll->get_vadjustment();
-    x->set_value(x->get_upper());
-}
-
 void ChatWindow::OnMessageSendFail(const std::string &nonce, float retry_after) {
-    for (auto [id, widget] : m_id_to_widget) {
-        if (auto *container = dynamic_cast<ChatMessageItemContainer *>(widget); container->Nonce == nonce) {
-            container->SetFailed();
-            break;
-        }
-    }
+    m_chat->SetFailedByNonce(nonce);
 }
 
 ChatWindow::type_signal_action_message_delete ChatWindow::signal_action_message_delete() {
@@ -401,10 +226,6 @@ ChatWindow::type_signal_action_insert_mention ChatWindow::signal_action_insert_m
     return m_signal_action_insert_mention;
 }
 
-ChatWindow::type_signal_action_open_user_menu ChatWindow::signal_action_open_user_menu() {
-    return m_signal_action_open_user_menu;
-}
-
 ChatWindow::type_signal_action_reaction_add ChatWindow::signal_action_reaction_add() {
     return m_signal_action_reaction_add;
 }