114 lines
4.0 KiB
Python
114 lines
4.0 KiB
Python
"""Square tile widget representing a single systemd unit."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
|
|
import gi
|
|
|
|
gi.require_version("Gtk", "4.0")
|
|
gi.require_version("Adw", "1")
|
|
gi.require_version("Gdk", "4.0")
|
|
|
|
from gi.repository import Adw, Gdk, GdkPixbuf, GLib, Gtk
|
|
|
|
from . import systemctl
|
|
|
|
LOADING_STATES = {"reloading", "activating", "deactivating", "maintenance"}
|
|
|
|
ICON_DIR = os.environ.get("SOVRAN_HUB_ICONS", "")
|
|
ICON_EXTENSIONS = [".svg", ".png"]
|
|
|
|
|
|
class ServiceTile(Gtk.Box):
|
|
|
|
def __init__(self, name, unit, scope="system", method="systemctl", icon_name="", **kw):
|
|
super().__init__(
|
|
orientation=Gtk.Orientation.VERTICAL,
|
|
spacing=6,
|
|
halign=Gtk.Align.CENTER,
|
|
valign=Gtk.Align.CENTER,
|
|
width_request=140,
|
|
height_request=160,
|
|
css_classes=["card", "sovran-tile"],
|
|
margin_top=6, margin_bottom=6, margin_start=6, margin_end=6,
|
|
**kw,
|
|
)
|
|
self._unit = unit
|
|
self._scope = scope
|
|
self._method = method
|
|
|
|
self._logo = Gtk.Image(pixel_size=48, margin_top=12, halign=Gtk.Align.CENTER)
|
|
self._set_logo(icon_name)
|
|
self.append(self._logo)
|
|
|
|
self.append(Gtk.Label(
|
|
label=name, css_classes=["heading"],
|
|
halign=Gtk.Align.CENTER, ellipsize=3, max_width_chars=14,
|
|
))
|
|
|
|
self._status_label = Gtk.Label(css_classes=["caption"], halign=Gtk.Align.CENTER)
|
|
self.append(self._status_label)
|
|
|
|
controls = Gtk.Box(
|
|
orientation=Gtk.Orientation.HORIZONTAL,
|
|
spacing=8, halign=Gtk.Align.CENTER, margin_bottom=8,
|
|
)
|
|
self._switch = Gtk.Switch(valign=Gtk.Align.CENTER)
|
|
self._switch.connect("state-set", self._on_toggled)
|
|
controls.append(self._switch)
|
|
|
|
restart_btn = Gtk.Button(
|
|
icon_name="view-refresh-symbolic", valign=Gtk.Align.CENTER,
|
|
tooltip_text="Restart", css_classes=["flat", "circular"],
|
|
)
|
|
restart_btn.connect("clicked", self._on_restart)
|
|
controls.append(restart_btn)
|
|
self.append(controls)
|
|
|
|
self.refresh()
|
|
|
|
def _set_logo(self, icon_name):
|
|
if icon_name and ICON_DIR:
|
|
for ext in ICON_EXTENSIONS:
|
|
path = os.path.join(ICON_DIR, f"{icon_name}{ext}")
|
|
if os.path.isfile(path):
|
|
pixbuf = GdkPixbuf.Pixbuf.new_from_file_at_scale(path, 48, 48, True)
|
|
texture = Gdk.Texture.new_for_pixbuf(pixbuf)
|
|
self._logo.set_from_paintable(texture)
|
|
return
|
|
self._logo.set_from_icon_name("system-run-symbolic")
|
|
|
|
def refresh(self):
|
|
active = systemctl.is_active(self._unit, self._scope)
|
|
enabled = systemctl.is_enabled(self._unit, self._scope)
|
|
is_on = active == "active"
|
|
is_loading = active in LOADING_STATES
|
|
is_failed = active == "failed"
|
|
|
|
self._switch.handler_block_by_func(self._on_toggled)
|
|
self._switch.set_active(is_on)
|
|
self._switch.handler_unblock_by_func(self._on_toggled)
|
|
self._switch.set_sensitive(not is_loading)
|
|
|
|
if is_failed:
|
|
self._status_label.set_label("● failed")
|
|
self._status_label.set_css_classes(["caption", "error"])
|
|
elif is_on:
|
|
self._status_label.set_label("● running")
|
|
self._status_label.set_css_classes(["caption", "success"])
|
|
elif is_loading:
|
|
self._status_label.set_label(f"● {active}")
|
|
self._status_label.set_css_classes(["caption", "warning"])
|
|
else:
|
|
self._status_label.set_label(f"● {active}")
|
|
self._status_label.set_css_classes(["caption", "dim-label"])
|
|
|
|
def _on_toggled(self, switch, state):
|
|
systemctl.run_action("start" if state else "stop", self._unit, self._scope, self._method)
|
|
GLib.timeout_add(1500, self.refresh)
|
|
return False
|
|
|
|
def _on_restart(self, _btn):
|
|
systemctl.run_action("restart", self._unit, self._scope, self._method)
|
|
GLib.timeout_add(1500, self.refresh) |