hyperion.ng/libsrc/webserver/WebSocketJsonHandler.cpp
2024-12-28 14:05:23 +01:00

64 lines
2.2 KiB
C++

#include "WebSocketJsonHandler.h"
#include <api/JsonAPI.h>
#include <api/JsonCallbacks.h>
#include <utils/JsonUtils.h>
#include <utils/NetOrigin.h>
#define NO_TRACE_SEND
#define NO_TRACE_RECEIVE
WebSocketJsonHandler::WebSocketJsonHandler(QWebSocket* websocket, QObject* parent)
: QObject(parent)
, _websocket(websocket)
, _log(Logger::getInstance("WEBSOCKET"))
{
connect(_websocket, &QWebSocket::textMessageReceived, this, &WebSocketJsonHandler::onTextMessageReceived);
connect(_websocket, &QWebSocket::binaryMessageReceived, this, &WebSocketJsonHandler::onBinaryMessageReceived);
connect(_websocket, &QWebSocket::disconnected, this, &WebSocketJsonHandler::onDisconnected);
_peerAddress = _websocket->peerAddress().toString();
_origin = websocket->origin();
Debug(_log, "New WebSocket connection from %s initiated via: %s", QSTRING_CSTR(_peerAddress), QSTRING_CSTR(_origin));
bool localConnection = NetOrigin::getInstance()->isLocalAddress(_websocket->peerAddress(), _websocket->localAddress());
// Json processor
_jsonAPI.reset(new JsonAPI(_peerAddress, _log, localConnection, this));
connect(_jsonAPI.get(), &JsonAPI::callbackReady, this, &WebSocketJsonHandler::sendMessage);
connect(_jsonAPI->getCallBack().get(), &JsonCallbacks::callbackReady, this, &WebSocketJsonHandler::sendMessage);
// Init JsonAPI
_jsonAPI->initialize();
}
void WebSocketJsonHandler::onTextMessageReceived(const QString& message)
{
#ifdef RECEIVE_TRACE
qDebug() << "[" << _peerAddress << "] WebSocket message received:" << message;
#endif
_jsonAPI->handleMessage(message);
}
void WebSocketJsonHandler::onBinaryMessageReceived(const QByteArray& message)
{
#ifdef RECEIVE_TRACE
qDebug() << "[" << _peerAddress << "] WebSocket message received:" << message.toHex();
#endif
Warning(_log,"Unexpected binary message received");
}
qint64 WebSocketJsonHandler::sendMessage(QJsonObject obj)
{
#ifdef TRACE_SEND
qDebug() << "[" << _peerAddress << "] WebSocket send message: " << obj;
#endif
return _websocket->sendTextMessage(JsonUtils::jsonValueToQString(obj));
}
void WebSocketJsonHandler::onDisconnected()
{
Debug(_log, "WebSocket disconnected from %s initiated via: %s", QSTRING_CSTR(_peerAddress), QSTRING_CSTR(_origin));
}