#ifndef QUERY_COLUMN_NAME_GENERATOR_HPP
#define QUERY_COLUMN_NAME_GENERATOR_HPP

#include "matador/utils/access.hpp"
#include "matador/utils/field_attributes.hpp"

#include <string>
#include <vector>

namespace matador::utils {
class identifiable;
}

namespace matador::sql {

class column_name_generator
{
private:
  explicit column_name_generator(std::vector<std::string> &column_names);

public:
  ~column_name_generator() = default;

  template < class Type >
  static std::vector<std::string> generate()
  {
    std::vector<std::string> columns;
    column_name_generator gen(columns);
    Type obj;
    matador::utils::access::process(gen, obj);
    return std::move(columns);
  }

  template < class V >
  void on_primary_key(const char *id, V &, typename std::enable_if<std::is_integral<V>::value && !std::is_same<bool, V>::value>::type* = 0)
  {
    column_names_.emplace_back(id);
  }
  void on_primary_key(const char *id, std::string &, size_t);
  void on_revision(const char *id, unsigned long long &/*rev*/);

  template<typename Type>
  void on_attribute(const char *id, Type &, const utils::field_attributes &/*attr*/ = utils::null_attributes)
  {
    column_names_.emplace_back(id);
  }

  void on_belongs_to(const char *id, utils::identifiable &, utils::cascade_type);
  void on_has_one(const char *id, utils::identifiable &, utils::cascade_type);
  template<class ContainerType>
  void on_has_many(const char *, ContainerType &, const char *, const char *, utils::cascade_type) {}
  template<class ContainerType>
  void on_has_many(const char *, ContainerType &, utils::cascade_type) {}

private:
  std::vector<std::string> &column_names_;
};

}

#endif //QUERY_COLUMN_NAME_GENERATOR_HPP