summaryrefslogtreecommitdiffstats
path: root/src/PYEnglishDatabase.h
blob: ca1407cd9e7f09b3b7c21343b79a823866d209ad (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
/* vim:set et ts=4 sts=4:
 *
 * ibus-libpinyin - Intelligent Pinyin engine based on libpinyin for IBus
 *
 * Copyright (c) 2021 Peng Wu <alexepico@gmail.com>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program 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 General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef __PY_ENGLISH_DATABASE_
#define __PY_ENGLISH_DATABASE_

#include <sqlite3.h>
#include <vector>
#include <memory>
#include "PYString.h"

namespace PY{

class EnglishDatabase{
public:
    static void init ();
    static EnglishDatabase & instance (void) { return *m_instance; }

public:
    EnglishDatabase();
    ~EnglishDatabase();

    gboolean isDatabaseExisted(const char *filename);
    gboolean createDatabase(const char *filename);

    gboolean openDatabase(const char *system_db, const char *user_db);
    gboolean hasWord(const char *word);
    gboolean listWords(const char *prefix, std::vector<std::string> & words);
    gboolean getUserWordInfo(const char *word, float & freq);
    gboolean insertUserWord(const char *word, float freq);
    gboolean updateUserWord(const char *word, float freq);
    gboolean deleteUserWord(const char *word);
    gboolean train(const char *word, float delta);

private:
    gboolean executeSQL(sqlite3 *sqlite);
    gboolean loadUserDB (void);
    gboolean saveUserDB (void);
    void modified (void);
    static gboolean timeoutCallback (gpointer data);

private:
    sqlite3 *m_sqlite;
    String m_sql;
    gchar *m_user_db;

    guint m_timeout_id;
    GTimer *m_timer;

private:
    static std::unique_ptr<EnglishDatabase> m_instance;
};

};

#endif