#ifndef QUERY_SESSION_HPP
#define QUERY_SESSION_HPP

#include "matador/sql/column_generator.hpp"
#include "matador/sql/connection.hpp"
#include "matador/sql/connection_pool.hpp"
#include "matador/sql/query_builder.hpp"
#include "matador/sql/query_intermediates.hpp"
#include "matador/sql/statement.hpp"
#include "matador/sql/table_repository.hpp"

#include <unordered_map>

namespace matador::sql {

class dialect;

class session
{
public:
  explicit session(connection_pool<connection> &pool);

  [[nodiscard]] query_result<record> fetch(const query_context &q) const;
//  [[nodiscard]] query_result<record> fetch(const std::string &sql) const;
  [[nodiscard]] size_t execute(const std::string &sql) const;
  statement prepare(query_context q) const;

  record describe_table(const std::string &table_name) const;
  bool table_exists(const std::string &table_name) const;

  template<typename Type>
  void attach(const std::string &table_name)
  {
    table_repository_.attach<Type>(table_name);
  }

  [[nodiscard]] const table_repository& tables() const;

  const class dialect& dialect() const;

private:
  friend class query_select_finish;

  [[nodiscard]] std::unique_ptr<query_result_impl> fetch(const std::string &sql) const;

private:
  connection_pool<connection> &pool_;
  const class dialect &dialect_;

  table_repository table_repository_;
  mutable std::unordered_map<std::string, record> prototypes_;
};

}
#endif //QUERY_SESSION_HPP