relation endpoint progress

include/matador/object/basic_object_info.hpp

@@ -2,6 +2,8 @@
 #define BASIC_PROTOTYPE_INFO_HPP
 
 #include "matador/object/object_definition.hpp"
+#include "matador/object/relation_endpoint.hpp"
+
 #include "matador/utils/identifier.hpp"
 
 #include <string>
@@ -13,6 +15,10 @@ class schema_node;
 
 class basic_object_info {
 public:
+  using t_endpoint_map = std::unordered_map<std::type_index, relation_endpoint>;
+  using endpoint_iterator = t_endpoint_map::iterator;
+  using const_endpoint_iterator = t_endpoint_map::const_iterator;
+
   virtual ~basic_object_info() = default;
 
   [[nodiscard]] std::type_index type_index() const;
@@ -20,6 +26,24 @@ public:
   [[nodiscard]] const object_definition& definition() const;
   [[nodiscard]] std::shared_ptr<attribute_definition> reference_column() const;
 
+  void register_relation_endpoint(const std::type_index &tindex, const relation_endpoint &endpoint);
+  void unregister_relation_endpoint(const std::type_index &tindex);
+
+  const_endpoint_iterator find_relation_endpoint(const std::type_index &tindex) const;
+  endpoint_iterator find_relation_endpoint(const std::type_index &tindex);
+
+  const_endpoint_iterator find_relation_endpoint(const std::string &field) const;
+  endpoint_iterator find_relation_endpoint(const std::string &field);
+
+  endpoint_iterator endpoint_begin();
+  const_endpoint_iterator endpoint_begin() const;
+
+  endpoint_iterator endpoint_end();
+  const_endpoint_iterator endpoint_end() const;
+
+  std::size_t endpoints_size() const;
+  bool endpoints_empty() const;
+
 protected:
   basic_object_info(std::shared_ptr<schema_node> node, std::type_index type_index, utils::identifier &&pk, std::shared_ptr<attribute_definition> &&pk_column, object_definition &&definition);
   basic_object_info(std::shared_ptr<schema_node> node, std::type_index type_index, utils::identifier &&pk, std::shared_ptr<attribute_definition> &&pk_column);
@@ -30,6 +54,7 @@ protected:
   object_definition definition_;
   std::optional<utils::identifier> identifier_;
   std::shared_ptr<attribute_definition> pk_column_;
+  t_endpoint_map relation_endpoints_;
 };
 
 using basic_object_info_ref = std::reference_wrapper<const basic_object_info>;

include/matador/object/relation_endpoint.hpp

@@ -0,0 +1,39 @@
+#ifndef MATADOR_RELATION_ENDPOINT_HPP
+#define MATADOR_RELATION_ENDPOINT_HPP
+
+#include "matador/object/schema_node.hpp"
+
+#include "matador/utils/enum_mapper.hpp"
+
+namespace matador::object {
+
+enum class relation_type : uint8_t {
+  BELONGS_TO,
+  HAS_ONE,
+  HAS_MANY
+};
+
+static const utils::enum_mapper<relation_type> relation_type_enum({
+                                                                    { relation_type::BELONGS_TO, "belongs_to" },
+                                                                    { relation_type::HAS_ONE, "has_one" },
+                                                                    { relation_type::HAS_MANY, "has_many" },
+                                                                  });
+
+class relation_endpoint {
+public:
+  relation_endpoint(std::string field_name, relation_type type, const std::shared_ptr<schema_node>& node);
+
+  [[nodiscard]] std::string field_name() const;
+  [[nodiscard]] relation_type type() const;
+  [[nodiscard]] const schema_node& node() const;
+
+private:
+  std::string field_name_;
+  relation_type type_;
+  std::shared_ptr<schema_node> node_;
+  std::weak_ptr<relation_endpoint> foreign_endpoint;
+};
+
+}
+
+#endif //MATADOR_RELATION_ENDPOINT_HPP