#pragma once

#include <drogon/WebSocketController.h>

using namespace drogon;

class WsClient : public drogon::WebSocketController<WsClient>
{
  public:
    virtual void handleNewMessage(const WebSocketConnectionPtr &,
                                  std::string &&,
                                  const WebSocketMessageType &) override;
    virtual void handleNewConnection(const HttpRequestPtr &,
                                     const WebSocketConnectionPtr &) override;
    virtual void handleConnectionClosed(
        const WebSocketConnectionPtr &) override;
    WS_PATH_LIST_BEGIN
    WS_PATH_ADD("/sub");
    WS_PATH_ADD("/pub");
    WS_PATH_LIST_END
};