Compare commits

...

4 commits

Author SHA1 Message Date
nerix a62b986fbd
Merge 417cdbc009 into 74a385dfee 2024-10-27 13:01:07 +01:00
Nerixyz 417cdbc009
good thing i added that 2024-10-26 18:27:47 +02:00
Nerixyz 7b5a21d088
fix: formatting and changelog xd 2024-10-26 18:11:05 +02:00
Nerixyz ad21098bf2
refactor: deduplicate irc parsing 2024-10-26 18:05:22 +02:00
20 changed files with 565 additions and 731 deletions

View file

@ -114,6 +114,7 @@
- Dev: Refactored static `MessageBuilder` helpers to standalone functions. (#5652) - Dev: Refactored static `MessageBuilder` helpers to standalone functions. (#5652)
- Dev: Decoupled reply parsing from `MessageBuilder`. (#5660, #5668) - Dev: Decoupled reply parsing from `MessageBuilder`. (#5660, #5668)
- Dev: Refactored IRC message building. (#5663) - Dev: Refactored IRC message building. (#5663)
- Dev: Unified parsing of historic and live IRC messages. (#5678)
## 2.5.1 ## 2.5.1

View file

@ -39,6 +39,7 @@ set(SOURCE_FILES
common/WindowDescriptors.cpp common/WindowDescriptors.cpp
common/WindowDescriptors.hpp common/WindowDescriptors.hpp
common/enums/MessageContext.hpp
common/enums/MessageOverflow.hpp common/enums/MessageOverflow.hpp
common/network/NetworkCommon.cpp common/network/NetworkCommon.cpp
@ -281,6 +282,9 @@ set(SOURCE_FILES
messages/MessageElement.cpp messages/MessageElement.cpp
messages/MessageElement.hpp messages/MessageElement.hpp
messages/MessageFlag.hpp messages/MessageFlag.hpp
messages/MessageSimilarity.cpp
messages/MessageSimilarity.hpp
messages/MessageSink.hpp
messages/MessageThread.cpp messages/MessageThread.cpp
messages/MessageThread.hpp messages/MessageThread.hpp
@ -526,6 +530,8 @@ set(SOURCE_FILES
util/Twitch.hpp util/Twitch.hpp
util/TypeName.hpp util/TypeName.hpp
util/Variant.hpp util/Variant.hpp
util/VectorMessageSink.cpp
util/VectorMessageSink.hpp
util/WidgetHelpers.cpp util/WidgetHelpers.cpp
util/WidgetHelpers.hpp util/WidgetHelpers.hpp
util/WindowsHelper.cpp util/WindowsHelper.cpp

View file

@ -3,7 +3,9 @@
#include "Application.hpp" #include "Application.hpp"
#include "messages/Message.hpp" #include "messages/Message.hpp"
#include "messages/MessageBuilder.hpp" #include "messages/MessageBuilder.hpp"
#include "messages/MessageSimilarity.hpp"
#include "providers/twitch/IrcMessageHandler.hpp" #include "providers/twitch/IrcMessageHandler.hpp"
#include "providers/twitch/TwitchIrcServer.hpp"
#include "singletons/Emotes.hpp" #include "singletons/Emotes.hpp"
#include "singletons/Logging.hpp" #include "singletons/Logging.hpp"
#include "singletons/Settings.hpp" #include "singletons/Settings.hpp"
@ -121,10 +123,10 @@ void Channel::addSystemMessage(const QString &contents)
this->addMessage(msg, MessageContext::Original); this->addMessage(msg, MessageContext::Original);
} }
void Channel::addOrReplaceTimeout(MessagePtr message) void Channel::addOrReplaceTimeout(MessagePtr message, QTime now)
{ {
addOrReplaceChannelTimeout( addOrReplaceChannelTimeout(
this->getMessageSnapshot(), std::move(message), QTime::currentTime(), this->getMessageSnapshot(), std::move(message), now,
[this](auto /*idx*/, auto msg, auto replacement) { [this](auto /*idx*/, auto msg, auto replacement) {
this->replaceMessage(msg, replacement); this->replaceMessage(msg, replacement);
}, },
@ -287,10 +289,15 @@ void Channel::clearMessages()
} }
MessagePtr Channel::findMessage(QString messageID) MessagePtr Channel::findMessage(QString messageID)
{
return this->findMessageByID(messageID);
}
MessagePtr Channel::findMessageByID(QStringView messageID)
{ {
MessagePtr res; MessagePtr res;
if (auto msg = this->messages_.rfind([&messageID](const MessagePtr &msg) { if (auto msg = this->messages_.rfind([messageID](const MessagePtr &msg) {
return msg->id == messageID; return msg->id == messageID;
}); });
msg) msg)
@ -301,6 +308,19 @@ MessagePtr Channel::findMessage(QString messageID)
return res; return res;
} }
void Channel::applySimilarityFilters(const MessagePtr &message) const
{
setSimilarityFlags(message, this->messages_.getSnapshot());
}
MessageSinkTraits Channel::sinkTraits() const
{
return {
MessageSinkTrait::AddMentionsToGlobalChannel,
MessageSinkTrait::RequiresKnownChannelPointReward,
};
}
bool Channel::canSendMessage() const bool Channel::canSendMessage() const
{ {
return false; return false;

View file

@ -1,8 +1,10 @@
#pragma once #pragma once
#include "common/enums/MessageContext.hpp"
#include "controllers/completion/TabCompletionModel.hpp" #include "controllers/completion/TabCompletionModel.hpp"
#include "messages/LimitedQueue.hpp" #include "messages/LimitedQueue.hpp"
#include "messages/MessageFlag.hpp" #include "messages/MessageFlag.hpp"
#include "messages/MessageSink.hpp"
#include <magic_enum/magic_enum.hpp> #include <magic_enum/magic_enum.hpp>
#include <pajlada/signals/signal.hpp> #include <pajlada/signals/signal.hpp>
@ -26,15 +28,7 @@ enum class TimeoutStackStyle : int {
Default = DontStackBeyondUserMessage, Default = DontStackBeyondUserMessage,
}; };
/// Context of the message being added to a channel class Channel : public std::enable_shared_from_this<Channel>, public MessageSink
enum class MessageContext {
/// This message is the original
Original,
/// This message is a repost of a message that has already been added in a channel
Repost,
};
class Channel : public std::enable_shared_from_this<Channel>
{ {
public: public:
// This is for Lua. See scripts/make_luals_meta.py // This is for Lua. See scripts/make_luals_meta.py
@ -55,7 +49,7 @@ public:
}; };
explicit Channel(const QString &name, Type type); explicit Channel(const QString &name, Type type);
virtual ~Channel(); ~Channel() override;
// SIGNALS // SIGNALS
pajlada::Signals::Signal<const QString &, const QString &, bool &> pajlada::Signals::Signal<const QString &, const QString &, bool &>
@ -85,8 +79,9 @@ public:
// overridingFlags can be filled in with flags that should be used instead // overridingFlags can be filled in with flags that should be used instead
// of the message's flags. This is useful in case a flag is specific to a // of the message's flags. This is useful in case a flag is specific to a
// type of split // type of split
void addMessage(MessagePtr message, MessageContext context, void addMessage(
std::optional<MessageFlags> overridingFlags = std::nullopt); MessagePtr message, MessageContext context,
std::optional<MessageFlags> overridingFlags = std::nullopt) final;
void addMessagesAtStart(const std::vector<MessagePtr> &messages_); void addMessagesAtStart(const std::vector<MessagePtr> &messages_);
void addSystemMessage(const QString &contents); void addSystemMessage(const QString &contents);
@ -94,8 +89,8 @@ public:
/// Inserts the given messages in order by Message::serverReceivedTime. /// Inserts the given messages in order by Message::serverReceivedTime.
void fillInMissingMessages(const std::vector<MessagePtr> &messages); void fillInMissingMessages(const std::vector<MessagePtr> &messages);
void addOrReplaceTimeout(MessagePtr message); void addOrReplaceTimeout(MessagePtr message, QTime now) final;
void disableAllMessages(); void disableAllMessages() final;
void replaceMessage(MessagePtr message, MessagePtr replacement); void replaceMessage(MessagePtr message, MessagePtr replacement);
void replaceMessage(size_t index, MessagePtr replacement); void replaceMessage(size_t index, MessagePtr replacement);
void deleteMessage(QString messageID); void deleteMessage(QString messageID);
@ -104,9 +99,14 @@ public:
void clearMessages(); void clearMessages();
MessagePtr findMessage(QString messageID); MessagePtr findMessage(QString messageID);
MessagePtr findMessageByID(QStringView messageID) final;
bool hasMessages() const; bool hasMessages() const;
void applySimilarityFilters(const MessagePtr &message) const final;
MessageSinkTraits sinkTraits() const final;
// CHANNEL INFO // CHANNEL INFO
virtual bool canSendMessage() const; virtual bool canSendMessage() const;
virtual bool isWritable() const; // whether split input will be usable virtual bool isWritable() const; // whether split input will be usable

View file

@ -0,0 +1,13 @@
#pragma once
namespace chatterino {
/// Context of the message being added to a channel
enum class MessageContext {
/// This message is the original
Original,
/// This message is a repost of a message that has already been added in a channel
Repost,
};
} // namespace chatterino

View file

@ -0,0 +1,51 @@
#include "messages/MessageSimilarity.hpp"
#include <algorithm>
#include <vector>
namespace chatterino::similarity::detail {
float relativeSimilarity(QStringView str1, QStringView str2)
{
using SizeType = QStringView::size_type;
// Longest Common Substring Problem
std::vector<std::vector<int>> tree(str1.size(),
std::vector<int>(str2.size(), 0));
int z = 0;
for (SizeType i = 0; i < str1.size(); ++i)
{
for (SizeType j = 0; j < str2.size(); ++j)
{
if (str1[i] == str2[j])
{
if (i == 0 || j == 0)
{
tree[i][j] = 1;
}
else
{
tree[i][j] = tree[i - 1][j - 1] + 1;
}
z = std::max(tree[i][j], z);
}
else
{
tree[i][j] = 0;
}
}
}
// ensure that no div by 0
if (z == 0)
{
return 0.F;
}
auto div = std::max<>({static_cast<SizeType>(1), str1.size(), str2.size()});
return float(z) / float(div);
}
} // namespace chatterino::similarity::detail

View file

@ -0,0 +1,73 @@
#pragma once
#include "Application.hpp"
#include "controllers/accounts/AccountController.hpp"
#include "messages/Message.hpp"
#include "providers/twitch/TwitchAccount.hpp"
#include "singletons/Settings.hpp"
#include <ranges>
namespace chatterino::similarity::detail {
float relativeSimilarity(QStringView str1, QStringView str2);
float inMessages(const MessagePtr &msg,
const std::ranges::bidirectional_range auto &messages)
{
float similarityPercent = 0.0F;
for (const auto &prevMsg :
messages | std::views::reverse |
std::views::take(getSettings()->hideSimilarMaxMessagesToCheck))
{
if (prevMsg->parseTime.secsTo(QTime::currentTime()) >=
getSettings()->hideSimilarMaxDelay)
{
break;
}
if (getSettings()->hideSimilarBySameUser &&
msg->loginName != prevMsg->loginName)
{
continue;
}
similarityPercent = std::max(
similarityPercent,
relativeSimilarity(msg->messageText, prevMsg->messageText));
}
return similarityPercent;
}
} // namespace chatterino::similarity::detail
namespace chatterino {
void setSimilarityFlags(const MessagePtr &message,
const std::ranges::bidirectional_range auto &messages)
{
if (getSettings()->similarityEnabled)
{
bool isMyself =
message->loginName ==
getApp()->getAccounts()->twitch.getCurrent()->getUserName();
bool hideMyself = getSettings()->hideSimilarMyself;
if (isMyself && !hideMyself)
{
return;
}
if (similarity::detail::inMessages(message, messages) >
getSettings()->similarityPercentage)
{
message->flags.set(MessageFlag::Similar);
if (getSettings()->colorSimilarDisabled)
{
message->flags.set(MessageFlag::Disabled);
}
}
}
}
} // namespace chatterino

View file

@ -0,0 +1,58 @@
#pragma once
#include "common/enums/MessageContext.hpp"
#include "messages/MessageFlag.hpp"
#include <memory>
class QStringView;
class QTime;
namespace chatterino {
struct Message;
using MessagePtr = std::shared_ptr<const Message>;
enum class MessageSinkTrait : uint8_t {
None = 0,
AddMentionsToGlobalChannel = 1 << 0,
RequiresKnownChannelPointReward = 1 << 1,
};
using MessageSinkTraits = FlagsEnum<MessageSinkTrait>;
/// A generic interface for a managed buffer of `Message`s
class MessageSink
{
public:
virtual ~MessageSink() = default;
/// Add a message to this sink
///
/// @param message The message to add (non-null)
/// @param ctx The context in which this message is being added.
/// @param overridingFlags
virtual void addMessage(
MessagePtr message, MessageContext ctx,
std::optional<MessageFlags> overridingFlags = std::nullopt) = 0;
/// Adds a timeout message or merges it into an existing one
virtual void addOrReplaceTimeout(MessagePtr clearchatMessage,
QTime now) = 0;
/// Flags all messages as `Disabled`
virtual void disableAllMessages() = 0;
/// Searches for similar messages and flags this message as similar
/// (based on the current settings).
virtual void applySimilarityFilters(const MessagePtr &message) const = 0;
/// @brief Searches for a message by an ID
///
/// If there is no message found, an empty shared-pointer is returned.
virtual MessagePtr findMessageByID(QStringView id) = 0;
///
virtual MessageSinkTraits sinkTraits() const = 0;
};
} // namespace chatterino

View file

@ -3,7 +3,9 @@
#include "common/Env.hpp" #include "common/Env.hpp"
#include "messages/MessageBuilder.hpp" #include "messages/MessageBuilder.hpp"
#include "providers/twitch/IrcMessageHandler.hpp" #include "providers/twitch/IrcMessageHandler.hpp"
#include "providers/twitch/TwitchChannel.hpp"
#include "util/Helpers.hpp" #include "util/Helpers.hpp"
#include "util/VectorMessageSink.hpp"
#include <QJsonArray> #include <QJsonArray>
#include <QUrlQuery> #include <QUrlQuery>
@ -40,7 +42,13 @@ std::vector<Communi::IrcMessage *> parseRecentMessages(
std::vector<MessagePtr> buildRecentMessages( std::vector<MessagePtr> buildRecentMessages(
std::vector<Communi::IrcMessage *> &messages, Channel *channel) std::vector<Communi::IrcMessage *> &messages, Channel *channel)
{ {
std::vector<MessagePtr> allBuiltMessages; VectorMessageSink sink({}, MessageFlag::RecentMessage);
auto *twitchChannel = dynamic_cast<TwitchChannel *>(channel);
if (!twitchChannel)
{
return {};
}
for (auto *message : messages) for (auto *message : messages)
{ {
@ -58,24 +66,16 @@ std::vector<MessagePtr> buildRecentMessages(
auto msg = makeSystemMessage( auto msg = makeSystemMessage(
QLocale().toString(msgDate, QLocale::LongFormat), QLocale().toString(msgDate, QLocale::LongFormat),
QTime(0, 0)); QTime(0, 0));
msg->flags.set(MessageFlag::RecentMessage); sink.addMessage(msg, MessageContext::Original);
allBuiltMessages.emplace_back(msg);
} }
} }
auto builtMessages = IrcMessageHandler::parseMessageWithReply( IrcMessageHandler::parseMessageInto(message, sink, twitchChannel);
channel, message, allBuiltMessages);
for (const auto &builtMessage : builtMessages)
{
builtMessage->flags.set(MessageFlag::RecentMessage);
allBuiltMessages.emplace_back(builtMessage);
}
message->deleteLater(); message->deleteLater();
} }
return allBuiltMessages; return std::move(sink).takeMessages();
} }
// Returns the URL to be used for querying the Recent Messages API for the // Returns the URL to be used for querying the Recent Messages API for the

View file

@ -7,24 +7,21 @@
#include "common/QLogging.hpp" #include "common/QLogging.hpp"
#include "controllers/accounts/AccountController.hpp" #include "controllers/accounts/AccountController.hpp"
#include "controllers/ignores/IgnoreController.hpp" #include "controllers/ignores/IgnoreController.hpp"
#include "messages/LimitedQueue.hpp"
#include "messages/Link.hpp" #include "messages/Link.hpp"
#include "messages/Message.hpp" #include "messages/Message.hpp"
#include "messages/MessageBuilder.hpp" #include "messages/MessageBuilder.hpp"
#include "messages/MessageColor.hpp" #include "messages/MessageColor.hpp"
#include "messages/MessageElement.hpp" #include "messages/MessageElement.hpp"
#include "messages/MessageSink.hpp"
#include "messages/MessageThread.hpp" #include "messages/MessageThread.hpp"
#include "providers/twitch/ChannelPointReward.hpp"
#include "providers/twitch/TwitchAccount.hpp" #include "providers/twitch/TwitchAccount.hpp"
#include "providers/twitch/TwitchAccountManager.hpp" #include "providers/twitch/TwitchAccountManager.hpp"
#include "providers/twitch/TwitchChannel.hpp" #include "providers/twitch/TwitchChannel.hpp"
#include "providers/twitch/TwitchHelpers.hpp" #include "providers/twitch/TwitchHelpers.hpp"
#include "providers/twitch/TwitchIrcServer.hpp" #include "providers/twitch/TwitchIrcServer.hpp"
#include "singletons/Resources.hpp"
#include "singletons/Settings.hpp" #include "singletons/Settings.hpp"
#include "singletons/StreamerMode.hpp" #include "singletons/StreamerMode.hpp"
#include "singletons/WindowManager.hpp" #include "singletons/WindowManager.hpp"
#include "util/ChannelHelpers.hpp"
#include "util/FormatTime.hpp" #include "util/FormatTime.hpp"
#include "util/Helpers.hpp" #include "util/Helpers.hpp"
#include "util/IrcHelpers.hpp" #include "util/IrcHelpers.hpp"
@ -34,7 +31,6 @@
#include <QStringBuilder> #include <QStringBuilder>
#include <memory> #include <memory>
#include <unordered_set>
using namespace chatterino::literals; using namespace chatterino::literals;
@ -165,50 +161,6 @@ ChannelPtr channelOrEmptyByTarget(const QString &target,
return server.getChannelOrEmpty(channelName); return server.getChannelOrEmpty(channelName);
} }
float relativeSimilarity(const QString &str1, const QString &str2)
{
// Longest Common Substring Problem
std::vector<std::vector<int>> tree(str1.size(),
std::vector<int>(str2.size(), 0));
int z = 0;
for (int i = 0; i < str1.size(); ++i)
{
for (int j = 0; j < str2.size(); ++j)
{
if (str1[i] == str2[j])
{
if (i == 0 || j == 0)
{
tree[i][j] = 1;
}
else
{
tree[i][j] = tree[i - 1][j - 1] + 1;
}
if (tree[i][j] > z)
{
z = tree[i][j];
}
}
else
{
tree[i][j] = 0;
}
}
}
// ensure that no div by 0
if (z == 0)
{
return 0.F;
}
auto div = std::max<int>(1, std::max(str1.size(), str2.size()));
return float(z) / float(div);
}
QMap<QString, QString> parseBadges(const QString &badgesString) QMap<QString, QString> parseBadges(const QString &badgesString)
{ {
QMap<QString, QString> badges; QMap<QString, QString> badges;
@ -232,106 +184,6 @@ struct ReplyContext {
MessagePtr parent; MessagePtr parent;
}; };
[[nodiscard]] ReplyContext getReplyContext(
TwitchChannel *channel, Communi::IrcMessage *message,
const std::vector<MessagePtr> &otherLoaded)
{
ReplyContext ctx;
const auto &tags = message->tags();
if (const auto it = tags.find("reply-thread-parent-msg-id");
it != tags.end())
{
const QString replyID = it.value().toString();
auto threadIt = channel->threads().find(replyID);
std::shared_ptr<MessageThread> rootThread;
if (threadIt != channel->threads().end())
{
auto owned = threadIt->second.lock();
if (owned)
{
// Thread already exists (has a reply)
checkThreadSubscription(tags, message->nick(), owned);
ctx.thread = owned;
rootThread = owned;
}
}
if (!rootThread)
{
MessagePtr foundMessage;
// Thread does not yet exist, find root reply and create thread.
// Linear search is justified by the infrequent use of replies
for (const auto &otherMsg : otherLoaded)
{
if (otherMsg->id == replyID)
{
// Found root reply message
foundMessage = otherMsg;
break;
}
}
if (!foundMessage)
{
// We didn't find the reply root message in the otherLoaded messages
// which are typically the already-parsed recent messages from the
// Recent Messages API. We could have a really old message that
// still exists being replied to, so check for that here.
foundMessage = channel->findMessage(replyID);
}
if (foundMessage)
{
std::shared_ptr<MessageThread> newThread =
std::make_shared<MessageThread>(foundMessage);
checkThreadSubscription(tags, message->nick(), newThread);
ctx.thread = newThread;
rootThread = newThread;
// Store weak reference to thread in channel
channel->addReplyThread(newThread);
}
}
if (const auto parentIt = tags.find("reply-parent-msg-id");
parentIt != tags.end())
{
const QString parentID = parentIt.value().toString();
if (replyID == parentID)
{
if (rootThread)
{
ctx.parent = rootThread->root();
}
}
else
{
auto parentThreadIt = channel->threads().find(parentID);
if (parentThreadIt != channel->threads().end())
{
auto thread = parentThreadIt->second.lock();
if (thread)
{
ctx.parent = thread->root();
}
}
else
{
auto parent = channel->findMessage(parentID);
if (parent)
{
ctx.parent = parent;
}
}
}
}
}
return ctx;
}
std::optional<ClearChatMessage> parseClearChatMessage( std::optional<ClearChatMessage> parseClearChatMessage(
Communi::IrcMessage *message) Communi::IrcMessage *message)
{ {
@ -370,9 +222,9 @@ std::optional<ClearChatMessage> parseClearChatMessage(
} }
/** /**
* Parse a single IRC NOTICE message into 0 or more Chatterino messages * Parse a single IRC NOTICE message into a Chatterino message
**/ **/
std::vector<MessagePtr> parseNoticeMessage(Communi::IrcNoticeMessage *message) MessagePtr parseNoticeMessage(Communi::IrcNoticeMessage *message)
{ {
assert(message != nullptr); assert(message != nullptr);
@ -400,7 +252,7 @@ std::vector<MessagePtr> parseNoticeMessage(Communi::IrcNoticeMessage *message)
linkColor) linkColor)
->setLink(accountsLink); ->setLink(accountsLink);
return {builder.release()}; return builder.release();
} }
if (message->content().startsWith("You are permanently banned ")) if (message->content().startsWith("You are permanently banned "))
@ -410,256 +262,19 @@ std::vector<MessagePtr> parseNoticeMessage(Communi::IrcNoticeMessage *message)
if (message->tags().value("msg-id") == "msg_timedout") if (message->tags().value("msg-id") == "msg_timedout")
{ {
std::vector<MessagePtr> builtMessage;
QString remainingTime = QString remainingTime =
formatTime(message->content().split(" ").value(5)); formatTime(message->content().split(" ").value(5));
QString formattedMessage = QString formattedMessage =
QString("You are timed out for %1.") QString("You are timed out for %1.")
.arg(remainingTime.isEmpty() ? "0s" : remainingTime); .arg(remainingTime.isEmpty() ? "0s" : remainingTime);
builtMessage.emplace_back(makeSystemMessage( return makeSystemMessage(formattedMessage,
formattedMessage, calculateMessageTime(message).time())); calculateMessageTime(message).time());
return builtMessage;
} }
// default case // default case
std::vector<MessagePtr> builtMessages; return makeSystemMessage(message->content(),
builtMessages.emplace_back(makeSystemMessage(
message->content(), calculateMessageTime(message).time()));
return builtMessages;
}
/**
* Parse a single IRC USERNOTICE message into 0 or more Chatterino messages
**/
std::vector<MessagePtr> parseUserNoticeMessage(Channel *channel,
Communi::IrcMessage *message)
{
assert(channel != nullptr);
assert(message != nullptr);
std::vector<MessagePtr> builtMessages;
auto tags = message->tags();
auto parameters = message->parameters();
QString msgType = tags.value("msg-id").toString();
bool mirrored = msgType == "sharedchatnotice";
if (mirrored)
{
msgType = tags.value("source-msg-id").toString();
}
else
{
auto rIt = tags.find("room-id");
auto sIt = tags.find("source-room-id");
if (rIt != tags.end() && sIt != tags.end())
{
mirrored = rIt.value().toString() != sIt.value().toString();
}
}
if (mirrored && msgType != "announcement")
{
// avoid confusing broadcasters with user payments to other channels
return {};
}
QString content;
if (parameters.size() >= 2)
{
content = parameters[1];
}
if (isIgnoredMessage({
.message = content,
.twitchUserID = tags.value("user-id").toString(),
.isMod = channel->isMod(),
.isBroadcaster = channel->isBroadcaster(),
}))
{
return {};
}
if (SPECIAL_MESSAGE_TYPES.contains(msgType))
{
// Messages are not required, so they might be empty
if (!content.isEmpty())
{
MessageParseArgs args;
args.trimSubscriberUsername = true;
args.allowIgnore = false;
auto [built, highlight] = MessageBuilder::makeIrcMessage(
channel, message, args, content, 0);
if (built)
{
built->flags.set(MessageFlag::Subscription);
built->flags.unset(MessageFlag::Highlighted);
if (mirrored)
{
built->flags.set(MessageFlag::SharedMessage);
}
builtMessages.emplace_back(std::move(built));
}
}
}
auto it = tags.find("system-msg");
if (it != tags.end())
{
// By default, we return value of system-msg tag
QString messageText = it.value().toString();
if (msgType == "bitsbadgetier")
{
messageText =
QString("%1 just earned a new %2 Bits badge!")
.arg(tags.value("display-name").toString(),
kFormatNumbers(
tags.value("msg-param-threshold").toInt()));
}
else if (msgType == "announcement")
{
messageText = "Announcement";
}
else if (msgType == "raid")
{
auto login = tags.value("login").toString();
auto displayName = tags.value("msg-param-displayName").toString();
if (!login.isEmpty() && !displayName.isEmpty())
{
MessageColor color = MessageColor::System;
if (auto colorTag = tags.value("color").value<QColor>();
colorTag.isValid())
{
color = MessageColor(colorTag);
}
auto b = MessageBuilder(
raidEntryMessage, parseTagString(messageText), login,
displayName, color, calculateMessageTime(message).time());
b->flags.set(MessageFlag::Subscription);
if (mirrored)
{
b->flags.set(MessageFlag::SharedMessage);
}
auto newMessage = b.release();
builtMessages.emplace_back(newMessage);
return builtMessages;
}
}
else if (msgType == "subgift")
{
if (auto monthsIt = tags.find("msg-param-gift-months");
monthsIt != tags.end())
{
int months = monthsIt.value().toInt();
if (months > 1)
{
auto plan = tags.value("msg-param-sub-plan").toString();
QString name =
ANONYMOUS_GIFTER_ID == tags.value("user-id").toString()
? "An anonymous user"
: tags.value("display-name").toString();
messageText =
QString("%1 gifted %2 months of a Tier %3 sub to %4!")
.arg(name, QString::number(months),
plan.isEmpty() ? '1' : plan.at(0),
tags.value("msg-param-recipient-display-name")
.toString());
if (auto countIt = tags.find("msg-param-sender-count");
countIt != tags.end())
{
int count = countIt.value().toInt();
if (count > months)
{
messageText +=
QString(
" They've gifted %1 months in the channel.")
.arg(QString::number(count));
}
}
}
}
}
else if (msgType == "sub" || msgType == "resub")
{
if (auto tenure = tags.find("msg-param-multimonth-tenure");
tenure != tags.end() && tenure.value().toInt() == 0)
{
int months =
tags.value("msg-param-multimonth-duration").toInt();
if (months > 1)
{
int tier = tags.value("msg-param-sub-plan").toInt() / 1000;
messageText =
QString(
"%1 subscribed at Tier %2 for %3 months in advance")
.arg(tags.value("display-name").toString(),
QString::number(tier),
QString::number(months));
if (msgType == "resub")
{
int cumulative =
tags.value("msg-param-cumulative-months").toInt();
messageText +=
QString(", reaching %1 months cumulatively so far!")
.arg(QString::number(cumulative));
}
else
{
messageText += "!";
}
}
}
}
auto b = MessageBuilder(systemMessage, parseTagString(messageText),
calculateMessageTime(message).time()); calculateMessageTime(message).time());
b->flags.set(MessageFlag::Subscription);
if (mirrored)
{
b->flags.set(MessageFlag::SharedMessage);
}
auto newMessage = b.release();
builtMessages.emplace_back(newMessage);
}
return builtMessages;
}
/**
* Parse a single IRC PRIVMSG into 0-1 Chatterino messages
*/
std::vector<MessagePtr> parsePrivMessage(Channel *channel,
Communi::IrcPrivateMessage *message)
{
assert(channel != nullptr);
assert(message != nullptr);
std::vector<MessagePtr> builtMessages;
MessageParseArgs args;
args.isAction = message->isAction();
auto [built, alert] = MessageBuilder::makeIrcMessage(channel, message, args,
message->content(), 0);
if (built)
{
builtMessages.emplace_back(std::move(built));
MessageBuilder::triggerHighlights(channel, alert);
}
return builtMessages;
} }
} // namespace } // namespace
@ -674,65 +289,27 @@ IrcMessageHandler &IrcMessageHandler::instance()
return instance; return instance;
} }
std::vector<MessagePtr> IrcMessageHandler::parseMessageWithReply( void IrcMessageHandler::parseMessageInto(Communi::IrcMessage *message,
Channel *channel, Communi::IrcMessage *message, MessageSink &sink,
std::vector<MessagePtr> &otherLoaded) TwitchChannel *channel)
{ {
std::vector<MessagePtr> builtMessages;
auto command = message->command(); auto command = message->command();
if (command == u"PRIVMSG"_s) if (command == u"PRIVMSG"_s)
{ {
auto *privMsg = dynamic_cast<Communi::IrcPrivateMessage *>(message); parsePrivMessageInto(
auto *tc = dynamic_cast<TwitchChannel *>(channel); dynamic_cast<Communi::IrcPrivateMessage *>(message), sink, channel);
if (!tc)
{
return parsePrivMessage(channel, privMsg);
} }
else if (command == u"USERNOTICE"_s)
QString content = privMsg->content();
int messageOffset = stripLeadingReplyMention(privMsg->tags(), content);
MessageParseArgs args;
auto tags = privMsg->tags();
if (const auto it = tags.find("custom-reward-id"); it != tags.end())
{ {
args.channelPointRewardId = it.value().toString(); parseUserNoticeMessageInto(message, sink, channel);
}
args.isAction = privMsg->isAction();
auto replyCtx = getReplyContext(tc, message, otherLoaded);
auto [built, alert] = MessageBuilder::makeIrcMessage(
channel, message, args, content, messageOffset, replyCtx.thread,
replyCtx.parent);
if (built)
{
builtMessages.emplace_back(built);
MessageBuilder::triggerHighlights(channel, alert);
}
if (message->tags().contains(u"pinned-chat-paid-amount"_s))
{
auto ptr = MessageBuilder::buildHypeChatMessage(privMsg);
if (ptr)
{
builtMessages.emplace_back(std::move(ptr));
}
}
return builtMessages;
}
if (command == u"USERNOTICE"_s)
{
return parseUserNoticeMessage(channel, message);
} }
if (command == u"NOTICE"_s) if (command == u"NOTICE"_s)
{ {
return parseNoticeMessage( sink.addMessage(parseNoticeMessage(
dynamic_cast<Communi::IrcNoticeMessage *>(message)); dynamic_cast<Communi::IrcNoticeMessage *>(message)),
MessageContext::Original);
} }
if (command == u"CLEARCHAT"_s) if (command == u"CLEARCHAT"_s)
@ -740,32 +317,20 @@ std::vector<MessagePtr> IrcMessageHandler::parseMessageWithReply(
auto cc = parseClearChatMessage(message); auto cc = parseClearChatMessage(message);
if (!cc) if (!cc)
{ {
return builtMessages; return;
} }
auto &clearChat = *cc; auto &clearChat = *cc;
if (clearChat.disableAllMessages) if (clearChat.disableAllMessages)
{ {
builtMessages.emplace_back(std::move(clearChat.message)); sink.addMessage(std::move(clearChat.message),
MessageContext::Original);
} }
else else
{ {
addOrReplaceChannelTimeout( sink.addOrReplaceTimeout(std::move(clearChat.message),
otherLoaded, std::move(clearChat.message), calculateMessageTime(message).time());
calculateMessageTime(message).time(),
[&](auto idx, auto /*msg*/, auto &&replacement) {
replacement->flags.set(MessageFlag::RecentMessage);
otherLoaded[idx] = replacement;
},
[&](auto &&msg) {
builtMessages.emplace_back(msg);
},
false);
} }
return builtMessages;
} }
return builtMessages;
} }
void IrcMessageHandler::handlePrivMessage(Communi::IrcPrivateMessage *message, void IrcMessageHandler::handlePrivMessage(Communi::IrcPrivateMessage *message,
@ -778,9 +343,18 @@ void IrcMessageHandler::handlePrivMessage(Communi::IrcPrivateMessage *message,
} }
auto *twitchChannel = dynamic_cast<TwitchChannel *>(chan.get()); auto *twitchChannel = dynamic_cast<TwitchChannel *>(chan.get());
if (!twitchChannel)
if (twitchChannel != nullptr)
{ {
return;
}
parsePrivMessageInto(message, *twitchChannel, twitchChannel);
}
void IrcMessageHandler::parsePrivMessageInto(
Communi::IrcPrivateMessage *message, MessageSink &sink,
TwitchChannel *channel)
{
auto currentUser = getApp()->getAccounts()->twitch.getCurrent(); auto currentUser = getApp()->getAccounts()->twitch.getCurrent();
if (message->tag("user-id") == currentUser->getUserId()) if (message->tag("user-id") == currentUser->getUserId())
{ {
@ -788,22 +362,22 @@ void IrcMessageHandler::handlePrivMessage(Communi::IrcPrivateMessage *message,
if (badgesTag.isValid()) if (badgesTag.isValid())
{ {
auto parsedBadges = parseBadges(badgesTag.toString()); auto parsedBadges = parseBadges(badgesTag.toString());
twitchChannel->setMod(parsedBadges.contains("moderator")); channel->setMod(parsedBadges.contains("moderator"));
twitchChannel->setVIP(parsedBadges.contains("vip")); channel->setVIP(parsedBadges.contains("vip"));
twitchChannel->setStaff(parsedBadges.contains("staff")); channel->setStaff(parsedBadges.contains("staff"));
}
} }
} }
this->addMessage(message, chan, unescapeZeroWidthJoiner(message->content()), addMessage(message, sink, channel,
twitchServer, false, message->isAction()); unescapeZeroWidthJoiner(message->content()), false,
message->isAction());
if (message->tags().contains(u"pinned-chat-paid-amount"_s)) if (message->tags().contains(u"pinned-chat-paid-amount"_s))
{ {
auto ptr = MessageBuilder::buildHypeChatMessage(message); auto ptr = MessageBuilder::buildHypeChatMessage(message);
if (ptr) if (ptr)
{ {
chan->addMessage(ptr, MessageContext::Original); sink.addMessage(ptr, MessageContext::Original);
} }
} }
} }
@ -900,7 +474,8 @@ void IrcMessageHandler::handleClearChatMessage(Communi::IrcMessage *message)
return; return;
} }
chan->addOrReplaceTimeout(std::move(clearChat.message)); chan->addOrReplaceTimeout(std::move(clearChat.message),
calculateMessageTime(message).time());
// refresh all // refresh all
getApp()->getWindows()->repaintVisibleChatWidgets(chan.get()); getApp()->getWindows()->repaintVisibleChatWidgets(chan.get());
@ -1039,11 +614,24 @@ void IrcMessageHandler::handleWhisperMessage(Communi::IrcMessage *ircMessage)
void IrcMessageHandler::handleUserNoticeMessage(Communi::IrcMessage *message, void IrcMessageHandler::handleUserNoticeMessage(Communi::IrcMessage *message,
ITwitchIrcServer &twitchServer) ITwitchIrcServer &twitchServer)
{
auto target = message->parameter(0);
auto *channel = dynamic_cast<TwitchChannel *>(
twitchServer.getChannelOrEmpty(target).get());
if (!channel)
{
return;
}
parseUserNoticeMessageInto(message, *channel, channel);
}
void IrcMessageHandler::parseUserNoticeMessageInto(Communi::IrcMessage *message,
MessageSink &sink,
TwitchChannel *channel)
{ {
auto tags = message->tags(); auto tags = message->tags();
auto parameters = message->parameters(); auto parameters = message->parameters();
auto target = parameters[0];
QString msgType = tags.value("msg-id").toString(); QString msgType = tags.value("msg-id").toString();
bool mirrored = msgType == "sharedchatnotice"; bool mirrored = msgType == "sharedchatnotice";
if (mirrored) if (mirrored)
@ -1072,12 +660,11 @@ void IrcMessageHandler::handleUserNoticeMessage(Communi::IrcMessage *message,
content = parameters[1]; content = parameters[1];
} }
auto chn = twitchServer.getChannelOrEmpty(target);
if (isIgnoredMessage({ if (isIgnoredMessage({
.message = content, .message = content,
.twitchUserID = tags.value("user-id").toString(), .twitchUserID = tags.value("user-id").toString(),
.isMod = chn->isMod(), .isMod = channel->isMod(),
.isBroadcaster = chn->isBroadcaster(), .isBroadcaster = channel->isBroadcaster(),
})) }))
{ {
return; return;
@ -1088,7 +675,7 @@ void IrcMessageHandler::handleUserNoticeMessage(Communi::IrcMessage *message,
// Messages are not required, so they might be empty // Messages are not required, so they might be empty
if (!content.isEmpty()) if (!content.isEmpty())
{ {
this->addMessage(message, chn, content, twitchServer, true, false); addMessage(message, sink, channel, content, true, false);
} }
} }
@ -1136,25 +723,7 @@ void IrcMessageHandler::handleUserNoticeMessage(Communi::IrcMessage *message,
} }
auto newMessage = b.release(); auto newMessage = b.release();
QString channelName; sink.addMessage(newMessage, MessageContext::Original);
if (message->parameters().size() < 1)
{
return;
}
if (!trimChannelName(message->parameter(0), channelName))
{
return;
}
auto chan = twitchServer.getChannelOrEmpty(channelName);
if (!chan->isEmpty())
{
chan->addMessage(newMessage, MessageContext::Original);
}
return; return;
} }
} }
@ -1235,33 +804,14 @@ void IrcMessageHandler::handleUserNoticeMessage(Communi::IrcMessage *message,
} }
auto newMessage = b.release(); auto newMessage = b.release();
QString channelName; sink.addMessage(newMessage, MessageContext::Original);
if (message->parameters().size() < 1)
{
return;
}
if (!trimChannelName(message->parameter(0), channelName))
{
return;
}
auto chan = twitchServer.getChannelOrEmpty(channelName);
if (!chan->isEmpty())
{
chan->addMessage(newMessage, MessageContext::Original);
}
} }
} }
void IrcMessageHandler::handleNoticeMessage(Communi::IrcNoticeMessage *message) void IrcMessageHandler::handleNoticeMessage(Communi::IrcNoticeMessage *message)
{ {
auto builtMessages = parseNoticeMessage(message); auto msg = parseNoticeMessage(message);
for (const auto &msg : builtMessages)
{
QString channelName; QString channelName;
if (!trimChannelName(message->target(), channelName) || if (!trimChannelName(message->target(), channelName) ||
channelName == "jtv") channelName == "jtv")
@ -1316,8 +866,8 @@ void IrcMessageHandler::handleNoticeMessage(Communi::IrcNoticeMessage *message)
{ {
hostedChannelName.chop(1); hostedChannelName.chop(1);
} }
channel->addMessage(MessageBuilder::makeHostingSystemMessage( channel->addMessage(
hostedChannelName, hostOn), MessageBuilder::makeHostingSystemMessage(hostedChannelName, hostOn),
MessageContext::Original); MessageContext::Original);
} }
else if (tags == "room_mods" || tags == "vips_success") else if (tags == "room_mods" || tags == "vips_success")
@ -1345,15 +895,14 @@ void IrcMessageHandler::handleNoticeMessage(Communi::IrcNoticeMessage *message)
.mid(1) // there is a space before the first user .mid(1) // there is a space before the first user
.split(", "); .split(", ");
users.sort(Qt::CaseInsensitive); users.sort(Qt::CaseInsensitive);
channel->addMessage(MessageBuilder::makeListOfUsersMessage( channel->addMessage(
msgParts.at(0), users, tc), MessageBuilder::makeListOfUsersMessage(msgParts.at(0), users, tc),
MessageContext::Original); MessageContext::Original);
} }
else else
{ {
channel->addMessage(msg, MessageContext::Original); channel->addMessage(msg, MessageContext::Original);
} }
}
} }
void IrcMessageHandler::handleJoinMessage(Communi::IrcMessage *message) void IrcMessageHandler::handleJoinMessage(Communi::IrcMessage *message)
@ -1405,76 +954,12 @@ void IrcMessageHandler::handlePartMessage(Communi::IrcMessage *message)
} }
} }
float IrcMessageHandler::similarity(
const MessagePtr &msg, const LimitedQueueSnapshot<MessagePtr> &messages)
{
float similarityPercent = 0.0F;
int checked = 0;
for (int i = 1; i <= messages.size(); ++i)
{
if (checked >= getSettings()->hideSimilarMaxMessagesToCheck)
{
break;
}
const auto &prevMsg = messages[messages.size() - i];
if (prevMsg->parseTime.secsTo(QTime::currentTime()) >=
getSettings()->hideSimilarMaxDelay)
{
break;
}
if (getSettings()->hideSimilarBySameUser &&
msg->loginName != prevMsg->loginName)
{
continue;
}
++checked;
similarityPercent = std::max(
similarityPercent,
relativeSimilarity(msg->messageText, prevMsg->messageText));
}
return similarityPercent;
}
void IrcMessageHandler::setSimilarityFlags(const MessagePtr &message,
const ChannelPtr &channel)
{
if (getSettings()->similarityEnabled)
{
bool isMyself =
message->loginName ==
getApp()->getAccounts()->twitch.getCurrent()->getUserName();
bool hideMyself = getSettings()->hideSimilarMyself;
if (isMyself && !hideMyself)
{
return;
}
if (IrcMessageHandler::similarity(message,
channel->getMessageSnapshot()) >
getSettings()->similarityPercentage)
{
message->flags.set(MessageFlag::Similar, true);
if (getSettings()->colorSimilarDisabled)
{
message->flags.set(MessageFlag::Disabled, true);
}
}
}
}
void IrcMessageHandler::addMessage(Communi::IrcMessage *message, void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
const ChannelPtr &chan, MessageSink &sink, TwitchChannel *channel,
const QString &originalContent, const QString &originalContent, bool isSub,
ITwitchIrcServer &server, bool isSub,
bool isAction) bool isAction)
{ {
if (chan->isEmpty()) assert(channel);
{
return;
}
MessageParseArgs args; MessageParseArgs args;
if (isSub) if (isSub)
@ -1483,14 +968,12 @@ void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
args.trimSubscriberUsername = true; args.trimSubscriberUsername = true;
} }
if (chan->isBroadcaster()) if (channel->isBroadcaster())
{ {
args.isStaffOrBroadcaster = true; args.isStaffOrBroadcaster = true;
} }
args.isAction = isAction; args.isAction = isAction;
auto *channel = dynamic_cast<TwitchChannel *>(chan.get());
const auto &tags = message->tags(); const auto &tags = message->tags();
QString rewardId; QString rewardId;
if (const auto it = tags.find("custom-reward-id"); it != tags.end()) if (const auto it = tags.find("custom-reward-id"); it != tags.end())
@ -1506,7 +989,10 @@ void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
rewardId = msgId; rewardId = msgId;
} }
} }
if (!rewardId.isEmpty() && !channel->isChannelPointRewardKnown(rewardId)) if (!rewardId.isEmpty() &&
sink.sinkTraits().has(
MessageSinkTrait::RequiresKnownChannelPointReward) &&
!channel->isChannelPointRewardKnown(rewardId))
{ {
// Need to wait for pubsub reward notification // Need to wait for pubsub reward notification
qCDebug(chatterinoTwitch) << "TwitchChannel reward added ADD " qCDebug(chatterinoTwitch) << "TwitchChannel reward added ADD "
@ -1539,7 +1025,7 @@ void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
else else
{ {
// Thread does not yet exist, find root reply and create thread. // Thread does not yet exist, find root reply and create thread.
auto root = channel->findMessage(replyID); auto root = sink.findMessageByID(replyID);
if (root) if (root)
{ {
// Found root reply message // Found root reply message
@ -1577,7 +1063,7 @@ void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
} }
else else
{ {
auto parent = channel->findMessage(parentID); auto parent = sink.findMessageByID(parentID);
if (parent) if (parent)
{ {
replyCtx.parent = parent; replyCtx.parent = parent;
@ -1600,7 +1086,7 @@ void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
msg->flags.unset(MessageFlag::Highlighted); msg->flags.unset(MessageFlag::Highlighted);
} }
IrcMessageHandler::setSimilarityFlags(msg, chan); sink.applySimilarityFilters(msg);
if (!msg->flags.has(MessageFlag::Similar) || if (!msg->flags.has(MessageFlag::Similar) ||
(!getSettings()->hideSimilar && (!getSettings()->hideSimilar &&
@ -1612,17 +1098,15 @@ void IrcMessageHandler::addMessage(Communi::IrcMessage *message,
const auto highlighted = msg->flags.has(MessageFlag::Highlighted); const auto highlighted = msg->flags.has(MessageFlag::Highlighted);
const auto showInMentions = msg->flags.has(MessageFlag::ShowInMentions); const auto showInMentions = msg->flags.has(MessageFlag::ShowInMentions);
if (highlighted && showInMentions) if (highlighted && showInMentions &&
sink.sinkTraits().has(MessageSinkTrait::AddMentionsToGlobalChannel))
{ {
server.getMentionsChannel()->addMessage(msg, getApp()->getTwitch()->getMentionsChannel()->addMessage(
MessageContext::Original); msg, MessageContext::Original);
} }
chan->addMessage(msg, MessageContext::Original); sink.addMessage(msg, MessageContext::Original);
if (auto *chatters = dynamic_cast<ChannelChatters *>(chan.get())) channel->addRecentChatter(msg->displayName);
{
chatters->addRecentChatter(msg->displayName);
}
} }
} }

View file

@ -16,6 +16,7 @@ struct Message;
using MessagePtr = std::shared_ptr<const Message>; using MessagePtr = std::shared_ptr<const Message>;
class TwitchChannel; class TwitchChannel;
class TwitchMessageBuilder; class TwitchMessageBuilder;
class MessageSink;
struct ClearChatMessage { struct ClearChatMessage {
MessagePtr message; MessagePtr message;
@ -33,30 +34,35 @@ public:
* Parse an IRC message into 0 or more Chatterino messages * Parse an IRC message into 0 or more Chatterino messages
* Takes previously loaded messages into consideration to add reply contexts * Takes previously loaded messages into consideration to add reply contexts
**/ **/
static std::vector<MessagePtr> parseMessageWithReply( static void parseMessageInto(Communi::IrcMessage *message,
Channel *channel, Communi::IrcMessage *message, MessageSink &sink, TwitchChannel *channel);
std::vector<MessagePtr> &otherLoaded);
void handlePrivMessage(Communi::IrcPrivateMessage *message, void handlePrivMessage(Communi::IrcPrivateMessage *message,
ITwitchIrcServer &twitchServer); ITwitchIrcServer &twitchServer);
static void parsePrivMessageInto(Communi::IrcPrivateMessage *message,
MessageSink &sink, TwitchChannel *channel);
void handleRoomStateMessage(Communi::IrcMessage *message); void handleRoomStateMessage(Communi::IrcMessage *message);
void handleClearChatMessage(Communi::IrcMessage *message); void handleClearChatMessage(Communi::IrcMessage *message);
void handleClearMessageMessage(Communi::IrcMessage *message); void handleClearMessageMessage(Communi::IrcMessage *message);
void handleUserStateMessage(Communi::IrcMessage *message); void handleUserStateMessage(Communi::IrcMessage *message);
void handleWhisperMessage(Communi::IrcMessage *ircMessage);
void handleWhisperMessage(Communi::IrcMessage *ircMessage);
void handleUserNoticeMessage(Communi::IrcMessage *message, void handleUserNoticeMessage(Communi::IrcMessage *message,
ITwitchIrcServer &twitchServer); ITwitchIrcServer &twitchServer);
static void parseUserNoticeMessageInto(Communi::IrcMessage *message,
MessageSink &sink,
TwitchChannel *channel);
void handleNoticeMessage(Communi::IrcNoticeMessage *message); void handleNoticeMessage(Communi::IrcNoticeMessage *message);
void handleJoinMessage(Communi::IrcMessage *message); void handleJoinMessage(Communi::IrcMessage *message);
void handlePartMessage(Communi::IrcMessage *message); void handlePartMessage(Communi::IrcMessage *message);
void addMessage(Communi::IrcMessage *message, const ChannelPtr &chan, static void addMessage(Communi::IrcMessage *message, MessageSink &sink,
const QString &originalContent, ITwitchIrcServer &server, TwitchChannel *channel,
bool isSub, bool isAction); const QString &originalContent, bool isSub,
bool isAction);
private: private:
static float similarity(const MessagePtr &msg, static float similarity(const MessagePtr &msg,

View file

@ -452,8 +452,8 @@ void TwitchChannel::addChannelPointReward(const ChannelPointReward &reward)
if (reward.id == msg.rewardID) if (reward.id == msg.rewardID)
{ {
IrcMessageHandler::instance().addMessage( IrcMessageHandler::instance().addMessage(
msg.message.get(), shared_from_this(), msg.message.get(), *this, this, msg.originalContent,
msg.originalContent, *server, false, false); false, false);
return true; return true;
} }
return false; return false;

View file

@ -312,7 +312,7 @@ void TwitchIrcServer::initialize()
postToThread([chan, action] { postToThread([chan, action] {
MessageBuilder msg(action); MessageBuilder msg(action);
msg->flags.set(MessageFlag::PubSub); msg->flags.set(MessageFlag::PubSub);
chan->addOrReplaceTimeout(msg.release()); chan->addOrReplaceTimeout(msg.release(), QTime::currentTime());
}); });
}); });

View file

@ -0,0 +1,86 @@
#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,
QTime 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::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

View file

@ -0,0 +1,36 @@
#pragma once
#include "messages/MessageSink.hpp"
namespace chatterino {
class VectorMessageSink final : public MessageSink
{
public:
VectorMessageSink(MessageSinkTraits traits = {},
MessageFlags additionalFlags = {});
~VectorMessageSink() override;
void addMessage(
MessagePtr message, MessageContext ctx,
std::optional<MessageFlags> overridingFlags = std::nullopt) override;
void addOrReplaceTimeout(MessagePtr clearchatMessage, QTime now) override;
void disableAllMessages() override;
void applySimilarityFilters(const MessagePtr &message) const override;
MessagePtr findMessageByID(QStringView id) override;
MessageSinkTraits sinkTraits() const override;
const std::vector<MessagePtr> &messages() const;
std::vector<MessagePtr> takeMessages() &&;
private:
std::vector<MessagePtr> messages_;
MessageFlags additionalFlags;
MessageSinkTraits traits;
};
} // namespace chatterino

View file

@ -180,6 +180,7 @@
} }
], ],
"flags": "Collapsed|Subscription", "flags": "Collapsed|Subscription",
"highlightColor": "#64c466ff",
"id": "8c26e1ab-b50c-4d9d-bc11-3fd57a941d90", "id": "8c26e1ab-b50c-4d9d-bc11-3fd57a941d90",
"localizedName": "", "localizedName": "",
"loginName": "supinic", "loginName": "supinic",

View file

@ -80,7 +80,7 @@
"trailingSpace": true, "trailingSpace": true,
"type": "SingleLineTextElement", "type": "SingleLineTextElement",
"words": [ "words": [
"a" "b"
] ]
}, },
{ {
@ -169,6 +169,7 @@
"localizedName": "", "localizedName": "",
"loginName": "nerixyz", "loginName": "nerixyz",
"messageText": "c", "messageText": "c",
"replyParent": "474f19ab-a1b0-410a-877a-5b0e2ae8be6d",
"replyThread": { "replyThread": {
"replies": [ "replies": [
"474f19ab-a1b0-410a-877a-5b0e2ae8be6d", "474f19ab-a1b0-410a-877a-5b0e2ae8be6d",

View file

@ -256,6 +256,7 @@
} }
], ],
"flags": "Collapsed|Subscription|SharedMessage", "flags": "Collapsed|Subscription|SharedMessage",
"highlightColor": "#64c466ff",
"id": "01cd601f-bc3f-49d5-ab4b-136fa9d6ec22", "id": "01cd601f-bc3f-49d5-ab4b-136fa9d6ec22",
"localizedName": "", "localizedName": "",
"loginName": "lahoooo", "loginName": "lahoooo",

View file

@ -243,6 +243,7 @@
} }
], ],
"flags": "Collapsed|Subscription", "flags": "Collapsed|Subscription",
"highlightColor": "#64c466ff",
"id": "db25007f-7a18-43eb-9379-80131e44d633", "id": "db25007f-7a18-43eb-9379-80131e44d633",
"localizedName": "", "localizedName": "",
"loginName": "ronni", "loginName": "ronni",

View file

@ -27,6 +27,7 @@
#include "singletons/Emotes.hpp" #include "singletons/Emotes.hpp"
#include "Test.hpp" #include "Test.hpp"
#include "util/IrcHelpers.hpp" #include "util/IrcHelpers.hpp"
#include "util/VectorMessageSink.hpp"
#include <IrcConnection> #include <IrcConnection>
#include <QDebug> #include <QDebug>
@ -572,19 +573,14 @@ TEST_P(TestIrcMessageHandlerP, Run)
{ {
auto channel = makeMockTwitchChannel(u"pajlada"_s, *snapshot); auto channel = makeMockTwitchChannel(u"pajlada"_s, *snapshot);
std::vector<MessagePtr> prevMessages; VectorMessageSink sink;
for (auto prevInput : snapshot->param("prevMessages").toArray()) for (auto prevInput : snapshot->param("prevMessages").toArray())
{ {
auto *ircMessage = Communi::IrcMessage::fromData( auto *ircMessage = Communi::IrcMessage::fromData(
prevInput.toString().toUtf8(), nullptr); prevInput.toString().toUtf8(), nullptr);
ASSERT_NE(ircMessage, nullptr); ASSERT_NE(ircMessage, nullptr);
auto builtMessages = IrcMessageHandler::parseMessageWithReply( IrcMessageHandler::parseMessageInto(ircMessage, sink, channel.get());
channel.get(), ircMessage, prevMessages);
for (const auto &builtMessage : builtMessages)
{
prevMessages.emplace_back(builtMessage);
}
delete ircMessage; delete ircMessage;
} }
@ -592,13 +588,13 @@ TEST_P(TestIrcMessageHandlerP, Run)
Communi::IrcMessage::fromData(snapshot->inputUtf8(), nullptr); Communi::IrcMessage::fromData(snapshot->inputUtf8(), nullptr);
ASSERT_NE(ircMessage, nullptr); ASSERT_NE(ircMessage, nullptr);
auto builtMessages = IrcMessageHandler::parseMessageWithReply( auto firstAddedMsg = sink.messages().size();
channel.get(), ircMessage, prevMessages); IrcMessageHandler::parseMessageInto(ircMessage, sink, channel.get());
QJsonArray got; QJsonArray got;
for (const auto &msg : builtMessages) for (auto i = firstAddedMsg; i < sink.messages().size(); i++)
{ {
got.append(msg->toJson()); got.append(sink.messages()[i]->toJson());
} }
delete ircMessage; delete ircMessage;