from gi.repository import IBus from gi.repository import GLib from gi.repository import GObject import os import sys import locale import re import logging import pathlib from comb.engine import Comb from comb.user_dict import UserDict from comb.system_dict import SystemDict MODE_KANA = 1 MODE_ALPHA = 2 num_keys = [] for n in range(1, 10): num_keys.append(getattr(IBus, str(n))) num_keys.append(getattr(IBus, '0')) del n numpad_keys = [] for n in range(1, 10): numpad_keys.append(getattr(IBus, 'KP_' + str(n))) numpad_keys.append(getattr(IBus, 'KP_0')) del n configdir = os.path.join(GLib.get_user_config_dir(), 'ibus-comb') pathlib.Path(configdir).mkdir(parents=True, exist_ok=True) logging.info(f"Loading user dictionary: {configdir}") user_dict = UserDict(os.path.join(configdir, 'user-dict.txt'), logging.getLogger('UserDict')) logging.info("Loaded user dictionary") system_dict = SystemDict() comb = Comb(logging.getLogger('Comb'), user_dict, system_dict) # ---------------------------------------------------------------------- # the engine # ---------------------------------------------------------------------- class CombIBusEngine(IBus.Engine): mode: bool __gtype_name__ = 'CombIBusEngine' def __init__(self): super(CombIBusEngine, self).__init__() self.is_invalidate = False # 未変換文字列。 self.preedit_string = '' self.lookup_table = IBus.LookupTable.new(10, 0, True, True) self.prop_list = IBus.PropList() self.comb = comb self.user_dict = user_dict self.logger = logging.getLogger(__name__) self.candidates = [] self.mode = MODE_KANA # カーソル変更をしたばっかりかどうかを、みるフラグ。 self.cursor_moved = False self.logger.debug("Create Comb engine OK") def set_lookup_table_cursor_pos_in_current_page(self, index): '''Sets the cursor in the lookup table to index in the current page Returns True if successful, False if not. ''' page_size = self.lookup_table.get_page_size() if index > page_size: return False page, pos_in_page = divmod(self.lookup_table.get_cursor_pos(), page_size) new_pos = page * page_size + index if new_pos > self.lookup_table.get_number_of_candidates(): return False self.lookup_table.set_cursor_pos(new_pos) return True def do_candidate_clicked(self, index, dummy_button, dummy_state): if self.set_lookup_table_cursor_pos_in_current_page(index): self.commit_candidate() def do_process_key_event(self, keyval, keycode, state): try: return self._do_process_key_event(keyval, keycode, state) except: self.logger.error(f"Cannot process key event: keyval={keyval} keycode={keycode} state={state}", exc_info=True) return False def _do_process_key_event(self, keyval, keycode, state): self.logger.debug("process_key_event(%04x, %04x, %04x)" % (keyval, keycode, state)) # ignore key release events is_press = ((state & IBus.ModifierType.RELEASE_MASK) == 0) if not is_press: return False # 入力モードの切り替え機能。 if keyval == IBus.Henkan: self.logger.info("Switch to kana mode") self.mode = MODE_KANA return True elif keyval == IBus.Muhenkan: self.logger.info("Switch to alpha mode") self.mode = MODE_ALPHA return True if self.preedit_string: if keyval in (IBus.Return, IBus.KP_Enter): if self.lookup_table.get_number_of_candidates() > 0: self.commit_candidate() else: self.commit_string(self.preedit_string) return True elif keyval == IBus.Escape: self.preedit_string = '' self.update_candidates() return True elif keyval == IBus.BackSpace: # サイゴの一文字をけずるが、子音が先行しているばあいは、子音もついでにとる。 self.preedit_string = re.sub('(?:z[hjkl.-]|[kstnhmyrwgzjdbp]?[aiueo]|.)$', '', self.preedit_string) self.invalidate() return True elif keyval in num_keys: index = num_keys.index(keyval) if self.set_lookup_table_cursor_pos_in_current_page(index): self.commit_candidate() return True return False elif keyval in numpad_keys: index = numpad_keys.index(keyval) if self.set_lookup_table_cursor_pos_in_current_page(index): self.commit_candidate() return True return False elif keyval in (IBus.Page_Up, IBus.KP_Page_Up, IBus.Left, IBus.KP_Left): self.page_up() return True elif keyval in (IBus.Page_Down, IBus.KP_Page_Down, IBus.Right, IBus.KP_Right): self.page_down() return True elif keyval in (IBus.Up, IBus.KP_Up): self.cursor_up() return True elif keyval in (IBus.Down, IBus.KP_Down): self.cursor_down() return True if keyval == IBus.space: if len(self.preedit_string) == 0: # もし、まだなにもはいっていなければ、ただの空白をそのままいれる。 return False else: self.logger.debug("cursor down") self.cursor_down() return True if self.mode == MODE_KANA: # Allow typing all ASCII letters and punctuation, except digits if ord('!') <= keyval < ord('0') or \ ord('9') < keyval <= ord('~'): if state & (IBus.ModifierType.CONTROL_MASK | IBus.ModifierType.MOD1_MASK) == 0: if self.cursor_moved: self.commit_candidate() self.preedit_string += chr(keyval) self.invalidate() return True else: if keyval < 128 and self.preedit_string: self.commit_string(self.preedit_string) else: return False return False def invalidate(self): if self.is_invalidate: return self.is_invalidate = True GLib.idle_add(self.update_candidates) def page_up(self): if self.lookup_table.page_up(): self.cursor_moved = True self._update_lookup_table() return True return False def page_down(self): if self.lookup_table.page_down(): self.cursor_moved = True self._update_lookup_table() return True return False def cursor_up(self): if self.lookup_table.cursor_up(): self.cursor_moved = True self._update_lookup_table() return True return False def cursor_down(self): if self.lookup_table.cursor_down(): self.cursor_moved = True self._update_lookup_table() return True return False def commit_string(self, text): self.cursor_moved = False self.user_dict.add_entry(self.preedit_string, text) self.commit_text(IBus.Text.new_from_string(text)) self.preedit_string = '' self.update_candidates() def commit_candidate(self): cursor_pos = self.lookup_table.get_cursor_pos() if cursor_pos < len(self.candidates): self.commit_string(self.candidates[cursor_pos]) else: # maybe, not happen, but happen.. why? self.logger.error( f"commit_candidate failure: cursor_pos={cursor_pos}, candidates={self.candidates}") self.commit_string('') def update_candidates(self): preedit_len = len(self.preedit_string) attrs = IBus.AttrList() self.lookup_table.clear() self.candidates = [] if preedit_len > 0: try: comb_results = self.comb.convert(self.preedit_string) self.logger.debug("HAHAHA %s, %s" % (str(self.preedit_string), str(comb_results))) for char_sequence, display_str in comb_results: candidate = IBus.Text.new_from_string(display_str) self.candidates.append(char_sequence) self.lookup_table.append_candidate(candidate) except: self.logger.error("cannot get kanji candidates %s" % (sys.exc_info()[0]), exc_info=True) first_candidate = self.candidates[0] if len(self.candidates) > 0 else self.preedit_string # にほんご ですね. text = IBus.Text.new_from_string(first_candidate) text.set_attributes(attrs) self.update_auxiliary_text(text, preedit_len > 0) attrs.append(IBus.Attribute.new(IBus.AttrType.UNDERLINE, IBus.AttrUnderline.SINGLE, 0, preedit_len)) text = IBus.Text.new_from_string(first_candidate) text.set_attributes(attrs) self.update_preedit_text(text, preedit_len, preedit_len > 0) self._update_lookup_table() self.is_invalidate = False def _update_lookup_table(self): visible = self.lookup_table.get_number_of_candidates() > 0 self.update_lookup_table(self.lookup_table, visible) def do_focus_in(self): self.logger.debug("focus_in") self.register_properties(self.prop_list) def do_focus_out(self): self.logger.debug("focus_out") self.do_reset() def do_reset(self): self.logger.debug("reset") self.preedit_string = '' def do_property_activate(self, prop_name): self.logger.debug("PropertyActivate(%s)" % prop_name) def do_page_up(self): return self.page_up() def do_page_down(self): return self.page_down() def do_cursor_up(self): return self.cursor_up() def do_cursor_down(self): return self.cursor_down()