Minor refactor for null rpc scheme handler

1 files changed, 44 insertions, 0 deletions

diff --git a/src/schemes/NullRpcSchemeHandler.cpp b/src/schemes/NullRpcSchemeHandler.cpp
new file mode 100644
index 0000000..0b4d993
--- /dev/null
+++ b/src/schemes/NullRpcSchemeHandler.cpp
@@ -0,0 +1,44 @@
+#include "NullRpcSchemeHandler.hpp"
+
+const QStringList NullRPCSchemeHandler::allowed_origin_schemes = {
+    "nullrpc",
+};
+
+void NullRPCSchemeHandler::requestStarted(QWebEngineUrlRequestJob *job) {
+  auto url = job->requestUrl();
+  auto origin = job->initiator();
+  qDebug() << "RPC Msg:" << url << " | From: " << job->initiator();
+
+  auto is_origin_allowed = allowed_origin_schemes.contains(origin.scheme());
+  if (!is_origin_allowed || url.host().isEmpty()) {
+    send_empty_response(job);
+    return;
+  }
+
+  if (url.host() == "noop") {
+    send_empty_response(job, "text/html");
+    return;
+  }
+
+  if (url.host() == "log") {
+    qInfo() << url.query();
+    send_empty_response(job);
+    return;
+  }
+
+  NullRPCMessage message{
+      .name = url.host(),
+      .params = QUrlQuery(url.query()),
+  };
+  emit message_received(message);
+
+  // TODO: responses managed with request ids
+  send_empty_response(job);
+}
+
+void NullRPCSchemeHandler::send_empty_response(QWebEngineUrlRequestJob *job,
+                                               const QByteArray &content_type) {
+  QBuffer *buffer = new QBuffer(job);
+  buffer->open(QIODevice::ReadOnly);
+  job->reply(content_type, new QBuffer(job));
+}