pack.h

Go to the documentation of this file.

#pragma once
#ifdef MSGPACK_DEFINE_MAP
#error "webcface/common/internal/pack.h must be included first"
#endif
 
#include <msgpack.hpp>
#include <string>
#include <cstdint>
#include <utf8.h>
#include "webcface/common/val_adaptor.h"
#include "webcface/common/num_vector.h"
#include "./base.h"
 
#ifdef WEBCFACE_COMPILER_IS_GCC
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wabi"
#endif
#include <spdlog/logger.h>
#ifdef WEBCFACE_COMPILER_IS_GCC
#pragma GCC diagnostic pop
#endif
 
MSGPACK_ADD_ENUM(webcface::ValType)
 
WEBCFACE_NS_BEGIN
namespace message {
struct Ping;
struct PingStatusReq;
 
std::string messageTrace(const std::string &message);
 
std::vector<std::pair<int, std::shared_ptr<void>>>
unpack(const std::string &message,
       const std::shared_ptr<spdlog::logger> &logger);
 
template <typename T>
std::string packSingle(const T &obj) {
    msgpack::type::tuple<int, T> src(static_cast<int>(T::kind), obj);
    std::stringstream buffer;
    msgpack::pack(buffer, src);
    return buffer.str();
}
 
template <typename T>
void pack(std::stringstream &buffer, int &len, const T &obj) {
    msgpack::pack(buffer, static_cast<int>(T::kind));
    msgpack::pack(buffer, obj);
    len += 2;
}
 
inline std::string packDone(std::stringstream &buffer, int len) {
    std::stringstream buffer2;
    msgpack::packer packer(buffer2);
    packer.pack_array(len);
    buffer2 << buffer.rdbuf();
    return buffer2.str();
}
 
} // namespace message
WEBCFACE_NS_END
 
namespace msgpack {
MSGPACK_API_VERSION_NAMESPACE(MSGPACK_DEFAULT_API_NS) {
    namespace adaptor {
 
    template <>
    struct pack<double> {
        template <typename Stream>
        packer<Stream> &operator()(msgpack::packer<Stream> &o, double v) const {
            if (v == v) { // check for nan
                // compare d to limits to avoid undefined behaviour
                if (v >= 0 &&
                    v <= double(std::numeric_limits<uint64_t>::max()) &&
                    v == static_cast<double>(static_cast<uint64_t>(v))) {
                    o.pack_uint64(static_cast<std::uint64_t>(v));
                    return o;
                } else if (v < 0 &&
                           v >= double(std::numeric_limits<int64_t>::min()) &&
                           v == static_cast<double>(static_cast<int64_t>(v))) {
                    o.pack_int64(static_cast<std::int64_t>(v));
                    return o;
                } else if (std::abs(v) <= std::numeric_limits<float>::max() &&
                           v == static_cast<double>(static_cast<float>(v))) {
                    o.pack_float(static_cast<float>(v));
                    return o;
                }
            }
            o.pack_double(v);
            return o;
        }
    };
 
    template <>
    struct convert<webcface::SharedString> {
        msgpack::object const &operator()(msgpack::object const &o,
                                          webcface::SharedString &v) const {
            v = webcface::SharedString::fromU8String(utf8::replace_invalid(
                std::string(o.via.bin.ptr, o.via.bin.size)));
            return o;
        }
    };
    template <>
    struct pack<webcface::SharedString> {
        template <typename Stream>
        msgpack::packer<Stream> &operator()(msgpack::packer<Stream> &o,
                                            const webcface::SharedString &v) {
            o.pack(std::string_view(v.u8StringView()));
            return o;
        }
    };
 
    template <>
    struct convert<webcface::ValAdaptor> {
        msgpack::object const &operator()(msgpack::object const &o,
                                          webcface::ValAdaptor &v) const {
            switch (o.type) {
            case msgpack::type::FLOAT32:
            case msgpack::type::FLOAT64:
                v = o.via.f64;
                break;
            case msgpack::type::POSITIVE_INTEGER:
                v = o.via.u64;
                break;
            case msgpack::type::NEGATIVE_INTEGER:
                v = o.via.i64;
                break;
            case msgpack::type::BOOLEAN:
                v = o.via.boolean;
                break;
            case msgpack::type::BIN:
            case msgpack::type::STR:
                v = webcface::SharedString::fromU8String(utf8::replace_invalid(
                    std::string(o.via.bin.ptr, o.via.bin.size)));
                break;
            default:
                throw msgpack::type_error();
            }
            return o;
        }
    };
    template <>
    struct pack<webcface::ValAdaptor> {
        template <typename Stream>
        msgpack::packer<Stream> &operator()(msgpack::packer<Stream> &o,
                                            const webcface::ValAdaptor &v) {
            switch (v.valType()) {
            case webcface::ValType::bool_:
                o.pack(static_cast<bool>(v));
                break;
            case webcface::ValType::int_:
                o.pack(static_cast<std::int64_t>(v));
                break;
            case webcface::ValType::float_:
                o.pack(static_cast<double>(v));
                break;
            case webcface::ValType::string_:
            default:
                o.pack(std::string_view(v.asU8StringView()));
                break;
            }
            return o;
        }
    };
 
    template <>
    struct convert<webcface::MutableNumVector> {
        msgpack::object const &operator()(msgpack::object const &o,
                                          webcface::MutableNumVector &v) const {
            if (o.type != msgpack::type::ARRAY) {
                throw msgpack::type_error();
            }
            if (o.via.array.size == 0) {
                v.assign(0);
            } else if (o.via.array.size == 1) {
                v.assign(o.via.array.ptr[0].as<double>());
            } else {
                std::vector<double> vec(o.via.array.size);
                for (std::size_t i = 0; i < o.via.array.size; i++) {
                    vec.at(i) = o.via.array.ptr[i].as<double>();
                }
                v.assign(std::move(vec));
            }
            return o;
        }
    };
    template <>
    struct pack<webcface::MutableNumVector> {
        template <typename Stream>
        msgpack::packer<Stream> &
        operator()(msgpack::packer<Stream> &o,
                   const webcface::MutableNumVector &v) {
            o.pack_array(static_cast<std::uint32_t>(v.size()));
            for (auto val : v) {
                o.pack(val);
            }
            return o;
        }
    };
 
    template <typename T>
    struct EmptyConvert {
        msgpack::object const &operator()(msgpack::object const &o, T &) const {
            return o;
        }
    };
    template <typename T>
    struct EmptyPack {
        template <typename Stream>
        msgpack::packer<Stream> &operator()(msgpack::packer<Stream> &o,
                                            const T &) {
            o.pack_map(0);
            return o;
        }
    };
    template <>
    struct convert<webcface::message::Ping>
        : public EmptyConvert<webcface::message::Ping> {};
    template <>
    struct convert<webcface::message::PingStatusReq>
        : public EmptyConvert<webcface::message::PingStatusReq> {};
    template <>
    struct pack<webcface::message::Ping>
        : public EmptyPack<webcface::message::Ping> {};
    template <>
    struct pack<webcface::message::PingStatusReq>
        : public EmptyPack<webcface::message::PingStatusReq> {};
 
    } // namespace adaptor
}
} // namespace msgpack