Merge pull request #6 from caelestia-dots/python-rework

feat: rewrite in python

1 files changed, 29 insertions, 0 deletions

diff --git a/src/caelestia/utils/hypr.py b/src/caelestia/utils/hypr.py
new file mode 100644
index 0000000..f89cd98
--- /dev/null
+++ b/src/caelestia/utils/hypr.py
@@ -0,0 +1,29 @@
+import json as j
+import os
+import socket
+
+socket_base = f"{os.getenv('XDG_RUNTIME_DIR')}/hypr/{os.getenv('HYPRLAND_INSTANCE_SIGNATURE')}"
+socket_path = f"{socket_base}/.socket.sock"
+socket2_path = f"{socket_base}/.socket2.sock"
+
+
+def message(msg: str, json: bool = True) -> str | dict[str, any]:
+    with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as sock:
+        sock.connect(socket_path)
+
+        if json:
+            msg = f"j/{msg}"
+        sock.send(msg.encode())
+
+        resp = sock.recv(8192).decode()
+        while True:
+            new_resp = sock.recv(8192)
+            if not new_resp:
+                break
+            resp += new_resp.decode()
+
+        return j.loads(resp) if json else resp
+
+
+def dispatch(dispatcher: str, *args: list[any]) -> bool:
+    return message(f"dispatch {dispatcher} {' '.join(map(str, args))}".rstrip(), json=False) == "ok"