__init__.py
Python script, ASCII text executable
1import os 2from pathlib import Path 3from pywayland.client import Display 4from pywayland.protocol.wayland import WlRegistry 5from pywayland.protocol.wlr_foreign_toplevel_management_unstable_v1 import ( 6ZwlrForeignToplevelManagerV1, 7ZwlrForeignToplevelHandleV1 8) 9import panorama_panel 10 11import gi 12 13gi.require_version("Gtk", "4.0") 14 15from gi.repository import Gtk, GLib, Gio, Gdk 16 17 18module_directory = Path(__file__).resolve().parent 19 20 21class WindowButton(Gtk.ToggleButton): 22def __init__(self, window_id, window_title, **kwargs): 23super().__init__(**kwargs) 24 25self.window_id = window_id 26self.window_title = window_title 27self.set_has_frame(False) 28 29self.set_label(self.window_title) 30 31self.last_state = False 32 33 34class WFWindowList(panorama_panel.Applet): 35name = "Wayfire window list" 36description = "Traditional window list (for Wayfire)" 37 38def __init__(self, orientation=Gtk.Orientation.HORIZONTAL, config=None): 39super().__init__(orientation=orientation, config=config) 40if config is None: 41config = {} 42 43self.toplevel_buttons: dict[ZwlrForeignToplevelHandleV1, WindowButton] = {} 44 45self.display = Display() 46self.display.connect() 47self.registry = self.display.get_registry() 48self.registry.dispatcher["global"] = self.on_global 49self.display.roundtrip() 50fd = self.display.get_fd() 51GLib.io_add_watch(fd, GLib.IO_IN, self.on_display_event) 52 53self.initial_button = Gtk.ToggleButton() 54 55self.context_menu = self.make_context_menu() 56panorama_panel.track_popover(self.context_menu) 57 58right_click_controller = Gtk.GestureClick() 59right_click_controller.set_button(3) 60right_click_controller.connect("pressed", self.show_context_menu) 61 62self.add_controller(right_click_controller) 63 64action_group = Gio.SimpleActionGroup() 65options_action = Gio.SimpleAction.new("options", None) 66options_action.connect("activate", self.show_options) 67action_group.add_action(options_action) 68self.insert_action_group("applet", action_group) 69 70self.options_window = None 71 72def on_display_event(self, source, condition): 73if condition == GLib.IO_IN: 74self.display.dispatch() 75return True 76 77def on_global(self, registry, name, interface, version): 78print(f"Global: {interface} (v{version})") 79if interface == "zwlr_foreign_toplevel_manager_v1": 80print("Interface registered") 81self.manager = registry.bind(name, ZwlrForeignToplevelManagerV1, version) 82self.manager.dispatcher["toplevel"] = self.on_new_toplevel 83self.manager.dispatcher["finished"] = lambda *a: print("Toplevel manager finished") 84self.display.roundtrip() 85self.display.flush() 86 87def on_new_toplevel(self, manager: ZwlrForeignToplevelManagerV1, 88handle: ZwlrForeignToplevelHandleV1): 89print("Toplevel received") 90handle.dispatcher["title"] = lambda h, title: self.on_title_changed(h, title) 91#handle.dispatcher["app_id"] = lambda h, app_id: self.on_app_id_changed(h, app_id) 92#handle.dispatcher["state"] = lambda h, states: self.on_state_changed(h, states) 93handle.dispatcher["closed"] = lambda h: self.on_closed(h) 94 95def on_title_changed(self, handle, title): 96print(f"Window title: {title}") 97if handle not in self.toplevel_buttons: 98button = WindowButton(id(handle), title) 99button.connect("clicked", lambda *a: self.on_button_click(handle)) 100self.toplevel_buttons[handle] = button 101self.append(button) 102 103def make_context_menu(self): 104menu = Gio.Menu() 105menu.append("Window list _options", "applet.options") 106context_menu = Gtk.PopoverMenu.new_from_model(menu) 107context_menu.set_has_arrow(False) 108context_menu.set_parent(self) 109context_menu.set_halign(Gtk.Align.START) 110context_menu.set_flags(Gtk.PopoverMenuFlags.NESTED) 111return context_menu 112 113def show_context_menu(self, gesture, n_presses, x, y): 114rect = Gdk.Rectangle() 115rect.x = int(x) 116rect.y = int(y) 117rect.width = 1 118rect.height = 1 119 120self.context_menu.set_pointing_to(rect) 121self.context_menu.popup() 122 123def show_options(self, _0=None, _1=None): 124pass 125 126def get_config(self): 127return {} 128