1 files changed, 114 insertions, 0 deletions
diff --git a/src/Database.h b/src/Database.h
new file mode 100644
index 0000000..95d7841
--- /dev/null
+++ b/src/Database.h
@@ -0,0 +1,114 @@
+/* vim:set et ts=4 sts=4:
+ *
+ * libpyzy - The Chinese PinYin and Bopomofo conversion library.
+ *
+ * Copyright (c) 2008-2010 Peng Huang <shawn.p.huang@gmail.com>
+ *
+ * This library is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as
+ * published by the Free Software Foundation; either version 2.1 of the
+ * License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
+ * USA
+ */
+#ifndef __PYZY_DATABASE_H_
+#define __PYZY_DATABASE_H_
+
+#include "PhraseArray.h"
+#include "String.h"
+#include "Types.h"
+#include "Util.h"
+
+typedef struct sqlite3 sqlite3;
+
+namespace PyZy {
+
+class PinyinArray;
+struct Phrase;
+
+class SQLStmt;
+typedef std::shared_ptr<SQLStmt> SQLStmtPtr;
+
+class Database;
+
+class Query {
+public:
+    Query (const PinyinArray    & pinyin,
+           size_t                 pinyin_begin,
+           size_t                 pinyin_len,
+           unsigned int           option);
+    ~Query (void);
+    int fill (PhraseArray &phrases, int count);
+
+private:
+    const PinyinArray & m_pinyin;
+    size_t m_pinyin_begin;
+    size_t m_pinyin_len;
+    unsigned int m_option;
+    SQLStmtPtr m_stmt;
+};
+
+class Database {
+public:
+    ~Database ();
+protected:
+    Database (const std::string & user_data_dir);
+
+public:
+    static void init (const std::string & data_dir);
+
+    SQLStmtPtr query (const PinyinArray   & pinyin,
+                      size_t                pinyin_begin,
+                      size_t                pinyin_len,
+                      int                   m,
+                      unsigned int          option);
+    void commit (const PhraseArray  & phrases);
+    void remove (const Phrase & phrase);
+
+    void conditionsDouble (void);
+    void conditionsTriple (void);
+
+    static void finalize (void);
+    static Database & instance (void)
+    {
+        if (m_instance == NULL) {
+            g_error ("Error: Please call InputContext::init () !");
+        }
+        return *m_instance;
+    }
+
+private:
+    bool open (void);
+    bool loadUserDB (void);
+    bool saveUserDB (void);
+    void prefetch (void);
+    void phraseSql (const Phrase & p, String & sql);
+    void phraseWhereSql (const Phrase & p, String & sql);
+    bool executeSQL (const char *sql, sqlite3 *db = NULL);
+    void modified (void);
+    static gboolean timeoutCallback (void * data);
+
+private:
+    sqlite3 *m_db;              /* sqlite3 database */
+
+    String m_sql;        /* sql stmt */
+    String m_buffer;     /* temp buffer */
+    unsigned int m_timeout_id;
+    GTimer *m_timer;
+    String m_user_data_dir;
+
+private:
+    static std::unique_ptr<Database> m_instance;
+};
+
+};  // namespace PyZy
+
+#endif  // __PYZY_DATABASE_H_