1 files changed, 108 insertions, 0 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;
+}