#ifndef QUERY_QUERY_RESULT_IMPL_HPP
#define QUERY_QUERY_RESULT_IMPL_HPP

#include "matador/utils/access.hpp"
#include "matador/utils/field_attributes.hpp"
#include "matador/utils/foreign_attributes.hpp"
#include "matador/utils/default_type_traits.hpp"
#include "matador/utils/identifier.hpp"

#include "matador/sql/interface/query_result_reader.hpp"

#include "matador/object/attribute_definition.hpp"

#include <iostream>
#include <memory>
#include <stack>
#include <string>
#include <typeindex>
#include <unordered_set>

namespace matador::utils {
class value;
}

namespace matador::sql {

namespace detail {
class pk_reader
{
public:
  explicit pk_reader(query_result_reader &reader);

  template<class Type>
  void read(Type &obj, const size_t column_index)
  {
    column_index_ = column_index;
    access::process(*this, obj);
  }

  template<typename ValueType>
  void on_primary_key(const char *id, ValueType &value, std::enable_if_t<std::is_integral_v<ValueType> && !std::is_same_v<bool, ValueType>>* = nullptr);
  void on_primary_key(const char *id, std::string &value, size_t size);
  void on_revision(const char * /*id*/, unsigned long long &/*rev*/) {}

  template < class Type >
  void on_attribute(const char * /*id*/, Type &/*x*/, const utils::field_attributes &/*attr*/ = utils::null_attributes) {}
  template < class Pointer >
  void on_belongs_to(const char * /*id*/, Pointer &/*x*/, const utils::foreign_attributes &/*attr*/) {}
  template < class Pointer >
  void on_has_one(const char * /*id*/, Pointer &/*x*/, const utils::foreign_attributes &/*attr*/) {}

  template<class ContainerType>
  void on_has_many(const char * /*id*/, ContainerType &, const char *, const utils::foreign_attributes &/*attr*/) {}
  template<class ContainerType>
  void on_has_many_to_many(const char *id, ContainerType &c, const char *join_column, const char *inverse_join_column, const utils::foreign_attributes &/*attr*/) {}
  template<class ContainerType>
  void on_has_many_to_many(const char *id, ContainerType &c, const utils::foreign_attributes &/*attr*/) {}

private:
  size_t column_index_{};
  query_result_reader &reader_;
};

}

class query_result_impl
{
public:
  query_result_impl(std::unique_ptr<query_result_reader> &&reader, std::vector<object::attribute_definition> &&prototype, size_t column_index = 0);
  query_result_impl(std::unique_ptr<query_result_reader> &&reader, const std::vector<object::attribute_definition> &prototype, size_t column_index = 0);

  template<typename ValueType>
  void on_primary_key(const char *id, ValueType &value, std::enable_if_t<std::is_integral_v<ValueType> && !std::is_same_v<bool, ValueType>>* = nullptr)
  {
    utils::data_type_traits<ValueType>::read_value(*reader_, id, column_index_++, value);
    if (type_stack_.size() == 1) {
      last_pk_ = current_pk_;
      current_pk_ = value;
    }
  }
  void on_primary_key(const char *id, std::string &value, size_t size);
  void on_revision(const char *id, uint64_t &rev);

  template < class Type >
  void on_attribute(const char *id, Type &x, const utils::field_attributes &/*attr*/ = utils::null_attributes)
  {
    utils::data_type_traits<Type>::read_value(*reader_, id, column_index_++, x);
  }
  void on_attribute(const char *id, char *value, const utils::field_attributes &attr = utils::null_attributes);
  void on_attribute(const char *id, std::string &value, const utils::field_attributes &attr = utils::null_attributes);
  void on_attribute(const char *id, utils::value &val, const utils::field_attributes &attr = utils::null_attributes);

  template < class Pointer >
  void on_belongs_to(const char * /*id*/, Pointer &x, const utils::foreign_attributes &attr) {
    if (x.empty()) {
      x.reset(new typename Pointer::value_type);
    }
    if (attr.fetch() == utils::fetch_type::LAZY) {
      pk_reader_.read(*x, column_index_++);
    } else if (const auto ti = std::type_index(typeid(*x)); processed_types_.count(ti) == 0) {
      processed_types_.insert(ti);
      type_stack_.push(ti);
      access::process(*this, *x);
      type_stack_.pop();
    }
  }
  template < class Pointer >
  void on_has_one(const char * /*id*/, Pointer &x, const utils::foreign_attributes &attr)
  {
    if (x.empty()) {
      x.reset(new typename Pointer::value_type);
    }
    if (attr.fetch() == utils::fetch_type::LAZY) {
      pk_reader_.read(*x, column_index_++);
    } else if (const auto ti = std::type_index(typeid(*x)); processed_types_.count(ti) == 0) {
      processed_types_.insert(ti);
    }
  }

  template<class ContainerType>
  void on_has_many_to_many(const char *, ContainerType &, const char * /*join_column*/, const char * /*inverse_join_column*/, const utils::foreign_attributes &/*attr*/) {}
  template<class ContainerType>
  void on_has_many_to_many(const char *, ContainerType &, const utils::foreign_attributes &/*attr*/) {}
  template<class ContainerType>
  void on_has_many(const char * /*id*/, ContainerType &cont, const char * /*join_column*/, const utils::foreign_attributes &attr) {
      if ( attr.fetch() == utils::fetch_type::LAZY ) {
          // pk_reader_.read(*id, column_index_++);
      } else /*if (const auto ti = std::type_index(typeid(typename ContainerType::value_type::value_type)); processed_types_.count(ti) == 0)*/ {
          const auto ti = std::type_index(typeid(typename ContainerType::value_type::value_type));
          processed_types_.insert(ti);
          auto obj = std::make_unique<typename ContainerType::value_type::value_type>();
          type_stack_.push(ti);
          access::process(*this, *obj);
          type_stack_.pop();
          auto ptr = typename ContainerType::value_type(obj.release());
          const auto pk = ptr.primary_key();
          if (ptr.primary_key().is_valid()) {
              cont.push_back(ptr);
          }
      }
  }
  template<class ContainerType>
  void on_has_many(const char *, ContainerType &, const utils::foreign_attributes &/*attr*/) {}

  template<class Type>
  void bind(const Type &obj)
  {
    reader_->bind(obj);
  }

  template<class Type>
  bool fetch(Type &obj) {
    bool result = false;
    do {
        column_index_ = reader_->start_column_index();
        auto fetched = reader_->fetch();
        if (!fetched.is_ok()) {
            return false;
        }
        if (!*fetched) {
            return false;
        }
        processed_types_.insert(typeid(Type));
        type_stack_.push(typeid(Type));
        access::process(*this, obj);
        type_stack_.pop();
        std::cout << "last pk: " << last_pk_.str() << std::endl;
        std::cout << "current pk: " << current_pk_.str() << std::endl;
        std::cout << "last == current: " << std::boolalpha << (last_pk_ == current_pk_) << std::endl;
    } while (last_pk_ == current_pk_);
    return true;
  }

  [[nodiscard]] const std::vector<object::attribute_definition>& prototype() const;

protected:
  size_t column_index_ = 0;
  std::vector<object::attribute_definition> prototype_;
  std::unique_ptr<query_result_reader> reader_;
  detail::pk_reader pk_reader_;
  std::unordered_set<std::type_index> processed_types_;
  std::stack<std::type_index> type_stack_;
  utils::identifier current_pk_{};
  utils::identifier last_pk_{};
};

namespace detail {

template<typename ValueType>
void pk_reader::on_primary_key(const char *id, ValueType &value, std::enable_if_t<std::is_integral_v<ValueType> && !std::is_same_v<bool, ValueType>> *)
{
  utils::data_type_traits<ValueType>::read_value(reader_, id, column_index_++, value);
}

}
}

#endif //QUERY_QUERY_RESULT_IMPL_HPP