#ifndef BASIC_PROTOTYPE_INFO_HPP
#define BASIC_PROTOTYPE_INFO_HPP

#include "matador/object/object_definition.hpp"
#include "matador/object/relation_endpoint.hpp"

#include "matador/utils/identifier.hpp"

#include <string>
#include <typeindex>

namespace matador::object {

class schema_node;
class relation_endpoint;

class basic_object_info {
public:
  using t_endpoint_map = std::unordered_map<std::type_index, std::shared_ptr<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;
  [[nodiscard]] std::string name() const;
  [[nodiscard]] const object_definition& definition() const;
  [[nodiscard]] std::shared_ptr<attribute_definition> reference_column() const;

  [[nodiscard]] bool has_primary_key() const;
  [[nodiscard]] const utils::identifier& primary_key() const;

  endpoint_iterator register_relation_endpoint(const std::type_index &type, const std::shared_ptr<relation_endpoint> &endpoint);
  void unregister_relation_endpoint(const std::type_index &type);

  [[nodiscard]] const_endpoint_iterator find_relation_endpoint(const std::type_index &type) const;
  endpoint_iterator find_relation_endpoint(const std::type_index &type);

  [[nodiscard]] const_endpoint_iterator find_relation_endpoint(const std::string &field) const;
  endpoint_iterator find_relation_endpoint(const std::string &field);

  endpoint_iterator endpoint_begin();
  [[nodiscard]] const_endpoint_iterator endpoint_begin() const;

  endpoint_iterator endpoint_end();
  [[nodiscard]] const_endpoint_iterator endpoint_end() const;

  [[nodiscard]] std::size_t endpoints_size() const;
  [[nodiscard]] bool endpoints_empty() const;

protected:
  basic_object_info(std::shared_ptr<schema_node> node, std::type_index type_index, utils::identifier &&pk, const 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, const std::shared_ptr<attribute_definition> &pk_column);
  basic_object_info(std::shared_ptr<schema_node> node, std::type_index type_index, object_definition &&definition);

protected:
  std::shared_ptr<schema_node> node_;            /**< prototype node of the represented object type */
  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>;

}

#endif //BASIC_PROTOTYPE_INFO_HPP