Plugins

src/utils/WebsocketNotifier.cpp

@@ -5,6 +5,7 @@
 #include "Utils.h"
 #include "utils/os/tails.h"
 #include "utils/os/whonix.h"
+#include "plugins/PluginRegistry.h"
 
 #include <QJsonObject>
 
@@ -13,6 +14,10 @@ WebsocketNotifier::WebsocketNotifier(QObject *parent)
     , websocketClient(new WebsocketClient(this))
 {
     connect(websocketClient, &WebsocketClient::WSMessage, this, &WebsocketNotifier::onWSMessage);
+
+    for (const auto& plugin : PluginRegistry::getPlugins()) {
+        m_pluginSubscriptions << plugin->socketData();
+    }
 }
 
 QPointer<WebsocketNotifier> WebsocketNotifier::m_instance(nullptr);
@@ -45,56 +50,20 @@ void WebsocketNotifier::onWSMessage(const QJsonObject &msg) {
         emit FiatRatesReceived(fiat_rates);
     }
 
-    else if(cmd == "reddit") {
-        QJsonArray reddit_data = msg.value("data").toArray();
-        this->onWSReddit(reddit_data);
-    }
-
-    else if(cmd == "ccs") {
-        auto ccs_data = msg.value("data").toArray();
-        this->onWSCCS(ccs_data);
-    }
-
-    else if(cmd == "bounties") {
-        auto data = msg.value("data").toArray();
-        this->onWSBounties(data);
-    }
-
     else if(cmd == "txFiatHistory") {
         auto txFiatHistory_data = msg.value("data").toObject();
         emit TxFiatHistoryReceived(txFiatHistory_data);
     }
 
-    else if(cmd == "revuo") {
-        auto revuo_data = msg.value("data").toArray();
-        this->onWSRevuo(revuo_data);
-    }
-
 #if defined(CHECK_UPDATES)
     else if (cmd == "updates") {
         this->onWSUpdates(msg.value("data").toObject());
     }
 #endif
 
-#if defined(HAS_XMRIG)
-    else if(cmd == "xmrig") {
-        this->onWSXMRigDownloads(msg.value("data").toObject());
+    else if (m_pluginSubscriptions.contains(cmd)) {
+        emit dataReceived(cmd, msg.value("data"));
     }
-#endif
-
-#if defined(HAS_LOCALMONERO)
-    else if (cmd == "localmonero_countries") {
-        emit LocalMoneroCountriesReceived(msg.value("data").toArray());
-    }
-
-    else if (cmd == "localmonero_currencies") {
-        emit LocalMoneroCurrenciesReceived(msg.value("data").toArray());
-    }
-
-    else if (cmd == "localmonero_payment_methods") {
-        emit LocalMoneroPaymentMethodsReceived(msg.value("data").toObject());
-    }
-#endif
 }
 
 void WebsocketNotifier::emitCache() {
@@ -136,103 +105,6 @@ void WebsocketNotifier::onWSNodes(const QJsonArray &nodes) {
     emit NodesReceived(l);
 }
 
-void WebsocketNotifier::onWSReddit(const QJsonArray& reddit_data) {
-    QList<QSharedPointer<RedditPost>> l;
-
-    for (auto &&entry: reddit_data) {
-        auto obj = entry.toObject();
-        auto redditPost = new RedditPost(
-                obj.value("title").toString(),
-                obj.value("author").toString(),
-                obj.value("permalink").toString(),
-                obj.value("comments").toInt());
-        QSharedPointer<RedditPost> r = QSharedPointer<RedditPost>(redditPost);
-        l.append(r);
-    }
-
-    emit RedditReceived(l);
-}
-
-void WebsocketNotifier::onWSCCS(const QJsonArray &ccs_data) {
-    QList<QSharedPointer<CCSEntry>> l;
-
-    for (const auto& entry: ccs_data) {
-        auto obj = entry.toObject();
-        auto c = QSharedPointer<CCSEntry>(new CCSEntry());
-
-        if (obj.value("state").toString() != "FUNDING-REQUIRED")
-            continue;
-
-        c->state = obj.value("state").toString();
-        c->address = obj.value("address").toString();
-        c->author = obj.value("author").toString();
-        c->date = obj.value("date").toString();
-        c->title = obj.value("title").toString();
-        c->target_amount = obj.value("target_amount").toDouble();
-        c->raised_amount = obj.value("raised_amount").toDouble();
-        c->percentage_funded = obj.value("percentage_funded").toDouble();
-        c->contributions = obj.value("contributions").toInt();
-        c->organizer = obj.value("organizer").toString();
-        c->currency = obj.value("currency").toString();
-
-        QString urlpath = obj.value("urlpath").toString();
-        if (c->organizer == "CCS") {
-            c->url = QString("https://ccs.getmonero.org/%1").arg(urlpath);
-        }
-        else if (c->organizer == "MAGIC") {
-            c->url = QString("https://monerofund.org/%1").arg(urlpath);
-        }
-        else {
-            continue;
-        }
-
-        l.append(c);
-    }
-
-    emit CCSReceived(l);
-}
-
-void WebsocketNotifier::onWSRevuo(const QJsonArray &revuo_data) {
-    QList<QSharedPointer<RevuoItem>> l;
-
-    for (auto &&entry: revuo_data) {
-        auto obj = entry.toObject();
-
-        QStringList newsbytes;
-        for (const auto &n : obj.value("newsbytes").toArray()) {
-            newsbytes.append(n.toString());
-        }
-
-        auto revuoItem = new RevuoItem(
-                obj.value("title").toString(),
-                obj.value("url").toString(),
-                newsbytes);
-
-        QSharedPointer<RevuoItem> r = QSharedPointer<RevuoItem>(revuoItem);
-        l.append(r);
-    }
-
-    emit RevuoReceived(l);
-}
-
-void WebsocketNotifier::onWSBounties(const QJsonArray &bounties_data) {
-    QList<QSharedPointer<BountyEntry>> l;
-
-    for (const auto& entry : bounties_data) {
-        QJsonObject obj = entry.toObject();
-        auto bounty = new BountyEntry(obj.value("votes").toInt(),
-                                      obj.value("title").toString(),
-                                      obj.value("amount").toDouble(),
-                                      obj.value("link").toString(),
-                                      obj.value("address").toString(),
-                                      obj.value("status").toString());
-        QSharedPointer<BountyEntry> b = QSharedPointer<BountyEntry>(bounty);
-        l.append(b);
-    }
-
-    emit BountyReceived(l);
-}
-
 void WebsocketNotifier::onWSUpdates(const QJsonObject &updates) {
     emit UpdatesReceived(updates);
 }