SylFilter

/* SylFilter - a message filter

 *

 * Copyright (C) 2011 Hiroyuki Yamamoto

 * Copyright (C) 2011 Sylpheed Development Team

 */

#include "config.h"

#ifdef USE_SQLITE

#include <sqlite3.h>

#include <stdio.h>

#include <stdlib.h>

#include <glib.h>

#include "filter-kvs.h"

#include "filter-kvs-sqlite.h"

static XFilterKVS *sqlite_open(const char *dbfile);

static int sqlite_close(XFilterKVS *kvs);

static int sqlite_insert(XFilterKVS *kvs, const char *key, void *value, int size);

static int sqlite_delete(XFilterKVS *kvs, const char *key);

static int sqlite_update(XFilterKVS *kvs, const char *key, void *value, int size);

static int sqlite_fetch(XFilterKVS *kvs, const char *key, void *vbuf, int vsize);

static int sqlite_begin(XFilterKVS *kvs);

static int sqlite_end(XFilterKVS *kvs);

static int sqlite_size(XFilterKVS *kvs);

static int sqlite_foreach(XFilterKVS *kvs, XFilterKVSForeachFunc func, void *data);

int xfilter_kvs_sqlite_set_engine(void)

{

        XFilterKVSEngine engine = {

                sqlite_open,

                sqlite_close,

                sqlite_insert,

                sqlite_delete,

                sqlite_update,

                sqlite_fetch,

                sqlite_begin,

                sqlite_end,

                sqlite_size,

                sqlite_foreach

        };

        return xfilter_kvs_set_engine(&engine);

}

static XFilterKVS *sqlite_open(const char *dbfile)

{

        sqlite3 *dbhandle = NULL;

        int ret;

        if (sqlite3_open(dbfile, &dbhandle) != SQLITE_OK) {

                sqlite3_close(dbhandle);

                return NULL;

        }

        if ((ret = sqlite3_exec(dbhandle, "CREATE TABLE kvs (key TEXT PRIMARY KEY, value INTEGER NOT NULL)", NULL, NULL, NULL)) != SQLITE_OK)

                printf("sqlite3_exec: returned %d\n", ret);

        return xfilter_kvs_new(dbfile, (void *)dbhandle);

}

static int sqlite_close(XFilterKVS *kvs)

{

        if (sqlite3_close((sqlite3 *)xfilter_kvs_get_handle(kvs)) != SQLITE_OK)

                return -1;

        return 0;

}

static int sqlite_insert(XFilterKVS *kvs, const char *key, void *value, int size)

{

        sqlite3 *db;

        char buf[1024];

        int ival;

        int ret;

        if (size != 4)

                return -1;

        ival = *((int *)value);

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        sqlite3_snprintf(sizeof(buf), buf,

                         "INSERT INTO kvs VALUES ('%q', %d)", key, ival);

        //g_print("%s\n", buf);

        if ((ret = sqlite3_exec(db, buf, NULL, NULL, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: insert: returned %d\n", ret);

                return -1;

        }

        return 0;

}

static int sqlite_delete(XFilterKVS *kvs, const char *key)

{

        sqlite3 *db;

        char buf[1024];

        int ret;

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        sqlite3_snprintf(sizeof(buf), buf,

                         "DELETE FROM kvs WHERE key = '%q'", key);

        g_print("%s\n", buf);

        if ((ret = sqlite3_exec(db, buf, NULL, NULL, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: delete: returned %d\n", ret);

                return -1;

        }

        return 0;

}

static int sqlite_update(XFilterKVS *kvs, const char *key, void *value, int size)

{

        sqlite3 *db;

        char buf[1024];

        int ival;

        int ret;

        if (size != 4)

                return -1;

        ival = *((int *)value);

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        sqlite3_snprintf(sizeof(buf), buf,

                         "UPDATE kvs SET value = %d WHERE key = '%q'", ival, key);

        //g_print("%s\n", buf);

        if ((ret = sqlite3_exec(db, buf, NULL, NULL, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: update: returned %d\n", ret);

                return -1;

        }

        return 0;

}

static int fetch_cb(void *data, int count, char **cols, char **names)

{

        char *val;

        val = cols[0];

        *((int *)data) = atoi(val);

        return 0;

}

static int sqlite_fetch(XFilterKVS *kvs, const char *key, void *vbuf, int vsize)

{

        sqlite3 *db;

        char buf[1024];

        int ival = -1;

        int ret;

        if (vsize != 4)

                return -1;

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        sqlite3_snprintf(sizeof(buf), buf,

                         "SELECT value FROM kvs WHERE key = '%q'", key);

        //g_print("%s\n", buf);

        if ((ret = sqlite3_exec(db, buf, fetch_cb, &ival, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: update: returned %d\n", ret);

                return -1;

        }

        if (ival == -1)

                return -1;

        *((int *)vbuf) = ival;

        return sizeof(ival);

}

static int sqlite_begin(XFilterKVS *kvs)

{

        sqlite3 *db;

        int ret;

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        if ((ret = sqlite3_exec(db, "BEGIN", NULL, NULL, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: begin: returned %d\n", ret);

                return -1;

        }

        return 0;

}

static int sqlite_end(XFilterKVS *kvs)

{

        sqlite3 *db;

        int ret;

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        if ((ret = sqlite3_exec(db, "COMMIT", NULL, NULL, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: commit: returned %d\n", ret);

                return -1;

        }

        return 0;

}

static int size_cb(void *data, int count, char **cols, char **names)

{

        char *val;

        val = cols[0];

        *((int *)data) = atoi(val);

        return 0;

}

static int sqlite_size(XFilterKVS *kvs)

{

        sqlite3 *db;

        char buf[1024];

        int ival = 0;

        int ret;

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        sqlite3_snprintf(sizeof(buf), buf, "SELECT count(key) FROM kvs");

        g_print("%s\n", buf);

        if ((ret = sqlite3_exec(db, buf, size_cb, &ival, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: update: returned %d\n", ret);

                return -1;

        }

        return ival;

}

struct cbdata

{

        XFilterKVS *kvs;

        XFilterKVSForeachFunc func;

        void *data;

};

static int foreach_cb(void *data, int count, char **cols, char **names)

{

        const char *key;

        const char *val;

        int ival;

        int r;

        struct cbdata *cbdata;

        cbdata = (struct cbdata *)data;

        key = cols[0];

        val = cols[1];

        ival = atoi(val);

        r = cbdata->func(cbdata->kvs, key, &ival, sizeof(ival), cbdata->data);

        if (r < 0)

                return -1;

        return 0;

}

static int sqlite_foreach(XFilterKVS *kvs, XFilterKVSForeachFunc func, void *data)

{

        sqlite3 *db;

        char buf[1024];

        int ret;

        struct cbdata cbdata;

        cbdata.func = func;

        cbdata.data = data;

        db = (sqlite3 *)xfilter_kvs_get_handle(kvs);

        sqlite3_snprintf(sizeof(buf), buf, "SELECT key, value FROM kvs");

        if ((ret = sqlite3_exec(db, buf, foreach_cb, &cbdata, NULL)) != SQLITE_OK) {

                printf("sqlite3_exec: update: returned %d\n", ret);

                if (ret != SQLITE_ABORT)

                        return -1;

        }

        return 0;

}

#endif /* USE_SQLITE */