mirror-chatterino2/src/providers/twitch/TwitchMessageBuilder.cpp

828 lines
28 KiB
C++
Raw Normal View History

2018-06-26 14:09:39 +02:00
#include "providers/twitch/TwitchMessageBuilder.hpp"
2018-06-26 14:09:39 +02:00
#include "Application.hpp"
#include "controllers/accounts/AccountController.hpp"
#include "controllers/highlights/HighlightController.hpp"
#include "controllers/ignores/IgnoreController.hpp"
#include "debug/Log.hpp"
#include "providers/twitch/TwitchChannel.hpp"
2018-06-28 19:46:45 +02:00
#include "singletons/Emotes.hpp"
#include "singletons/Resources.hpp"
#include "singletons/Settings.hpp"
2018-06-28 20:03:04 +02:00
#include "singletons/Theme.hpp"
2018-06-26 14:09:39 +02:00
#include "singletons/WindowManager.hpp"
#include "util/IrcHelpers.hpp"
2017-04-12 17:46:44 +02:00
#include <QApplication>
#include <QDebug>
#include <QMediaPlayer>
2017-04-14 17:52:22 +02:00
namespace chatterino {
2017-12-31 00:50:07 +01:00
TwitchMessageBuilder::TwitchMessageBuilder(Channel *_channel,
2017-07-02 18:12:11 +02:00
const Communi::IrcPrivateMessage *_ircMessage,
const MessageParseArgs &_args)
2017-07-02 18:12:11 +02:00
: channel(_channel)
2017-12-31 00:50:07 +01:00
, twitchChannel(dynamic_cast<TwitchChannel *>(_channel))
2017-07-02 18:12:11 +02:00
, ircMessage(_ircMessage)
, args(_args)
, tags(this->ircMessage->tags())
2018-07-06 19:23:47 +02:00
, originalMessage_(_ircMessage->content())
, action_(_ircMessage->isAction())
{
auto app = getApp();
2018-07-06 19:23:47 +02:00
this->usernameColor_ = app->themes->messages.textColors.system;
}
TwitchMessageBuilder::TwitchMessageBuilder(Channel *_channel,
2018-06-07 15:49:24 +02:00
const Communi::IrcMessage *_ircMessage,
const MessageParseArgs &_args, QString content,
bool isAction)
: channel(_channel)
, twitchChannel(dynamic_cast<TwitchChannel *>(_channel))
, ircMessage(_ircMessage)
, args(_args)
, tags(this->ircMessage->tags())
2018-07-06 19:23:47 +02:00
, originalMessage_(content)
, action_(isAction)
2017-04-12 17:46:44 +02:00
{
auto app = getApp();
2018-07-06 19:23:47 +02:00
this->usernameColor_ = app->themes->messages.textColors.system;
2017-04-12 17:46:44 +02:00
}
2018-01-23 21:33:49 +01:00
bool TwitchMessageBuilder::isIgnored() const
2017-04-12 17:46:44 +02:00
{
auto app = getApp();
2018-05-13 19:24:32 +02:00
// TODO(pajlada): Do we need to check if the phrase is valid first?
for (const auto &phrase : app->ignores->phrases.getVector()) {
2018-07-06 19:23:47 +02:00
if (phrase.isMatch(this->originalMessage_)) {
2018-06-26 17:06:17 +02:00
Log("Blocking message because it contains ignored phrase {}", phrase.getPattern());
2018-01-23 21:33:49 +01:00
return true;
}
}
if (app->settings->enableTwitchIgnoredUsers && this->tags.contains("user-id")) {
auto sourceUserID = this->tags.value("user-id").toString();
2018-05-26 20:26:25 +02:00
for (const auto &user : app->accounts->twitch.getCurrent()->getIgnores()) {
if (sourceUserID == user.id) {
2018-06-26 17:06:17 +02:00
Log("Blocking message because it's from blocked user {}", user.name);
return true;
}
}
}
2018-01-23 21:33:49 +01:00
return false;
2017-04-12 17:46:44 +02:00
}
2018-01-23 23:28:06 +01:00
MessagePtr TwitchMessageBuilder::build()
2017-04-12 17:46:44 +02:00
{
auto app = getApp();
// PARSING
this->parseUsername();
if (this->userName == this->channel->name) {
this->senderIsBroadcaster = true;
}
//#ifdef XD
// if (this->originalMessage.length() > 100) {
// this->message->flags |= Message::Collapsed;
// this->emplace<EmoteElement>(getApp()->resources->badgeCollapsed,
// MessageElement::Collapsed);
// }
//#endif
2018-07-06 19:23:47 +02:00
this->message_->flags |= Message::Collapsed;
2018-01-23 23:28:06 +01:00
// PARSING
this->parseMessageID();
2018-01-23 23:28:06 +01:00
this->parseRoomID();
2018-01-23 23:28:06 +01:00
this->appendChannelName();
2018-01-23 23:28:06 +01:00
// timestamp
bool isPastMsg = this->tags.contains("historical");
2017-12-28 17:56:00 +01:00
if (isPastMsg) {
// This may be architecture dependent(datatype)
qint64 ts = this->tags.value("tmi-sent-ts").toLongLong();
QDateTime dateTime = QDateTime::fromMSecsSinceEpoch(ts);
this->emplace<TimestampElement>(dateTime.time());
} else {
this->emplace<TimestampElement>();
}
2017-04-12 17:46:44 +02:00
bool addModerationElement = true;
if (this->senderIsBroadcaster) {
addModerationElement = false;
} else {
bool hasUserType = this->tags.contains("user-type");
if (hasUserType) {
QString userType = this->tags.value("user-type").toString();
if (userType == "mod") {
if (!args.isStaffOrBroadcaster) {
addModerationElement = false;
}
}
}
}
if (addModerationElement) {
this->emplace<TwitchModerationElement>();
}
2017-04-12 17:46:44 +02:00
2018-01-23 23:28:06 +01:00
this->appendTwitchBadges();
2017-04-12 17:46:44 +02:00
2018-01-23 23:28:06 +01:00
this->appendChatterinoBadges();
2017-04-12 17:46:44 +02:00
this->appendUsername();
2017-04-12 17:46:44 +02:00
// highlights
if (/*app->settings->enableHighlights &&*/ !isPastMsg) {
this->parseHighlights();
}
2017-04-12 17:46:44 +02:00
QString bits;
2017-07-02 18:12:11 +02:00
auto iterator = this->tags.find("bits");
if (iterator != this->tags.end()) {
2017-04-12 17:46:44 +02:00
bits = iterator.value().toString();
}
// twitch emotes
2018-06-26 17:06:17 +02:00
std::vector<std::pair<long, EmoteData>> twitchEmotes;
2017-04-12 17:46:44 +02:00
2017-07-02 18:12:11 +02:00
iterator = this->tags.find("emotes");
if (iterator != this->tags.end()) {
QStringList emoteString = iterator.value().toString().split('/');
2017-04-12 17:46:44 +02:00
for (QString emote : emoteString) {
2017-12-17 02:18:13 +01:00
this->appendTwitchEmote(ircMessage, emote, twitchEmotes);
2017-04-12 17:46:44 +02:00
}
2018-05-24 15:42:06 +02:00
std::sort(twitchEmotes.begin(), twitchEmotes.end(),
[](const auto &a, const auto &b) { return a.first < b.first; });
2017-04-12 17:46:44 +02:00
}
auto currentTwitchEmote = twitchEmotes.begin();
// words
2018-07-06 19:23:47 +02:00
QStringList splits = this->originalMessage_.split(' ');
2017-04-12 17:46:44 +02:00
long int i = 0;
for (QString split : splits) {
MessageColor textColor =
2018-07-06 19:23:47 +02:00
this->action_ ? MessageColor(this->usernameColor_) : MessageColor(MessageColor::Text);
2017-08-12 12:07:53 +02:00
2017-04-12 17:46:44 +02:00
// twitch emote
if (currentTwitchEmote != twitchEmotes.end() && currentTwitchEmote->first == i) {
auto emoteImage = currentTwitchEmote->second;
this->emplace<EmoteElement>(emoteImage, MessageElement::TwitchEmote);
2017-04-12 17:46:44 +02:00
i += split.length() + 1;
currentTwitchEmote = std::next(currentTwitchEmote);
continue;
}
// split words
2018-06-26 17:06:17 +02:00
std::vector<std::tuple<EmoteData, QString>> parsed;
2017-04-12 17:46:44 +02:00
2017-07-02 17:37:17 +02:00
// Parse emojis and take all non-emojis and put them in parsed as full text-words
2018-06-05 18:53:49 +02:00
app->emotes->emojis.parse(parsed, split);
2017-04-12 17:46:44 +02:00
for (const auto &tuple : parsed) {
2018-06-26 17:06:17 +02:00
const EmoteData &emoteData = std::get<0>(tuple);
2017-04-12 17:46:44 +02:00
if (!emoteData.isValid()) { // is text
2017-04-12 17:46:44 +02:00
QString string = std::get<1>(tuple);
if (!bits.isEmpty() && this->tryParseCheermote(string)) {
// This string was parsed as a cheermote
2017-04-12 17:46:44 +02:00
continue;
}
// TODO: Implement ignored emotes
// Format of ignored emotes:
// Emote name: "forsenPuke" - if string in ignoredEmotes
// Will match emote regardless of source (i.e. bttv, ffz)
// Emote source + name: "bttv:nyanPls"
2017-07-23 09:53:50 +02:00
if (this->tryAppendEmote(string)) {
// Successfully appended an emote
2017-04-12 17:46:44 +02:00
continue;
}
2017-07-23 09:53:50 +02:00
// Actually just text
2017-08-05 18:44:14 +02:00
QString linkString = this->matchLink(string);
2018-07-04 13:55:03 +02:00
auto fontStyle = FontStyle::ChatMedium;
2017-04-12 17:46:44 +02:00
2018-07-04 13:59:57 +02:00
if (string.startsWith('@') && app->settings->usernameBold) {
2018-07-04 13:55:03 +02:00
fontStyle = FontStyle::ChatMediumBold;
2018-07-04 12:22:01 +02:00
}
2017-08-12 12:09:26 +02:00
Link link;
if (linkString.isEmpty()) {
link = Link();
} else {
if (app->settings->lowercaseLink) {
QRegularExpression httpRegex("\\bhttps?://",
QRegularExpression::CaseInsensitiveOption);
QRegularExpression ftpRegex("\\bftps?://",
QRegularExpression::CaseInsensitiveOption);
QRegularExpression getDomain("\\/\\/([^\\/]*)");
QString tempString = string;
if (!string.contains(httpRegex)) {
if (!string.contains(ftpRegex)) {
tempString.insert(0, "http://");
}
}
QString domain = getDomain.match(tempString).captured(1);
string.replace(domain, domain.toLower());
}
2017-08-12 12:09:26 +02:00
link = Link(Link::Url, linkString);
textColor = MessageColor(MessageColor::Link);
2017-08-12 12:09:26 +02:00
}
2017-08-05 18:44:14 +02:00
this->emplace<TextElement>(string, MessageElement::Text, textColor,
2018-07-04 13:54:01 +02:00
fontStyle) //
->setLink(link);
2017-04-12 17:46:44 +02:00
} else { // is emoji
this->emplace<EmoteElement>(emoteData, EmoteElement::EmojiAll);
2017-04-12 17:46:44 +02:00
}
}
2017-08-17 22:46:16 +02:00
for (int j = 0; j < split.size(); j++) {
i++;
if (split.at(j).isHighSurrogate()) {
j++;
}
}
i++;
2017-04-12 17:46:44 +02:00
}
2018-07-06 19:23:47 +02:00
this->message_->searchText = this->userName + ": " + this->originalMessage_;
2018-01-17 14:25:03 +01:00
return this->getMessage();
2017-07-02 18:12:11 +02:00
}
void TwitchMessageBuilder::parseMessageID()
{
auto iterator = this->tags.find("id");
if (iterator != this->tags.end()) {
this->messageID = iterator.value().toString();
}
2017-04-12 17:46:44 +02:00
}
2017-07-02 18:12:11 +02:00
void TwitchMessageBuilder::parseRoomID()
{
2017-12-31 00:50:07 +01:00
if (this->twitchChannel == nullptr) {
return;
}
2017-07-02 18:12:11 +02:00
auto iterator = this->tags.find("room-id");
2017-07-02 18:12:11 +02:00
if (iterator != std::end(this->tags)) {
2018-07-06 19:23:47 +02:00
this->roomID_ = iterator.value().toString();
2017-07-02 18:12:11 +02:00
if (this->twitchChannel->roomID.isEmpty()) {
2018-07-06 19:23:47 +02:00
this->twitchChannel->roomID = this->roomID_;
2017-07-02 18:12:11 +02:00
}
}
}
2018-01-23 23:28:06 +01:00
void TwitchMessageBuilder::appendChannelName()
2017-07-02 18:12:11 +02:00
{
2017-07-23 09:53:50 +02:00
QString channelName("#" + this->channel->name);
Link link(Link::Url, this->channel->name + "\n" + this->messageID);
this->emplace<TextElement>(channelName, MessageElement::ChannelName, MessageColor::System) //
->setLink(link);
2017-07-02 18:12:11 +02:00
}
void TwitchMessageBuilder::parseUsername()
{
auto iterator = this->tags.find("color");
if (iterator != this->tags.end()) {
2018-07-06 19:23:47 +02:00
this->usernameColor_ = QColor(iterator.value().toString());
2017-07-02 18:12:11 +02:00
}
// username
this->userName = this->ircMessage->nick();
2017-07-02 18:12:11 +02:00
if (this->userName.isEmpty() || this->args.trimSubscriberUsername) {
2017-07-02 18:12:11 +02:00
this->userName = this->tags.value(QLatin1String("login")).toString();
}
2017-12-17 17:48:46 +01:00
2018-06-04 12:23:23 +02:00
// display name
// auto displayNameVariant = this->tags.value("display-name");
// if (displayNameVariant.isValid()) {
// this->userName = displayNameVariant.toString() + " (" + this->userName + ")";
// }
2018-07-06 19:23:47 +02:00
this->message_->loginName = this->userName;
}
2017-07-02 18:12:11 +02:00
void TwitchMessageBuilder::appendUsername()
{
auto app = getApp();
QString username = this->userName;
2018-07-06 19:23:47 +02:00
this->message_->loginName = username;
QString localizedName;
2017-07-02 18:12:11 +02:00
auto iterator = this->tags.find("display-name");
if (iterator != this->tags.end()) {
2018-06-26 17:06:17 +02:00
QString displayName = parseTagString(iterator.value().toString()).trimmed();
if (QString::compare(displayName, this->userName, Qt::CaseInsensitive) == 0) {
username = displayName;
2018-07-06 19:23:47 +02:00
this->message_->displayName = displayName;
} else {
localizedName = displayName;
2018-07-06 19:23:47 +02:00
this->message_->displayName = username;
this->message_->localizedName = displayName;
}
2017-07-02 18:12:11 +02:00
}
bool hasLocalizedName = !localizedName.isEmpty();
// The full string that will be rendered in the chat widget
QString usernameText;
pajlada::Settings::Setting<int> usernameDisplayMode(
"/appearance/messages/usernameDisplayMode", UsernameDisplayMode::UsernameAndLocalizedName);
switch (usernameDisplayMode.getValue()) {
case UsernameDisplayMode::Username: {
usernameText = username;
} break;
case UsernameDisplayMode::LocalizedName: {
if (hasLocalizedName) {
usernameText = localizedName;
} else {
usernameText = username;
}
} break;
default:
case UsernameDisplayMode::UsernameAndLocalizedName: {
if (hasLocalizedName) {
usernameText = username + "(" + localizedName + ")";
} else {
usernameText = username;
}
} break;
}
2017-07-02 18:12:11 +02:00
if (this->args.isSentWhisper) {
// TODO(pajlada): Re-implement
// userDisplayString += IrcManager::getInstance().getUser().getUserName();
} else if (this->args.isReceivedWhisper) {
// Sender username
2018-07-06 19:23:47 +02:00
this->emplace<TextElement>(usernameText, MessageElement::Text, this->usernameColor_,
2018-05-23 04:22:17 +02:00
FontStyle::ChatMediumBold)
->setLink({Link::UserInfo, this->userName});
2018-05-26 20:26:25 +02:00
auto currentUser = app->accounts->twitch.getCurrent();
// Separator
this->emplace<TextElement>("->", MessageElement::Text,
2018-05-23 04:22:17 +02:00
app->themes->messages.textColors.system, FontStyle::ChatMedium);
QColor selfColor = currentUser->color;
if (!selfColor.isValid()) {
selfColor = app->themes->messages.textColors.system;
}
2017-07-02 18:12:11 +02:00
// Your own username
this->emplace<TextElement>(currentUser->getUserName() + ":", MessageElement::Text,
2018-05-23 04:22:17 +02:00
selfColor, FontStyle::ChatMediumBold);
} else {
2018-07-06 19:23:47 +02:00
if (!this->action_) {
usernameText += ":";
}
2017-07-02 18:12:11 +02:00
2018-07-06 19:23:47 +02:00
this->emplace<TextElement>(usernameText, MessageElement::Text, this->usernameColor_,
2018-05-23 04:22:17 +02:00
FontStyle::ChatMediumBold)
->setLink({Link::UserInfo, this->userName});
2017-07-02 18:12:11 +02:00
}
}
void TwitchMessageBuilder::parseHighlights()
{
static auto player = new QMediaPlayer;
static QUrl currentPlayerUrl;
auto app = getApp();
2018-05-26 20:26:25 +02:00
auto currentUser = app->accounts->twitch.getCurrent();
QString currentUsername = currentUser->getUserName();
if (this->ircMessage->nick() == currentUsername) {
2018-07-06 19:23:47 +02:00
currentUser->color = this->usernameColor_;
// Do nothing. Highlights cannot be triggered by yourself
return;
}
// update the media player url if necessary
QUrl highlightSoundUrl;
if (app->settings->customHighlightSound) {
highlightSoundUrl = QUrl(app->settings->pathHighlightSound.getValue());
} else {
highlightSoundUrl = QUrl("qrc:/sounds/ping2.wav");
}
if (currentPlayerUrl != highlightSoundUrl) {
player->setMedia(highlightSoundUrl);
currentPlayerUrl = highlightSoundUrl;
}
// TODO: This vector should only be rebuilt upon highlights being changed
// fourtf: should be implemented in the HighlightsController
2018-06-26 17:06:17 +02:00
std::vector<HighlightPhrase> activeHighlights = app->highlights->phrases.getVector();
std::vector<HighlightPhrase> userHighlights = app->highlights->highlightedUsers.getVector();
if (app->settings->enableHighlightsSelf && currentUsername.size() > 0) {
2018-06-26 17:06:17 +02:00
HighlightPhrase selfHighlight(currentUsername, app->settings->enableHighlightTaskbar,
app->settings->enableHighlightSound, false);
activeHighlights.emplace_back(std::move(selfHighlight));
}
bool doHighlight = false;
bool playSound = false;
bool doAlert = false;
bool hasFocus = (QApplication::focusWidget() != nullptr);
if (!app->highlights->blacklistContains(this->ircMessage->nick())) {
2018-06-26 17:06:17 +02:00
for (const HighlightPhrase &highlight : activeHighlights) {
2018-07-06 19:23:47 +02:00
if (highlight.isMatch(this->originalMessage_)) {
Log("Highlight because {} matches {}", this->originalMessage_,
2018-06-26 17:06:17 +02:00
highlight.getPattern());
doHighlight = true;
2018-05-06 12:52:47 +02:00
if (highlight.getAlert()) {
doAlert = true;
}
2018-05-06 12:52:47 +02:00
if (highlight.getSound()) {
playSound = true;
}
if (playSound && doAlert) {
// Break if no further action can be taken from other highlights
// This might change if highlights can have custom colors/sounds/actions
break;
}
}
}
for (const HighlightPhrase &userHighlight : userHighlights) {
if (userHighlight.isMatch(this->ircMessage->nick())) {
Log("Highlight because user {} sent a message", this->ircMessage->nick());
doHighlight = true;
if (userHighlight.getAlert()) {
doAlert = true;
}
if (userHighlight.getSound()) {
playSound = true;
}
if (playSound && doAlert) {
// Break if no further action can be taken from other usernames
// Mostly used for regex stuff
break;
}
}
}
this->setHighlight(doHighlight);
if (playSound && (!hasFocus || app->settings->highlightAlwaysPlaySound)) {
player->play();
}
if (doAlert) {
QApplication::alert(getApp()->windows->getMainWindow().window(), 2500);
}
2018-01-06 03:48:56 +01:00
if (doHighlight) {
2018-07-06 19:23:47 +02:00
this->message_->flags |= Message::Highlighted;
2018-01-06 03:48:56 +01:00
}
}
}
void TwitchMessageBuilder::appendTwitchEmote(const Communi::IrcMessage *ircMessage,
const QString &emote,
2018-06-26 17:06:17 +02:00
std::vector<std::pair<long int, EmoteData>> &vec)
{
auto app = getApp();
if (!emote.contains(':')) {
return;
}
QStringList parameters = emote.split(':');
if (parameters.length() < 2) {
return;
}
const auto &id = parameters.at(0);
QStringList occurences = parameters.at(1).split(',');
for (QString occurence : occurences) {
QStringList coords = occurence.split('-');
if (coords.length() < 2) {
return;
}
2018-06-07 13:24:07 +02:00
int start = coords.at(0).toInt();
int end = coords.at(1).toInt();
2018-07-06 19:23:47 +02:00
if (start >= end || start < 0 || end > this->originalMessage_.length()) {
return;
}
2018-07-06 19:23:47 +02:00
QString name = this->originalMessage_.mid(start, end - start + 1);
2018-06-26 17:06:17 +02:00
vec.push_back(
std::pair<long int, EmoteData>(start, app->emotes->twitch.getEmoteById(id, name)));
}
}
2017-07-23 09:53:50 +02:00
bool TwitchMessageBuilder::tryAppendEmote(QString &emoteString)
{
auto app = getApp();
2018-06-26 17:06:17 +02:00
EmoteData emoteData;
2017-07-23 09:53:50 +02:00
2018-01-27 21:13:22 +01:00
auto appendEmote = [&](MessageElement::Flags flags) {
2018-01-22 22:38:44 +01:00
this->emplace<EmoteElement>(emoteData, flags);
return true;
};
2018-06-05 17:39:49 +02:00
if (app->emotes->bttv.globalEmotes.tryGet(emoteString, emoteData)) {
2017-07-23 09:53:50 +02:00
// BTTV Global Emote
2018-01-22 22:38:44 +01:00
return appendEmote(MessageElement::BttvEmote);
2017-12-31 00:50:07 +01:00
} else if (this->twitchChannel != nullptr &&
this->twitchChannel->bttvChannelEmotes->tryGet(emoteString, emoteData)) {
2017-07-23 09:53:50 +02:00
// BTTV Channel Emote
2018-01-22 22:38:44 +01:00
return appendEmote(MessageElement::BttvEmote);
2018-06-05 18:07:17 +02:00
} else if (app->emotes->ffz.globalEmotes.tryGet(emoteString, emoteData)) {
2017-07-23 09:53:50 +02:00
// FFZ Global Emote
2018-01-22 22:38:44 +01:00
return appendEmote(MessageElement::FfzEmote);
2017-12-31 00:50:07 +01:00
} else if (this->twitchChannel != nullptr &&
this->twitchChannel->ffzChannelEmotes->tryGet(emoteString, emoteData)) {
2017-07-23 09:53:50 +02:00
// FFZ Channel Emote
2018-01-22 22:38:44 +01:00
return appendEmote(MessageElement::FfzEmote);
2017-07-23 09:53:50 +02:00
}
return false;
}
// fourtf: this is ugly
// maybe put the individual badges into a map instead of this mess
2018-01-23 23:28:06 +01:00
void TwitchMessageBuilder::appendTwitchBadges()
{
auto app = getApp();
2018-07-06 19:23:47 +02:00
const auto &channelResources = app->resources->channels[this->roomID_];
2017-07-02 18:12:11 +02:00
auto iterator = this->tags.find("badges");
if (iterator == this->tags.end()) {
// No badges in this message
return;
}
QStringList badges = iterator.value().toString().split(',');
for (QString badge : badges) {
if (badge.isEmpty()) {
continue;
}
if (badge.startsWith("bits/")) {
if (!app->resources->dynamicBadgesLoaded) {
2017-06-15 23:13:01 +02:00
// Do nothing
continue;
}
QString cheerAmountQS = badge.mid(5);
std::string versionKey = cheerAmountQS.toStdString();
QString tooltip = QString("Twitch cheer ") + cheerAmountQS;
2017-06-15 23:13:01 +02:00
2018-01-12 22:38:00 +01:00
// Try to fetch channel-specific bit badge
2017-06-16 08:03:13 +02:00
try {
2018-01-12 22:38:00 +01:00
const auto &badge = channelResources.badgeSets.at("bits").versions.at(versionKey);
2018-05-24 15:42:06 +02:00
this->emplace<ImageElement>(badge.badgeImage1x, MessageElement::BadgeVanity)
2018-05-24 16:06:34 +02:00
->setTooltip(tooltip);
2018-01-12 22:38:00 +01:00
continue;
} catch (const std::out_of_range &) {
// Channel does not contain a special bit badge for this version
}
2017-06-16 08:03:13 +02:00
2018-01-12 22:38:00 +01:00
// Use default bit badge
try {
const auto &badge = app->resources->badgeSets.at("bits").versions.at(versionKey);
2018-05-24 16:06:34 +02:00
this->emplace<ImageElement>(badge.badgeImage1x, MessageElement::BadgeVanity)
->setTooltip(tooltip);
2018-01-12 22:38:00 +01:00
} catch (const std::out_of_range &) {
2018-06-26 17:06:17 +02:00
Log("No default bit badge for version {} found", versionKey);
2018-01-12 22:38:00 +01:00
continue;
2017-06-16 08:03:13 +02:00
}
} else if (badge == "staff/1") {
this->emplace<ImageElement>(app->resources->badgeStaff,
MessageElement::BadgeGlobalAuthority)
->setTooltip("Twitch Staff");
} else if (badge == "admin/1") {
this->emplace<ImageElement>(app->resources->badgeAdmin,
MessageElement::BadgeGlobalAuthority)
->setTooltip("Twitch Admin");
} else if (badge == "global_mod/1") {
this->emplace<ImageElement>(app->resources->badgeGlobalModerator,
MessageElement::BadgeGlobalAuthority)
->setTooltip("Twitch Global Moderator");
} else if (badge == "moderator/1") {
// TODO: Implement custom FFZ moderator badge
this->emplace<ImageElement>(app->resources->badgeModerator,
MessageElement::BadgeChannelAuthority)
->setTooltip("Twitch Channel Moderator");
} else if (badge == "turbo/1") {
this->emplace<ImageElement>(app->resources->badgeTurbo,
MessageElement::BadgeGlobalAuthority)
->setTooltip("Twitch Turbo Subscriber");
} else if (badge == "broadcaster/1") {
this->emplace<ImageElement>(app->resources->badgeBroadcaster,
MessageElement::BadgeChannelAuthority)
->setTooltip("Twitch Broadcaster");
} else if (badge == "premium/1") {
this->emplace<ImageElement>(app->resources->badgePremium, MessageElement::BadgeVanity)
->setTooltip("Twitch Prime Subscriber");
} else if (badge.startsWith("partner/")) {
int index = badge.midRef(8).toInt();
switch (index) {
case 1: {
this->emplace<ImageElement>(app->resources->badgeVerified,
MessageElement::BadgeVanity)
->setTooltip("Twitch Verified");
} break;
default: {
printf("[TwitchMessageBuilder] Unhandled partner badge index: %d\n", index);
} break;
}
} else if (badge.startsWith("subscriber/")) {
2017-07-02 15:11:33 +02:00
if (channelResources.loaded == false) {
// qDebug() << "Channel resources are not loaded, can't add the subscriber badge";
2017-07-02 15:11:33 +02:00
continue;
}
auto badgeSetIt = channelResources.badgeSets.find("subscriber");
if (badgeSetIt == channelResources.badgeSets.end()) {
// Fall back to default badge
this->emplace<ImageElement>(app->resources->badgeSubscriber,
MessageElement::BadgeSubscription)
->setTooltip("Twitch Subscriber");
continue;
}
2017-07-02 15:11:33 +02:00
const auto &badgeSet = badgeSetIt->second;
2017-07-02 15:11:33 +02:00
std::string versionKey = badge.mid(11).toStdString();
2017-07-02 15:11:33 +02:00
auto badgeVersionIt = badgeSet.versions.find(versionKey);
if (badgeVersionIt == badgeSet.versions.end()) {
// Fall back to default badge
this->emplace<ImageElement>(app->resources->badgeSubscriber,
MessageElement::BadgeSubscription)
->setTooltip("Twitch Subscriber");
continue;
}
2017-07-02 15:11:33 +02:00
auto &badgeVersion = badgeVersionIt->second;
this->emplace<ImageElement>(badgeVersion.badgeImage1x,
MessageElement::BadgeSubscription)
->setTooltip("Twitch " + QString::fromStdString(badgeVersion.title));
} else {
if (!app->resources->dynamicBadgesLoaded) {
// Do nothing
continue;
}
QStringList parts = badge.split('/');
if (parts.length() != 2) {
qDebug() << "Bad number of parts: " << parts.length() << " in " << parts;
continue;
}
MessageElement::Flags badgeType = MessageElement::Flags::BadgeVanity;
std::string badgeSetKey = parts[0].toStdString();
std::string versionKey = parts[1].toStdString();
try {
auto &badgeSet = app->resources->badgeSets.at(badgeSetKey);
try {
auto &badgeVersion = badgeSet.versions.at(versionKey);
this->emplace<ImageElement>(badgeVersion.badgeImage1x, badgeType)
->setTooltip("Twitch " + QString::fromStdString(badgeVersion.title));
} catch (const std::exception &e) {
qDebug() << "Exception caught:" << e.what()
<< "when trying to fetch badge version " << versionKey.c_str();
}
} catch (const std::exception &e) {
qDebug() << "No badge set with key" << badgeSetKey.c_str()
<< ". Exception: " << e.what();
}
}
}
}
2018-01-23 23:28:06 +01:00
void TwitchMessageBuilder::appendChatterinoBadges()
2017-08-12 13:20:52 +02:00
{
auto app = getApp();
auto &badges = app->resources->chatterinoBadges;
2017-08-12 13:20:52 +02:00
auto it = badges.find(this->userName.toStdString());
if (it == badges.end()) {
return;
}
const auto badge = it->second;
this->emplace<ImageElement>(badge->image, MessageElement::BadgeChatterino)
->setTooltip(QString::fromStdString(badge->tooltip));
2017-08-12 13:20:52 +02:00
}
bool TwitchMessageBuilder::tryParseCheermote(const QString &string)
{
auto app = getApp();
// Try to parse custom cheermotes
2018-07-06 19:23:47 +02:00
const auto &channelResources = app->resources->channels[this->roomID_];
if (channelResources.loaded) {
for (const auto &cheermoteSet : channelResources.cheermoteSets) {
auto match = cheermoteSet.regex.match(string);
if (!match.hasMatch()) {
continue;
}
QString amount = match.captured(1);
bool ok = false;
int numBits = amount.toInt(&ok);
if (!ok) {
2018-06-26 17:06:17 +02:00
Log("Error parsing bit amount in tryParseCheermote");
return false;
}
auto savedIt = cheermoteSet.cheermotes.end();
// Fetch cheermote that matches our numBits
for (auto it = cheermoteSet.cheermotes.begin(); it != cheermoteSet.cheermotes.end();
++it) {
if (numBits >= it->minBits) {
savedIt = it;
} else {
break;
}
}
if (savedIt == cheermoteSet.cheermotes.end()) {
2018-06-26 17:06:17 +02:00
Log("Error getting a cheermote from a cheermote set for the bit amount {}",
numBits);
return false;
}
const auto &cheermote = *savedIt;
this->emplace<EmoteElement>(cheermote.emoteDataAnimated, EmoteElement::BitsAnimated);
this->emplace<TextElement>(amount, EmoteElement::Text, cheermote.color);
return true;
}
}
return false;
}
} // namespace chatterino