By using this site, you agree to have cookies stored on your device, strictly for functional purposes, such as storing your session and preferences.

Dismiss

 __init__.py

View raw Download
text/plain • 11.32 kiB
Python script, ASCII text executable
        
            
1
"""
2
The MIT License (MIT)
3
4
Copyright (c) 2025 Scott Moreau <oreaus@gmail.com>, modified by <root@roundabout-host.com>
5
6
Permission is hereby granted, free of charge, to any person obtaining a copy
7
of this software and associated documentation files (the "Software"), to deal
8
in the Software without restriction, including without limitation the rights
9
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10
copies of the Software, and to permit persons to whom the Software is
11
furnished to do so, subject to the following conditions:
12
13
The above copyright notice and this permission notice shall be included in
14
all copies or substantial portions of the Software.
15
16
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22
SOFTWARE.
23
"""
24
25
import os
26
from pathlib import Path
27
import locale
28
import panorama_panel
29
30
import gi
31
gi.require_version("Gtk", "4.0")
32
gi.require_version("Gtk4LayerShell", "1.0")
33
34
from gi.repository import Gtk, GLib, Gtk4LayerShell, Gio, Gdk, Pango
35
import subprocess
36
37
38
LOGOUT_BUTTON_SIZE = 128
39
LOGOUT_BUTTON_MARGIN = 12
40
41
def create_logout_ui_button(icon_name, label_text):
42
layout = Gtk.Box()
43
image = Gtk.Image()
44
label = Gtk.Label()
45
button = Gtk.Button()
46
button.set_size_request(LOGOUT_BUTTON_SIZE, LOGOUT_BUTTON_SIZE)
47
image.set_from_icon_name(icon_name)
48
label.set_text(label_text)
49
layout.set_orientation(Gtk.Orientation.VERTICAL)
50
layout.set_halign(Gtk.Align.CENTER)
51
layout.append(image)
52
image.set_icon_size(Gtk.IconSize.LARGE)
53
image.set_vexpand(True)
54
layout.append(label)
55
button.set_child(layout)
56
return button
57
58
class WayfireLogoutUI(Gtk.Window):
59
60
def __init__(self):
61
super().__init__()
62
63
hbox = Gtk.CenterBox()
64
main_layout = Gtk.Grid()
65
suspend = create_logout_ui_button("emblem-synchronizing", "Suspend")
66
suspend.connect("clicked", self.on_suspend_click)
67
main_layout.attach(suspend, 0, 0, 1, 1)
68
69
hibernate = create_logout_ui_button("weather-clear-night", "Hibernate")
70
hibernate.connect("clicked", self.on_hibernate_click)
71
main_layout.attach(hibernate, 1, 0, 1, 1)
72
73
switchuser = create_logout_ui_button("system-users", "Switch User")
74
switchuser.connect("clicked", self.on_switchuser_click)
75
main_layout.attach(switchuser, 2, 0, 1, 1)
76
77
logout = create_logout_ui_button("system-log-out", "Log Out")
78
logout.connect("clicked", self.on_logout_click)
79
main_layout.attach(logout, 0, 1, 1, 1)
80
81
reboot = create_logout_ui_button("system-reboot", "Reboot")
82
reboot.connect("clicked", self.on_reboot_click)
83
main_layout.attach(reboot, 1, 1, 1, 1)
84
85
shutdown = create_logout_ui_button("system-shutdown", "Shut Down")
86
shutdown.connect("clicked", self.on_shutdown_click)
87
main_layout.attach(shutdown, 2, 1, 1, 1)
88
89
cancel_button = Gtk.Button()
90
cancel_button.set_size_request(100, 50)
91
cancel_button.set_label("Cancel")
92
main_layout.attach(cancel_button, 1, 2, 1, 1)
93
cancel_button.connect("clicked", self.on_cancel_click)
94
95
main_layout.set_row_spacing(LOGOUT_BUTTON_MARGIN)
96
main_layout.set_column_spacing(LOGOUT_BUTTON_MARGIN)
97
# Make surfaces layer shell
98
Gtk4LayerShell.init_for_window(self)
99
Gtk4LayerShell.set_namespace(self, "com.roundabout_host.panorama.logout")
100
Gtk4LayerShell.set_layer(self, Gtk4LayerShell.Layer.OVERLAY)
101
102
Gtk4LayerShell.set_anchor(self, Gtk4LayerShell.Edge.TOP, True)
103
Gtk4LayerShell.set_anchor(self, Gtk4LayerShell.Edge.BOTTOM, True)
104
Gtk4LayerShell.set_anchor(self, Gtk4LayerShell.Edge.LEFT, True)
105
Gtk4LayerShell.set_anchor(self, Gtk4LayerShell.Edge.RIGHT, True)
106
main_layout.set_valign(Gtk.Align.CENTER)
107
hbox.set_center_widget(main_layout)
108
hbox.set_hexpand(True)
109
hbox.set_vexpand(True)
110
self.set_child(hbox)
111
self.get_style_context().add_class("logout")
112
display = self.get_display()
113
css_provider = Gtk.CssProvider()
114
css_provider.load_from_data("window.logout { background-color: rgba(0, 0, 0, 0.5); }")
115
Gtk.StyleContext.add_provider_for_display(display, css_provider, Gtk.STYLE_PROVIDER_PRIORITY_USER)
116
117
def on_suspend_click(self, button):
118
GLib.spawn_command_line_async("systemctl suspend")
119
120
def on_hibernate_click(self, button):
121
GLib.spawn_command_line_async("systemctl hibernate")
122
123
def on_switchuser_click(self, button):
124
GLib.spawn_command_line_async("dm-tool switch-to-greeter")
125
126
def on_logout_click(self, button):
127
GLib.spawn_command_line_async("wayland-logout")
128
129
def on_reboot_click(self, button):
130
GLib.spawn_command_line_async("systemctl reboot")
131
132
def on_shutdown_click(self, button):
133
GLib.spawn_command_line_async("systemctl poweroff")
134
135
def on_cancel_click(self, button):
136
self.hide()
137
138
class MenuItemButton(Gtk.Button):
139
140
def __init__(self, name, desc, exe):
141
super().__init__()
142
self.name = name
143
self.desc = desc
144
self.exe = exe
145
146
class SoreausMenu(panorama_panel.Applet):
147
name = "Soreau's menu"
148
description = "Flowbox app menu"
149
150
def __init__(self, orientation=Gtk.Orientation.HORIZONTAL, config=None):
151
super().__init__()
152
if config is None:
153
config = {}
154
155
self.button = Gtk.MenuButton()
156
157
self.menu_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
158
self.image = Gtk.Image.new_from_icon_name(config.get("icon_name", "wayfire"))
159
self.image.set_pixel_size(config.get("icon_size", 24))
160
self.append(self.button)
161
self.button.set_child(self.image)
162
self.popover = Gtk.Popover()
163
self.popover.set_parent(self)
164
self.flowbox = Gtk.FlowBox()
165
self.flowbox.set_valign(Gtk.Align.START)
166
self.flowbox.set_homogeneous(True)
167
self.flowbox_item_focus_signal = self.flowbox.connect("selected-children-changed", self.on_flowbox_item_focus)
168
self.app_buttons = []
169
self.scrolled_window = Gtk.ScrolledWindow()
170
self.scrolled_window.set_size_request(-1, config.get("height", 360))
171
self.scrolled_window.set_child(self.flowbox)
172
self.logout_box = Gtk.Box(orientation=Gtk.Orientation.HORIZONTAL)
173
self.logout_button = Gtk.Button()
174
self.logout_button.set_margin_top(6)
175
self.logout_button.set_margin_bottom(6)
176
self.logout_button.set_margin_start(6)
177
self.logout_button.set_margin_end(6)
178
self.logout_button.set_child(Gtk.Image.new_from_icon_name("system-shutdown"))
179
self.logout_button.connect("clicked", self.on_logout_button_clicked)
180
self.logout_box.set_halign(Gtk.Align.END)
181
self.logout_box.append(self.logout_button)
182
self.search_entry = Gtk.SearchEntry()
183
self.search_entry.connect("search-changed", self.on_search_changed)
184
self.search_entry.connect("activate", self.click_first_button)
185
self.menu_box.append(self.search_entry)
186
self.menu_box.append(self.scrolled_window)
187
self.menu_box.append(self.logout_box)
188
self.popover.set_child(self.menu_box)
189
self.popover.set_size_request(config.get("width", 400), -1)
190
self.popover.connect("show", self.on_popover_popup)
191
self.button.set_popover(self.popover)
192
self.logout_ui = WayfireLogoutUI()
193
self.populate_menu_entries()
194
195
def click_first_button(self, search_entry):
196
flowbox_child = self.flowbox.get_first_child()
197
if flowbox_child is not None:
198
button = flowbox_child.get_first_child()
199
button.emit("clicked")
200
201
def on_search_changed(self, search_entry):
202
self.flowbox.remove_all()
203
self.populate_menu_entries()
204
205
def on_logout_button_clicked(self, button):
206
self.logout_ui.present()
207
self.popover.popdown()
208
209
def on_flowbox_item_focus(self, flowbox):
210
selected_children = flowbox.get_selected_children()
211
if len(selected_children) >= 1:
212
selected_children[0].get_child().grab_focus()
213
214
def get_config(self):
215
return {
216
"width": self.popover.get_size_request()[0],
217
"height": self.scrolled_window.get_size_request()[1],
218
"icon_name": self.image.get_icon_name(),
219
"icon_size": self.image.get_pixel_size(),
220
}
221
222
def on_popover_popup(self, parent):
223
for child in self.flowbox.get_selected_children():
224
self.flowbox.unselect_child(child)
225
self.search_entry.set_text("")
226
self.popover.popup()
227
228
def app_button_clicked(self, app_button):
229
subprocess.Popen(app_button.command, start_new_session=True)
230
self.popover.popdown()
231
232
def populate_menu_entries(self):
233
if self.search_entry.get_text():
234
desktop_names = Gio.DesktopAppInfo.search(self.search_entry.get_text())
235
236
app_infos = []
237
238
for sublist in desktop_names:
239
for desktop_name in sublist:
240
try:
241
app_infos.append(Gio.DesktopAppInfo.new(desktop_name))
242
except TypeError:
243
# The constructor may return a C NULL if the file is unable to be launched,
244
# so skip the app
245
pass
246
else:
247
app_infos = Gio.AppInfo.get_all()
248
app_infos.sort(key=lambda app_info: app_info.get_display_name())
249
250
for app_info in app_infos:
251
app_categories = app_info.get_categories()
252
if app_categories == None:
253
continue
254
app_name = app_info.get_display_name()
255
command = app_info.get_executable()
256
app_button = MenuItemButton(app_name, app_info.get_description(), command)
257
app_button.command = command
258
app_button.set_tooltip_text(app_name)
259
app_label = Gtk.Label(label=app_name)
260
app_label.set_ellipsize(Pango.EllipsizeMode.END)
261
app_label.set_max_width_chars(7)
262
app_button_box = Gtk.Box(orientation=Gtk.Orientation.VERTICAL)
263
app_button_box.append(app_label)
264
app_button.set_child(app_button_box)
265
app_button.set_has_frame(False)
266
267
image = Gtk.Image()
268
image.set_icon_size(Gtk.IconSize.LARGE)
269
icon = app_info.get_icon()
270
if icon:
271
icon_name = icon.to_string()
272
if not Gtk.IconTheme.get_for_display(Gdk.Display.get_default()).has_icon(icon_name):
273
continue
274
if icon_name[0] == '/':
275
image.set_from_file(icon_name)
276
else:
277
image.set_from_icon_name(icon_name)
278
else:
279
if not Gtk.IconTheme.get_for_display(Gdk.Display.get_default()).has_icon(app_name.lower()):
280
continue
281
image.set_from_icon_name(app_name.lower())
282
283
self.flowbox.append(app_button)
284
app_button.connect("clicked", self.app_button_clicked)
285
app_button_box.prepend(image)
286
self.app_buttons.append(app_button)
287