89 lines
2.8 KiB
Python
89 lines
2.8 KiB
Python
import operator
|
|
from fabric.widgets.wayland import WaylandWindow as Window
|
|
from fabric.widgets.box import Box
|
|
from fabric.widgets.label import Label
|
|
from fabric.widgets.entry import Entry
|
|
from gi.repository import Gdk
|
|
from bar.services.wlr.service import WaylandWindowTracker, Window as WaylandWindow
|
|
from pywayland.client import Display
|
|
|
|
|
|
class FuzzyWindowFinder(Window):
|
|
def __init__(
|
|
self,
|
|
monitor: int = 0,
|
|
):
|
|
super().__init__(
|
|
name="finder",
|
|
anchor="center",
|
|
monitor=monitor,
|
|
keyboard_mode="on-demand",
|
|
type="popup",
|
|
visible=False,
|
|
)
|
|
self.window_tracker = WaylandWindowTracker()
|
|
self.window_tracker.ready_signal.connect(lambda *_: print("Tracker is ready"))
|
|
self._all_windows: list[WaylandWindow] = []
|
|
|
|
self.viewport = Box(name="viewport", spacing=4, orientation="v")
|
|
|
|
self.search_entry = Entry(
|
|
name="search-entry",
|
|
placeholder="Search Windows...",
|
|
h_expand=True,
|
|
editable=True,
|
|
notify_text=self.notify_text,
|
|
on_activate=lambda entry, *_: self.on_search_entry_activate(
|
|
entry.get_text()
|
|
),
|
|
on_key_press_event=self.on_search_entry_key_press,
|
|
)
|
|
self.picker_box = Box(
|
|
name="picker-box",
|
|
spacing=4,
|
|
orientation="v",
|
|
children=[self.search_entry, self.viewport],
|
|
)
|
|
|
|
self.add(self.picker_box)
|
|
self.arrange_viewport("")
|
|
|
|
def open(self):
|
|
self._all_windows = self.window_tracker.windows
|
|
print(self._all_windows[0])
|
|
self.arrange_viewport("")
|
|
self.show()
|
|
|
|
def notify_text(self, entry, *_):
|
|
text = entry.get_text()
|
|
self.arrange_viewport(text) # Update list on typing
|
|
print(text)
|
|
|
|
def on_search_entry_key_press(self, widget, event):
|
|
# if event.keyval in (Gdk.KEY_Up, Gdk.KEY_Down, Gdk.KEY_Left, Gdk.KEY_Right):
|
|
# self.move_selection_2d(event.keyval)
|
|
# return True
|
|
print(event.keyval)
|
|
if event.keyval == Gdk.KEY_Return:
|
|
self.window_tracker.activate_window(self._filtered[0])
|
|
if event.keyval in [Gdk.KEY_Escape, 103]:
|
|
self.hide()
|
|
return True
|
|
return False
|
|
|
|
def on_search_entry_activate(self, text):
|
|
print(f"activate {text}")
|
|
|
|
def arrange_viewport(self, query: str = ""):
|
|
self.viewport.children = [] # Clear previous entries
|
|
|
|
self._filtered = [
|
|
w for w in self._all_windows if query.lower() in w.title.lower()
|
|
]
|
|
titles = [w.title for w in self._filtered]
|
|
|
|
for window in titles:
|
|
self.viewport.add(
|
|
Box(name="slot-box", orientation="h", children=[Label(label=window)])
|
|
)
|