// Copyright contributors to the nodejs-idb-connector project
// SPDX-License-Identifier: MIT

#pragma once

#include "dbconn.h"
#include <vector>


struct db2ColumnDescription
{
  SQLCHAR *name;
  SQLSMALLINT nameLength;
  SQLSMALLINT sqlType;
  SQLINTEGER colPrecise;
  SQLSMALLINT colScale;
  SQLSMALLINT colNull;
  SQLINTEGER rlength;
  SQLINTEGER clobLoc;
};

struct db2ParameterDescription
{
  SQLSMALLINT valueType;
  SQLSMALLINT paramType;
  SQLINTEGER paramSize;
  SQLSMALLINT decDigits;
  SQLSMALLINT nullable;
  int io;
  SQLINTEGER ind;
  void *buf;
};

struct resultSetItem
{
  SQLCHAR *data;
  SQLINTEGER rlength;
};

class DbStmt : public Napi::ObjectWrap<DbStmt>
{
  // classes that do the async versions of the workflows
  friend class ExecAsyncWorker;
  friend class PrepareAsyncWorker;
  friend class BindParamAsyncWorker;
  friend class BindParametersAsyncWorker;
  friend class ExecuteAsyncWorker;
  friend class FetchAsyncWorker;
  friend class FetchAllAsyncWorker;

public:
  static Napi::Object Init(Napi::Env env, Napi::Object exports);
  DbStmt(const Napi::CallbackInfo &info);

private:
  static Napi::FunctionReference constructor;

  Napi::Value SetStmtAttr(const Napi::CallbackInfo &info);
  Napi::Value GetStmtAttr(const Napi::CallbackInfo &info);

  void Exec(const Napi::CallbackInfo &info);
  Napi::Value ExecSync(const Napi::CallbackInfo &info);

  void Prepare(const Napi::CallbackInfo &info);
  void PrepareSync(const Napi::CallbackInfo &info);

  void BindParam(const Napi::CallbackInfo &info);
  void BindParamSync(const Napi::CallbackInfo &info);

  void BindParameters(const Napi::CallbackInfo &info);
  void BindParametersSync(const Napi::CallbackInfo &info);

  void Execute(const Napi::CallbackInfo &info);
  Napi::Value ExecuteSync(const Napi::CallbackInfo &info);

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

  void Fetch(const Napi::CallbackInfo &info);
  Napi::Value FetchSync(const Napi::CallbackInfo &info);

  void FetchAll(const Napi::CallbackInfo &info);
  Napi::Value FetchAllSync(const Napi::CallbackInfo &info);

  Napi::Value CloseCursor(const Napi::CallbackInfo &info);
  void Reset(const Napi::CallbackInfo &info);
  Napi::Value Commit(const Napi::CallbackInfo &info);
  Napi::Value Rollback(const Napi::CallbackInfo &info);
  Napi::Value Close(const Napi::CallbackInfo &info);

  Napi::Value NumFields(const Napi::CallbackInfo &info);
  Napi::Value NumRows(const Napi::CallbackInfo &info);
  Napi::Value FieldType(const Napi::CallbackInfo &info);
  Napi::Value FieldWidth(const Napi::CallbackInfo &info);
  Napi::Value FieldName(const Napi::CallbackInfo &info);
  Napi::Value FieldPrecise(const Napi::CallbackInfo &info);
  Napi::Value FieldScale(const Napi::CallbackInfo &info);
  Napi::Value FieldNullable(const Napi::CallbackInfo &info);
  Napi::Value FieldInfo(const Napi::CallbackInfo &info);

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

  void StmtError(const Napi::CallbackInfo &info);

  // unexposed helper functions
  int populateColumnDescriptions(Napi::Env env);
  void freeColumnDescriptions();
  int bindColData(Napi::Env env);
  int fetchData();
  int buildJsObject(Napi::Env env, Napi::Array *array);
  void freeBindingRow();
  void freeColumns();
  void freeSp();
  int bindParams(Napi::Env env, Napi::Array *params, std::string &error);
  int fetchSp(Napi::Env env, Napi::Array *array);
  int fetch(Napi::Env env, Napi::Object *row);

  bool stmtAllocated = false;
  bool resultSetAvailable = false;
  bool colDescAllocated = false;
  bool bindingRowAllocated = false;

  bool isDebug = false;
  bool asNumber = false;

  static SQLHENV envh;
  SQLHDBC connh;
  SQLHSTMT stmth;
  DbConn *myDbConn;

  char *xmlOut;
  char *spIn[SP_PARAM_MAX];
  char *spOut[SP_PARAM_MAX];
  int spInNum[SP_PARAM_MAX];
  int spOutNum[SP_PARAM_MAX];
  SQLINTEGER indicator[SP_PARAM_MAX];
  int spInCount = 0;
  int spOutCount = 0;
  int spInNumCount;
  int spOutNumCount;

  SQLSMALLINT colCount = 0;
  db2ColumnDescription *dbColumn;
  SQLCHAR **bindingRowInC;
  std::vector<resultSetItem *> resultSetInC;

  db2ParameterDescription *param = NULL;
  int paramCount = 0;

  char msg[SQL_MAX_MESSAGE_LENGTH + SQL_SQLSTATE_SIZE + 10];
};