minor niri updates

configs/niri/config.kdl

@@ -16,6 +16,11 @@ window-rule {
   match app-id="^brave-.*-Default$"
  // Extensions popup  open-floating true
 }
+// Full width window
+window-rule {
+  match app-id="beekeeper-studio"
+  open-maximized true
+}
 window-rule {
   match app-id="^org\\.keepassxc\\.KeePassXC$"
   match app-id="^org\\.gnome\\.World\\.Secrets$"
@@ -51,6 +56,7 @@ layer-rule {
 // ╭─────────────────────────────────────────────────────────╮
 // │                     Auto Start Apps                     │
 // ╰─────────────────────────────────────────────────────────╯
+spawn-at-startup "./fix_float.py"
 spawn-at-startup "/usr/lib/pam_kwallet_init"
 spawn-at-startup "udieskie"
 spawn-at-startup "brave"

configs/niri/fix_float.py

@@ -0,0 +1,124 @@
+#!/usr/bin/python3
+"""
+Like open-float, but dynamically. Floats a window when it matches the rules.
+
+Some windows don't have the right title and app-id when they open, and only set
+them afterward. This script is like open-float for those windows.
+
+Usage: fill in the RULES array below, then run the script.
+"""
+
+import json
+import os
+import re
+from dataclasses import dataclass, field
+from socket import AF_UNIX, SHUT_WR, socket
+
+
+@dataclass(kw_only=True)
+class Match:
+    title: str | None = None
+    app_id: str | None = None
+
+    def matches(self, window):
+        if self.title is None and self.app_id is None:
+            return False
+
+        matched = True
+
+        if self.title is not None:
+            matched &= re.search(self.title, window["title"]) is not None
+        if self.app_id is not None:
+            matched &= re.search(self.app_id, window["app_id"]) is not None
+
+        return matched
+
+
+@dataclass
+class Rule:
+    match: list[Match] = field(default_factory=list)
+    exclude: list[Match] = field(default_factory=list)
+
+    def matches(self, window):
+        if len(self.match) > 0 and not any(m.matches(window) for m in self.match):
+            return False
+        if any(m.matches(window) for m in self.exclude):
+            return False
+
+        return True
+
+
+# Write your rules here. One Rule() = one window-rule {}.
+RULES = [
+    # window-rule {} with one match.
+    Rule([Match(title="Bitwarden", app_id="^brave-.*-Default$")]),
+    # window-rule {} with one match and one exclude.
+    # Rule(
+    #     [Match(title="rs")],
+    #     exclude=[Match(app_id="Alacritty")],
+    # ),
+    # window-rule {} with two matches.
+    # Rule(
+    #     [
+    #         Match(app_id="^foot$"),
+    #         Match(app_id="^mpv$"),
+    #     ]
+    # ),
+]
+
+
+if len(RULES) == 0:
+    print("fill in the RULES list, then run the script")
+    exit()
+
+
+niri_socket = socket(AF_UNIX)
+niri_socket.connect(os.environ["NIRI_SOCKET"])
+file = niri_socket.makefile("rw")
+
+_ = file.write('"EventStream"')
+file.flush()
+niri_socket.shutdown(SHUT_WR)
+
+windows = {}
+
+
+def send(request):
+    with socket(AF_UNIX) as niri_socket:
+        niri_socket.connect(os.environ["NIRI_SOCKET"])
+        file = niri_socket.makefile("rw")
+        _ = file.write(json.dumps(request))
+        file.flush()
+
+
+def float(id: int):
+    send({"Action": {"MoveWindowToFloating": {"id": id}}})
+    send({"Action": {"SetWindowWidth": {"id": id, "change": {"SetProportion": 20}}}})
+    send({"Action": {"SetWindowHeight": {"id": id, "change": {"SetProportion": 50}}}})
+
+
+def update_matched(win):
+    win["matched"] = False
+    if existing := windows.get(win["id"]):
+        win["matched"] = existing["matched"]
+
+    matched_before = win["matched"]
+    win["matched"] = any(r.matches(win) for r in RULES)
+    if win["matched"] and not matched_before:
+        print(f"floating title={win['title']}, app_id={win['app_id']}")
+        float(win["id"])
+
+
+for line in file:
+    event = json.loads(line)
+
+    if changed := event.get("WindowsChanged"):
+        for win in changed["windows"]:
+            update_matched(win)
+        windows = {win["id"]: win for win in changed["windows"]}
+    elif changed := event.get("WindowOpenedOrChanged"):
+        win = changed["window"]
+        update_matched(win)
+        windows[win["id"]] = win
+    elif changed := event.get("WindowClosed"):
+        del windows[changed["id"]]