Support Layout Mapping and Cyrillic characters

readme.md

@@ -199,6 +199,14 @@ FILE_PICKER_CMD = "ranger --choosefile={file_path}"
 MAILCAP_FILE = os.path.expanduser("~/.config/mailcap")
 
 DOWNLOAD_DIR = os.path.expanduser("~/Downloads/")  # copy file to this dir
+
+# to map key bindings to a different language or a different keyboard layout
+LAYOUT_MAPPING = {
+    eng: rus for (eng, rus) in zip(
+        "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ",
+        "фисвуапршолдьтщзйкыегмцчняФИСВУАПРШОЛДЬТЩЗЙКЫЕГМЦЧНЯ"
+    )
+}
 ```
 
 ### Mailcap file

tg/config.py

@@ -80,6 +80,13 @@
 
 DOWNLOAD_DIR = os.path.expanduser("~/Downloads/")
 
+LAYOUT_MAPPING = {
+    eng: rus for (eng, rus) in zip(
+        "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ",
+        "фисвуапршолдьтщзйкыегмцчняФИСВУАПРШОЛДЬТЩЗЙКЫЕГМЦЧНЯ"
+    )
+}
+
 if os.path.isfile(CONFIG_FILE):
     config_params = runpy.run_path(CONFIG_FILE)  # type: ignore
     for param, value in config_params.items():

tg/controllers.py

@@ -5,7 +5,7 @@
 from functools import partial, wraps
 from queue import Queue
 from tempfile import NamedTemporaryFile
-from typing import Any, Callable, Dict, List, Optional
+from typing import Any, Callable, Dict, List, Optional, Set
 
 from telegram.utils import AsyncResult
 
@@ -37,6 +37,17 @@
 
 chat_handler: Dict[str, HandlerType] = {}
 msg_handler: Dict[str, HandlerType] = {}
+layout_maps: Set[str] = set()
+
+
+def map_key_to_layout(key:str, lang_map: Dict[str, str]):
+    mapped_key = ""
+    for char in key:
+        if char in lang_map:
+            mapped_key += lang_map[char]
+        else:
+            mapped_key += char
+    return mapped_key
 
 
 def bind(
@@ -61,6 +72,10 @@ def _no_repeat_factor(self: "Controller", _: bool) -> Optional[str]:
             ), f"Key {key} already binded to {binding[key]}"
             binding[key] = fun if repeat_factor else _no_repeat_factor  # type: ignore
 
+            mapped_key = map_key_to_layout(key, config.LAYOUT_MAPPING)
+            binding[mapped_key] = fun if repeat_factor else _no_repeat_factor
+            layout_maps.add(mapped_key)
+
         return wrapper
 
     return decorator
@@ -148,6 +163,7 @@ def format_help(bindings: Dict[str, HandlerType]) -> str:
         return "\n".join(
             f"{key}\t{fun.__name__}\t{fun.__doc__ or ''}"
             for key, fun in sorted(bindings.items())
+            if key not in layout_maps
         )
 
     @bind(chat_handler, ["?"])

tg/views.py

@@ -26,6 +26,14 @@
     "ng",
     "bp",
 )
+if config.LAYOUT_MAPPING:
+    mapped_bindings = []
+    for binding in MULTICHAR_KEYBINDINGS:
+        mapped_binding = ""
+        for char in binding:
+            mapped_binding += config.LAYOUT_MAPPING[char]
+        mapped_bindings.append(mapped_binding)
+    MULTICHAR_KEYBINDINGS = (*MULTICHAR_KEYBINDINGS, *mapped_bindings)
 
 
 class Win:
@@ -83,15 +91,19 @@ def get_keys(self) -> Tuple[int, str]:
 
         for _ in range(MAX_KEYBINDING_LENGTH):
             ch = self.stdscr.getch()
-            log.info("raw ch without unctrl: %s", ch)
-            try:
-                key = curses.unctrl(ch).decode()
-            except Exception:
-                log.warning("cant uncrtl: %s", ch)
-                break
-            if key.isdigit():
-                repeat_factor += key
-                continue
+            if ch == 208 or ch == 209:
+                second_ch = self.stdscr.getch()
+                key = (ch.to_bytes() + second_ch.to_bytes()).decode('utf-8')
+            else:
+                log.info("raw ch without unctrl: %s", ch)
+                try:
+                    key = curses.unctrl(ch).decode()
+                except Exception:
+                    log.warning("cant uncrtl: %s", ch)
+                    break
+                if key.isdigit():
+                    repeat_factor += key
+                    continue
             keys += key
             # if match found or there are not any shortcut matches at all
             if all(