1 files changed, 227 insertions, 0 deletions
diff --git a/include/sqdb.hpp b/include/sqdb.hpp
new file mode 100644
index 0000000..b92cac6
--- /dev/null
+++ b/include/sqdb.hpp
@@ -0,0 +1,227 @@
+#pragma once
+
+#include <string>
+
+#include "sqlite3.h"
+
+#ifdef _WIN32
+#  include <tchar.h>
+#  define SQDB_MAKE_TEXT(x) _TEXT(x)
+#  define SQDB_STRLEN _tcslen
+#  define SQDB_STRDUP _tcsdup
+#else
+#  define SQDB_MAKE_TEXT(x) (x)
+#  define SQDB_STRLEN strlen
+#  define SQDB_STRDUP strdup
+#endif
+
+#if !defined(SQDB_UTF16) && !defined(SQDB_UTF8)
+#  ifdef _WIN32
+#    if defined(UNICODE) || defined(_UNICODE)
+#      define SQDB_UTF16
+#    else
+#      define SQDB_UTF8
+#    endif
+#  else
+#    define SQDB_UTF8
+#  endif
+#endif
+
+#ifdef SQDB_UTF8
+#  define SQDB_CHAR char
+#  define SQDB_STD_STRING std::string
+#endif
+
+#ifdef SQDB_UTF16
+#  define SQDB_CHAR TCHAR
+#  define SQDB_STD_STRING std::wstring
+#endif
+
+namespace sqdb {
+
+    class Exception {
+    public:
+        Exception(sqlite3 *db);
+
+        Exception(sqlite3 *db, int errorCode);
+
+        Exception(const SQDB_CHAR *errorMsg);
+
+        ~Exception();
+
+        int GetErrorCode() const;
+
+        const SQDB_CHAR *GetErrorMsg() const;
+
+    private:
+        int m_errorCode;
+        SQDB_CHAR *m_errorMsg;
+    };
+
+#define CHECK(db, returnCode) \
+  if ( (returnCode) != SQLITE_OK ) throw Exception(db, returnCode)
+
+    class RefCount {
+    protected:
+        RefCount();
+
+        RefCount(const RefCount &x);
+
+        RefCount &operator=(const RefCount &x);
+
+        void IncRef();
+
+        unsigned DecRef();
+
+    private:
+        unsigned *m_refCount;
+    };
+
+    class Blob : public RefCount {
+    public:
+        Blob(const void *data, int size);
+
+        Blob(const Blob &x);
+
+        Blob &operator=(const Blob &x);
+
+        int GetSize() const;
+
+        const char *GetData() const;
+
+        ~Blob();
+
+    private:
+        char *m_data;
+        int m_size;
+    };
+
+    class Convertor {
+    public:
+        Convertor(sqlite3 *db, sqlite3_stmt *stmt, int field);
+
+        operator int() const;
+
+        operator unsigned long() const;
+
+        operator long long() const;
+
+        operator double() const;
+
+        operator SQDB_STD_STRING() const;
+
+        operator const SQDB_CHAR *() const;
+
+        operator Blob() const;
+
+        int GetInt() const;
+
+        unsigned long GetUnsignedLong() const;
+
+        long long GetLongLong() const;
+
+        double GetDouble() const;
+
+        SQDB_STD_STRING GetString() const;
+
+        const SQDB_CHAR *GetText() const;
+
+        Blob GetBlob() const;
+
+    private:
+        sqlite3 *m_db;
+        sqlite3_stmt *m_stmt;
+        int m_field;
+    };
+
+    class Statement : public RefCount {
+    public:
+        Statement(sqlite3 *db, sqlite3_stmt *stmt);
+
+        Statement(const Statement &x);
+
+        Statement &operator=(const Statement &x);
+
+        bool Next();
+
+        Convertor GetField(int field) const;
+
+        template<class T>
+        void Bind(int i, const T &value) {
+            if (m_needReset)
+                Reset();
+            DoBind(i, value);
+        }
+
+        void BindBlob(int i, const void *value, int n);
+
+        void BindNull(int i);
+
+        ~Statement();
+
+    private:
+        void DoBind(int i, int value);
+
+        void DoBind(int i, long long value);
+
+        void DoBind(int i, double value);
+
+        void DoBind(int i, const SQDB_STD_STRING &value);
+
+        void DoBind(int i, const SQDB_CHAR *value);
+
+        // Bind blob.
+        void DoBind(int i, const void *value, int n);
+
+        // Bind null.
+        void DoBind(int i);
+
+        // Reset binders so that new values can be bound.
+        void Reset();
+
+        sqlite3 *m_db;
+        sqlite3_stmt *m_stmt;
+        bool m_needReset;
+    };
+
+    class QueryStr {
+    public:
+        QueryStr();
+
+        const SQDB_CHAR *Format(const SQDB_CHAR *fmt, ...);
+
+        const SQDB_CHAR *Get() const;
+
+        ~QueryStr();
+
+    private:
+        SQDB_CHAR *m_buf;
+    };
+
+    class Db : public RefCount {
+    public:
+        Db(const SQDB_CHAR *fileName);
+
+        void BeginTransaction();
+
+        void CommitTransaction();
+
+        void RollbackTransaction();
+
+        bool TableExists(const SQDB_CHAR *tableName);
+
+        Statement Query(const SQDB_CHAR *queryStr);
+
+        long long LastId();
+
+        Db(const Db &x);
+
+        Db &operator=(const Db &x);
+
+        ~Db();
+
+    private:
+        sqlite3 *m_db;
+    };
+
+}