/* SylFilter - a message filter
 *
 * Copyright (C) 2011 Hiroyuki Yamamoto
 * Copyright (C) 2011 Sylpheed Development Team
 */

#include <depot.h>
#include <stdlib.h>

#include "filter-kvs.h"
#include "filter-kvs-qdbm.h"

static XFilterKVS *qdbm_open(const char *dbfile);
static int qdbm_close(XFilterKVS *kvs);
static int qdbm_insert(XFilterKVS *kvs, const char *key, void *value, int size);
static int qdbm_delete(XFilterKVS *kvs, const char *key);
static int qdbm_update(XFilterKVS *kvs, const char *key, void *value, int size);
static int qdbm_fetch(XFilterKVS *kvs, const char *key, void *vbuf, int vsize);
static int qdbm_size(XFilterKVS *kvs);
static int qdbm_foreach(XFilterKVS *kvs, XFilterKVSForeachFunc func, void *data);


int xfilter_kvs_qdbm_set_engine(void)
{
	XFilterKVSEngine engine = {
		qdbm_open,
		qdbm_close,
		qdbm_insert,
		qdbm_delete,
		qdbm_update,
		qdbm_fetch,
		qdbm_size,
		qdbm_foreach
	};

	return xfilter_kvs_set_engine(&engine);
}

static XFilterKVS *qdbm_open(const char *dbfile)
{
	DEPOT *depot;

	depot = dpopen(dbfile, DP_OREADER|DP_OWRITER|DP_OCREAT, 0);
	if (!depot)
		return NULL;
	return xfilter_kvs_new(dbfile, (void *)depot);
}

static int qdbm_close(XFilterKVS *kvs)
{
	if (!dpclose((DEPOT *)xfilter_kvs_get_handle(kvs)))
		return -1;
	return 0;
}

static int qdbm_insert(XFilterKVS *kvs, const char *key, void *value, int size)
{
	DEPOT *depot;

	depot = (DEPOT *)xfilter_kvs_get_handle(kvs);
	if (!dpput(depot, key, -1, (const char *)value, size, DP_DKEEP))
		return -1;
	return 0;
}

static int qdbm_delete(XFilterKVS *kvs, const char *key)
{
	DEPOT *depot;

	depot = (DEPOT *)xfilter_kvs_get_handle(kvs);
	if (!dpout(depot, key, -1))
		return -1;

	return 0;
}

static int qdbm_update(XFilterKVS *kvs, const char *key, void *value, int size)
{
	DEPOT *depot;

	depot = (DEPOT *)xfilter_kvs_get_handle(kvs);
	if (!dpput(depot, key, -1, (const char *)value, size, DP_DOVER))
		return -1;

	return 0;
}

static int qdbm_fetch(XFilterKVS *kvs, const char *key, void *vbuf, int vsize)
{
	DEPOT *depot;
	int size;

	depot = (DEPOT *)xfilter_kvs_get_handle(kvs);
	size = dpgetwb(depot, key, -1, 0, vsize, (char *)vbuf);
	if (size < 0)
		return -1;

	return size;
}

static int qdbm_size(XFilterKVS *kvs)
{
	DEPOT *depot;

	depot = (DEPOT *)xfilter_kvs_get_handle(kvs);
	return dprnum(depot);
}

static int qdbm_foreach(XFilterKVS *kvs, XFilterKVSForeachFunc func, void *data)
{
	DEPOT *depot;
	char *key;
	int ks;
	int size;
	char vbuf[1024];

	depot = (DEPOT *)xfilter_kvs_get_handle(kvs);
	if (!dpiterinit(depot))
		return -1;
	while ((key = dpiternext(depot, &ks)) != NULL) {
		int r;

		size = dpgetwb(depot, key, ks, 0, sizeof(vbuf), vbuf);
		if (size < 0) {
			free(key);
			break;
		}
		r = func(kvs, key, vbuf, size, data);
		free(key);
		if (r < 0)
			break;
	}

	return 0;
}