1 files changed, 71 insertions, 0 deletions
diff --git a/test/sqlite.h b/test/sqlite.h
new file mode 100644
index 0000000..883a2c0
--- a/dev/null
+++ b/test/sqlite.h
@@ -0,0 +1,71 @@
+#include <sqlite3.h>
+
+class sqlite3_t {
+    public:
+	sqlite3 *_D;
+
+	sqlite3_t(const char *f)
+	    : _D(0) {
+		int r = sqlite3_open(f,&_D);
+		if(r!=SQLITE_OK) {
+		    string msg = sqlite3_errmsg(_D); sqlite3_close(_D);
+		    throw opkele::exception(OPKELE_CP_ "Failed to open SQLite database: "+msg);
+		}
+	    }
+	~sqlite3_t() {
+	    if(_D) sqlite3_close(_D);
+	}
+
+	operator const sqlite3*(void) const { return _D; }
+	operator sqlite3*(void) { return _D; }
+
+	void exec(const char *sql) {
+	    assert(_D);
+	    char *errm;
+	    if(sqlite3_exec(_D,sql,NULL,NULL,&errm)!=SQLITE_OK)
+		throw opkele::exception(OPKELE_CP_ string("Failed to sqlite3_exec():")+errm);
+	}
+	void get_table(const char *sql,char ***resp,int *nr,int *nc) {
+	    assert(_D);
+	    char *errm;
+	    if(sqlite3_get_table(_D,sql,resp,nr,nc,&errm)!=SQLITE_OK)
+		throw opkele::exception(OPKELE_CP_ string("Failed to sqlite3_get_table():")+errm);
+	}
+};
+
+template<typename T>
+class sqlite3_mem_t {
+    public:
+	T _M;
+
+	sqlite3_mem_t(T M) :_M(M) { }
+	~sqlite3_mem_t() { if(_M) sqlite3_free(_M); }
+
+	operator const T&(void) const { return _M; }
+	operator T&(void) { return _M; }
+
+	sqlite3_mem_t operator=(T M) {
+	    if(_M) sqlite3_free(_M);
+	    _M = M;
+	}
+};
+
+class sqlite3_table_t {
+    public:
+	char **_T;
+
+	sqlite3_table_t() : _T(0) { }
+	sqlite3_table_t(char **T) : _T(T) { }
+	~sqlite3_table_t() { if(_T) sqlite3_free_table(_T); }
+
+	operator char**&(void) { return _T; }
+
+	operator char ***(void) {
+	    if(_T) sqlite3_free_table(_T);
+	    return &_T; }
+
+	const char *get(int r,int c,int nc) {
+	    assert(_T);
+	    return _T[r*nc+c];
+	}
+};