Initial commit

2023-11-23 08:19:32 +00:00
commit 852d59b0b3
5 changed files with 229 additions and 0 deletions
--- a/stompframe.cpp
+++ b/stompframe.cpp
@@ -0,0 +1,151 @@
+#include "stompframe.h"
+
+STOMPFrame::STOMPFrame(Command command, std::initializer_list<QPair<QString, QString>> headers, const QByteArray& body):
+    command_(command), body_(body) {
+    for (const auto& header: headers) {
+        headers_[header.first] = header.second;
+    }
+}
+
+qint64 STOMPFrame::send(QIODevice& device) {
+    QByteArray encoded;
+    QTextStream out {&encoded};
+    out << command_ << "\n";
+    for (const auto& key: headers_.keys()) {
+        out << key << ":" << headers_[key] << "\n";
+    }
+    if (!headers_.contains("content-length")) {
+        out << "content-length" << ":" << body_.size() << "\n";
+    }
+    out << "\n" << body_ << '\0';
+    out.flush();
+    return device.write(encoded);
+}
+
+STOMPFrame STOMPFrame::receive(QIODevice& device) {
+    STOMPFrame frame;
+    QTextStream in(&device);
+
+    QString commandStr;
+    if (!in.readLineInto(&commandStr)) {
+        return {};
+    }
+    if (commandStr == "STOMP") {
+        frame.command_ = STOMPFrame::STOMP;
+    } else if (commandStr == "SEND") {
+        frame.command_ = STOMPFrame::SEND;
+    } else if (commandStr == "SUBSCRIBE") {
+        frame.command_ = STOMPFrame::SUBSCRIBE;
+    } else if (commandStr == "UNSUBSCRIBE") {
+        frame.command_ = STOMPFrame::UNSUBSCRIBE;
+    } else if (commandStr == "BEGIN") {
+        frame.command_ = STOMPFrame::BEGIN;
+    } else if (commandStr == "COMMIT") {
+        frame.command_ = STOMPFrame::COMMIT;
+    } else if (commandStr == "ABORT") {
+        frame.command_ = STOMPFrame::ABORT;
+    } else if (commandStr == "ACK") {
+        frame.command_ = STOMPFrame::ACK;
+    } else if (commandStr == "NACK") {
+        frame.command_ = STOMPFrame::NACK;
+    } else if (commandStr == "DISCONNECT") {
+        frame.command_ = STOMPFrame::DISCONNECT;
+    } else if (commandStr == "CONNECTED") {
+        frame.command_ = STOMPFrame::CONNECTED;
+    } else if (commandStr == "MESSAGE") {
+        frame.command_ = STOMPFrame::MESSAGE;
+    } else if (commandStr == "RECEIPT") {
+        frame.command_ = STOMPFrame::RECEIPT;
+    } else if (commandStr == "ERROR") {
+        frame.command_ = STOMPFrame::ERROR;
+    } else {
+        return {};
+    }
+
+    QString headerLine;
+    while (in.readLineInto(&headerLine) && headerLine != "") {
+        auto components = headerLine.split(':');
+        if (components.count() != 2) {
+            return {};
+        }
+        frame.headers_.insert(components[0].trimmed(), components[1].trimmed());
+    }
+
+    if (frame.headers_.contains("content-length")) {
+        bool conversionOk;
+        qint64 length = frame.headers_["content-length"].toLongLong(&conversionOk);
+        if (!conversionOk) {
+            return {};
+        }
+        while (frame.body_.length() < length) {
+            frame.body_ += in.read(length - frame.body_.length()).toUtf8();
+        }
+    } else {
+        in >> frame.body_;
+    }
+
+    return frame;
+}
+
+QTextStream& operator <<(QTextStream& out, STOMPFrame::Command command) {
+    switch (command) {
+        case STOMPFrame::INVALID:
+            break;
+
+        case STOMPFrame::STOMP:
+            out << "STOMP";
+            break;
+
+        case STOMPFrame::SEND:
+            out << "SEND";
+            break;
+
+        case STOMPFrame::SUBSCRIBE:
+            out << "SUBSCRIBE";
+            break;
+
+        case STOMPFrame::UNSUBSCRIBE:
+            out << "UNSUBSCRIBE";
+            break;
+
+        case STOMPFrame::BEGIN:
+            out << "BEGIN";
+            break;
+
+        case STOMPFrame::COMMIT:
+            out << "COMMIT";
+            break;
+
+        case STOMPFrame::ABORT:
+            out << "ABORT";
+            break;
+
+        case STOMPFrame::ACK:
+            out << "ACK";
+            break;
+
+        case STOMPFrame::NACK:
+            out << "NACK";
+            break;
+
+        case STOMPFrame::DISCONNECT:
+            out << "DISCONNECT";
+            break;
+
+        case STOMPFrame::CONNECTED:
+            out << "CONNECTED";
+            break;
+
+        case STOMPFrame::MESSAGE:
+            out << "MESSAGE";
+            break;
+
+        case STOMPFrame::RECEIPT:
+            out << "RECEIPT";
+            break;
+        case STOMPFrame::ERROR:
+            out << "ERROR";
+            break;
+    }
+    return out;
+}