renamed *Manager files

src/singletons/NativeMessaging.cpp

@@ -0,0 +1,241 @@
+#include "NativeMessaging.hpp"
+
+#include "Application.hpp"
+#include "providers/twitch/TwitchServer.hpp"
+#include "singletons/Paths.hpp"
+#include "util/PostToThread.hpp"
+
+#include <QCoreApplication>
+#include <QFile>
+#include <QJsonArray>
+#include <QJsonDocument>
+#include <QJsonObject>
+#include <QJsonValue>
+
+#include <boost/interprocess/ipc/message_queue.hpp>
+
+namespace ipc = boost::interprocess;
+
+#ifdef Q_OS_WIN
+#include <QProcess>
+
+#include <Windows.h>
+#include "singletons/WindowManager.hpp"
+#include "widgets/AttachedWindow.hpp"
+#endif
+
+#include <iostream>
+
+#define EXTENSION_ID "glknmaideaikkmemifbfkhnomoknepka"
+#define MESSAGE_SIZE 1024
+
+namespace chatterino {
+
+// fourtf: don't add this class to the application class
+NativeMessagingManager::NativeMessagingManager()
+{
+    qDebug() << "init NativeMessagingManager";
+}
+
+void NativeMessagingManager::writeByteArray(QByteArray a)
+{
+    char *data = a.data();
+    uint32_t size;
+    size = a.size();
+    std::cout.write(reinterpret_cast<char *>(&size), 4);
+    std::cout.write(data, a.size());
+    std::cout.flush();
+}
+
+void NativeMessagingManager::registerHost()
+{
+    auto app = getApp();
+
+    if (app->paths->isPortable()) {
+        return;
+    }
+
+    auto getBaseDocument = [&] {
+        QJsonObject obj;
+        obj.insert("name", "com.chatterino.chatterino");
+        obj.insert("description", "Browser interaction with chatterino.");
+        obj.insert("path", QCoreApplication::applicationFilePath());
+        obj.insert("type", "stdio");
+
+        return obj;
+    };
+
+    auto registerManifest = [&](const QString &manifestFilename, const QString &registryKeyName,
+                                const QJsonDocument &document) {
+        // save the manifest
+        QString manifestPath = app->paths->miscDirectory + manifestFilename;
+        QFile file(manifestPath);
+        file.open(QIODevice::WriteOnly | QIODevice::Truncate);
+        file.write(document.toJson());
+        file.flush();
+
+#ifdef Q_OS_WIN
+        // clang-format off
+        QProcess::execute("REG ADD \"" + registryKeyName + "\" /ve /t REG_SZ /d \"" + manifestPath + "\" /f");
+// clang-format on
+#endif
+    };
+
+    // chrome
+    {
+        QJsonDocument document;
+
+        auto obj = getBaseDocument();
+        QJsonArray allowed_origins_arr = {"chrome-extension://" EXTENSION_ID "/"};
+        obj.insert("allowed_origins", allowed_origins_arr);
+        document.setObject(obj);
+
+        registerManifest(
+            "/native-messaging-manifest-chrome.json",
+            "HKCU\\Software\\Google\\Chrome\\NativeMessagingHosts\\com.chatterino.chatterino",
+            document);
+    }
+
+    // firefox
+    {
+        QJsonDocument document;
+
+        auto obj = getBaseDocument();
+        QJsonArray allowed_extensions = {"chatterino_native@chatterino.com"};
+        obj.insert("allowed_extensions", allowed_extensions);
+        document.setObject(obj);
+
+        registerManifest("/native-messaging-manifest-firefox.json",
+                         "HKCU\\Software\\Mozilla\\NativeMessagingHosts\\com.chatterino.chatterino",
+                         document);
+    }
+}
+
+void NativeMessagingManager::openGuiMessageQueue()
+{
+    static ReceiverThread thread;
+
+    if (thread.isRunning()) {
+        thread.exit();
+    }
+
+    thread.start();
+}
+
+void NativeMessagingManager::sendToGuiProcess(const QByteArray &array)
+{
+    try {
+        ipc::message_queue messageQueue(ipc::open_only, "chatterino_gui");
+
+        messageQueue.try_send(array.data(), array.size(), 1);
+    } catch (ipc::interprocess_exception &ex) {
+        qDebug() << "send to gui process:" << ex.what();
+    }
+}
+
+void NativeMessagingManager::ReceiverThread::run()
+{
+    ipc::message_queue::remove("chatterino_gui");
+
+    ipc::message_queue messageQueue(ipc::open_or_create, "chatterino_gui", 100, MESSAGE_SIZE);
+
+    while (true) {
+        try {
+            std::unique_ptr<char> buf(static_cast<char *>(malloc(MESSAGE_SIZE)));
+            ipc::message_queue::size_type retSize;
+            unsigned int priority;
+
+            messageQueue.receive(buf.get(), MESSAGE_SIZE, retSize, priority);
+
+            QJsonDocument document =
+                QJsonDocument::fromJson(QByteArray::fromRawData(buf.get(), retSize));
+
+            this->handleMessage(document.object());
+        } catch (ipc::interprocess_exception &ex) {
+            qDebug() << "received from gui process:" << ex.what();
+        }
+    }
+}
+
+void NativeMessagingManager::ReceiverThread::handleMessage(const QJsonObject &root)
+{
+    auto app = getApp();
+
+    QString action = root.value("action").toString();
+
+    if (action.isNull()) {
+        qDebug() << "NM action was null";
+        return;
+    }
+
+    if (action == "select") {
+        QString _type = root.value("type").toString();
+        bool attach = root.value("attach").toBool();
+        QString name = root.value("name").toString();
+
+        qDebug() << attach;
+
+#ifdef USEWINSDK
+        AttachedWindow::GetArgs args;
+        args.winId = root.value("winId").toString();
+        args.yOffset = root.value("yOffset").toInt(-1);
+        args.width = root.value("size").toObject().value("width").toInt(-1);
+        args.height = root.value("size").toObject().value("height").toInt(-1);
+
+        if (_type.isNull() || args.winId.isNull()) {
+            qDebug() << "NM type, name or winId missing";
+            attach = false;
+            return;
+        }
+#endif
+
+        if (_type == "twitch") {
+            postToThread([=] {
+                if (!name.isEmpty()) {
+                    app->twitch.server->watchingChannel.update(
+                        app->twitch.server->getOrAddChannel(name));
+                }
+
+                if (attach) {
+#ifdef USEWINSDK
+                    //                    if (args.height != -1) {
+                    auto *window = AttachedWindow::get(::GetForegroundWindow(), args);
+                    if (!name.isEmpty()) {
+                        window->setChannel(app->twitch.server->getOrAddChannel(name));
+                    }
+//                    }
+//                    window->show();
+#endif
+                }
+            });
+
+        } else {
+            qDebug() << "NM unknown channel type";
+        }
+    } else if (action == "detach") {
+        QString winId = root.value("winId").toString();
+
+        if (winId.isNull()) {
+            qDebug() << "NM winId missing";
+            return;
+        }
+
+#ifdef USEWINSDK
+        postToThread([winId] {
+            qDebug() << "NW detach";
+            AttachedWindow::detach(winId);
+        });
+#endif
+    } else {
+        qDebug() << "NM unknown action " + action;
+    }
+}
+
+std::string &NativeMessagingManager::getGuiMessageQueueName()
+{
+    static std::string name =
+        "chatterino_gui" + PathManager::getInstance()->applicationFilePathHash.toStdString();
+    return name;
+}
+
+}  // namespace chatterino