Add JSON output to RC

1 files changed, 89 insertions, 5 deletions

diff --git a/lib/RemoteControl.cpp b/lib/RemoteControl.cpp
index 30dcb60..6e9af20 100644
--- a/lib/RemoteControl.cpp
+++ b/lib/RemoteControl.cpp
@@ -25,6 +25,8 @@
 #include <list>
 #include <string>
 #include <iostream>
+#include <sstream>
+#include <iomanip>
 #include <string>
 #include <algorithm>
 
@@ -102,6 +104,74 @@ std::list< std::vector<std::string> > RemoteControllers::get_param_list_values(c
     return allparams;
 }
 
+
+static std::string escape_json(const std::string &s) {
+    std::ostringstream o;
+    for (auto c = s.cbegin(); c != s.cend(); c++) {
+        switch (*c) {
+        case '"': o << "\\\""; break;
+        case '\\': o << "\\\\"; break;
+        case '\b': o << "\\b"; break;
+        case '\f': o << "\\f"; break;
+        case '\n': o << "\\n"; break;
+        case '\r': o << "\\r"; break;
+        case '\t': o << "\\t"; break;
+        default:
+            if ('\x00' <= *c && *c <= '\x1f') {
+                o << "\\u"
+                  << std::hex << std::setw(4) << std::setfill('0') << static_cast<int>(*c);
+            } else {
+                o << *c;
+            }
+        }
+    }
+    return o.str();
+}
+
+std::string RemoteControllers::get_params_json(const std::string& name) {
+    RemoteControllable* controllable = get_controllable_(name);
+    const auto& values = controllable->get_all_values();
+
+    std::ostringstream ss;
+    ss << "{ ";
+    size_t ix = 0;
+    for (const auto& element : values) {
+        if (ix > 0) {
+            ss << ",";
+        }
+
+        ss << "\"" << escape_json(element.first) << "\": ";
+
+        const auto& value = element.second;
+        if (std::holds_alternative<string>(value)) {
+            ss << "\"" << escape_json(std::get<string>(value)) << "\"";
+        }
+        else if (std::holds_alternative<double>(value)) {
+            ss << std::defaultfloat << std::get<double>(value);
+        }
+        else if (std::holds_alternative<ssize_t>(value)) {
+            ss << std::get<ssize_t>(value);
+        }
+        else if (std::holds_alternative<size_t>(value)) {
+            ss << std::get<size_t>(value);
+        }
+        else if (std::holds_alternative<bool>(value)) {
+            ss << (std::get<bool>(value) ? "true" : "false");
+        }
+        else if (std::holds_alternative<std::nullopt_t>(value)) {
+            ss << "null";
+        }
+        else {
+            throw std::logic_error("variant alternative not handled");
+        }
+
+        ix++;
+    }
+    ss << " }";
+
+    return ss.str();
+}
+
 std::string RemoteControllers::get_param(const std::string& name, const std::string& param) {
     RemoteControllable* controllable = get_controllable_(name);
     return controllable->get_parameter(param);
@@ -427,10 +497,15 @@ void RemoteControllerZmq::recv_all(zmq::socket_t& pSocket, std::vector<std::stri
     bool more = true;
     do {
         zmq::message_t msg;
-        pSocket.recv(msg);
-        std::string incoming((char*)msg.data(), msg.size());
-        message.push_back(incoming);
-        more = msg.more();
+        const auto zresult = pSocket.recv(msg);
+        if (zresult) {
+            std::string incoming((char*)msg.data(), msg.size());
+            message.push_back(incoming);
+            more = msg.more();
+        }
+        else {
+            more = false;
+        }
     } while (more);
 }
 
@@ -532,6 +607,15 @@ void RemoteControllerZmq::process()
                         send_fail_reply(repSocket, err.what());
                     }
                 }
+                else if (msg.size() == 2 && command == "showjson") {
+                    std::string module((char*) msg[1].data(), msg[1].size());
+                    std::string json = rcs.get_params_json(module);
+
+                    zmq::message_t zmsg(json.size());
+                    memcpy(zmsg.data(), json.data(), json.size());
+
+                    repSocket.send(zmsg, zmq::send_flags::none);
+                }
                 else if (msg.size() == 3 && command == "get") {
                     std::string module((char*) msg[1].data(), msg[1].size());
                     std::string parameter((char*) msg[2].data(), msg[2].size());
@@ -561,7 +645,7 @@ void RemoteControllerZmq::process()
                 }
                 else {
                     send_fail_reply(repSocket,
-                            "Unsupported command. commands: list, show, get, set");
+                            "Unsupported command. commands: list, show, get, set, showjson");
                 }
             }
         }