#pragma once

#include "main/lbug.h"

#include "node_connection.h"

#include <napi.h>

using namespace lbug::main;

class NodePreparedStatement : public Napi::ObjectWrap<NodePreparedStatement> {

friend class NodeConnection;

friend class PreparedStatementInitAsyncWorker;

public:

static Napi::Object Init(Napi::Env env, Napi::Object exports);

explicit NodePreparedStatement(const Napi::CallbackInfo& info);

~NodePreparedStatement() override = default;

private:

Napi::Value InitAsync(const Napi::CallbackInfo& info);

Napi::Value InitSync(const Napi::CallbackInfo& info);

void InitCppPreparedStatement();

Napi::Value IsSuccess(const Napi::CallbackInfo& info);

Napi::Value GetErrorMessage(const Napi::CallbackInfo& info);

private:

std::shared_ptr<PreparedStatement> preparedStatement;

std::shared_ptr<Connection> connection;

std::string queryString;

};

class PreparedStatementInitAsyncWorker : public Napi::AsyncWorker {

public:

PreparedStatementInitAsyncWorker(Napi::Function& callback,

NodePreparedStatement* nodePreparedStatement)

: AsyncWorker(callback), nodePreparedStatement(nodePreparedStatement) {}

~PreparedStatementInitAsyncWorker() override = default;

inline void Execute() override {

try {

nodePreparedStatement->InitCppPreparedStatement();

} catch (const std::exception& exc) {

SetError(std::string(exc.what()));

}

}

inline void OnOK() override { Callback().Call({Env().Null()}); }

inline void OnError(Napi::Error const& error) override { Callback().Call({error.Value()}); }

private:

NodePreparedStatement* nodePreparedStatement;

};