ws_server_adapter.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2024, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include "kcenon/network/interfaces/i_protocol_server.h"
 
#include <memory>
#include <mutex>
#include <string>
#include <unordered_map>
 
namespace kcenon::network::core {
    class messaging_ws_server;
    class ws_connection;
}
 
namespace kcenon::network::internal::adapters {
 
class ws_session_wrapper : public interfaces::i_session {
public:
    explicit ws_session_wrapper(std::shared_ptr<core::ws_connection> connection);
 
    [[nodiscard]] auto id() const -> std::string_view override;
    [[nodiscard]] auto is_connected() const -> bool override;
    [[nodiscard]] auto send(std::vector<uint8_t>&& data) -> VoidResult override;
    auto close() -> void override;
 
private:
    std::shared_ptr<core::ws_connection> connection_;
    std::string id_cache_;
};
 
class ws_server_adapter : public interfaces::i_protocol_server {
public:
    explicit ws_server_adapter(std::string_view server_id);
 
    ~ws_server_adapter() override;
 
    // Non-copyable, non-movable
    ws_server_adapter(const ws_server_adapter&) = delete;
    ws_server_adapter& operator=(const ws_server_adapter&) = delete;
    ws_server_adapter(ws_server_adapter&&) = delete;
    ws_server_adapter& operator=(ws_server_adapter&&) = delete;
 
    // =========================================================================
    // Path Configuration
    // =========================================================================
 
    auto set_path(std::string_view path) -> void;
 
    // =========================================================================
    // i_network_component interface implementation
    // =========================================================================
 
    [[nodiscard]] auto is_running() const -> bool override;
    auto wait_for_stop() -> void override;
 
    // =========================================================================
    // i_protocol_server interface implementation
    // =========================================================================
 
    [[nodiscard]] auto start(uint16_t port) -> VoidResult override;
    [[nodiscard]] auto stop() -> VoidResult override;
    [[nodiscard]] auto connection_count() const -> size_t override;
 
    auto set_connection_callback(connection_callback_t callback) -> void override;
    auto set_disconnection_callback(disconnection_callback_t callback) -> void override;
    auto set_receive_callback(receive_callback_t callback) -> void override;
    auto set_error_callback(error_callback_t callback) -> void override;
 
private:
    auto setup_internal_callbacks() -> void;
 
    auto get_or_create_session(std::shared_ptr<core::ws_connection> connection)
        -> std::shared_ptr<interfaces::i_session>;
 
    std::string server_id_;
    std::string path_{"/"}; // Default WebSocket path
    std::shared_ptr<core::messaging_ws_server> server_;
 
    mutable std::mutex callbacks_mutex_;
    connection_callback_t connection_callback_;
    disconnection_callback_t disconnection_callback_;
    receive_callback_t receive_callback_;
    error_callback_t error_callback_;
 
    mutable std::mutex sessions_mutex_;
    std::unordered_map<std::string, std::shared_ptr<ws_session_wrapper>> sessions_;
};
 
} // namespace kcenon::network::internal::adapters