mirror-chatterino2/channel.cpp

#include "channel.h"
#include "emotes.h"
#include "messages/message.h"
#include "windows.h"

#include <QJsonArray>
#include <QJsonDocument>
#include <QJsonObject>
#include <QJsonValue>
#include <QNetworkAccessManager>
#include <QNetworkReply>
#include <QNetworkRequest>
#include <memory>

namespace chatterino {

Channel::Channel(const QString &channel)
    : messages()
    , name((channel.length() > 0 && channel[0] == '#') ? channel.mid(1)
                                                       : channel)
    , bttvChannelEmotes()
    , ffzChannelEmotes()
    , subLink("https://www.twitch.tv/" + name +
              "/subscribe?ref=in_chat_subscriber_link")
    , channelLink("https://twitch.tv/" + name)
    , popoutPlayerLink("https://player.twitch.tv/?channel=" + name)
{
    reloadChannelEmotes();
}

void
Channel::reloadBttvEmotes()
{
    // bttv
    QNetworkAccessManager *manager = new QNetworkAccessManager();

    QUrl url("https://api.betterttv.net/2/channels/" + this->name);
    QNetworkRequest request(url);

    QNetworkReply *reply = manager->get(request);

    QObject::connect(reply, &QNetworkReply::finished, [=] {
        if (reply->error() == QNetworkReply::NetworkError::NoError) {
            QByteArray data = reply->readAll();
            QJsonDocument jsonDoc(QJsonDocument::fromJson(data));
            QJsonObject root = jsonDoc.object();

            auto emotes = root.value("emotes").toArray();

            QString _template = "https:" + root.value("urlTemplate").toString();

            for (const QJsonValue &emote : emotes) {
                QString id = emote.toObject().value("id").toString();
                QString code = emote.toObject().value("code").toString();
                // emote.value("imageType").toString();

                QString tmp = _template;
                tmp.detach();
                QString url =
                    tmp.replace("{{id}}", id).replace("{{image}}", "1x");

                this->getBttvChannelEmotes().insert(
                    code,
                    Emotes::getBttvChannelEmoteFromCaches().getOrAdd(
                        id, [&code, &url] {
                            return new messages::LazyLoadedImage(
                                url, 1, code, code + "\nChannel Bttv Emote");
                        }));
            }
        }

        reply->deleteLater();
        manager->deleteLater();
    });
}

void
Channel::reloadFfzEmotes()
{
    QNetworkAccessManager *manager = new QNetworkAccessManager();

    QUrl url("http://api.frankerfacez.com/v1/room/" + this->name);
    QNetworkRequest request(url);

    QNetworkReply *reply = manager->get(request);

    QObject::connect(reply, &QNetworkReply::finished, [=] {
        if (reply->error() == QNetworkReply::NetworkError::NoError) {
            QByteArray data = reply->readAll();
            QJsonDocument jsonDoc(QJsonDocument::fromJson(data));
            QJsonObject root = jsonDoc.object();

            auto sets = root.value("sets").toObject();

            for (const QJsonValue &set : sets) {
                auto emoticons = set.toObject().value("emoticons").toArray();

                for (const QJsonValue &emote : emoticons) {
                    QJsonObject object = emote.toObject();

                    // margins

                    int id = object.value("id").toInt();
                    QString code = object.value("name").toString();

                    QJsonObject urls = object.value("urls").toObject();
                    QString url1 = "http:" + urls.value("1").toString();

                    this->getFfzChannelEmotes().insert(
                        code,
                        Emotes::getFfzChannelEmoteFromCaches().getOrAdd(
                            id, [&code, &url1] {
                                return new messages::LazyLoadedImage(
                                    url1, 1, code, code + "\nGlobal Ffz Emote");
                            }));
                }
            }
        }

        reply->deleteLater();
        manager->deleteLater();
    });
}

void
Channel::addMessage(std::shared_ptr<messages::Message> message)
{
    std::shared_ptr<messages::Message> deleted;

    if (this->messages.appendItem(message, deleted)) {
        this->messageRemovedFromStart(deleted);
    }

    this->messageAppended(message);

    Windows::repaintVisibleChatWidgets(this);
}
}