init

bar/river/widgets.py

@@ -0,0 +1,99 @@
+from fabric.core.service import Property
+from fabric.widgets.button import Button
+from fabric.widgets.box import Box
+from fabric.widgets.eventbox import EventBox
+from fabric.utils.helpers import bulk_connect
+from .service import River
+
+
+from gi.repository import Gdk
+
+_connection: River | None = None
+
+
+def get_river_connection() -> River:
+    global _connection
+    if not _connection:
+        _connection = River()
+    return _connection
+
+
+class RiverWorkspaceButton(Button):
+    @Property(int, "readable")
+    def id(self) -> int:
+        return self._id
+
+    @Property(bool, "read-write", default_value=False)
+    def active(self) -> bool:
+        return self._active
+
+    @active.setter
+    def active(self, value: bool):
+        self._active = value
+        (self.remove_style_class if not value else self.add_style_class)("active")
+
+    @Property(bool, "read-write", default_value=False)
+    def empty(self) -> bool:
+        return self._empty
+
+    @empty.setter
+    def empty(self, value: bool):
+        self._empty = value
+        (self.remove_style_class if not value else self.add_style_class)("empty")
+
+    def __init__(self, id: int, label: str = None, **kwargs):
+        super().__init__(label or str(id), **kwargs)
+        self._id = id
+        self._active = False
+        self._empty = True
+
+
+class RiverWorkspaces(EventBox):
+    def __init__(self, output_id: int, max_tags: int = 9, **kwargs):
+        super().__init__(events="scroll")
+        self.output_id = output_id
+        self.max_tags = max_tags
+        self.service = get_river_connection()
+        self._box = Box(**kwargs)
+        self.children = self._box
+
+        self._buttons = {i: RiverWorkspaceButton(i) for i in range(max_tags)}
+        for btn in self._buttons.values():
+            btn.connect("clicked", self.on_workspace_click)
+            self._box.add(btn)
+
+        # hook into River signals
+        self.service.connect(f"event::focused_tags::{output_id}", self.on_focus_change)
+        self.service.connect(f"event::view_tags::{output_id}", self.on_view_change)
+        if self.service.ready:
+            self.on_ready(None)
+        else:
+            self.service.connect("event::ready", self.on_ready)
+
+        self.connect("scroll-event", self.on_scroll)
+
+    def on_ready(self, _):
+        print(self.service.outputs)
+
+    def on_focus_change(self, _, tags: list[int]):
+        print(tags)
+        for i, btn in self._buttons.items():
+            btn.active = i in tags
+
+    def on_view_change(self, _, tags: list[int]):
+        print(tags)
+        for i, btn in self._buttons.items():
+            btn.empty = i not in tags
+
+    def on_workspace_click(self, btn: RiverWorkspaceButton):
+        import subprocess
+
+        subprocess.run(["riverctl", "tag", str(btn.id)])
+        return
+
+    def on_scroll(self, _, event: Gdk.EventScroll):
+        direction = event.direction  # UP or DOWN
+        cmd = "tag +1" if direction == Gdk.ScrollDirection.DOWN else "tag -1"
+        import subprocess
+
+        subprocess.run(["riverctl", *cmd.split()])