#include "db_utils.h"
#include <stdio.h>
#include <stdlib.h>
#include <sqlite3.h>

void db_initialize() {
  sqlite3 *db;
  char *err_msg = 0;
  int rc = sqlite3_open("database.db", &db);

  if (rc != SQLITE_OK) {
    fprintf(stderr, "Error: Cannot open database: %s\n", sqlite3_errmsg(db));
    return;
  }

  const char *sql =
      "CREATE TABLE IF NOT EXISTS kv_store (key TEXT PRIMARY KEY, value TEXT);";
  rc = sqlite3_exec(db, sql, 0, 0, &err_msg);

  if (rc != SQLITE_OK) {
    fprintf(stderr, "SQL error: %s\n", err_msg);
    sqlite3_free(err_msg);
  }

  sqlite3_close(db);
}

void test_db_set() {
  json_object *result = db_set("test_key", "test_value");
  if (result) {
    printf("db_set: %s\n", json_object_get_string(result));
    json_object_put(result);
  } else {
    printf("db_set failed\n");
  }
}

void test_db_get() {
  json_object *result = db_get("test_key");
  if (result) {
    printf("db_get: %s\n", json_object_to_json_string(result));
    json_object_put(result);
  } else {
    printf("db_get failed\n");
  }
}

void test_db_query() {
  json_object *result = db_query("SELECT * FROM kv_store");
  if (result) {
    printf("db_query: %s\n", json_object_to_json_string(result));
    json_object_put(result);
  } else {
    printf("db_query failed\n");
  }
}

int main() {
  db_initialize();
  test_db_set();
  test_db_get();
  test_db_query();
  return 0;
}