add mention/emoji/channel completion
This commit is contained in:
ouwou
323
components/completer.cpp
Normal file
323
components/completer.cpp
Normal file
@@ -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;
|
||||
}
|
||||
Reference in New Issue
Block a user