summaryrefslogtreecommitdiff
path: root/kvmd/plugins/hid/ch9329/keyboard.py
diff options
context:
space:
mode:
Diffstat (limited to 'kvmd/plugins/hid/ch9329/keyboard.py')
-rw-r--r--kvmd/plugins/hid/ch9329/keyboard.py71
1 files changed, 71 insertions, 0 deletions
diff --git a/kvmd/plugins/hid/ch9329/keyboard.py b/kvmd/plugins/hid/ch9329/keyboard.py
new file mode 100644
index 00000000..70ccf547
--- /dev/null
+++ b/kvmd/plugins/hid/ch9329/keyboard.py
@@ -0,0 +1,71 @@
+# ========================================================================== #
+# #
+# KVMD - The main PiKVM daemon. #
+# #
+# Copyright (C) 2018-2022 Maxim Devaev <[email protected]> #
+# #
+# 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 <https://www.gnu.org/licenses/>. #
+# #
+# ========================================================================== #
+
+from .... import aiomulti
+
+from ....keyboard.mappings import KEYMAP
+
+
+class Keyboard:
+ def __init__(self) -> None:
+
+ self.__notifier = aiomulti.AioProcessNotifier()
+ self.__leds = aiomulti.AioSharedFlags({
+ "num": False,
+ "caps": False,
+ "scroll": False,
+ }, self.__notifier, type=bool)
+
+ self.__active_keys: list[list] = []
+
+ def key(self, key: str, state: bool) -> list[int]:
+ if state:
+ self.__active_keys.append([key, self.__is_modifier(key)])
+ else:
+ self.__active_keys.remove([key, self.__is_modifier(key)])
+ return self.__key()
+
+ async def leds(self) -> dict:
+ leds = await self.__leds.get()
+ return leds
+
+ def set_leds(self, led_byte: int) -> None:
+ num = bool(led_byte & 1)
+ caps = bool((led_byte >> 1) & 1)
+ scroll = bool((led_byte >> 2) & 1)
+ self.__leds.update(num=num, caps=caps, scroll=scroll)
+
+ def __key(self) -> list[int]:
+ cmd = [0x00, 0x02, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00]
+ counter = 0
+ for key in self.__active_keys:
+ if key[1]:
+ cmd[3 + counter] = self.__keycode(key[0])
+ else:
+ cmd[5 + counter] = self.__keycode(key[0])
+ counter += 1
+ return cmd
+
+ def __keycode(self, key: str) -> int:
+ return KEYMAP[key].usb.code
+
+ def __is_modifier(self, key: str) -> bool:
+ return KEYMAP[key].usb.is_modifier