Sovran_SystemsOS/app/sovran_systemsos_hub/service_tile.py

"""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 Gdk, GdkPixbuf, GLib, Gtk, Pango

from . import systemctl

LOADING_STATES = {"reloading", "activating", "deactivating", "maintenance"}

ICON_DIR = os.environ.get("SOVRAN_HUB_ICONS", "")
ICON_EXTENSIONS = [".svg", ".png"]

# ── Locked tile dimensions ───────────────────────────────────────
TILE_W = 180
TILE_H = 210
ICON_PX = 48
LABEL_W = TILE_W - 24   # 12px padding each side


class ServiceTile(Gtk.Box):

    def __init__(self, name, unit, scope="system", method="systemctl",
                 icon_name="", enabled=True, **kw):
        super().__init__(
            orientation=Gtk.Orientation.VERTICAL,
            spacing=2,
            halign=Gtk.Align.CENTER,
            valign=Gtk.Align.START,
            css_classes=["card", "sovran-tile"],
            **kw,
        )
        self.set_size_request(TILE_W, TILE_H)
        self.set_hexpand(False)
        self.set_vexpand(False)

        self._unit = unit
        self._scope = scope
        self._method = method
        self._enabled = enabled

        # ── Icon ─────────────────────────────────────────────────
        self._logo = Gtk.Image(
            pixel_size=ICON_PX,
            margin_top=18,
            halign=Gtk.Align.CENTER,
        )
        self._set_logo(icon_name)
        self.append(self._logo)

        # ── Name label ───────────────────────────────────────────
        self._name_label = Gtk.Label(
            label=name,
            css_classes=["tile-name"],
            halign=Gtk.Align.CENTER,
            justify=Gtk.Justification.CENTER,
            wrap=True,
            wrap_mode=Pango.WrapMode.WORD_CHAR,
            lines=2,
            ellipsize=Pango.EllipsizeMode.END,
            margin_start=12,
            margin_end=12,
            margin_top=6,
        )
        self._name_label.set_size_request(LABEL_W, -1)
        self._name_label.set_max_width_chars(1)
        self.append(self._name_label)

        # ── Status label ─────────────────────────────────────────
        self._status_label = Gtk.Label(
            label="● …",
            css_classes=["caption", "tile-status", "dim-label"],
            halign=Gtk.Align.CENTER,
            margin_top=2,
        )
        self.append(self._status_label)

        # ── Spacer ───────────────────────────────────────────────
        spacer = Gtk.Box(vexpand=True)
        self.append(spacer)

        # ── Controls ───────────────<E29480><E29480><EFBFBD>─────────────────────────────
        controls = Gtk.Box(
            orientation=Gtk.Orientation.HORIZONTAL,
            spacing=10,
            halign=Gtk.Align.CENTER,
            margin_bottom=14,
        )
        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)

        if not self._enabled:
            self._switch.set_active(False)
            self._switch.set_sensitive(False)
            self._status_label.set_label("○ disabled")
            self._status_label.set_css_classes(["caption", "tile-status", "disabled-label"])
            self._logo.set_opacity(0.35)
            self.set_tooltip_text(f"{name} is not enabled in custom.nix")

    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):
                    try:
                        pixbuf = GdkPixbuf.Pixbuf.new_from_file_at_scale(
                            path, ICON_PX, ICON_PX, True)
                        texture = Gdk.Texture.new_for_pixbuf(pixbuf)
                        self._logo.set_from_paintable(texture)
                        return
                    except Exception:
                        break
        self._logo.set_from_icon_name("system-run-symbolic")

    def refresh(self):
        if not self._enabled:
            return

        active = systemctl.is_active(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", "tile-status", "error"])
        elif is_on:
            self._status_label.set_label("● running")
            self._status_label.set_css_classes(["caption", "tile-status", "success"])
        elif is_loading:
            self._status_label.set_label(f"● {active}")
            self._status_label.set_css_classes(["caption", "tile-status", "warning"])
        else:
            self._status_label.set_label(f"● {active}")
            self._status_label.set_css_classes(["caption", "tile-status", "dim-label"])

    def _on_toggled(self, switch, state):
        if not self._enabled:
            return True
        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):
        if not self._enabled:
            return
        systemctl.run_action("restart", self._unit, self._scope, self._method)
        GLib.timeout_add(1500, self.refresh)