noury/mirror-chatterino2
1
0
Fork 0
mirror of https://github.com/Chatterino/chatterino2.git synced 2024-11-21 22:24:07 +01:00
Code Issues Projects Releases Packages Wiki Activity

fix replacement with emotes issues

Browse source
This commit is contained in:
hemirt 2018-09-23 19:52:03 +02:00
parent 646268ab18
commit 1acb1278aa
2 changed files with 119 additions and 51 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
src/controllers/ignores/IgnorePhrase.hpp
View file

@ -41,16 +41,6 @@ public:
regex_.setPatternOptions(QRegularExpression::CaseInsensitiveOption |
QRegularExpression::UseUnicodePropertiesOption);
}
const auto &accvec = getApp()->accounts->twitch.accounts.getVector();
for (const auto &acc : accvec) {
const auto &accemotes = *acc->accessEmotes();
for (const auto &emote : accemotes.emotes) {
if (this->replace_.contains(emote.first.string, Qt::CaseSensitive)) {
this->emotes_.emplace(emote.first, emote.second);
}
}
}
}
const QString &getPattern() const
@ -107,6 +97,18 @@ public:
bool containsEmote() const
{
if (!this->emotesChecked_) {
const auto &accvec = getApp()->accounts->twitch.accounts.getVector();
for (const auto &acc : accvec) {
const auto &accemotes = *acc->accessEmotes();
for (const auto &emote : accemotes.emotes) {
if (this->replace_.contains(emote.first.string, Qt::CaseSensitive)) {
this->emotes_.emplace(emote.first, emote.second);
}
}
}
this->emotesChecked_ = true;
}
return !this->emotes_.empty();
}
@ -117,7 +119,8 @@ private:
bool isBlock_;
QString replace_;
bool isCaseSensitive_;
std::unordered_map<EmoteName, EmotePtr> emotes_;
mutable std::unordered_map<EmoteName, EmotePtr> emotes_;
mutable bool emotesChecked_{false};
};
} // namespace chatterino

145
src/providers/twitch/TwitchMessageBuilder.cpp
View file

@ -19,6 +19,7 @@
#include <QApplication>
#include <QDebug>
#include <QMediaPlayer>
#include <QStringRef>
#include <boost/variant.hpp>
namespace chatterino {
@ -156,36 +157,39 @@ MessagePtr TwitchMessageBuilder::build()
}
auto app = getApp();
const auto &phrases = app->ignores->phrases.getVector();
auto removeEmotesInRange = [&twitchEmotes](int pos, int len) mutable {
auto it = std::remove_if(
twitchEmotes.begin(), twitchEmotes.end(), [&pos, &len](const auto &item) {
return ((std::get<0>(item) >= pos) && std::get<0>(item) < (pos + len));
});
for (; it != twitchEmotes.end(); ++it) {
if (std::get<1>(*it) == nullptr) {
log("remem nullptr {}", std::get<2>(*it).string);
auto removeEmotesInRange =
[](int pos, int len,
std::vector<std::tuple<int, EmotePtr, EmoteName>> &twitchEmotes) mutable {
auto it = std::partition(
twitchEmotes.begin(), twitchEmotes.end(), [pos, len](const auto &item) {
return !((std::get<0>(item) >= pos) && std::get<0>(item) < (pos + len));
});
for (auto copy = it; copy != twitchEmotes.end(); ++copy) {
if (std::get<1>(*copy) == nullptr) {
log("remem nullptr {}", std::get<2>(*copy).string);
}
}
}
std::vector<std::tuple<int, EmotePtr, EmoteName>> v(it, twitchEmotes.end());
twitchEmotes.erase(it, twitchEmotes.end());
return v;
};
std::vector<std::tuple<int, EmotePtr, EmoteName>> v(it, twitchEmotes.end());
twitchEmotes.erase(it, twitchEmotes.end());
return v;
};
auto shiftIndicesAfter = [&twitchEmotes](int pos, int by) mutable {
auto it = std::find_if(twitchEmotes.begin(), twitchEmotes.end(),
[&pos](const auto &item) { return std::get<0>(item) >= pos; });
while (it != twitchEmotes.end()) {
std::get<0>(*it) += by;
++it;
for (auto &item : twitchEmotes) {
auto &index = std::get<0>(item);
if (index >= pos) {
index += by;
}
}
};
auto addReplEmotes = [&twitchEmotes](const IgnorePhrase &phrase, const QString &midrepl,
auto addReplEmotes = [&twitchEmotes](const IgnorePhrase &phrase, const QStringRef &midrepl,
int startIndex) mutable {
if (!phrase.containsEmote()) {
return;
}
QStringList words = midrepl.split(' ');
QVector<QStringRef> words = midrepl.split(' ');
int pos = 0;
for (const auto &word : words) {
for (const auto &emote : phrase.getEmotes()) {
@ -214,17 +218,14 @@ MessagePtr TwitchMessageBuilder::build()
int from = 0;
while ((from = this->originalMessage_.indexOf(regex, from, &match)) != -1) {
int len = match.capturedLength();
auto vret = removeEmotesInRange(from, len);
auto vret = removeEmotesInRange(from, len, twitchEmotes);
auto mid = this->originalMessage_.mid(from, len);
mid.replace(regex, phrase.getReplace());
// hemirt
// doesnt check for own emotes in the Replace part
// mb in IgnoredPhrase ??
for (auto &tup : vret) {
/*for (auto &tup : vret) {
if (std::get<1>(tup) == nullptr) {
log("v nullptr {}", std::get<2>(tup).string);
continue;
}
int index = 0;
const auto &emote = std::get<2>(tup);
@ -233,14 +234,45 @@ MessagePtr TwitchMessageBuilder::build()
index += emote.string.size();
twitchEmotes.push_back(tup);
}
}*/
int midsize = mid.size();
this->originalMessage_.replace(from, len, mid);
int pos1 = from;
while (pos1 > 0) {
if (this->originalMessage_[pos1 - 1] == ' ') {
break;
}
--pos1;
}
int pos2 = from + midsize;
while (pos2 < this->originalMessage_.length()) {
if (this->originalMessage_[pos2] == ' ') {
break;
}
++pos2;
}
addReplEmotes(phrase, mid, from);
shiftIndicesAfter(from + len, midsize - len);
auto midExtendedRef = this->originalMessage_.midRef(pos1, pos2 - pos1);
for (auto &tup : vret) {
if (std::get<1>(tup) == nullptr) {
log("v nullptr {}", std::get<2>(tup).string);
continue;
}
int index = 0;
QString emote = " " + std::get<2>(tup).string + " ";
while ((index = midExtendedRef.indexOf(emote, index)) != -1) {
std::get<0>(tup) = from + index + 1;
index += emote.size() - 1;
twitchEmotes.push_back(tup);
}
}
addReplEmotes(phrase, midExtendedRef, pos1);
this->originalMessage_.replace(from, len, mid);
int midsize = mid.size();
from += midsize;
shiftIndicesAfter(from, midsize - len);
}
} else {
const auto &pattern = phrase.getPattern();
@ -251,16 +283,13 @@ MessagePtr TwitchMessageBuilder::build()
while ((from = this->originalMessage_.indexOf(pattern, from,
phrase.caseSensitivity())) != -1) {
int len = pattern.size();
auto vret = removeEmotesInRange(from, len);
auto vret = removeEmotesInRange(from, len, twitchEmotes);
auto replace = phrase.getReplace();
// hemirt
// doesnt check for own emotes in the Replace part
// mb in IgnoredPhrase ??
for (auto &tup : vret) {
/*for (auto &tup : vret) {
if (std::get<1>(tup) == nullptr) {
log("v nullptr {}", std::get<2>(tup).string);
continue;
}
int index = 0;
const auto &emote = std::get<2>(tup);
@ -269,14 +298,47 @@ MessagePtr TwitchMessageBuilder::build()
index += emote.string.size();
twitchEmotes.push_back(tup);
}
}*/
int replacesize = replace.size();
this->originalMessage_.replace(from, len, replace);
int pos1 = from;
while (pos1 > 0) {
if (this->originalMessage_[pos1 - 1] == ' ') {
break;
}
--pos1;
}
int pos2 = from + replacesize;
while (pos2 < this->originalMessage_.length()) {
if (this->originalMessage_[pos2] == ' ') {
break;
}
++pos2;
}
addReplEmotes(phrase, replace, from);
shiftIndicesAfter(from + len, replacesize - len);
auto midExtendedRef = this->originalMessage_.midRef(pos1, pos2 - pos1);
for (auto &tup : vret) {
if (std::get<1>(tup) == nullptr) {
log("v nullptr {}", std::get<2>(tup).string);
continue;
}
int index = 0;
QString emote = " " + std::get<2>(tup).string + " ";
while ((index = midExtendedRef.indexOf(emote, index)) != -1) {
std::get<0>(tup) = from + index + 1;
index += emote.size() - 1;
twitchEmotes.push_back(tup);
}
}
addReplEmotes(phrase, midExtendedRef, pos1);
this->originalMessage_.replace(from, len, replace);
int replacesize = replace.size();
from += replacesize;
shiftIndicesAfter(from, replacesize - len);
}
}
}
@ -308,6 +370,9 @@ void TwitchMessageBuilder::addWords(
for (const auto &word : words) {
// check if it's a twitch emote twitch emote
while (currentTwitchEmote != twitchEmotes.end() && std::get<0>(*currentTwitchEmote) < i) {
++currentTwitchEmote;
}
if (currentTwitchEmote != twitchEmotes.end() && std::get<0>(*currentTwitchEmote) == i) {
auto emoteImage = std::get<1>(*currentTwitchEmote);
if (emoteImage == nullptr) {