#ifndef QUERY_COLUMN_DEFINITION_HPP
#define QUERY_COLUMN_DEFINITION_HPP

#include "matador/utils/basic_types.hpp"
#include "matador/utils/default_type_traits.hpp"
#include "matador/utils/field_attributes.hpp"
#include "matador/utils/value.hpp"

#include <memory>
#include <vector>

namespace matador::object {

enum class null_option_type : uint8_t {
  NULLABLE, NOT_NULL
};

class object_definition;

struct attribute_options {
  utils::field_attributes attributes;
  null_option_type null_option{null_option_type::NOT_NULL};
  int index{-1};
};

class attribute_definition {
public:
  explicit attribute_definition(const char *name); // NOLINT(*-explicit-constructor)
  explicit attribute_definition(std::string name); // NOLINT(*-explicit-constructor)

  attribute_definition(const attribute_definition&) = default;
  attribute_definition& operator=(const attribute_definition&) = default;
  attribute_definition(attribute_definition&&) noexcept = default;
  attribute_definition& operator=(attribute_definition&&) noexcept = default;

  attribute_definition() = default;

  template<typename Type>
   attribute_definition(std::string name, const utils::field_attributes& attr)
  : attribute_definition(std::move(name), utils::data_type_traits<Type>::type(attr.size()), attr)
  {}

  template<typename Type>
  attribute_definition(std::string name, const Type &, const utils::field_attributes& attr, null_option_type null_opt)
  : attribute_definition(std::move(name), utils::data_type_traits<Type>::type(attr.size()), attr, null_opt)
  {}

  template<size_t SIZE>
  attribute_definition(std::string name, const char (&)[SIZE], const utils::field_attributes& attr, const null_option_type null_opt)
  : attribute_definition(std::move(name), utils::data_type_traits<const char*>::type(attr.size()), attr, null_opt)
  {}

  attribute_definition(std::string name, utils::basic_type type, const utils::field_attributes&, null_option_type null_opt, int index = 0);

  template<typename Type>
  attribute_definition(std::string name, const std::shared_ptr<attribute_definition> &ref_column, const utils::field_attributes& attr, null_option_type null_opt)
  : attribute_definition(std::move(name), utils::data_type_traits<Type>::type(attr.size()), ref_column, attr, null_opt)
  {}

  attribute_definition(std::string name, utils::basic_type type, int index, const std::shared_ptr<attribute_definition> &ref_column, const utils::field_attributes& attr, null_option_type null_opt);

  attribute_definition(std::string name, utils::basic_type type, const std::shared_ptr<attribute_definition> &ref_column = {});
  attribute_definition(std::string name, utils::basic_type type, const attribute_options& options, const std::shared_ptr<object_definition>& obj = {}, const std::shared_ptr<attribute_definition> &ref_column = {});

  [[nodiscard]] const std::string& name() const;
  void name(const std::string& n);
  [[nodiscard]] std::string full_name() const;
  [[nodiscard]] std::shared_ptr<object_definition> object() const;
  void object(const std::shared_ptr<object_definition> &obj);
  [[nodiscard]] int index() const;
  [[nodiscard]] const utils::field_attributes& attributes() const;
  [[nodiscard]] utils::field_attributes& attributes();
  [[nodiscard]] bool is_nullable() const;
  [[nodiscard]] utils::basic_type type() const;
  void change_type(utils::basic_type type, const utils::field_attributes &attr = utils::null_attributes);
  template < typename Type >
  void change_type(const utils::field_attributes &attr = utils::null_attributes) {
      type_ = utils::data_type_traits<Type>::type(attr.size());
  }
  [[nodiscard]] std::shared_ptr<attribute_definition> reference_column() const;
  [[nodiscard]] bool is_foreign_reference() const;

  [[nodiscard]] bool is_integer() const;
  [[nodiscard]] bool is_floating_point() const;
  [[nodiscard]] bool is_bool() const;
  [[nodiscard]] bool is_string() const;
  [[nodiscard]] bool is_varchar() const;
  [[nodiscard]] bool is_date() const;
  [[nodiscard]] bool is_time() const;
  [[nodiscard]] bool is_blob() const;
  [[nodiscard]] bool is_null() const;

  template< typename Type >
  [[nodiscard]] bool is_type_of() const {
    return type() == utils::data_type_traits<Type>::type(attributes().size());
  }

private:
  using data_type_index = std::vector<utils::basic_type>;

private:
  friend class object_definition;

  static const data_type_index data_type_index_;

  std::string name_;
  std::shared_ptr<object_definition> object_{};
  attribute_options options_;
  utils::basic_type type_{utils::basic_type::type_null};

  std::shared_ptr<attribute_definition> reference_column_;
};

/**
 * User defined literal to have a shortcut creating a column object
 * @param name Name of the column
 * @param type
 * @param attr Length of the column name
 * @param null_opt
 * @return A column object with a given name
 */
attribute_definition make_column(const std::string &name, utils::basic_type type, utils::field_attributes attr = utils::null_attributes, null_option_type null_opt = null_option_type::NOT_NULL);

template < typename Type >
attribute_definition make_column(const std::string &name, utils::field_attributes attr = utils::null_attributes, null_option_type null_opt = null_option_type::NOT_NULL)
{
    return make_column(name, utils::data_type_traits<Type>::type(0), attr, null_opt);
}
template <>
attribute_definition make_column<std::string>(const std::string &name, utils::field_attributes attr, null_option_type null_opt);

template < typename Type >
attribute_definition make_pk_column(const std::string &name, size_t size = 0)
{
    return make_column<Type>(name, { size, utils::constraints::PRIMARY_KEY });
}

template <>
attribute_definition make_pk_column<std::string>(const std::string &name, size_t size);

template < typename Type >
attribute_definition make_fk_column(const std::string &name, size_t size, const std::shared_ptr<attribute_definition> &ref_column) {
    return {name, utils::data_type_traits<Type>::type(size), ref_column, { size, utils::constraints::FOREIGN_KEY }};
}

template < typename Type >
[[maybe_unused]] attribute_definition make_fk_column(const std::string &name, const std::shared_ptr<attribute_definition> &ref_column)
{
    return {name, utils::data_type_traits<Type>::type(0), 0, ref_column, { 0, utils::constraints::FOREIGN_KEY }, null_option_type::NOT_NULL};
}

template <>
[[maybe_unused]] attribute_definition make_fk_column<std::string>(const std::string &name, size_t size, const std::shared_ptr<attribute_definition> &ref_column);

template < typename Type >
[[maybe_unused]] attribute_definition make_fk_column(const std::string &name, size_t size, const std::string &ref_table_name, const std::string &ref_column_name) {
  return {
    name, utils::data_type_traits<Type>::type(size), 0,
    std::make_shared<attribute_definition>(ref_column_name, ref_table_name, utils::data_type_traits<Type>::type(size), utils::constraints::FOREIGN_KEY),
    { 0, utils::constraints::FOREIGN_KEY }, null_option_type::NOT_NULL
  };
}

template < typename Type >
[[maybe_unused]] attribute_definition make_fk_column(const std::string &name, const std::string &ref_table_name, const std::string &ref_column_name) {
  return {
    name, utils::data_type_traits<Type>::type(0), 0,
    std::make_shared<attribute_definition>(ref_column_name, ref_table_name, utils::data_type_traits<Type>::type(0), utils::constraints::FOREIGN_KEY),
    { 0, utils::constraints::FOREIGN_KEY }, null_option_type::NOT_NULL
  };
}

template <>
[[maybe_unused]] attribute_definition make_fk_column<std::string>(const std::string &name, const std::string &ref_table_name, const std::string &ref_column_name);

template <>
[[maybe_unused]] attribute_definition make_fk_column<std::string>(const std::string &name, size_t size, const std::string &ref_table_name, const std::string &ref_column_name);

}

#endif //QUERY_COLUMN_DEFINITION_HPP