#include <stdio.h>
#include <assert.h>
#include <stdlib.h>
#include <sqlite3.h>
#include <glib.h>

#include "json-c/json.h"

static sqlite3 *db;

static int callback(void *param, int argc, char **argv, char **azColName)
{
    int i;
    json_object *j_array = (json_object *)param;
    json_object *j_cfg = json_object_new_object();

    for (i = 0; i < argc; i++) {
        if (azColName[i][0] == 'i') {
            json_object_object_add(j_cfg, azColName[i], json_object_new_int(atoi(argv[i])));
        } else {
            json_object_object_add(j_cfg, azColName[i], json_object_new_string(argv[i] ? argv[i] : ""));
        }
    }
    json_object_array_add(j_array, j_cfg);

    return 0;
}

char *rkdb_sql(char *sql)
{
    int rc;
    char *zErrMsg = 0;
    json_object *j_cfg = json_object_new_object();
    json_object *j_array = json_object_new_array();

    rc = sqlite3_exec(db, sql, callback, (void*)j_array, &zErrMsg);
    json_object_object_add(j_cfg, "iReturn", json_object_new_int(rc));
    if (rc != SQLITE_OK) {
        printf("SQL error: %s\n", zErrMsg);
        json_object_object_add(j_cfg, "sErrMsg", json_object_new_string(zErrMsg));
        sqlite3_free(zErrMsg);
    } else {
        json_object_object_add(j_cfg, "sErrMsg", json_object_new_string(""));
        //printf("Operation done successfully\n");
    }
    json_object_object_add(j_cfg, "jData", j_array);
    char *ret = g_strdup((char *)json_object_get_string(j_cfg));
    json_object_put(j_cfg);

    return ret;
}

char *rkdb_drop(char *table)
{
    char *sql = g_strdup_printf("DROP TABLE %s;", table);

    char *ret = rkdb_sql(sql);
    g_free(sql);

    return ret;
}

char *rkdb_create(char *table, char *col_para)
{
    char *sql = g_strdup_printf("CREATE TABLE %s(%s);", table, col_para);

    char *ret = rkdb_sql(sql);
    g_free(sql);

    return ret;
}

char *rkdb_insert(char *table, char *cols, char *vals)
{
    char *sql = g_strdup_printf("INSERT INTO %s (%s) VALUES (%s);", table, cols, vals);

    char *ret = rkdb_sql(sql);
    g_free(sql);

    return ret;
}

char *rkdb_update(char *table, char *set, char *where)
{
    char *sql = g_strdup_printf("UPDATE %s SET %s WHERE %s;", table, set, where);

    char *ret = rkdb_sql(sql);
    g_free(sql);

    return ret;
}

char *rkdb_delete(char *table, char *where)
{
    char *sql;

    if (where)
        sql = g_strdup_printf("DELETE from %s WHERE %s;", table, where);
    else
        sql = g_strdup_printf("DELETE from %s;", table);

    char *ret = rkdb_sql(sql);
    g_free(sql);

    return ret;
}

char *rkdb_select(char *table, char *colname, char *where, char *order, char *limit)
{
    char *sql = NULL;
    char *tmp;

    if (colname)
        sql = g_strdup_printf("SELECT %s from %s", colname, table);
    else
        sql = g_strdup_printf("SELECT * from %s", table);

    if (where) {
        tmp = sql;
        sql = g_strdup_printf("%s WHERE %s", tmp, where);
        g_free(tmp);
    }

    if (order) {
        tmp = sql;
        sql = g_strdup_printf("%s ORDER BY %s", tmp, order);
        g_free(tmp);
    }

    if (limit) {
        tmp = sql;
        sql = g_strdup_printf("%s LIMIT %s", tmp, limit);
        g_free(tmp);
    }

    char *ret = rkdb_sql(sql);
    g_free(sql);

    return ret;
}

int rkdb_init(char *file)
{
    int rc;
    char *zErrMsg = 0;

    rc = sqlite3_open(file, &db);
    assert(!rc);

    return 0;
}

void rkdb_deinit(void)
{
    if (db)
        sqlite3_close(db);

    db = NULL;
}