/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
*
* Copyright 2013 - 2022, nymea GmbH
* Contact: contact@nymea.io
*
* This file is part of nymea.
* This project including source code and documentation is protected by copyright law, and
* remains the property of nymea GmbH. All rights, including reproduction, publication,
* editing and translation, are reserved. The use of this project is subject to the terms of a
* license agreement to be concluded with nymea GmbH in accordance with the terms
* of use of nymea GmbH, available under https://nymea.io/license
*
* GNU Lesser General Public License Usage
* Alternatively, this project may be redistributed and/or modified under the terms of the GNU
* Lesser General Public License as published by the Free Software Foundation; version 3.
* this project is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License along with this project.
* If not, see <https://www.gnu.org/licenses/>.
*
* For any further details and any questions please contact us under contact@nymea.io
* or see our FAQ/Licensing Information on https://nymea.io/license/faq
*
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#include "mqttwebsocketclienttransport.h"

MqttWebSocketClientTransport::MqttWebSocketClientTransport(const QNetworkRequest &request, QObject *parent):
    MqttClientTransport(parent),
    m_request(request)
{
    m_socket = new QWebSocket(QString(), QWebSocketProtocol::VersionLatest, this);

    connect(m_socket, &QWebSocket::connected, this, &MqttClientTransport::connected);
    connect(m_socket, &QWebSocket::disconnected, this, &MqttClientTransport::disconnected);
    connect(m_socket, &QWebSocket::stateChanged, this, &MqttClientTransport::stateChanged);
    connect(m_socket, &QWebSocket::sslErrors, this, &MqttClientTransport::sslErrors);
    typedef void (QWebSocket:: *errorSignal)(QAbstractSocket::SocketError);
    connect(m_socket, static_cast<errorSignal>(&QWebSocket::error), this, &MqttClientTransport::errorSignal);

    connect(m_socket, &QWebSocket::textMessageReceived, this, &MqttWebSocketClientTransport::onTextMessageReceived);
    connect(m_socket, &QWebSocket::binaryMessageReceived, this, &MqttWebSocketClientTransport::onBinaryMessageReceived);
}

void MqttWebSocketClientTransport::connectToHost()
{
#if QT_VERSION < QT_VERSION_CHECK(5, 6, 0)
    if (!m_request.rawHeaderList().isEmpty()) {
        qCWarning(dbgClient) << "Qt versions older than 5.6 do not support HTTP request headers with web sockets. The connection may fail.";
    }
    m_socket->open(m_request.url());
#else
    m_socket->open(m_request);
#endif
}

void MqttWebSocketClientTransport::abort()
{
    m_socket->abort();
}

bool MqttWebSocketClientTransport::isOpen() const
{
    return m_socket->isValid();
}

bool MqttWebSocketClientTransport::write(const QByteArray &data)
{
    int ret = m_socket->sendBinaryMessage(data);
    return ret == data.length();
}

void MqttWebSocketClientTransport::flush()
{
    m_socket->flush();
}

void MqttWebSocketClientTransport::disconnectFromHost()
{
    m_socket->close();
}

QAbstractSocket::SocketState MqttWebSocketClientTransport::state() const
{
    return m_socket->state();
}

void MqttWebSocketClientTransport::ignoreSslErrors()
{
    m_socket->ignoreSslErrors();
}

void MqttWebSocketClientTransport::onTextMessageReceived(const QString &message)
{
    qCDebug(dbgClient()) << "Text message received:" << message;
    qCWarning(dbgClient()) << "Received a text message from the server. Doesn't look like MQTT. Closing connection.";
    m_socket->close(QWebSocketProtocol::CloseCodeProtocolError);
}

void MqttWebSocketClientTransport::onBinaryMessageReceived(const QByteArray &message)
{
    emit dataReceived(message);
}