From 5a898afc4dc9cf4f463ca19b35f5c3bc5cc455a7 Mon Sep 17 00:00:00 2001
From: "Dr. Patrick Urbanke" <patrick@getml.com>
Date: Thu, 26 Dec 2024 16:31:31 +0100
Subject: [PATCH] First draft for the schema generator

---
 include/rfl/capnproto/schema/Type.hpp |   9 ++-
 src/reflectcpp_capnproto.cpp          |   1 +
 src/rfl/capnproto/Type.cpp            | 109 ++++++++++++++++++++++++++
 3 files changed, 115 insertions(+), 4 deletions(-)
 create mode 100644 src/rfl/capnproto/Type.cpp

diff --git a/include/rfl/capnproto/schema/Type.hpp b/include/rfl/capnproto/schema/Type.hpp
index 6d27c92b..8b20ed2d 100644
--- a/include/rfl/capnproto/schema/Type.hpp
+++ b/include/rfl/capnproto/schema/Type.hpp
@@ -1,10 +1,10 @@
 #ifndef RFL_CAPNPROTO_SCHEMA_TYPE_HPP_
 #define RFL_CAPNPROTO_SCHEMA_TYPE_HPP_
 
+#include <iostream>
 #include <map>
 #include <memory>
 #include <optional>
-#include <sstream>
 #include <string>
 
 #include "../../Literal.hpp"
@@ -57,12 +57,12 @@ struct Type {
   };
 
   struct Reference {
-    std::string type;
+    std::string type_name;
   };
 
   using ReflectionType =
       rfl::Variant<Void, Bool, Int8, Int16, Int32, Int64, UInt8, UInt16, UInt32,
-                   UInt64, Float32, Float64, Data, Text, Struct, List, Map,
+                   UInt64, Float32, Float64, Data, Text, Struct, List, /*Map,*/
                    Reference>;
 
   const auto& reflection() const { return value; }
@@ -70,6 +70,7 @@ struct Type {
   ReflectionType value;
 };
 
-}  // namespace rfl::capnproto::schema
+std::ostream& operator<<(std::ostream& os, const Type& _t);
 
+}  // namespace rfl::capnproto::schema
 #endif
diff --git a/src/reflectcpp_capnproto.cpp b/src/reflectcpp_capnproto.cpp
index eeea9c72..65defc58 100644
--- a/src/reflectcpp_capnproto.cpp
+++ b/src/reflectcpp_capnproto.cpp
@@ -37,5 +37,6 @@ SOFTWARE.
 #include "rfl/avro/to_schema.cpp"*/
 #include "rfl/capnproto/Reader.cpp"
 #include "rfl/capnproto/SchemaImpl.cpp"
+#include "rfl/capnproto/Type.cpp"
 #include "rfl/capnproto/Writer.cpp"
 
diff --git a/src/rfl/capnproto/Type.cpp b/src/rfl/capnproto/Type.cpp
new file mode 100644
index 00000000..ef0ad5d2
--- /dev/null
+++ b/src/rfl/capnproto/Type.cpp
@@ -0,0 +1,109 @@
+/*
+
+MIT License
+
+Copyright (c) 2023-2024 Code17 GmbH
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+*/
+
+#include "rfl/capnproto/schema/Type.hpp"
+
+namespace rfl::capnproto::schema {
+
+std::ostream& operator<<(std::ostream& _os, const Type::Void&) {
+  return _os << "Void";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Bool&) {
+  return _os << "Bool";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Int8&) {
+  return _os << "Int8";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Int16&) {
+  return _os << "Int16";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Int32&) {
+  return _os << "Int32";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Int64&) {
+  return _os << "Int64";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::UInt8&) {
+  return _os << "UInt8";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::UInt16&) {
+  return _os << "UInt16";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::UInt32&) {
+  return _os << "UInt32";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::UInt64&) {
+  return _os << "UInt64";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Float32&) {
+  return _os << "Float32";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Float64&) {
+  return _os << "Float64";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Data&) {
+  return _os << "Data";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Text&) {
+  return _os << "Text";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Struct& _s) {
+  _os << "struct " << _s.name << " {" << std::endl;
+  for (size_t i = 0; i < _s.fields.size(); ++i) {
+    const auto& [name, type] = _s.fields[i];
+    _os << "  " << name << " @" << i << " :" << type << ";" << std::endl;
+  }
+  return _os << "}" << std::endl;
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::List& _l) {
+  return _os << "List(" << *_l.type << ")";
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type::Reference& _r) {
+  return _os << _r.type_name;
+}
+
+std::ostream& operator<<(std::ostream& _os, const Type& _t) {
+  return _t.reflection().visit(
+      [&](const auto& _r) -> std::ostream& { return _os << _r; });
+}
+
+}  // namespace rfl::capnproto::schema