#include "duckdb.hpp"
#include "result_iterator.h"
#include <memory>
#include <napi.h>
#include <string>

namespace NodeDuckDB {
class AsyncExecutor : public Napi::AsyncWorker {
public:
  AsyncExecutor(Napi::Env &env, std::string &query,
                std::shared_ptr<duckdb::Connection> &connection,
                Napi::Promise::Deferred &deferred, bool forceMaterialized,
                ResultFormat &rowResultFormat,
                std::shared_ptr<std::vector<ResultIterator *>> results);
  ~AsyncExecutor();
  void Execute() override;
  void OnOK() override;
  void OnError(const Napi::Error &e) override;

private:
  std::string query;
  ResultFormat rowResultFormat;
  std::shared_ptr<duckdb::Connection> connection;
  std::unique_ptr<duckdb::QueryResult> result;
  std::shared_ptr<std::vector<ResultIterator *>> results;
  Napi::Promise::Deferred deferred;
  bool forceMaterialized;
};
} // namespace NodeDuckDB