#ifndef QUERY_CONNECTION_IMPL_HPP
#define QUERY_CONNECTION_IMPL_HPP

#include <memory>

#include "matador/sql/column_definition.hpp"
#include "matador/sql/connection_info.hpp"
#include "matador/sql/query_context.hpp"
#include "matador/utils/error.hpp"
#include "matador/utils/result.hpp"
#include "matador/utils/version.hpp"

namespace matador::utils {
using blob = std::vector<unsigned char>;
}

namespace matador::sql {

class query_result_impl;
class statement_impl;

class connection_impl
{
public:
  virtual ~connection_impl() = default;

  virtual utils::result<void, utils::error> open() = 0;
  virtual utils::result<void, utils::error> close() = 0;
  [[nodiscard]] virtual utils::result<bool, utils::error> is_open() const = 0;
  [[nodiscard]] virtual utils::result<bool, utils::error> is_valid() const = 0;

  [[nodiscard]] virtual utils::result<utils::version, utils::error> client_version() const = 0;
  [[nodiscard]] virtual utils::result<utils::version, utils::error> server_version() const = 0;

  virtual utils::result<size_t, utils::error> execute(const std::string &stmt) = 0;
  virtual utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(const query_context &context) = 0;
  virtual utils::result<std::unique_ptr<statement_impl>, utils::error> prepare(const query_context &context) = 0;

  virtual utils::result<std::vector<column_definition>, utils::error> describe(const std::string &table) = 0;
  virtual utils::result<bool, utils::error> exists(const std::string &schema_name, const std::string &table_name) = 0;

  [[nodiscard]] const class dialect &dialect() const;

  [[nodiscard]] virtual std::string to_escaped_string(const utils::blob &value) const = 0;

protected:
  explicit connection_impl(const connection_info &info);

  [[nodiscard]] const connection_info &info() const;

private:
  std::reference_wrapper<const connection_info> info_;
  std::reference_wrapper<const class dialect> dialect_;
};

}
#endif //QUERY_CONNECTION_IMPL_HPP