File: write.hpp

package info (click to toggle)
reflect-cpp 0.21.0%2Bds-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 13,128 kB
  • sloc: cpp: 50,336; python: 139; makefile: 30; sh: 3
file content (69 lines) | stat: -rw-r--r-- 2,009 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#ifndef RFL_JSON_WRITE_HPP_
#define RFL_JSON_WRITE_HPP_

#include <stdexcept>
#if __has_include(<yyjson.h>)
#include <yyjson.h>
#else
#include "../thirdparty/yyjson.h"
#endif

#include <ostream>
#include <sstream>
#include <string>

#include "../Processors.hpp"
#include "../parsing/Parent.hpp"
#include "Parser.hpp"

namespace rfl {
namespace json {

/// Convenient alias for the YYJSON pretty flag
inline constexpr yyjson_write_flag pretty = YYJSON_WRITE_PRETTY;

/// Returns a JSON string.
template <class... Ps>
std::string write(const auto& _obj, const yyjson_write_flag _flag = 0) {
  using T = std::remove_cvref_t<decltype(_obj)>;
  using ParentType = parsing::Parent<Writer>;
  auto w = Writer(yyjson_mut_doc_new(NULL));
  Parser<T, Processors<Ps...>>::write(w, _obj, typename ParentType::Root{});
  yyjson_write_err err;
  const char* json_c_str =
      yyjson_mut_write_opts(w.doc_, _flag, NULL, NULL, &err);
  if (!json_c_str) {
    throw std::runtime_error("An error occured while writing to JSON: " +
                             std::string(err.msg));
  }
  const auto json_str = std::string(json_c_str);
  free((void*)json_c_str);
  yyjson_mut_doc_free(w.doc_);
  return json_str;
}

/// Writes a JSON into an ostream.
template <class... Ps>
std::ostream& write(const auto& _obj, std::ostream& _stream,
                    const yyjson_write_flag _flag = 0) {
  using T = std::remove_cvref_t<decltype(_obj)>;
  using ParentType = parsing::Parent<Writer>;
  auto w = Writer(yyjson_mut_doc_new(NULL));
  Parser<T, Processors<Ps...>>::write(w, _obj, typename ParentType::Root{});
  yyjson_write_err err;
  const char* json_c_str =
      yyjson_mut_write_opts(w.doc_, _flag, NULL, NULL, &err);
  if (!json_c_str) {
    throw std::runtime_error("An error occured while writing to JSON: " +
                             std::string(err.msg));
  }
  _stream << json_c_str;
  free((void*)json_c_str);
  yyjson_mut_doc_free(w.doc_);
  return _stream;
}

}  // namespace json
}  // namespace rfl

#endif