write class LuaConverterCandidates

2 files changed, 121 insertions, 3 deletions

diff --git a/src/PYPLuaConverterCandidates.cc b/src/PYPLuaConverterCandidates.cc
new file mode 100644
index 0000000..e4608f4
--- /dev/null
+++ b/src/PYPLuaConverterCandidates.cc
@@ -0,0 +1,108 @@
+/* vim:set et ts=4 sts=4:
+ *
+ * ibus-libpinyin - Intelligent Pinyin engine based on libpinyin for IBus
+ *
+ * Copyright (c) 2018 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 2, 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+ */
+
+#include "PYPLuaConverterCandidates.h"
+#include <assert.h>
+#include "PYString.h"
+#include "PYConfig.h"
+#include "PYPPhoneticEditor.h"
+
+using namespace PY;
+
+LuaConverterCandidates::LuaConverterCandidates (PhoneticEditor *editor)
+{
+    m_editor = editor;
+
+    m_lua_plugin = ibus_engine_plugin_new ();
+
+    loadLuaScript ( ".." G_DIR_SEPARATOR_S "lua" G_DIR_SEPARATOR_S "base.lua")||
+        loadLuaScript (PKGDATADIR G_DIR_SEPARATOR_S "base.lua");
+
+    gchar * path = g_build_filename (g_get_user_config_dir (),
+                             "ibus", "libpinyin", "user.lua", NULL);
+    loadLuaScript(path);
+    g_free(path);
+}
+
+int
+LuaConverterCandidates::loadLuaScript (std::string filename)
+{
+    return !ibus_engine_plugin_load_lua_script
+        (m_lua_plugin, filename.c_str ());
+}
+
+gboolean
+LuaConverterCandidates::setConverter (const char * lua_function_name)
+{
+    return ibus_engine_plugin_set_converter (m_lua_plugin, lua_function_name);
+}
+
+gboolean
+LuaConverterCandidates::processCandidates (std::vector<EnhancedCandidate> & candidates)
+{
+    m_candidates.clear ();
+
+    const char * converter = ibus_engine_plugin_get_converter (m_lua_plugin);
+
+    if (NULL == converter)
+        return FALSE;
+
+    for (guint i = 0; i < candidates.size (); i++) {
+        EnhancedCandidate & enhanced = candidates[i];
+
+        m_candidates.push_back (enhanced);
+
+        enhanced.m_candidate_type = CANDIDATE_LUA_CONVERTER;
+        enhanced.m_candidate_id = i;
+
+        ibus_engine_plugin_call (m_lua_plugin, converter,
+                                enhanced.m_display_string.c_str ());
+        enhanced.m_display_string =
+                    ibus_engine_plugin_get_first_result (m_lua_plugin);
+    }
+
+    return TRUE;
+}
+
+SelectCandidateAction
+LuaConverterCandidates::selectCandidate (EnhancedCandidate & enhanced)
+{
+    guint id = enhanced.m_candidate_id;
+    assert (CANDIDATE_LUA_CONVERTER == enhanced.m_candidate_type);
+
+    const char * converter = ibus_engine_plugin_get_converter (m_lua_plugin);
+    assert (NULL != converter);
+
+    if (G_UNLIKELY (id >= m_candidates.size ()))
+        return SELECT_CANDIDATE_ALREADY_HANDLED;
+
+    SelectCandidateAction action = SELECT_CANDIDATE_ALREADY_HANDLED;
+    action = m_editor->selectCandidateInternal (m_candidates[id]);
+
+    if (SELECT_CANDIDATE_MODIFY_IN_PLACE_AND_COMMIT == action) {
+        ibus_engine_plugin_call (m_lua_plugin, converter,
+                                 enhanced.m_display_string.c_str ());
+        enhanced.m_display_string =
+            ibus_engine_plugin_get_first_result (m_lua_plugin);
+    }
+
+    return action;
+}
diff --git a/src/PYPLuaConverterCandidates.h b/src/PYPLuaConverterCandidates.h
index 9a19bbd..6659d77 100644
--- a/src/PYPLuaConverterCandidates.h
+++ b/src/PYPLuaConverterCandidates.h
@@ -22,24 +22,34 @@
 #ifndef __PY_LIB_PINYIN_LUA_CONVERTER_CANDIDATES_H_
 #define __PY_LIB_PINYIN_LUA_CONVERTER_CANDIDATES_H_
 
+extern "C" {
+#include "lua-plugin.h"
+}
+
 #include <vector>
+#include "PYPointer.h"
 #include "PYPEnhancedCandidates.h"
 
 namespace PY {
 
 class LuaConverterCandidates : public EnhancedCandidates {
 public:
-    LuaConverterCandidates (PhoneticEditor *editor) {
-        m_editor = editor;
-    }
+    LuaConverterCandidates (PhoneticEditor *editor);
 
 public:
+    gboolean setConverter (const char * lua_function_name);
+
     gboolean processCandidates (std::vector<EnhancedCandidate> & candidates);
 
     SelectCandidateAction selectCandidate (EnhancedCandidate & enhanced);
 
 protected:
     std::vector<EnhancedCandidate> m_candidates;
+
+    int loadLuaScript (std::string filename);
+
+    Pointer<IBusEnginePlugin> m_lua_plugin;
+
 };
 
 };