#include "sqlite_statement.h"
#include <iostream>

// Constructor: Prepares the SQL statement
SQLiteStatement::SQLiteStatement(sqlite3 *db, const std::string &query)
    : db_(db), stmt_(nullptr) {
  if (db == nullptr) {
    throw std::invalid_argument("Database connection is null");
  }

  if (sqlite3_prepare_v2(db, query.c_str(), -1, &stmt_, nullptr) != SQLITE_OK) {
    std::cerr << "Error preparing SQL statement:" << "\"" + query + "\""
              << std::endl;
    std::cerr << "Error message:" << sqlite3_errmsg(db) << std::endl;

    if (stmt_) {
      sqlite3_finalize(stmt_);
    }

    throw std::runtime_error("Failed to prepare SQLite statement: " + query);
  }
}

// Destructor: Finalizes the statement
SQLiteStatement::~SQLiteStatement() { finalize(); }

// Bind integer value
bool SQLiteStatement::bind(int index, int value) {
  return sqlite3_bind_int(stmt_, index, value) == SQLITE_OK;
}

// Bind double value
bool SQLiteStatement::bind(int index, double value) {
  return sqlite3_bind_double(stmt_, index, value) == SQLITE_OK;
}

// Bind text value
bool SQLiteStatement::bind(int index, const std::string &value) {
  return sqlite3_bind_text(stmt_, index, value.c_str(), -1, SQLITE_TRANSIENT) ==
         SQLITE_OK;
}

// Bind NULL value
bool SQLiteStatement::bindNull(int index) {
  return sqlite3_bind_null(stmt_, index) == SQLITE_OK;
}

// Run the statement
// step() returns true if there are more rows
bool SQLiteStatement::execute() { return step() == false; }

// Execute the statement
bool SQLiteStatement::step() {
  int rc = sqlite3_step(stmt_);

  if (rc == SQLITE_ROW) {
    return true;
  }
  if (rc == SQLITE_DONE) {
    return false;
  }

  std::cerr << "Failed to step:" << sqlite3_errmsg(db_) << std::endl;
  return false;
}

// Get the number of rows affected
int SQLiteStatement::changes() { return sqlite3_changes(db_); }

// Reset the statement for re-use
void SQLiteStatement::reset() { sqlite3_reset(stmt_); }

// Finalize and clean up the statement
void SQLiteStatement::finalize() {
  if (stmt_) {
    sqlite3_finalize(stmt_);
    stmt_ = nullptr;
  }
}

// Adapter for sqlite3_column_count
int SQLiteStatement::getColumnCount() const {
  return sqlite3_column_count(stmt_);
}

// Adapter for sqlite3_column_text
std::string SQLiteStatement::getColumnText(int column_index) const {
  const unsigned char *text = sqlite3_column_text(stmt_, column_index);
  return text ? reinterpret_cast<const char *>(text) : "";
}