rpc.h

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
#ifndef PACKIO_JSON_RPC_RPC_H
#define PACKIO_JSON_RPC_RPC_H
 
#include <array>
#include <queue>
 
#include <boost/json.hpp>
 
#include "../arg.h"
#include "../args_specs.h"
#include "../internal/config.h"
#include "../internal/expected.h"
#include "../internal/log.h"
#include "../internal/rpc.h"
#include "converters.h"
#include "hash.h"
 
namespace packio {
namespace json_rpc {
namespace internal {
 
template <typename... Args>
constexpr bool positional_args_v = (!is_arg_v<Args> && ...);
 
template <typename... Args>
constexpr bool named_args_v = sizeof...(Args) > 0 && (is_arg_v<Args> && ...);
 
using id_type = boost::json::value;
using native_type = boost::json::value;
using string_type = boost::json::string;
using packio::internal::expected;
using packio::internal::unexpected;
 
struct request {
    call_type type;
    internal::id_type id;
    std::string method;
    native_type args;
};
 
struct response {
    id_type id;
    native_type result;
    native_type error;
};
 
class incremental_parser {
public:
    incremental_parser()
        : parser_{std::make_unique<boost::json::stream_parser>()}
    { //
        parser_->reset();
    }
 
    expected<request, std::string> get_request()
    {
        if (parsed_.empty()) {
            return unexpected{"no request parsed"};
        }
        auto object = std::move(parsed_.front());
        parsed_.pop();
        return parse_request(std::move(object.as_object()));
    }
 
    expected<response, std::string> get_response()
    {
        if (parsed_.empty()) {
            return unexpected{"no response parsed"};
        }
        auto object = std::move(parsed_.front());
        parsed_.pop();
        return parse_response(std::move(object.as_object()));
    }
 
    char* buffer()
    { //
        return buffer_.data();
    }
 
    std::size_t buffer_capacity() const
    { //
        return buffer_.size();
    }
 
    void buffer_consumed(std::size_t bytes)
    { //
        std::size_t parsed = 0;
        while (parsed < bytes) {
            parsed += parser_->write_some(
                buffer_.data() + parsed, bytes - parsed);
            if (parser_->done()) {
                parsed_.push(parser_->release());
                parser_->reset();
            }
        }
    }
 
    void reserve_buffer(std::size_t bytes)
    { //
        buffer_.resize(bytes);
    }
 
private:
    static expected<response, std::string> parse_response(boost::json::object&& res)
    {
        auto id_it = res.find("id");
        auto result_it = res.find("result");
        auto error_it = res.find("error");
 
        if (id_it == res.end()) {
            return unexpected{"missing id field"};
        }
        if (result_it == res.end() && error_it == res.end()) {
            return unexpected{"missing error and result field"};
        }
 
        response parsed;
        parsed.id = std::move(id_it->value());
        if (error_it != res.end()) {
            parsed.error = std::move(error_it->value());
        }
        if (result_it != res.end()) {
            parsed.result = std::move(result_it->value());
        }
        return {std::move(parsed)};
    }
 
    static expected<request, std::string> parse_request(boost::json::object&& req)
    {
        auto id_it = req.find("id");
        auto method_it = req.find("method");
        auto params_it = req.find("params");
 
        if (method_it == req.end()) {
            return unexpected{"missing method field"};
        }
        if (!method_it->value().is_string()) {
            return unexpected{"method field is not a string"};
        }
 
        request parsed;
        parsed.method = std::string{
            method_it->value().get_string().data(),
            method_it->value().get_string().size(),
        };
        if (params_it == req.end() || params_it->value().is_null()) {
            parsed.args = boost::json::array{};
        }
        else if (!params_it->value().is_array() && !params_it->value().is_object()) {
            return unexpected{"non-structured arguments are not supported"};
        }
        else {
            parsed.args = std::move(params_it->value());
        }
 
        if (id_it == req.end() || id_it->value().is_null()) {
            parsed.type = call_type::notification;
        }
        else {
            parsed.type = call_type::request;
            parsed.id = std::move(id_it->value());
        }
        return {std::move(parsed)};
    }
 
    std::vector<char> buffer_;
    std::queue<boost::json::value> parsed_;
    std::unique_ptr<boost::json::stream_parser> parser_;
};
 
} // internal
 
class rpc {
public:
    using id_type = internal::id_type;
 
    using native_type = internal::native_type;
 
    using request_type = internal::request;
 
    using response_type = internal::response;
 
    using incremental_parser_type = internal::incremental_parser;
 
    static std::string format_id(const id_type& id)
    { //
        return boost::json::serialize(id);
    }
 
    template <typename... Args>
    static auto serialize_notification(std::string_view method, Args&&... args)
        -> std::enable_if_t<internal::positional_args_v<Args...>, std::string>
    {
        auto res = boost::json::serialize(boost::json::object({
            {"jsonrpc", "2.0"},
            {"method", method},
            {"params", {boost::json::value_from(std::forward<Args>(args))...}},
        }));
        PACKIO_TRACE("notification: " + res);
        return res;
    }
 
    template <typename... Args>
    static auto serialize_notification(std::string_view method, Args&&... args)
        -> std::enable_if_t<internal::named_args_v<Args...>, std::string>
    {
        auto res = boost::json::serialize(boost::json::object({
            {"jsonrpc", "2.0"},
            {"method", method},
            {"params", {{args.name, boost::json::value_from(args.value)}...}},
        }));
        PACKIO_TRACE("notification: " + res);
        return res;
    }
 
    template <typename... Args>
    static auto serialize_notification(std::string_view, Args&&...) -> std::enable_if_t<
        !internal::positional_args_v<Args...> && !internal::named_args_v<Args...>,
        std::string>
    {
        static_assert(
            internal::positional_args_v<Args...> || internal::named_args_v<Args...>,
            "JSON-RPC does not support mixed named and unnamed arguments");
    }
 
    template <typename... Args>
    static auto serialize_request(
        const id_type& id,
        std::string_view method,
        Args&&... args)
        -> std::enable_if_t<internal::positional_args_v<Args...>, std::string>
    {
        auto res = boost::json::serialize(boost::json::object({
            {"jsonrpc", "2.0"},
            {"method", method},
            {"params", {boost::json::value_from(std::forward<Args>(args))...}},
            {"id", id},
        }));
        PACKIO_TRACE("request: " + res);
        return res;
    }
 
    template <typename... Args>
    static auto serialize_request(
        const id_type& id,
        std::string_view method,
        Args&&... args)
        -> std::enable_if_t<internal::named_args_v<Args...>, std::string>
    {
        auto res = boost::json::serialize(boost::json::object({
            {"jsonrpc", "2.0"},
            {"method", method},
            {"params", {{args.name, boost::json::value_from(args.value)}...}},
            {"id", id},
        }));
        PACKIO_TRACE("request: " + res);
        return res;
    }
 
    template <typename... Args>
    static auto serialize_request(const id_type&, std::string_view, Args&&...)
        -> std::enable_if_t<
            !internal::positional_args_v<Args...> && !internal::named_args_v<Args...>,
            std::string>
    {
        static_assert(
            internal::positional_args_v<Args...> || internal::named_args_v<Args...>,
            "JSON-RPC does not support mixed named and unnamed arguments");
    }
 
    static std::string serialize_response(const id_type& id)
    {
        return serialize_response(id, boost::json::value{});
    }
 
    template <typename T>
    static std::string serialize_response(const id_type& id, T&& value)
    {
        auto res = boost::json::serialize(boost::json::object({
            {"jsonrpc", "2.0"},
            {"id", id},
            {"result", boost::json::value_from(std::forward<T>(value))},
        }));
        PACKIO_TRACE("response: " + res);
        return res;
    }
 
    template <typename T>
    static std::string serialize_error_response(const id_type& id, T&& value)
    {
        auto res = boost::json::serialize(boost::json::object({
            {"jsonrpc", "2.0"},
            {"id", id},
            {"error",
             [&]() {
                 boost::json::object error = {
                     {"code", -32000}, // -32000 is an implementation-defined error
                     {"data", std::forward<T>(value)},
                 };
                 if (error["data"].is_string()) {
                     error["message"] = error["data"];
                 }
                 else {
                     error["message"] = "unknown error";
                 }
                 return error;
             }()},
        }));
        PACKIO_TRACE("response: " + res);
        return res;
    }
 
    static net::const_buffer buffer(const std::string& buf)
    {
        return net::const_buffer(buf.data(), buf.size());
    }
 
    template <typename T, typename F>
    static internal::expected<T, std::string> extract_args(
        boost::json::value&& args,
        const args_specs<F>& specs)
    {
        try {
            if (args.is_array()) {
                return convert_positional_args<T>(args.get_array(), specs);
            }
            else if (args.is_object()) {
                return convert_named_args<T>(args.get_object(), specs);
            }
            else {
                throw std::runtime_error{"arguments are not a structured type"};
            }
        }
        catch (const std::exception& exc) {
            return internal::unexpected{
                std::string{"cannot convert arguments: "} + exc.what()};
        }
    }
 
private:
    template <typename T, typename F>
    static constexpr T convert_positional_args(
        const boost::json::array& array,
        const args_specs<F>& specs)
    {
        return convert_positional_args<T>(
            array, specs, std::make_index_sequence<args_specs<F>::size()>());
    }
 
    template <typename T, typename F, std::size_t... Idxs>
    static constexpr T convert_positional_args(
        const boost::json::array& array,
        const args_specs<F>& specs,
        std::index_sequence<Idxs...>)
    {
        if (!specs.options().allow_extra_arguments
            && array.size() > std::tuple_size_v<T>) {
            throw std::runtime_error{"too many arguments"};
        }
        return {[&]() {
            if (Idxs < array.size()) {
                try {
                    return boost::json::value_to<std::tuple_element_t<Idxs, T>>(
                        array.at(Idxs));
                }
                catch (const boost::json::system_error&) {
                    throw std::runtime_error{
                        "invalid type for argument "
                        + specs.template get<Idxs>().name()};
                }
            }
            if (const auto& value = specs.template get<Idxs>().default_value()) {
                return *value;
            }
            throw std::runtime_error{
                "no value for argument " + specs.template get<Idxs>().name()};
        }()...};
    }
 
    template <typename T, typename F>
    static constexpr T convert_named_args(
        const boost::json::object& args,
        const args_specs<F>& specs)
    {
        return convert_named_args<T>(
            args, specs, std::make_index_sequence<args_specs<F>::size()>());
    }
 
    template <typename T, typename F, std::size_t... Idxs>
    static constexpr T convert_named_args(
        const boost::json::object& args,
        const args_specs<F>& specs,
        std::index_sequence<Idxs...>)
    {
        if (!specs.options().allow_extra_arguments) {
            const std::array<const std::string*, sizeof...(Idxs)>
                available_arguments = {&specs.template get<Idxs>().name()...};
            for (const auto& item : args) {
                auto it = std::find_if(
                    available_arguments.begin(),
                    available_arguments.end(),
                    [&](const std::string* arg) { return *arg == item.key(); });
                if (it == available_arguments.end()) {
                    throw std::runtime_error{
                        "unexpected argument " + std::string(item.key())};
                }
            }
        }
 
        return T{[&]() {
            auto it = args.find(specs.template get<Idxs>().name());
            if (it != args.end()) {
                try {
                    return boost::json::value_to<std::tuple_element_t<Idxs, T>>(
                        it->value());
                }
                catch (const boost::json::system_error&) {
                    throw std::runtime_error{
                        "invalid type for argument "
                        + specs.template get<Idxs>().name()};
                }
            }
            if (const auto& value = specs.template get<Idxs>().default_value()) {
                return *value;
            }
            throw std::runtime_error{
                "no value for argument " + specs.template get<Idxs>().name()};
        }()...};
    }
};
 
} // json_rpc
} // packio
 
#endif // PACKIO_JSON_RPC_RPC_H