diff options
Diffstat (limited to 'components/completer.cpp')
| -rw-r--r-- | components/completer.cpp | 323 |
1 files changed, 323 insertions, 0 deletions
diff --git a/components/completer.cpp b/components/completer.cpp new file mode 100644 index 0000000..d673d40 --- /dev/null +++ b/components/completer.cpp @@ -0,0 +1,323 @@ +#include "completer.hpp" +#include "../abaddon.hpp" +#include "../util.hpp" + +constexpr const int CompleterHeight = 150; +constexpr const int MaxCompleterEntries = 15; + +Completer::Completer() { + set_reveal_child(false); + set_transition_type(Gtk::REVEALER_TRANSITION_TYPE_NONE); // only SLIDE_UP and NONE work decently + + m_scroll.set_policy(Gtk::POLICY_NEVER, Gtk::POLICY_AUTOMATIC); + m_scroll.set_max_content_height(CompleterHeight); + m_scroll.set_size_request(-1, CompleterHeight); + m_scroll.set_placement(Gtk::CORNER_BOTTOM_LEFT); + + m_list.set_adjustment(m_scroll.get_vadjustment()); + m_list.set_focus_vadjustment(m_scroll.get_vadjustment()); + m_list.get_style_context()->add_class("completer"); + m_list.set_activate_on_single_click(true); + + m_list.set_focus_on_click(false); + set_can_focus(false); + + m_list.signal_row_activated().connect(sigc::mem_fun(*this, &Completer::OnRowActivate)); + + m_scroll.add(m_list); + add(m_scroll); + show_all(); +} + +Completer::Completer(const Glib::RefPtr<Gtk::TextBuffer> &buf) + : Completer() { + SetBuffer(buf); +} + +void Completer::SetBuffer(const Glib::RefPtr<Gtk::TextBuffer> &buf) { + m_buf = buf; + m_buf->signal_changed().connect(sigc::mem_fun(*this, &Completer::OnTextBufferChanged)); +} + +bool Completer::ProcessKeyPress(GdkEventKey *e) { + if (!IsShown()) return false; + if (e->type != GDK_KEY_PRESS) return false; + + switch (e->keyval) { + case GDK_KEY_Down: { + if (m_entries.size() == 0) return true; + const int index = m_list.get_selected_row()->get_index(); + if (index >= m_entries.size() - 1) return true; + m_list.select_row(*m_entries[index + 1]); + ScrollListBoxToSelected(m_list); + } + return true; + case GDK_KEY_Up: { + if (m_entries.size() == 0) return true; + const int index = m_list.get_selected_row()->get_index(); + if (index == 0) return true; + m_list.select_row(*m_entries[index - 1]); + ScrollListBoxToSelected(m_list); + } + return true; + case GDK_KEY_Return: { + if (m_entries.size() == 0) return true; + DoCompletion(m_list.get_selected_row()); + } + return true; + default: + break; + } + + return false; +} + +void Completer::SetGetRecentAuthors(get_recent_authors_cb cb) { + m_recent_authors_cb = cb; +} + +void Completer::SetGetChannelID(get_channel_id_cb cb) { + m_channel_id_cb = cb; +} + +bool Completer::IsShown() const { + return get_child_revealed(); +} + +CompleterEntry *Completer::CreateEntry(const Glib::ustring &completion) { + auto entry = Gtk::manage(new CompleterEntry(completion, m_entries.size())); + m_entries.push_back(entry); + entry->show_all(); + m_list.add(*entry); + return entry; +} + +void Completer::CompleteMentions(const Glib::ustring &term) { + if (!m_recent_authors_cb) + return; + + const auto &discord = Abaddon::Get().GetDiscordClient(); + + Snowflake channel_id; + if (m_channel_id_cb) + channel_id = m_channel_id_cb(); + auto author_ids = m_recent_authors_cb(); + if (channel_id.IsValid()) { + const auto chan = discord.GetChannel(channel_id); + if (chan->GuildID.has_value()) { + const auto members = discord.GetUsersInGuild(*chan->GuildID); + for (const auto x : members) + if (std::find(author_ids.begin(), author_ids.end(), x) == author_ids.end()) + author_ids.push_back(x); + } + } + const auto me = discord.GetUserData().ID; + int i = 0; + for (const auto id : author_ids) { + if (id == me) continue; + const auto author = discord.GetUser(id); + if (!author.has_value()) continue; + if (!StringContainsCaseless(author->Username, term)) continue; + if (i++ > 15) break; + + auto entry = CreateEntry(author->GetMention()); + + entry->SetText(author->Username + "#" + author->Discriminator); + + if (channel_id.IsValid()) { + const auto chan = discord.GetChannel(channel_id); + if (chan.has_value() && chan->GuildID.has_value()) { + const auto role_id = discord.GetMemberHoistedRole(*chan->GuildID, id, true); + if (role_id.IsValid()) { + const auto role = discord.GetRole(role_id); + if (role.has_value()) + entry->SetTextColor(role->Color); + } + } + } + + auto &img = Abaddon::Get().GetImageManager(); + const auto placeholder = img.GetPlaceholder(24); + if (author->HasAvatar()) { + auto pb = img.GetFromURLIfCached(author->GetAvatarURL()); + if (pb) { + entry->SetImage(pb); + } else { + entry->SetImage(placeholder); + img.LoadFromURL(author->GetAvatarURL(), sigc::mem_fun(*entry, &CompleterEntry::SetImage)); + } + } else + entry->SetImage(placeholder); + } +} + +void Completer::CompleteEmojis(const Glib::ustring &term) { + if (!m_channel_id_cb) + return; + + const auto &discord = Abaddon::Get().GetDiscordClient(); + const auto channel_id = m_channel_id_cb(); + const auto channel = discord.GetChannel(channel_id); + if (!channel->GuildID.has_value()) return; + const auto guild = discord.GetGuild(*channel->GuildID); + + const auto make_entry = [&](const Glib::ustring &name, const Glib::ustring &completion, const Glib::ustring &url = "") -> CompleterEntry * { + const auto entry = CreateEntry(completion); + entry->SetText(name); + if (url == "") return entry; + auto &img = Abaddon::Get().GetImageManager(); + const auto placeholder = img.GetPlaceholder(24); + const auto pb = img.GetFromURLIfCached(url); + if (pb) + entry->SetImage(pb); + else { + entry->SetImage(placeholder); + img.LoadFromURL(url, sigc::mem_fun(*entry, &CompleterEntry::SetImage)); + } + return entry; + }; + + int i = 0; + for (const auto tmp : guild->Emojis) { + const auto emoji = discord.GetEmoji(tmp.ID); + if (!emoji.has_value()) continue; + if (emoji->IsAnimated.has_value() && *emoji->IsAnimated) continue; + if (term.size() > 0) + if (!StringContainsCaseless(emoji->Name, term)) continue; + if (i++ > MaxCompleterEntries) break; + + const auto entry = make_entry(emoji->Name, "<:" + emoji->Name + ":" + std::to_string(emoji->ID) + ">", emoji->GetURL()); + } + + // if <15 guild emojis match then load up stock + if (i < 15) { + auto &emojis = Abaddon::Get().GetEmojis(); + const auto &shortcodes = emojis.GetShortCodes(); + for (const auto &[shortcode, pattern] : shortcodes) { + if (!StringContainsCaseless(shortcode, term)) continue; + if (i++ > 15) break; + const auto &pb = emojis.GetPixBuf(pattern); + if (!pb) continue; + const auto entry = make_entry(shortcode, pattern); + entry->SetImage(pb); + } + } +} + +void Completer::CompleteChannels(const Glib::ustring &term) { + if (!m_channel_id_cb) + return; + + const auto &discord = Abaddon::Get().GetDiscordClient(); + const auto channel_id = m_channel_id_cb(); + const auto channel = discord.GetChannel(channel_id); + if (!channel->GuildID.has_value()) return; + const auto channels = discord.GetChannelsInGuild(*channel->GuildID); + int i = 0; + for (const auto chan_id : channels) { + const auto chan = discord.GetChannel(chan_id); + if (chan->Type == ChannelType::GUILD_VOICE || chan->Type == ChannelType::GUILD_CATEGORY) continue; + if (!StringContainsCaseless(*chan->Name, term)) continue; + if (i++ > MaxCompleterEntries) break; + const auto entry = CreateEntry("<#" + std::to_string(chan_id) + ">"); + entry->SetText("#" + *chan->Name); + } +} + +void Completer::DoCompletion(Gtk::ListBoxRow *row) { + const int index = row->get_index(); + const auto completion = m_entries[index]->GetCompletion(); + const auto it = m_buf->erase(m_start, m_end); // entry is deleted here + m_buf->insert(it, completion + " "); +} + +void Completer::OnRowActivate(Gtk::ListBoxRow *row) { + DoCompletion(row); +} + +void Completer::OnTextBufferChanged() { + const auto term = GetTerm(); + + for (auto it = m_entries.begin(); it != m_entries.end();) { + delete *it; + it = m_entries.erase(it); + } + + switch (term[0]) { + case '@': + CompleteMentions(term.substr(1)); + break; + case ':': + CompleteEmojis(term.substr(1)); + break; + case '#': + CompleteChannels(term.substr(1)); + break; + default: + break; + } + if (m_entries.size() > 0) { + m_list.select_row(*m_entries[0]); + set_reveal_child(true); + } else { + set_reveal_child(false); + } +} + +Glib::ustring Completer::GetTerm() { + const auto iter = m_buf->get_insert()->get_iter(); + Gtk::TextBuffer::iterator dummy; + if (!iter.backward_search(" ", Gtk::TEXT_SEARCH_TEXT_ONLY, m_start, dummy)) + m_buf->get_bounds(m_start, dummy); + else + m_start.forward_char(); // 1 behind + if (!iter.forward_search(" ", Gtk::TEXT_SEARCH_TEXT_ONLY, dummy, m_end)) + m_buf->get_bounds(dummy, m_end); + return m_start.get_text(m_end); +} + +CompleterEntry::CompleterEntry(const Glib::ustring &completion, int index) + : m_index(index) + , m_completion(completion) + , m_box(Gtk::ORIENTATION_HORIZONTAL) { + set_halign(Gtk::ALIGN_START); + get_style_context()->add_class("completer-entry"); + set_can_focus(false); + set_focus_on_click(false); + m_box.show(); + add(m_box); +} + +void CompleterEntry::SetTextColor(int color) { + if (m_text == nullptr) return; + const auto cur = m_text->get_text(); + m_text->set_markup("<span color=\"#" + IntToCSSColor(color) + "\">" + Glib::Markup::escape_text(cur) + "</span>"); +} + +void CompleterEntry::SetText(const Glib::ustring &text) { + if (m_text == nullptr) { + m_text = Gtk::manage(new Gtk::Label); + m_text->get_style_context()->add_class("completer-entry-label"); + m_text->show(); + m_box.pack_end(*m_text); + } + m_text->set_label(text); +} + +void CompleterEntry::SetImage(const Glib::RefPtr<Gdk::Pixbuf> &pb) { + if (m_img == nullptr) { + m_img = Gtk::manage(new Gtk::Image); + m_img->get_style_context()->add_class("completer-entry-image"); + m_img->show(); + m_box.pack_start(*m_img); + } + m_img->property_pixbuf() = pb->scale_simple(24, 24, Gdk::INTERP_BILINEAR); +} + +int CompleterEntry::GetIndex() const { + return m_index; +} + +Glib::ustring CompleterEntry::GetCompletion() const { + return m_completion; +} |