1 files changed, 93 insertions, 0 deletions
diff --git a/modules/desktop/eww/workspaces.py b/modules/desktop/eww/workspaces.py
new file mode 100644
index 0000000..b8fae1b
--- /dev/null
+++ b/modules/desktop/eww/workspaces.py
@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+"""workspaces: emit workspace list with each workspace's focused-app icon,
+driven by niri's event-stream instead of polling."""
+import json
+import subprocess
+import gi
+from gi.repository import Gio, Gtk
+gi.require_version("Gtk", "3.0")
+
+icon_theme = Gtk.IconTheme.get_default()
+_icon_cache: dict[str, str] = {}
+
+
+def resolve_icon_path(app_id: str) -> str:
+    if not app_id:
+        return ""
+    if app_id in _icon_cache:
+        return _icon_cache[app_id]
+
+    app_info = Gio.DesktopAppInfo.new(f"{app_id}.desktop") \
+        or Gio.DesktopAppInfo.new(f"{app_id.lower()}.desktop")
+
+    if app_info is None:
+        for info in Gio.AppInfo.get_all():
+            if isinstance(info, Gio.DesktopAppInfo):
+                wm_class = info.get_startup_wm_class()
+                if wm_class and wm_class.lower() == app_id.lower():
+                    app_info = info
+                    break
+
+    icon_path = ""
+    gicon = app_info.get_icon() if app_info else None
+    if gicon is not None:
+        lookup = icon_theme.lookup_by_gicon(gicon, 24, Gtk.IconLookupFlags.FORCE_SIZE)
+        if lookup:
+            icon_path = lookup.get_filename() or ""
+    if not icon_path:
+        lookup = icon_theme.lookup_icon(app_id, 24, Gtk.IconLookupFlags.FORCE_SIZE)
+        if lookup:
+            icon_path = lookup.get_filename() or ""
+
+    _icon_cache[app_id] = icon_path
+    return icon_path
+
+
+def niri_json(*args):
+    out = subprocess.run(["niri", "msg", "-j", *args], capture_output=True, text=True, check=True)
+    return json.loads(out.stdout)
+
+
+def emit_workspaces():
+    workspaces = niri_json("workspaces")
+    windows = {w["id"]: w for w in niri_json("windows")}
+
+    result = []
+    for ws in sorted(workspaces, key=lambda w: w["idx"]):
+        active_win = windows.get(ws.get("active_window_id"))
+        app_id = (active_win or {}).get("app_id") or ""
+        result.append({
+            "idx": ws["idx"],
+            "id": ws["id"],
+            "is_active": ws.get("is_active", False),
+            "icon": resolve_icon_path(app_id),
+        })
+    print(json.dumps(result), flush=True)
+
+
+RELEVANT_EVENTS = {
+    "WorkspacesChanged", "WorkspaceActivated", "WorkspaceUrgencyChanged",
+    "WorkspaceActiveWindowChanged", "WindowOpenedOrChanged", "WindowClosed",
+}
+
+
+def main():
+    emit_workspaces()
+    proc = subprocess.Popen(
+        ["niri", "msg", "-j", "event-stream"],
+        stdout=subprocess.PIPE, text=True, bufsize=1,
+    )
+    for line in proc.stdout:
+        try:
+            key = next(iter(json.loads(line)))
+        except (json.JSONDecodeError, StopIteration):
+            continue
+        if key in RELEVANT_EVENTS:
+            try:
+                emit_workspaces()
+            except subprocess.CalledProcessError:
+                pass
+
+
+if __name__ == "__main__":
+    main()