#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));
}