draconisplusplus/include/rfl/json/Writer.hpp

#ifndef RFL_JSON_WRITER_HPP_
#define RFL_JSON_WRITER_HPP_

#include <exception>
#include <map>
#include <sstream>
#include <stdexcept>
#include <string>
#include <string_view>
#include <type_traits>
#include <vector>
#include <yyjson.h>

#include "../Result.hpp"
#include "../always_false.hpp"

namespace rfl {
  namespace json {

    class Writer {
     public:
      struct YYJSONOutputArray {
        YYJSONOutputArray(yyjson_mut_val* _val) : val_(_val) {}
        yyjson_mut_val* val_;
      };

      struct YYJSONOutputObject {
        YYJSONOutputObject(yyjson_mut_val* _val) : val_(_val) {}
        yyjson_mut_val* val_;
      };

      struct YYJSONOutputVar {
        YYJSONOutputVar(yyjson_mut_val* _val) : val_(_val) {}

        YYJSONOutputVar(YYJSONOutputArray _arr) : val_(_arr.val_) {}

        YYJSONOutputVar(YYJSONOutputObject _obj) : val_(_obj.val_) {}

        yyjson_mut_val* val_;
      };

      using OutputArrayType  = YYJSONOutputArray;
      using OutputObjectType = YYJSONOutputObject;
      using OutputVarType    = YYJSONOutputVar;

      Writer(yyjson_mut_doc* _doc) : doc_(_doc) {}

      ~Writer() = default;

      OutputArrayType array_as_root(const size_t _size) const noexcept {
        const auto arr = yyjson_mut_arr(doc_);
        yyjson_mut_doc_set_root(doc_, arr);
        return OutputArrayType(arr);
      }

      OutputObjectType object_as_root(const size_t _size) const noexcept {
        const auto obj = yyjson_mut_obj(doc_);
        yyjson_mut_doc_set_root(doc_, obj);
        return OutputObjectType(obj);
      }

      OutputVarType null_as_root() const noexcept {
        const auto null = yyjson_mut_null(doc_);
        yyjson_mut_doc_set_root(doc_, null);
        return OutputVarType(null);
      }

      template <class T>
      OutputVarType value_as_root(const T& _var) const noexcept {
        const auto val = from_basic_type(_var);
        yyjson_mut_doc_set_root(doc_, val.val_);
        return OutputVarType(val);
      }

      OutputArrayType add_array_to_array(
          const size_t     _size,
          OutputArrayType* _parent
      ) const noexcept {
        const auto arr = yyjson_mut_arr(doc_);
        yyjson_mut_arr_add_val(_parent->val_, arr);
        return OutputArrayType(arr);
      }

      OutputArrayType add_array_to_object(
          const std::string_view& _name,
          const size_t            _size,
          OutputObjectType*       _parent
      ) const noexcept {
        const auto arr = yyjson_mut_arr(doc_);
        yyjson_mut_obj_add(
            _parent->val_, yyjson_mut_strcpy(doc_, _name.data()), arr
        );
        return OutputArrayType(arr);
      }

      OutputObjectType add_object_to_array(
          const size_t     _size,
          OutputArrayType* _parent
      ) const noexcept {
        const auto obj = yyjson_mut_obj(doc_);
        yyjson_mut_arr_add_val(_parent->val_, obj);
        return OutputObjectType(obj);
      }

      OutputObjectType add_object_to_object(
          const std::string_view& _name,
          const size_t            _size,
          OutputObjectType*       _parent
      ) const noexcept {
        const auto obj = yyjson_mut_obj(doc_);
        yyjson_mut_obj_add(
            _parent->val_, yyjson_mut_strcpy(doc_, _name.data()), obj
        );
        return OutputObjectType(obj);
      }

      template <class T>
      OutputVarType add_value_to_array(const T& _var, OutputArrayType* _parent)
          const noexcept {
        const auto val = from_basic_type(_var);
        yyjson_mut_arr_add_val(_parent->val_, val.val_);
        return OutputVarType(val);
      }

      template <class T>
      OutputVarType add_value_to_object(
          const std::string_view& _name,
          const T&                _var,
          OutputObjectType*       _parent
      ) const noexcept {
        const auto val = from_basic_type(_var);
        yyjson_mut_obj_add(
            _parent->val_, yyjson_mut_strcpy(doc_, _name.data()), val.val_
        );
        return OutputVarType(val);
      }

      OutputVarType add_null_to_array(OutputArrayType* _parent) const noexcept {
        const auto null = yyjson_mut_null(doc_);
        yyjson_mut_arr_add_val(_parent->val_, null);
        return OutputVarType(null);
      }

      OutputVarType add_null_to_object(
          const std::string_view& _name,
          OutputObjectType*       _parent
      ) const noexcept {
        const auto null = yyjson_mut_null(doc_);
        yyjson_mut_obj_add(
            _parent->val_, yyjson_mut_strcpy(doc_, _name.data()), null
        );
        return OutputVarType(null);
      }

      void end_array(OutputArrayType* _arr) const noexcept {}

      void end_object(OutputObjectType* _obj) const noexcept {}

     private:
      template <class T>
      OutputVarType from_basic_type(const T& _var) const noexcept {
        if constexpr (std::is_same<std::remove_cvref_t<T>, std::string>()) {
          return OutputVarType(yyjson_mut_strcpy(doc_, _var.c_str()));
        } else if constexpr (std::is_same<std::remove_cvref_t<T>, bool>()) {
          return OutputVarType(yyjson_mut_bool(doc_, _var));
        } else if constexpr (std::is_floating_point<std::remove_cvref_t<T>>()) {
          return OutputVarType(yyjson_mut_real(doc_, static_cast<double>(_var))
          );
        } else if constexpr (std::is_unsigned<std::remove_cvref_t<T>>()) {
          return OutputVarType(
              yyjson_mut_uint(doc_, static_cast<uint64_t>(_var))
          );
        } else if constexpr (std::is_integral<std::remove_cvref_t<T>>()) {
          return OutputVarType(yyjson_mut_int(doc_, static_cast<int64_t>(_var))
          );
        } else {
          static_assert(rfl::always_false_v<T>, "Unsupported type.");
        }
      }

     public:
      yyjson_mut_doc* doc_;
    };

  } // namespace json
} // namespace rfl

#endif // JSON_PARSER_HPP_