Made sure uint8_t is properly handled by YAML; #270

getml · Dec 22, 2024 · 37a0be8 · 37a0be8
1 parent a1920a1
commit 37a0be8
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 4 deletions.
diff --git a/include/rfl/yaml/Reader.hpp b/include/rfl/yaml/Reader.hpp
@@ -61,9 +61,10 @@ struct Reader {
     try {
       if constexpr (std::is_same<std::remove_cvref_t<T>, std::string>() ||
                     std::is_same<std::remove_cvref_t<T>, bool>() ||
-                    std::is_floating_point<std::remove_cvref_t<T>>() ||
-                    std::is_integral<std::remove_cvref_t<T>>()) {
+                    std::is_floating_point<std::remove_cvref_t<T>>()) {
         return _var.node_.as<std::remove_cvref_t<T>>();
+      } else if constexpr (std::is_integral<std::remove_cvref_t<T>>()) {
+        return static_cast<T>(_var.node_.as<std::remove_cvref_t<int64_t>>());
       } else {
         static_assert(rfl::always_false_v<T>, "Unsupported type.");
       }

diff --git a/include/rfl/yaml/Writer.hpp b/include/rfl/yaml/Writer.hpp
@@ -86,7 +86,18 @@ class Writer {
   template <class T>
   OutputVarType insert_value(const std::string_view& _name,
                              const T& _var) const noexcept {
-    (*out_) << YAML::Key << _name.data() << YAML::Value << _var;
+    if constexpr (std::is_same<std::remove_cvref_t<T>, std::string>() ||
+                  std::is_same<std::remove_cvref_t<T>, bool>() ||
+                  std::is_floating_point<std::remove_cvref_t<T>>() ||
+                  std::is_same<std::remove_cvref_t<T>,
+                               std::remove_cvref_t<decltype(YAML::Null)>>()) {
+      (*out_) << YAML::Key << _name.data() << YAML::Value << _var;
+    } else if constexpr (std::is_integral<std::remove_cvref_t<T>>()) {
+      (*out_) << YAML::Key << _name.data() << YAML::Value
+              << static_cast<int64_t>(_var);
+    } else {
+      static_assert(rfl::always_false_v<T>, "Unsupported type.");
+    }
     return OutputVarType{};
   }
 

diff --git a/tests/yaml/test_snake_case_to_pascal_case.cpp b/tests/yaml/test_snake_case_to_pascal_case.cpp
@@ -13,7 +13,7 @@ struct Person {
   std::vector<Person> children;
 };
 
-TEST(bson, test_snake_case_to_pascal_case) {
+TEST(yaml, test_snake_case_to_pascal_case) {
   const auto bart = Person{
       .first_name = "Bart", .last_name = "Simpson", .birthday = "1987-04-19"};