WebsocketClient.h

/****
 * Sming Framework Project - Open Source framework for high efficiency native ESP8266 development.
 * Created 2015 by Skurydin Alexey
 * http://github.com/anakod/Sming
 * All files of the Sming Core are provided under the LGPL v3 license.
 *
 * WebsocketClient.h
 *
 * @authors:
 *       Originally - hrsavla <https://github.com/hrsavla>
 *       Refactored - Alexander V, Ribchansky <https://github.com/avr39-ripe>
 *       Refactored - Slavey Karadzhov <slav@attachix.com>
 *
 ****/

//TODO: Add stream support for sending big chunks of data via websockets.

#ifndef _SMING_CORE_NETWORK_WEBSOCKET_CLIENT_H_
#define _SMING_CORE_NETWORK_WEBSOCKET_CLIENT_H_

#include "Http/HttpClientConnection.h"
#include "Http/Websocket/WebsocketConnection.h"

class WebsocketClient : protected WebsocketConnection
{
public:
    WebsocketClient() : WebsocketConnection(new HttpClientConnection)
    {
    }

    WebsocketClient(HttpConnection* connection) : WebsocketConnection(connection)
    {
    }

    using WebsocketConnection::setBinaryHandler;
    using WebsocketConnection::setConnectionHandler;
    using WebsocketConnection::setDisconnectionHandler;
    using WebsocketConnection::setMessageHandler;

    HttpConnection* getHttpConnection();

    bool connect(const Url& url, uint32_t sslOptions = 0);

    using WebsocketConnection::send;
    using WebsocketConnection::sendBinary;
    using WebsocketConnection::sendString;

    void sendPing(const String& payload = nullptr)
    {
        debug_d("Sending PING");
        WebsocketConnection::send(payload.c_str(), payload.length(), WS_FRAME_PING);
    }

    void sendPong(const String& payload = nullptr)
    {
        debug_d("Sending PONG");
        WebsocketConnection::send(payload.c_str(), payload.length(), WS_FRAME_PONG);
    }

    using WebsocketConnection::close;
    using WebsocketConnection::getState;

    void disconnect() SMING_DEPRECATED
    {
        close();
    }

protected:
    int verifyKey(HttpConnection& connection, HttpResponse& response);

private:
    Url uri;
    String key;
};

#endif /* _SMING_CORE_NETWORK_WEBSOCKET_CLIENT_H_ */