mirror-chatterino2/src/messages/MessageBuilder.cpp

200 lines
5.6 KiB
C++
Raw Normal View History

2018-06-26 14:09:39 +02:00
#include "MessageBuilder.hpp"
#include "common/LinkParser.hpp"
2018-06-28 19:46:45 +02:00
#include "singletons/Emotes.hpp"
#include "singletons/Resources.hpp"
2018-06-28 20:03:04 +02:00
#include "singletons/Theme.hpp"
2018-08-07 01:35:24 +02:00
#include "util/FormatTime.hpp"
#include "util/IrcHelpers.hpp"
2017-04-12 17:46:44 +02:00
#include <QDateTime>
2017-04-14 17:52:22 +02:00
namespace chatterino {
2017-04-12 17:46:44 +02:00
2018-08-07 01:35:24 +02:00
MessagePtr makeSystemMessage(const QString &text)
{
return MessageBuilder(systemMessage, text).release();
}
2017-04-12 17:46:44 +02:00
MessageBuilder::MessageBuilder()
2018-08-07 01:35:24 +02:00
: message_(std::make_unique<Message>())
{
}
MessageBuilder::MessageBuilder(const QString &text)
: MessageBuilder()
2017-04-12 17:46:44 +02:00
{
2018-08-07 01:35:24 +02:00
this->emplace<TimestampElement>();
2018-08-07 07:55:31 +02:00
this->emplace<TextElement>(text, MessageElementFlag::Text,
2018-08-07 01:35:24 +02:00
MessageColor::System);
this->message().searchText = text;
2017-04-12 17:46:44 +02:00
}
2018-08-07 01:35:24 +02:00
MessageBuilder::MessageBuilder(SystemMessageTag, const QString &text)
: MessageBuilder()
2017-04-12 17:46:44 +02:00
{
2018-08-07 01:35:24 +02:00
this->emplace<TimestampElement>();
2018-08-07 07:55:31 +02:00
this->emplace<TextElement>(text, MessageElementFlag::Text,
2018-08-07 01:35:24 +02:00
MessageColor::System);
2018-08-07 07:55:31 +02:00
this->message().flags.set(MessageFlag::System);
this->message().flags.set(MessageFlag::DoNotTriggerNotification);
2018-08-07 01:35:24 +02:00
this->message().searchText = text;
2017-04-12 17:46:44 +02:00
}
2018-08-07 01:35:24 +02:00
MessageBuilder::MessageBuilder(TimeoutMessageTag, const QString &username,
const QString &durationInSeconds,
const QString &reason, bool multipleTimes)
: MessageBuilder()
2017-04-12 17:46:44 +02:00
{
2018-08-07 01:35:24 +02:00
QString text;
text.append(username);
if (!durationInSeconds.isEmpty()) {
text.append(" has been timed out");
// TODO: Implement who timed the user out
text.append(" for ");
bool ok = true;
int timeoutSeconds = durationInSeconds.toInt(&ok);
if (ok) {
text.append(formatTime(timeoutSeconds));
}
} else {
text.append(" has been permanently banned");
}
if (reason.length() > 0) {
text.append(": \"");
text.append(parseTagString(reason));
text.append("\"");
}
text.append(".");
if (multipleTimes) {
text.append(" (multiple times)");
}
2018-08-07 07:55:31 +02:00
this->message().flags.set(MessageFlag::System);
this->message().flags.set(MessageFlag::Timeout);
this->message().flags.set(MessageFlag::DoNotTriggerNotification);
2018-08-07 01:35:24 +02:00
this->message().timeoutUser = username;
2017-04-12 17:46:44 +02:00
}
2018-08-07 01:35:24 +02:00
MessageBuilder::MessageBuilder(const BanAction &action, uint32_t count)
2017-04-12 17:46:44 +02:00
{
2018-08-07 01:35:24 +02:00
this->emplace<TimestampElement>();
2018-08-07 07:55:31 +02:00
this->message().flags.set(MessageFlag::System);
this->message().flags.set(MessageFlag::Timeout);
2018-08-07 01:35:24 +02:00
this->message().timeoutUser = action.target.name;
this->message().count = count;
QString text;
if (action.isBan()) {
if (action.reason.isEmpty()) {
text = QString("%1 banned %2.") //
.arg(action.source.name)
.arg(action.target.name);
} else {
text = QString("%1 banned %2: \"%3\".") //
.arg(action.source.name)
.arg(action.target.name)
.arg(action.reason);
}
} else {
if (action.reason.isEmpty()) {
text = QString("%1 timed out %2 for %3.") //
.arg(action.source.name)
.arg(action.target.name)
.arg(formatTime(action.duration));
} else {
text = QString("%1 timed out %2 for %3: \"%4\".") //
.arg(action.source.name)
.arg(action.target.name)
.arg(formatTime(action.duration))
.arg(action.reason);
}
if (count > 1) {
text.append(QString(" (%1 times)").arg(count));
}
}
2018-08-07 07:55:31 +02:00
this->emplace<TextElement>(text, MessageElementFlag::Text,
2018-08-07 01:35:24 +02:00
MessageColor::System);
this->message().searchText = text;
2017-04-12 17:46:44 +02:00
}
2018-08-07 01:35:24 +02:00
MessageBuilder::MessageBuilder(const UnbanAction &action)
2017-07-31 00:57:42 +02:00
{
2018-08-07 01:35:24 +02:00
this->emplace<TimestampElement>();
2018-08-07 07:55:31 +02:00
this->message().flags.set(MessageFlag::System);
this->message().flags.set(MessageFlag::Untimeout);
2018-08-07 01:35:24 +02:00
this->message().timeoutUser = action.target.name;
QString text;
if (action.wasBan()) {
text = QString("%1 unbanned %2.") //
.arg(action.source.name)
.arg(action.target.name);
} else {
2018-08-07 01:35:24 +02:00
text = QString("%1 untimedout %2.") //
.arg(action.source.name)
.arg(action.target.name);
}
2018-08-07 01:35:24 +02:00
2018-08-07 07:55:31 +02:00
this->emplace<TextElement>(text, MessageElementFlag::Text,
2018-08-07 01:35:24 +02:00
MessageColor::System);
this->message().searchText = text;
}
Message *MessageBuilder::operator->()
{
return this->message_.get();
}
Message &MessageBuilder::message()
{
return *this->message_;
}
MessagePtr MessageBuilder::release()
{
return MessagePtr(this->message_.release());
}
2018-08-07 01:35:24 +02:00
void MessageBuilder::append(std::unique_ptr<MessageElement> element)
2017-04-12 17:46:44 +02:00
{
2018-08-07 01:35:24 +02:00
this->message().elements.push_back(std::move(element));
2017-04-12 17:46:44 +02:00
}
QString MessageBuilder::matchLink(const QString &string)
{
LinkParser linkParser(string);
2018-08-06 21:17:03 +02:00
static QRegularExpression httpRegex(
"\\bhttps?://", QRegularExpression::CaseInsensitiveOption);
static QRegularExpression ftpRegex(
"\\bftps?://", QRegularExpression::CaseInsensitiveOption);
static QRegularExpression spotifyRegex(
"\\bspotify:", QRegularExpression::CaseInsensitiveOption);
2017-08-12 12:09:26 +02:00
if (!linkParser.hasMatch()) {
2017-08-12 12:09:26 +02:00
return QString();
}
2017-08-05 18:44:14 +02:00
QString captured = linkParser.getCaptured();
2018-08-06 21:17:03 +02:00
if (!captured.contains(httpRegex) && !captured.contains(ftpRegex) &&
!captured.contains(spotifyRegex)) {
captured.insert(0, "http://");
}
2017-09-24 18:43:24 +02:00
return captured;
2017-04-12 17:46:44 +02:00
}
} // namespace chatterino