rename #2

2018-06-26 13:24:55 +02:00
parent ec04f10895
commit c9722b9780
301 changed files with 0 additions and 0 deletions
@@ -0,0 +1,261 @@
+#include "providers/twitch/twitchemotes.hpp"
+
+#include "debug/log.hpp"
+#include "messages/image.hpp"
+#include "util/benchmark.hpp"
+#include "util/rapidjson-helpers.hpp"
+#include "util/urlfetch.hpp"
+
+#define TWITCH_EMOTE_TEMPLATE "https://static-cdn.jtvnw.net/emoticons/v1/{id}/{scale}"
+
+namespace chatterino {
+namespace providers {
+namespace twitch {
+
+namespace {
+
+QString getEmoteLink(const QString &id, const QString &emoteScale)
+{
+    QString value = TWITCH_EMOTE_TEMPLATE;
+
+    value.detach();
+
+    return value.replace("{id}", id).replace("{scale}", emoteScale);
+}
+
+QString cleanUpCode(const QString &dirtyEmoteCode)
+{
+    QString cleanCode = dirtyEmoteCode;
+    // clang-format off
+    static QMap<QString, QString> emoteNameReplacements{
+        {"[oO](_|\\.)[oO]", "O_o"}, {"\\&gt\\;\\(", "&gt;("},   {"\\&lt\\;3", "&lt;3"},
+        {"\\:-?(o|O)", ":O"},       {"\\:-?(p|P)", ":P"},       {"\\:-?[\\\\/]", ":/"},
+        {"\\:-?[z|Z|\\|]", ":Z"},   {"\\:-?\\(", ":("},         {"\\:-?\\)", ":)"},
+        {"\\:-?D", ":D"},           {"\\;-?(p|P)", ";P"},       {"\\;-?\\)", ";)"},
+        {"R-?\\)", "R)"},           {"B-?\\)", "B)"},
+    };
+    // clang-format on
+
+    auto it = emoteNameReplacements.find(dirtyEmoteCode);
+    if (it != emoteNameReplacements.end()) {
+        cleanCode = it.value();
+    }
+
+    cleanCode.replace("&lt;", "<");
+    cleanCode.replace("&gt;", ">");
+
+    return cleanCode;
+}
+
+void loadSetData(std::shared_ptr<TwitchEmotes::EmoteSet> emoteSet)
+{
+    debug::Log("Load twitch emote set data for {}", emoteSet->key);
+    util::NetworkRequest req("https://braize.pajlada.com/chatterino/twitchemotes/set/" +
+                             emoteSet->key + "/");
+
+    req.setRequestType(util::NetworkRequest::GetRequest);
+
+    req.onError([](int errorCode) -> bool {
+        debug::Log("Emote sets on ERROR {}", errorCode);
+        return true;
+    });
+
+    req.onSuccess([emoteSet](const rapidjson::Document &root) -> bool {
+        debug::Log("Emote sets on success");
+        if (!root.IsObject()) {
+            return false;
+        }
+
+        std::string emoteSetID;
+        QString channelName;
+        if (!rj::getSafe(root, "channel_name", channelName)) {
+            return false;
+        }
+
+        emoteSet->channelName = channelName;
+
+        return true;
+    });
+
+    req.execute();
+}
+
+}  // namespace
+
+TwitchEmotes::TwitchEmotes()
+{
+    {
+        EmoteSet emoteSet;
+        emoteSet.key = "19194";
+        emoteSet.text = "Twitch Prime Emotes";
+        this->staticEmoteSets[emoteSet.key] = std::move(emoteSet);
+    }
+
+    {
+        EmoteSet emoteSet;
+        emoteSet.key = "0";
+        emoteSet.text = "Twitch Global Emotes";
+        this->staticEmoteSets[emoteSet.key] = std::move(emoteSet);
+    }
+}
+
+// id is used for lookup
+// emoteName is used for giving a name to the emote in case it doesn't exist
+util::EmoteData TwitchEmotes::getEmoteById(const QString &id, const QString &emoteName)
+{
+    QString _emoteName = emoteName;
+    _emoteName.replace("<", "&lt;");
+    _emoteName.replace(">", "&gt;");
+
+    // clang-format off
+    static QMap<QString, QString> emoteNameReplacements{
+        {"[oO](_|\\.)[oO]", "O_o"}, {"\\&gt\\;\\(", "&gt;("},   {"\\&lt\\;3", "&lt;3"},
+        {"\\:-?(o|O)", ":O"},       {"\\:-?(p|P)", ":P"},       {"\\:-?[\\\\/]", ":/"},
+        {"\\:-?[z|Z|\\|]", ":Z"},   {"\\:-?\\(", ":("},         {"\\:-?\\)", ":)"},
+        {"\\:-?D", ":D"},           {"\\;-?(p|P)", ";P"},       {"\\;-?\\)", ";)"},
+        {"R-?\\)", "R)"},           {"B-?\\)", "B)"},
+    };
+    // clang-format on
+
+    auto it = emoteNameReplacements.find(_emoteName);
+    if (it != emoteNameReplacements.end()) {
+        _emoteName = it.value();
+    }
+
+    return _twitchEmoteFromCache.getOrAdd(id, [&emoteName, &_emoteName, &id] {
+        util::EmoteData newEmoteData;
+        auto cleanCode = cleanUpCode(emoteName);
+        newEmoteData.image1x = new messages::Image(getEmoteLink(id, "1.0"), 1, emoteName,
+                                                   _emoteName + "<br/>Twitch Emote 1x");
+        newEmoteData.image1x->setCopyString(cleanCode);
+
+        newEmoteData.image2x = new messages::Image(getEmoteLink(id, "2.0"), .5, emoteName,
+                                                   _emoteName + "<br/>Twitch Emote 2x");
+        newEmoteData.image2x->setCopyString(cleanCode);
+
+        newEmoteData.image3x = new messages::Image(getEmoteLink(id, "3.0"), .25, emoteName,
+                                                   _emoteName + "<br/>Twitch Emote 3x");
+
+        newEmoteData.image3x->setCopyString(cleanCode);
+
+        return newEmoteData;
+    });
+}
+
+void TwitchEmotes::refresh(const std::shared_ptr<TwitchAccount> &user)
+{
+    debug::Log("Loading Twitch emotes for user {}", user->getUserName());
+
+    const auto &roomID = user->getUserId();
+    const auto &clientID = user->getOAuthClient();
+    const auto &oauthToken = user->getOAuthToken();
+
+    if (clientID.isEmpty() || oauthToken.isEmpty()) {
+        debug::Log("Missing Client ID or OAuth token");
+        return;
+    }
+
+    TwitchAccountEmoteData &emoteData = this->emotes[roomID];
+
+    if (emoteData.filled) {
+        debug::Log("Emotes are already loaded for room id {}", roomID);
+        return;
+    }
+
+    QString url("https://api.twitch.tv/kraken/users/" + roomID + "/emotes");
+
+    auto loadEmotes = [=, &emoteData](const QJsonObject &root) {
+        emoteData.emoteSets.clear();
+        emoteData.emoteCodes.clear();
+
+        auto emoticonSets = root.value("emoticon_sets").toObject();
+        for (QJsonObject::iterator it = emoticonSets.begin(); it != emoticonSets.end(); ++it) {
+            auto emoteSet = std::make_shared<EmoteSet>();
+
+            emoteSet->key = it.key();
+
+            loadSetData(emoteSet);
+
+            for (QJsonValue emoteValue : it.value().toArray()) {
+                QJsonObject emoticon = emoteValue.toObject();
+                QString id = QString::number(emoticon["id"].toInt());
+                QString code = emoticon["code"].toString();
+
+                auto cleanCode = cleanUpCode(code);
+                emoteSet->emotes.emplace_back(id, cleanCode);
+                emoteData.emoteCodes.push_back(cleanCode);
+
+                util::EmoteData emote = this->getEmoteById(id, code);
+                emoteData.emotes.insert(code, emote);
+            }
+
+            emoteData.emoteSets.emplace_back(emoteSet);
+        }
+
+        emoteData.filled = true;
+    };
+
+    util::twitch::getAuthorized(url, clientID, oauthToken, QThread::currentThread(), loadEmotes);
+}
+
+void TwitchEmotes::loadSetData(std::shared_ptr<TwitchEmotes::EmoteSet> emoteSet)
+{
+    if (!emoteSet) {
+        debug::Log("null emote set sent");
+        return;
+    }
+
+    auto staticSetIt = this->staticEmoteSets.find(emoteSet->key);
+    if (staticSetIt != this->staticEmoteSets.end()) {
+        const auto &staticSet = staticSetIt->second;
+        emoteSet->channelName = staticSet.channelName;
+        emoteSet->text = staticSet.text;
+        return;
+    }
+
+    debug::Log("Load twitch emote set data for {}..", emoteSet->key);
+    util::NetworkRequest req("https://braize.pajlada.com/chatterino/twitchemotes/set/" +
+                             emoteSet->key + "/");
+
+    req.setRequestType(util::NetworkRequest::GetRequest);
+
+    req.onError([](int errorCode) -> bool {
+        debug::Log("Emote sets on ERROR {}", errorCode);
+        return true;
+    });
+
+    req.onSuccess([emoteSet](const rapidjson::Document &root) -> bool {
+        if (!root.IsObject()) {
+            return false;
+        }
+
+        std::string emoteSetID;
+        QString channelName;
+        QString type;
+        if (!rj::getSafe(root, "channel_name", channelName)) {
+            return false;
+        }
+
+        if (!rj::getSafe(root, "type", type)) {
+            return false;
+        }
+
+        debug::Log("Loaded twitch emote set data for {}!", emoteSet->key);
+
+        if (type == "sub") {
+            emoteSet->text = QString("Twitch Subscriber Emote (%1)").arg(channelName);
+        } else {
+            emoteSet->text = QString("Twitch Account Emote (%1)").arg(channelName);
+        }
+
+        emoteSet->channelName = channelName;
+
+        return true;
+    });
+
+    req.execute();
+}
+
+}  // namespace twitch
+}  // namespace providers
+}  // namespace chatterino