#include <cstring>

#include "util/log.h"

#include "DB.h"

using namespace std;
using namespace crepe;

DB::DB(const string & path) {
	dbg_trace();
	int ret;

	// init database struct
	libdb::DB * db;
	if ((ret = libdb::db_create(&db, NULL, 0)) != 0)
		throw nullptr; // TODO: exception
	this->db = { db, [] (libdb::DB * db) { db->close(db, 0); } };

	// load or create database file
	if ((ret = this->db->open(this->db.get(), NULL, path.c_str(), NULL, libdb::DB_BTREE, DB_CREATE, 0)) != 0) {
		throw nullptr;
	}

	// create cursor
	libdb::DBC * cursor;
	if ((ret = this->db->cursor(this->db.get(), NULL, &cursor, 0)) != 0) {
		throw nullptr;
	}
	this->cursor = { cursor, [] (libdb::DBC * cursor) { cursor->close(cursor); } };
}


libdb::DBT DB::to_thing(const string & thing) {
	libdb::DBT thang;
	memset(&thang, 0, sizeof(libdb::DBT));
	thang.data = (void *) thing.data();
	thang.size = thing.size();
	return thang;
}

string DB::get(const string & key) {
	libdb::DBT db_key = this->to_thing(key);
	libdb::DBT db_val;
	memset(&db_val, 0, sizeof(libdb::DBT));

	// int ret = this->cursor->get(this->cursor.get(), NULL, &db_key, &db_val);
	return "";
}

void DB::set(const string & key, const string & value) {
	libdb::DBT db_key = this->to_thing(key);
	libdb::DBT db_val = this->to_thing(value);
	int ret = this->db->put(this->db.get(), NULL, &db_key, &db_val, 0);
	// TODO: check flags
	// TODO: check ret
}

bool DB::has(const std::string & key) {
	try {
		this->get(key);
	} catch (...) {
		return false;
	}
	return true;
}