websocket_transport.hpp

Go to the documentation of this file.

// Copyright (c) 2025 [arookieofc]
 
#pragma once
 
 
#include <chrono>
#include <cstddef>
#include <cstdint>
#include <memory>
#include <optional>
#include <string>
#include <string_view>
#include <unordered_map>
 
#include "cxxmcp/core/result.hpp"
#include "cxxmcp/protocol/types.hpp"
#include "cxxmcp/roles.hpp"
#include "cxxmcp/transport/transport.hpp"
 
namespace mcp::transport {
 
struct WebSocketClientTransportOptions {
  std::string uri;
 
  std::string host = "127.0.0.1";
 
  int port = 80;
 
  std::string path = "/mcp";
 
  std::unordered_map<std::string, std::string> headers;
 
  std::optional<std::string> auth_header;
 
  std::chrono::milliseconds timeout{30000};
 
  bool auto_reconnect = true;
 
  int max_reconnect_attempts = 0;
 
  std::chrono::milliseconds reconnect_initial_delay{1000};
 
  std::chrono::milliseconds reconnect_max_delay{30000};
 
  double reconnect_backoff_multiplier = 2.0;
 
  time_t ping_interval_sec = 30;
 
  int max_missed_pongs = 3;
};
 
struct WebSocketServerTransportOptions {
  std::string listen_host = "127.0.0.1";
 
  int listen_port = 0;
 
  std::string path = "/mcp";
 
  std::size_t max_connections = 1024;
 
  time_t ping_interval_sec = 30;
 
  int max_missed_pongs = 3;
};
 
class WebSocketClientTransport final : public ClientTransport {
 public:
  explicit WebSocketClientTransport(WebSocketClientTransportOptions options);
 
  ~WebSocketClientTransport() override;
 
  WebSocketClientTransport(const WebSocketClientTransport&) = delete;
  WebSocketClientTransport& operator=(const WebSocketClientTransport&) = delete;
 
  std::string_view name() const noexcept override;
 
  protocol::Json diagnostics() const override;
 
  core::Result<core::Unit> send(TxMessage message) override;
 
  core::Result<std::optional<RxMessage>> receive() override;
 
  core::Result<core::Unit> close() override;
 
 private:
  class Impl;
  std::unique_ptr<Impl> impl_;
};
 
class WebSocketServerTransport final : public ServerTransport {
 public:
  explicit WebSocketServerTransport(WebSocketServerTransportOptions options);
 
  ~WebSocketServerTransport() override;
 
  WebSocketServerTransport(const WebSocketServerTransport&) = delete;
  WebSocketServerTransport& operator=(const WebSocketServerTransport&) = delete;
 
  std::string_view name() const noexcept override;
 
  protocol::Json diagnostics() const override;
 
  core::Result<core::Unit> send(TxMessage message) override;
 
  core::Result<std::optional<RxMessage>> receive() override;
 
  core::Result<core::Unit> close() override;
 
  void wait_until_ready() override;
 
 private:
  class Impl;
  std::unique_ptr<Impl> impl_;
};
 
}  // namespace mcp::transport