pajlada
104 lines
2.7 KiB
C++
104 lines
2.7 KiB
C++
#include "util/VectorMessageSink.hpp"
|
|
|
|
#include "messages/MessageSimilarity.hpp"
|
|
#include "util/ChannelHelpers.hpp"
|
|
|
|
#include <cassert>
|
|
|
|
namespace chatterino {
|
|
|
|
VectorMessageSink::VectorMessageSink(MessageSinkTraits traits,
|
|
MessageFlags additionalFlags)
|
|
: additionalFlags(additionalFlags)
|
|
, traits(traits)
|
|
{
|
|
}
|
|
|
|
VectorMessageSink::~VectorMessageSink() = default;
|
|
|
|
void VectorMessageSink::addMessage(MessagePtr message, MessageContext ctx,
|
|
std::optional<MessageFlags> overridingFlags)
|
|
{
|
|
assert(!overridingFlags.has_value());
|
|
assert(ctx == MessageContext::Original);
|
|
|
|
message->flags.set(this->additionalFlags);
|
|
this->messages_.emplace_back(std::move(message));
|
|
}
|
|
|
|
void VectorMessageSink::addOrReplaceTimeout(MessagePtr clearchatMessage,
|
|
const QDateTime &now)
|
|
{
|
|
addOrReplaceChannelTimeout(
|
|
this->messages_, std::move(clearchatMessage), now,
|
|
[&](auto idx, auto /*msg*/, auto &&replacement) {
|
|
replacement->flags.set(this->additionalFlags);
|
|
this->messages_[idx] = replacement;
|
|
},
|
|
[&](auto &&msg) {
|
|
this->messages_.emplace_back(msg);
|
|
},
|
|
false);
|
|
}
|
|
|
|
void VectorMessageSink::addOrReplaceClearChat(MessagePtr clearchatMessage,
|
|
const QDateTime &now)
|
|
{
|
|
addOrReplaceChannelClear(
|
|
this->messages_, std::move(clearchatMessage), now,
|
|
[&](auto idx, auto /*msg*/, auto &&replacement) {
|
|
replacement->flags.set(this->additionalFlags);
|
|
this->messages_[idx] = replacement;
|
|
},
|
|
[&](auto &&msg) {
|
|
this->messages_.emplace_back(msg);
|
|
});
|
|
}
|
|
|
|
void VectorMessageSink::disableAllMessages()
|
|
{
|
|
if (this->additionalFlags.has(MessageFlag::RecentMessage))
|
|
{
|
|
return; // don't disable recent messages
|
|
}
|
|
|
|
for (const auto &msg : this->messages_)
|
|
{
|
|
msg->flags.set(MessageFlag::Disabled);
|
|
}
|
|
}
|
|
|
|
void VectorMessageSink::applySimilarityFilters(const MessagePtr &message) const
|
|
{
|
|
setSimilarityFlags(message, this->messages_);
|
|
}
|
|
|
|
MessagePtr VectorMessageSink::findMessageByID(QStringView id)
|
|
{
|
|
for (const auto &msg : this->messages_ | std::views::reverse)
|
|
{
|
|
if (msg->id == id)
|
|
{
|
|
return msg;
|
|
}
|
|
}
|
|
return {};
|
|
}
|
|
|
|
const std::vector<MessagePtr> &VectorMessageSink::messages() const
|
|
{
|
|
return this->messages_;
|
|
}
|
|
|
|
std::vector<MessagePtr> VectorMessageSink::takeMessages() &&
|
|
{
|
|
return std::move(this->messages_);
|
|
}
|
|
|
|
MessageSinkTraits VectorMessageSink::sinkTraits() const
|
|
{
|
|
return this->traits;
|
|
}
|
|
|
|
} // namespace chatterino
|