Daniel Sage
86 lines
2.5 KiB
C++
86 lines
2.5 KiB
C++
#include "controllers/completion/strategies/ClassicEmoteStrategy.hpp"
|
|
|
|
#include "singletons/Settings.hpp"
|
|
#include "util/Helpers.hpp"
|
|
|
|
namespace chatterino::completion {
|
|
|
|
void ClassicEmoteStrategy::apply(const std::vector<EmoteItem> &items,
|
|
std::vector<EmoteItem> &output,
|
|
const QString &query) const
|
|
{
|
|
QString normalizedQuery = query;
|
|
if (normalizedQuery.startsWith(':'))
|
|
{
|
|
normalizedQuery = normalizedQuery.mid(1);
|
|
}
|
|
|
|
// First pass: filter by contains match
|
|
for (const auto &item : items)
|
|
{
|
|
if (item.searchName.contains(normalizedQuery, Qt::CaseInsensitive))
|
|
{
|
|
output.push_back(item);
|
|
}
|
|
}
|
|
|
|
// Second pass: if there is an exact match, put that emote first
|
|
for (size_t i = 1; i < output.size(); i++)
|
|
{
|
|
auto emoteText = output.at(i).searchName;
|
|
|
|
// test for match or match with colon at start for emotes like ":)"
|
|
if (emoteText.compare(normalizedQuery, Qt::CaseInsensitive) == 0 ||
|
|
emoteText.compare(":" + normalizedQuery, Qt::CaseInsensitive) == 0)
|
|
{
|
|
auto emote = output[i];
|
|
output.erase(output.begin() + int(i));
|
|
output.insert(output.begin(), emote);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
struct CompletionEmoteOrder {
|
|
bool operator()(const EmoteItem &a, const EmoteItem &b) const
|
|
{
|
|
return compareEmoteStrings(a.searchName, b.searchName);
|
|
}
|
|
};
|
|
|
|
void ClassicTabEmoteStrategy::apply(const std::vector<EmoteItem> &items,
|
|
std::vector<EmoteItem> &output,
|
|
const QString &query) const
|
|
{
|
|
bool emojiOnly = false;
|
|
QString normalizedQuery = query;
|
|
if (normalizedQuery.startsWith(':'))
|
|
{
|
|
normalizedQuery = normalizedQuery.mid(1);
|
|
// tab completion with : prefix should do emojis only
|
|
emojiOnly = true;
|
|
}
|
|
|
|
std::set<EmoteItem, CompletionEmoteOrder> emotes;
|
|
|
|
for (const auto &item : items)
|
|
{
|
|
if (emojiOnly ^ item.isEmoji)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (startsWithOrContains(item.searchName, normalizedQuery,
|
|
Qt::CaseInsensitive,
|
|
getSettings()->prefixOnlyEmoteCompletion))
|
|
{
|
|
emotes.insert(item);
|
|
}
|
|
}
|
|
|
|
output.reserve(emotes.size());
|
|
output.assign(emotes.begin(), emotes.end());
|
|
}
|
|
|
|
} // namespace chatterino::completion
|