Skip to content

Commit

Permalink
refactoring
Browse files Browse the repository at this point in the history
  • Loading branch information
mdevaev committed Nov 2, 2024
1 parent 8192b1f commit d6b61cb
Show file tree
Hide file tree
Showing 10 changed files with 251 additions and 227 deletions.
16 changes: 1 addition & 15 deletions kvmd/apps/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,8 +33,6 @@

from .. import tools

from ..mouse import MouseRange

from ..plugins import UnknownPluginError
from ..plugins.auth import get_auth_service_class
from ..plugins.hid import get_hid_class
Expand Down Expand Up @@ -407,19 +405,7 @@ def _get_config_scheme() -> dict:

"hid": {
"type": Option("", type=valid_stripped_string_not_empty),

"keymap": Option("/usr/share/kvmd/keymaps/en-us", type=valid_abs_file),
"ignore_keys": Option([], type=functools.partial(valid_string_list, subval=valid_hid_key)),

"mouse_x_range": {
"min": Option(MouseRange.MIN, type=valid_hid_mouse_move),
"max": Option(MouseRange.MAX, type=valid_hid_mouse_move),
},
"mouse_y_range": {
"min": Option(MouseRange.MIN, type=valid_hid_mouse_move),
"max": Option(MouseRange.MAX, type=valid_hid_mouse_move),
},

"keymap": Option("/usr/share/kvmd/keymaps/en-us", type=valid_abs_file),
# Dynamic content
},

Expand Down
5 changes: 1 addition & 4 deletions kvmd/apps/kvmd/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ def main(argv: (list[str] | None)=None) -> None:
if config.kvmd.msd.type == "otg":
msd_kwargs["gadget"] = config.otg.gadget # XXX: Small crutch to pass gadget name to the plugin

hid_kwargs = config.kvmd.hid._unpack(ignore=["type", "keymap", "ignore_keys", "mouse_x_range", "mouse_y_range"])
hid_kwargs = config.kvmd.hid._unpack(ignore=["type", "keymap"])
if config.kvmd.hid.type == "otg":
hid_kwargs["udc"] = config.otg.udc # XXX: Small crutch to pass UDC to the plugin

Expand Down Expand Up @@ -103,9 +103,6 @@ def main(argv: (list[str] | None)=None) -> None:
),

keymap_path=config.hid.keymap,
ignore_keys=config.hid.ignore_keys,
mouse_x_range=(config.hid.mouse_x_range.min, config.hid.mouse_x_range.max),
mouse_y_range=(config.hid.mouse_y_range.min, config.hid.mouse_y_range.max),

stream_forever=config.streamer.forever,
).run(**config.server._unpack())
Expand Down
84 changes: 20 additions & 64 deletions kvmd/apps/kvmd/api/hid.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,13 +25,12 @@
import functools
import struct

from typing import Iterable
from typing import Callable

from aiohttp.web import Request
from aiohttp.web import Response

from ....mouse import MouseRange

from ....keyboard.keysym import build_symmap
from ....keyboard.printer import text_to_web_keys

Expand Down Expand Up @@ -59,12 +58,7 @@ class HidApi:
def __init__(
self,
hid: BaseHid,

keymap_path: str,
ignore_keys: list[str],

mouse_x_range: tuple[int, int],
mouse_y_range: tuple[int, int],
) -> None:

self.__hid = hid
Expand All @@ -73,11 +67,6 @@ def __init__(
self.__default_keymap_name = os.path.basename(keymap_path)
self.__ensure_symmap(self.__default_keymap_name)

self.__ignore_keys = ignore_keys

self.__mouse_x_range = mouse_x_range
self.__mouse_y_range = mouse_y_range

# =====

@exposed_http("GET", "/hid")
Expand Down Expand Up @@ -134,7 +123,7 @@ async def __print_handler(self, req: Request) -> Response:
if limit > 0:
text = text[:limit]
symmap = self.__ensure_symmap(req.query.get("keymap", self.__default_keymap_name))
self.__hid.send_key_events(text_to_web_keys(text, symmap))
self.__hid.send_key_events(text_to_web_keys(text, symmap), no_ignore_keys=True)
return make_json_response()

def __ensure_symmap(self, keymap_name: str) -> dict[int, dict[int, str]]:
Expand Down Expand Up @@ -162,8 +151,7 @@ async def __ws_bin_key_handler(self, _: WsSession, data: bytes) -> None:
state = valid_bool(data[0])
except Exception:
return
if key not in self.__ignore_keys:
self.__hid.send_key_events([(key, state)])
self.__hid.send_key_event(key, state)

@exposed_ws(2)
async def __ws_bin_mouse_button_handler(self, _: WsSession, data: bytes) -> None:
Expand All @@ -182,17 +170,17 @@ async def __ws_bin_mouse_move_handler(self, _: WsSession, data: bytes) -> None:
to_y = valid_hid_mouse_move(to_y)
except Exception:
return
self.__send_mouse_move_event(to_x, to_y)
self.__hid.send_mouse_move_event(to_x, to_y)

@exposed_ws(4)
async def __ws_bin_mouse_relative_handler(self, _: WsSession, data: bytes) -> None:
self.__process_ws_bin_delta_request(data, self.__hid.send_mouse_relative_event)
self.__process_ws_bin_delta_request(data, self.__hid.send_mouse_relative_events)

@exposed_ws(5)
async def __ws_bin_mouse_wheel_handler(self, _: WsSession, data: bytes) -> None:
self.__process_ws_bin_delta_request(data, self.__hid.send_mouse_wheel_event)
self.__process_ws_bin_delta_request(data, self.__hid.send_mouse_wheel_events)

def __process_ws_bin_delta_request(self, data: bytes, handler: Callable[[int, int], None]) -> None:
def __process_ws_bin_delta_request(self, data: bytes, handler: Callable[[Iterable[tuple[int, int]], bool], None]) -> None:
try:
squash = valid_bool(data[0])
data = data[1:]
Expand All @@ -202,7 +190,7 @@ def __process_ws_bin_delta_request(self, data: bytes, handler: Callable[[int, in
deltas.append((valid_hid_mouse_delta(delta_x), valid_hid_mouse_delta(delta_y)))
except Exception:
return
self.__send_mouse_delta_event(deltas, squash, handler)
handler(deltas, squash)

# =====

Expand All @@ -213,8 +201,7 @@ async def __ws_key_handler(self, _: WsSession, event: dict) -> None:
state = valid_bool(event["state"])
except Exception:
return
if key not in self.__ignore_keys:
self.__hid.send_key_events([(key, state)])
self.__hid.send_key_event(key, state)

@exposed_ws("mouse_button")
async def __ws_mouse_button_handler(self, _: WsSession, event: dict) -> None:
Expand All @@ -232,17 +219,17 @@ async def __ws_mouse_move_handler(self, _: WsSession, event: dict) -> None:
to_y = valid_hid_mouse_move(event["to"]["y"])
except Exception:
return
self.__send_mouse_move_event(to_x, to_y)
self.__hid.send_mouse_move_event(to_x, to_y)

@exposed_ws("mouse_relative")
async def __ws_mouse_relative_handler(self, _: WsSession, event: dict) -> None:
self.__process_ws_delta_event(event, self.__hid.send_mouse_relative_event)
self.__process_ws_delta_event(event, self.__hid.send_mouse_relative_events)

@exposed_ws("mouse_wheel")
async def __ws_mouse_wheel_handler(self, _: WsSession, event: dict) -> None:
self.__process_ws_delta_event(event, self.__hid.send_mouse_wheel_event)
self.__process_ws_delta_event(event, self.__hid.send_mouse_wheel_events)

def __process_ws_delta_event(self, event: dict, handler: Callable[[int, int], None]) -> None:
def __process_ws_delta_event(self, event: dict, handler: Callable[[Iterable[tuple[int, int]], bool], None]) -> None:
try:
raw_delta = event["delta"]
deltas = [
Expand All @@ -252,19 +239,18 @@ def __process_ws_delta_event(self, event: dict, handler: Callable[[int, int], No
squash = valid_bool(event.get("squash", False))
except Exception:
return
self.__send_mouse_delta_event(deltas, squash, handler)
handler(deltas, squash)

# =====

@exposed_http("POST", "/hid/events/send_key")
async def __events_send_key_handler(self, req: Request) -> Response:
key = valid_hid_key(req.query.get("key"))
if key not in self.__ignore_keys:
if "state" in req.query:
state = valid_bool(req.query["state"])
self.__hid.send_key_events([(key, state)])
else:
self.__hid.send_key_events([(key, True), (key, False)])
if "state" in req.query:
state = valid_bool(req.query["state"])
self.__hid.send_key_event(key, state)
else:
self.__hid.send_key_events([(key, True), (key, False)])
return make_json_response()

@exposed_http("POST", "/hid/events/send_mouse_button")
Expand All @@ -282,7 +268,7 @@ async def __events_send_mouse_button_handler(self, req: Request) -> Response:
async def __events_send_mouse_move_handler(self, req: Request) -> Response:
to_x = valid_hid_mouse_move(req.query.get("to_x"))
to_y = valid_hid_mouse_move(req.query.get("to_y"))
self.__send_mouse_move_event(to_x, to_y)
self.__hid.send_mouse_move_event(to_x, to_y)
return make_json_response()

@exposed_http("POST", "/hid/events/send_mouse_relative")
Expand All @@ -298,33 +284,3 @@ def __process_http_delta_event(self, req: Request, handler: Callable[[int, int],
delta_y = valid_hid_mouse_delta(req.query.get("delta_y"))
handler(delta_x, delta_y)
return make_json_response()

# =====

def __send_mouse_move_event(self, to_x: int, to_y: int) -> None:
if self.__mouse_x_range != MouseRange.RANGE:
to_x = MouseRange.remap(to_x, *self.__mouse_x_range)
if self.__mouse_y_range != MouseRange.RANGE:
to_y = MouseRange.remap(to_y, *self.__mouse_y_range)
self.__hid.send_mouse_move_event(to_x, to_y)

def __send_mouse_delta_event(
self,
deltas: list[tuple[int, int]],
squash: bool,
handler: Callable[[int, int], None],
) -> None:

if squash:
prev = (0, 0)
for cur in deltas:
if abs(prev[0] + cur[0]) > 127 or abs(prev[1] + cur[1]) > 127:
handler(*prev)
prev = cur
else:
prev = (prev[0] + cur[0], prev[1] + cur[1])
if prev[0] or prev[1]:
handler(*prev)
else:
for xy in deltas:
handler(*xy)
5 changes: 1 addition & 4 deletions kvmd/apps/kvmd/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -173,9 +173,6 @@ def __init__( # pylint: disable=too-many-arguments,too-many-locals
snapshoter: Snapshoter,

keymap_path: str,
ignore_keys: list[str],
mouse_x_range: tuple[int, int],
mouse_y_range: tuple[int, int],

stream_forever: bool,
) -> None:
Expand All @@ -189,7 +186,7 @@ def __init__( # pylint: disable=too-many-arguments,too-many-locals

self.__stream_forever = stream_forever

self.__hid_api = HidApi(hid, keymap_path, ignore_keys, mouse_x_range, mouse_y_range) # Ugly hack to get keymaps state
self.__hid_api = HidApi(hid, keymap_path) # Ugly hack to get keymaps state
self.__apis: list[object] = [
self,
AuthApi(auth_manager),
Expand Down
Loading

0 comments on commit d6b61cb

Please sign in to comment.