Initial commit

Lts/Proxies/Buildings/web_socket_handler.cpp

@@ -0,0 +1,219 @@
+/**
+  * Websocket interface to participant positions
+  * $Id$
+  * $HeadURL$
+  * $Author$
+  * Created by Justus Hertel, SLSE, 2020-07-08
+  */
+
+#include "web_socket_handler.h"
+#include "building_process.h"
+
+#include <QWebSocket>
+#include <QJsonObject>
+#include <QJsonDocument>
+#include <QJsonArray>
+
+//#include <ParticipantManagement/ppm.hpp>
+
+namespace Buildings
+{
+
+WebSocketHandler::WebSocketHandler(const std::shared_ptr<BuildingControlHandler> &buildingControlHandler, QObject *parent)
+  : QObject(parent)
+  , m_buildingControlHandler(buildingControlHandler)
+  , m_webSocketServer(new QWebSocketServer(QStringLiteral("BuildingsSocket"), QWebSocketServer::NonSecureMode, this))
+  , m_logger(getLogger(QStringLiteral("WebSocketHandler")))
+{}
+
+WebSocketHandler::~WebSocketHandler()
+{
+  for (const auto& ws : qAsConst(m_webSockets))
+  {
+    ws->disconnect(this);
+  }
+  m_webSockets.clear();
+}
+
+bool WebSocketHandler::init(quint16 listenPort)
+{
+  DEBUG_LOG(m_logger) << "init";
+  if (!m_webSocketServer->listen(QHostAddress::Any, listenPort))
+  {
+    WARNING_LOG(m_logger) << "Can not listen on port " << listenPort << " - aborting";
+    return false;
+  }
+
+  connect(m_webSocketServer, &QWebSocketServer::newConnection,
+          this, &WebSocketHandler::onNewConnection);
+
+
+  INFO_LOG(m_logger) << "Listening on port " << m_webSocketServer->serverPort();
+  return true;
+}
+
+bool WebSocketHandler::init(const QPair<quint16, quint16> &portRange)
+{
+  for (quint16 currentPort = portRange.first; currentPort != portRange.second; ++currentPort)
+  {
+    if (init(currentPort))
+      return true;
+  }
+
+  return init();
+}
+
+bool WebSocketHandler::sendTelegramInternal() const
+{
+  DEBUG_LOG(m_logger) << "sendTelegramInternal";
+  QJsonObject obj;
+  obj[QStringLiteral("participantId")]   = 1;
+  obj[QStringLiteral("equipmentNumber")] = 2;
+  obj[QStringLiteral("lat")]             = 3.0;
+  obj[QStringLiteral("lon")]             = 4.0;
+  obj[QStringLiteral("height")]          = 5.0;
+
+  const QString jsonStr = QString::fromUtf8(QJsonDocument(obj).toJson(QJsonDocument::Compact));
+
+  bool ret = false;
+  for (const auto &wsw : qAsConst(m_webSockets))
+  {
+    if (wsw->sendTextMessage(jsonStr))
+      ret = true;
+  }
+  return ret;
+}
+
+bool WebSocketHandler::sendbuildingInfos() const
+{
+  DEBUG_LOG(m_logger) << "sendbuildingInfos";
+  QJsonObject obj;
+  QJsonArray arr;
+
+
+  QMapIterator<int, BuildingProcess *> i(m_buildingControlHandler->getBuildings());
+  while (i.hasNext()) {
+    i.next();
+    BuildingProcess* buildingProcess = i.value();
+    BuildingInfo buildingInfo = buildingProcess->buildingInfo();
+    quint16 port = buildingProcess->getWebsocketPort();
+    DEBUG_LOG(m_logger) << "key=" << i.key()
+                         << " port= " << port
+                         << " buildingInfo.name=" << buildingInfo.name;
+    QJsonObject objIter;
+    objIter[QStringLiteral("building_name")]   = buildingInfo.name;
+    objIter[QStringLiteral("port")]   = port;
+    arr.push_back(objIter);
+  }
+  obj[QStringLiteral("type")] = QStringLiteral("Buildings");
+  obj[QStringLiteral("data")] = arr;
+
+
+
+  const QString jsonStr = QString::fromUtf8(QJsonDocument(obj).toJson(QJsonDocument::Compact));
+
+  bool ret = false;
+  for (const auto &wsw : qAsConst(m_webSockets))
+  {
+    if (wsw->sendTextMessage(jsonStr))
+      ret = true;
+  }
+  return ret;
+}
+
+void WebSocketHandler::onNewMessage(const QString &message)
+{
+  DEBUG_LOG(m_logger) << "onNewMessage='" << message << "'";
+  DEBUG_LOG(m_logger) << "m_buildingControlHandler->m_buildings.size()=" << m_buildingControlHandler->getBuildings().size();
+
+  if (message.indexOf(QStringLiteral("buildingRequest")) > 0)
+  {
+    sendbuildingInfos();
+  }
+  if (message.indexOf(QStringLiteral("triggerFlashlight")) > 0)
+  {
+    DEBUG_LOG(m_logger) << "triggerFlashlight" ;
+    triggerFlashlight(message);
+  }
+}
+
+QJsonObject WebSocketHandler::ObjectFromString(const QString& in)
+{
+    QJsonObject obj;
+
+    QJsonDocument doc = QJsonDocument::fromJson(in.toUtf8());
+
+    // check validity of the document
+    if(!doc.isNull())
+    {
+        if(doc.isObject())
+        {
+            obj = doc.object();
+        }
+        else
+        {
+            WARNING_LOG(m_logger) << "Document is not an object";
+        }
+    }
+    else
+    {
+        WARNING_LOG(m_logger) << "Invalid JSON...\n" << in ;
+    }
+
+    return obj;
+}
+
+
+void WebSocketHandler::triggerFlashlight(const QString &message)
+{
+  DEBUG_LOG(m_logger) << "triggerFlashlight message=" << message;
+  QJsonObject obj = ObjectFromString(message);
+  QJsonObject dataObj = obj.value(QStringLiteral("data")).toObject();
+  int structureElementId = dataObj.value(QStringLiteral("structureElementId")).toInt();
+  QString hostnameBsm = dataObj.value(QStringLiteral("hostnameBsm")).toString();
+
+  DEBUG_LOG(m_logger) << "hostnameBsm=" << hostnameBsm
+                      << " structureElementId=" << structureElementId;
+
+}
+
+void WebSocketHandler::onNewConnection()
+{
+  DEBUG_LOG(m_logger) << "onNewConnection";
+  QWebSocket *ws = m_webSocketServer->nextPendingConnection();
+
+  if (!ws)
+    return;
+
+  connect(ws, &QWebSocket::disconnected,
+          this, &WebSocketHandler::onWebSocketDisconnected);
+  connect(ws, &QWebSocket::textMessageReceived,
+          this, &WebSocketHandler::onNewMessage);
+
+  m_webSockets += ws;
+
+  INFO_LOG(m_logger) << "New websocket connected";
+}
+
+void WebSocketHandler::onWebSocketDisconnected()
+{
+  DEBUG_LOG(m_logger) << "onWebSocketDisconnected";
+  QWebSocket *ws = qobject_cast<QWebSocket *>(sender());
+  if (!ws)
+    return;
+
+  m_webSockets.removeAll(ws);
+  ws->deleteLater();
+
+  INFO_LOG(m_logger) << "A websocket disconnected";
+}
+
+bool WebSocketHandler::sendPosition()
+{
+  if (!isConnected() || m_webSockets.isEmpty())
+    return false;
+
+  return sendTelegramInternal();
+}
+
+}