#pragma once #include "baserule.hpp" #include "websocket.hpp" #include #include #include #include namespace crow { class WebSocketRule : public BaseRule { using self_t = WebSocketRule; public: explicit WebSocketRule(const std::string& ruleIn) : BaseRule(ruleIn) {} void validate() override {} void handle(const Request& /*req*/, const std::shared_ptr& asyncResp, const std::vector& /*params*/) override { asyncResp->res.result(boost::beast::http::status::not_found); } #ifndef BMCWEB_ENABLE_SSL void handleUpgrade(const Request& req, const std::shared_ptr& /*asyncResp*/, boost::asio::ip::tcp::socket&& adaptor) override { BMCWEB_LOG_DEBUG("Websocket handles upgrade"); std::shared_ptr< crow::websocket::ConnectionImpl> myConnection = std::make_shared< crow::websocket::ConnectionImpl>( req, req.url(), std::move(adaptor), openHandler, messageHandler, messageExHandler, closeHandler, errorHandler); myConnection->start(); } #else void handleUpgrade(const Request& req, const std::shared_ptr& /*asyncResp*/, boost::beast::ssl_stream&& adaptor) override { BMCWEB_LOG_DEBUG("Websocket handles upgrade"); std::shared_ptr>> myConnection = std::make_shared>>( req, req.url(), std::move(adaptor), openHandler, messageHandler, messageExHandler, closeHandler, errorHandler); myConnection->start(); } #endif template self_t& onopen(Func f) { openHandler = f; return *this; } template self_t& onmessage(Func f) { messageHandler = f; return *this; } template self_t& onmessageex(Func f) { messageExHandler = f; return *this; } template self_t& onclose(Func f) { closeHandler = f; return *this; } template self_t& onerror(Func f) { errorHandler = f; return *this; } protected: std::function openHandler; std::function messageHandler; std::function&& whenComplete)> messageExHandler; std::function closeHandler; std::function errorHandler; }; } // namespace crow