json_rpc.hpp

Go to the documentation of this file.

 
#ifndef PLOTLY_DETAILS_JSON_RPC_HPP
#define PLOTLY_DETAILS_JSON_RPC_HPP
 
#include "websockets_endpoint.hpp"
#include <nlohmann/json.hpp>
 
#include <cstdint>
#include <functional>
#include <future>
#include <memory>
#include <mutex>
#include <optional>
#include <string>
#include <unordered_set>
#include <utility>
 
namespace plotly::detail {

enum class JsonRpcErrorCode : std::int16_t {
  PARSE_ERROR = -32700,      
  INVALID_REQUEST = -32600,  
  METHOD_NOT_FOUND = -32601, 
  INVALID_PARAMS = -32602,   
  INTERNAL_ERROR = -32603,   
  SERVER_ERROR = -32000      
};

struct JsonRpcError {
  int code;            
  std::string message; 
  nlohmann::json data; 
};

struct JsonRpcResponse {
  std::optional<nlohmann::json> id; 
  std::optional<nlohmann::json>
      result; 
  std::optional<JsonRpcError> error; 
  std::string jsonrpc = "2.0";       

  [[nodiscard]] auto toJson() const -> nlohmann::json;
};

class JsonRpc {
public:
  explicit JsonRpc(std::unique_ptr<WebsocketEndpointInterface> wsEndpoint);

  ~JsonRpc();
 
  // Delete copy constructor and assignment operator
  JsonRpc(const JsonRpc &) = delete;
  auto operator=(const JsonRpc &) -> JsonRpc & = delete;
 
  // Move constructor and assignment operator
  JsonRpc(JsonRpc &&other) noexcept;
  auto operator=(JsonRpc &&other) noexcept -> JsonRpc &;

  void registerHandler(
      const std::string &method,
      const std::function<nlohmann::json(const nlohmann::json &)> &handler);
 
  void
  registerNotification(const std::string &method,
                       std::function<void(const nlohmann::json &)> handler);
 
  void unregisterHandler(const std::string &method);

  void registerCallbackWithWebsocket(
      const std::string &callbackName,
      std::function<void(const std::string_view)> callback);

  void unregisterCallbackFromWebsocket(const std::string &callbackName);

  void unregisterAllCallbacksFromWebsockets();

  auto call(const std::string &method, const nlohmann::json &params)
      -> std::pair<std::future<nlohmann::json>, std::function<void()>>;

  void notify(const std::string &method, const nlohmann::json &params);

  [[nodiscard]] auto getWebsocketEndpoint() const
      -> WebsocketEndpointInterface *;
 
private:
  std::unique_ptr<WebsocketEndpointInterface> _wsEndpoint;
 
  void handleIncomingMessage(const std::string_view &message);
  void sendSuccessResponse(const nlohmann::json &requestId,
                           const nlohmann::json &result);
  void sendErrorResponse(const nlohmann::json &requestId,
                         JsonRpcErrorCode errorCode,
                         const std::string &message);
 
  std::unordered_map<std::string,
                     std::function<nlohmann::json(const nlohmann::json &)>>
      _handlers;
  std::unordered_map<std::string, std::function<void(const nlohmann::json &)>>
      _notifications;
  std::unordered_set<std::string> _registeredCallbacks;
  mutable std::mutex _registeredCallbacksMutex;
};
 
} // namespace plotly::detail
#endif // PLOTLY_DETAILS_JSON_RPC_HPP