websocket_base.hpp

Go to the documentation of this file.

#pragma once
 
#include <roar/websocket/read_result.hpp>
#include <roar/error.hpp>
#include <roar/detail/promise_compat.hpp>
#include <roar/detail/stream_type.hpp>
 
#include <promise-cpp/promise.hpp>
#include <boost/beast/ssl/ssl_stream.hpp>
#include <boost/beast/core/tcp_stream.hpp>
#include <boost/beast/websocket.hpp>
#include <boost/beast/websocket/ssl.hpp>
 
#include <variant>
 
namespace Roar
{
    class WebsocketBase : public std::enable_shared_from_this<WebsocketBase>
    {
      public:
        WebsocketBase(std::variant<Detail::StreamType, boost::beast::ssl_stream<Detail::StreamType>>&& stream);
        WebsocketBase(std::variant<
                      boost::beast::websocket::stream<boost::beast::ssl_stream<Detail::StreamType>>,
                      boost::beast::websocket::stream<Detail::StreamType>>&& ws);
        WebsocketBase(WebsocketBase&&) = delete;
        WebsocketBase& operator=(WebsocketBase&&) = delete;
        WebsocketBase(WebsocketBase const&) = delete;
        WebsocketBase& operator=(WebsocketBase const&) = delete;
        virtual ~WebsocketBase();
 
        template <typename FunctionT>
        auto withStreamDo(FunctionT&& func)
        {
            return std::visit(std::forward<FunctionT>(func), ws_);
        }
 
        template <typename FunctionT>
        auto withStreamDo(FunctionT&& func) const
        {
            return std::visit(std::forward<FunctionT>(func), ws_);
        }
 
        void readMessageMax(std::size_t amount);
 
        std::size_t readMessageMax() const;
 
        void binary(bool enable);
 
        bool binary() const;
 
        Detail::PromiseTypeBind<Detail::PromiseTypeBindThen<std::size_t>, Detail::PromiseTypeBindFail<Error const&>>
        send(std::string message);
 
        Detail::
            PromiseTypeBind<Detail::PromiseTypeBindThen<WebsocketReadResult>, Detail::PromiseTypeBindFail<Error const&>>
            read();
 
        Detail::
            PromiseTypeBind<Detail::PromiseTypeBindThen<WebsocketReadResult>, Detail::PromiseTypeBindFail<Error const&>>
            read_some(std::size_t limit = 4096);
 
        void autoFragment(bool enable);
 
        bool autoFragment() const;
 
        bool close(std::chrono::seconds closeWaitTimeout = std::chrono::seconds{3});
 
      protected:
        std::variant<
            boost::beast::websocket::stream<boost::beast::ssl_stream<Detail::StreamType>>,
            boost::beast::websocket::stream<Detail::StreamType>>
            ws_;
    };
}