FrameBridge 0.3.1 community source release

source/expanded/framebridge_expanded/native_host.py

@@ -0,0 +1,111 @@
+"""Preserve the captured native protocol while guarding actual X11 dispatch."""
+from __future__ import annotations
+import importlib.util
+import json
+import os
+from pathlib import Path
+import sys
+from . import display_context as display
+
+META = ("input_may_have_been_sent", "do_not_replay", "performed", "display_context")
+
+def _input_primitive(argv):
+    args = list(argv)
+    if len(args) < 2 or Path(str(args[0])).name != "xdotool":
+        return False
+    return args[1] not in {"getactivewindow", "getdisplaygeometry", "getmouselocation",
+                          "getwindowgeometry", "getwindowname", "getwindowpid",
+                          "search", "version", "help"}
+
+def install_cua_guard(cua, *, env=None, checker=None):
+    env = os.environ if env is None else env
+    checker = display.check if checker is None else checker
+    original_run = cua._run
+    original_handle = cua.handle_cua_message
+    state = {"active": False, "input_started": False, "failed_code": ""}
+
+    def verify():
+        try:
+            checker(env)
+        except display.DisplayContextError as exc:
+            state["failed_code"] = str(exc)
+            raise cua.CuaTransportError(str(exc), str(exc)) from None
+
+    def guard_run(run, track_input):
+        def guarded(argv, timeout=15):
+            # The captured pointer helper calls its base runner after an internal
+            # query. Guard that final runner too, immediately before subprocess.
+            verify()
+            if track_input and _input_primitive(argv):
+                state["input_started"] = True
+            return run(argv, timeout=timeout)
+        return guarded
+
+    primitive_run = getattr(cua, "_run_v3_base", None)
+    if primitive_run is not None:
+        cua._run_v3_base = guard_run(primitive_run, True)
+    guarded_run = guard_run(original_run, primitive_run is None)
+
+    def guarded_handle(message):
+        state.update(active=True, input_started=False, failed_code="")
+        verify()
+        result = original_handle(message)
+        verify()
+        return result
+
+    cua._run = guarded_run
+    cua.handle_cua_message = guarded_handle
+    return state
+
+def decorate_response(response, state, env):
+    if not state.get("active") or not isinstance(response, dict):
+        return response
+    response = dict(response)
+    may_have_sent = bool(state.get("input_started"))
+    response["display_context"] = display.describe(env)
+    response["input_may_have_been_sent"] = may_have_sent
+    response["do_not_replay"] = bool(may_have_sent and
+                                    (state.get("failed_code") or not response.get("executed")))
+    if state.get("failed_code"):
+        details = {"ok": False, "error_code": state["failed_code"],
+                   "input_may_have_been_sent": may_have_sent,
+                   "do_not_replay": may_have_sent, "result_discarded": True}
+        if not may_have_sent:
+            details["performed"] = False
+        response.update(executed=False, exit_code=52, output=json.dumps(details),
+                        error=json.dumps(details))
+        if may_have_sent:
+            response.pop("performed", None)
+        else:
+            response["performed"] = False
+    elif response.get("executed"):
+        response["performed"] = may_have_sent
+    elif not may_have_sent:
+        response["performed"] = False
+    state["active"] = False
+    return response
+
+def _load(name, path):
+    spec = importlib.util.spec_from_file_location(name, path)
+    module = importlib.util.module_from_spec(spec)
+    sys.modules[name] = module
+    spec.loader.exec_module(module)
+    return module
+
+def main():
+    prefix = Path(os.environ.get("FRAMEBRIDGE_EXPANDED_PREFIX", Path(__file__).resolve().parents[2]))
+    host = prefix / "original/asus/framebridge-native/host/framebridge_host.py"
+    if not host.is_file():
+        host = prefix / "sources/original-asus/framebridge-native/host/framebridge_host.py"
+    if not host.is_file():
+        raise SystemExit("CAPTURED_NATIVE_HOST_NOT_FOUND")
+    sys.path.insert(0, str(host.parent))
+    cua = _load("framebridge_cua_linux", host.with_name("framebridge_cua_linux.py"))
+    state = install_cua_guard(cua)
+    original = _load("_framebridge_original_native_host", host)
+    send = original.send_message
+    original.send_message = lambda response: send(decorate_response(response, state, os.environ))
+    original.main()
+
+if __name__ == "__main__":
+    main()