WebsocketClient.h

/* 
 * File:   WebsocketClient.h
 * Author: https://github.com/hrsavla
 *
 * Created on August 4, 2015, 1:37 PM
 * This Websocket Client library is ported by hrsavla into Sming from
 * https://github.com/MORA99/Stokerbot/tree/master/Libraries/WebSocketClient 
 * 
 * Refactor and improve by https://github.com/avr39-ripe - Alexander V, Ribchansky
 * 
 */

#ifndef WEBSOCKETCLIENT_H
#define WEBSOCKETCLIENT_H

#include "../Wiring/WiringFrameworkIncludes.h"
#include "TcpClient.h"
#include "../Delegate.h"
#include "URL.h"
#include "../../Services/WebHelpers/aw-sha1.h"
#include "../../Services/WebHelpers/base64.h"
#include "../../Wiring/WString.h"
#include "../Digital.h"
#include <Network/WebsocketFrame.h>

class WebsocketClient;
enum class wsMode : uint8_t
{
    Disconnected = 0, Connecting, Connected
};

typedef Delegate<void(WebsocketClient& wsClient, const String message)> WebSocketClientMessageDelegate;
typedef Delegate<void(WebsocketClient& wsClient, uint8_t* data, size_t size)> WebSocketClientBinaryDelegate;
typedef Delegate<void(WebsocketClient& wsClient, bool  successful)> WebSocketClientDisconnectDelegate;
typedef Delegate<void(WebsocketClient& wsClient, wsMode  Mode)> WebSocketClientConnectedDelegate;
class WebsocketClient: protected TcpClient
{
public:
    WebsocketClient(bool autoDestruct = false) :TcpClient(autoDestruct) {};
    virtual ~WebsocketClient() {};
    void setWebSocketMessageHandler(WebSocketClientMessageDelegate handler);
    void setWebSocketDisconnectedHandler(WebSocketClientDisconnectDelegate handler);
    void setWebSocketConnectedHandler(WebSocketClientConnectedDelegate handler);
    void setWebSocketBinaryHandler(WebSocketClientBinaryDelegate handler);
    bool connect(String url, uint32_t sslOptions = 0);
    void sendPing();
    void sendPong();
    void disconnect();
    void sendMessage(char* msg, uint16_t length);
    void sendMessage(String str);
    void sendBinary(uint8_t* msg, uint16_t length);
    wsMode getWSMode();
#ifdef ENABLE_SSL
    using TcpClient::addSslOptions;
    using TcpClient::setSslFingerprint;
    using TcpClient::pinCertificate;
    using TcpClient::setSslClientKeyCert;
    using TcpClient::freeSslClientKeyCert;
    using TcpClient::getSsl;
#endif

protected:
    virtual void onFinished(TcpClientState finishState);
    virtual void onError(err_t err);
    virtual err_t onReceive(pbuf *buf);
    bool _verifyKey(char *buf, int size);
    void _sendFrame(WSFrameType frameType, uint8_t* msg, uint16_t length);
private:
    URL _uri;
    String _url;
    wsMode _mode = wsMode::Disconnected;
    WebSocketClientMessageDelegate wsMessage = nullptr;
    WebSocketClientBinaryDelegate wsBinary = nullptr;
    WebSocketClientDisconnectDelegate wsDisconnect = nullptr;
    WebSocketClientConnectedDelegate wsConnect = nullptr;
    String _key;
};

#endif  /* WEBSOCKETCLIENT_H */