104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
import json
|
|
from fabric.utils import exec_shell_command_async, get_relative_path, invoke_repeater
|
|
from fabric.widgets.box import Box
|
|
from fabric.widgets.centerbox import CenterBox
|
|
from fabric.widgets.label import Label
|
|
from fabric.widgets.stack import Stack
|
|
import snippets.iconss as icons
|
|
from loguru import logger
|
|
from gi.repository import Gtk
|
|
from fabric.widgets.button import Button
|
|
|
|
|
|
class UpdatesWidget(Button):
|
|
"""A widget to display the number of available updates."""
|
|
|
|
def __init__(
|
|
self,
|
|
**kwargs,
|
|
):
|
|
# Initialize the button with specific name and style
|
|
super().__init__(name="button-bar", **kwargs)
|
|
|
|
self.script_file = get_relative_path("../../../config/scripts/systemupdates.sh")
|
|
|
|
self.update_level_label = Label(
|
|
name="update-label",
|
|
label="0",
|
|
visible=False, # Initially hidden
|
|
)
|
|
|
|
self.update_icon = Label(name="update-icon", markup=icons.update)
|
|
self.updated_icon = Label(name="updated-icon", markup=icons.updated)
|
|
|
|
self.update_box = CenterBox(
|
|
center_children=[self.update_icon, self.update_level_label],
|
|
)
|
|
|
|
self.updated_box = CenterBox(
|
|
center_children=[self.updated_icon],
|
|
)
|
|
|
|
self.stack = Stack()
|
|
self.stack.add_named(self.update_box, "update_box")
|
|
self.stack.add_named(self.updated_box, "updated_box")
|
|
self.stack.set_visible_child_name("update_box")
|
|
|
|
self.children = Box(
|
|
children=self.stack,
|
|
)
|
|
|
|
# Ensure the stack resizes to the current visible child
|
|
self.stack.set_homogeneous(False)
|
|
self.stack.set_transition_type(Gtk.StackTransitionType.CROSSFADE)
|
|
|
|
# Connect the button press event
|
|
self.connect("button-press-event", self.on_button_press)
|
|
|
|
# Set up a repeater to call the update method at specified intervals
|
|
invoke_repeater(60000, self.update, initial_call=True)
|
|
|
|
def update_values(self, value: str):
|
|
logger.info(f"[Updates] Received update JSON: {value}")
|
|
|
|
try:
|
|
value = json.loads(value)
|
|
except json.JSONDecodeError:
|
|
logger.error("[Updates] Failed to parse update JSON")
|
|
return False
|
|
|
|
update_count = str(value["total"])
|
|
self.update_level_label.set_label(update_count)
|
|
|
|
if update_count == "0":
|
|
self.stack.set_visible_child_name("updated_box")
|
|
self.update_level_label.set_visible(False)
|
|
else:
|
|
self.stack.set_visible_child_name("update_box")
|
|
self.update_level_label.set_visible(True)
|
|
|
|
self.set_tooltip_text(value.get("tooltip", ""))
|
|
return True
|
|
|
|
def on_button_press(self, _, event):
|
|
if event.button == 1:
|
|
exec_shell_command_async(
|
|
f"{self.script_file} -arch -up",
|
|
lambda _: None,
|
|
)
|
|
self.update()
|
|
return True
|
|
else:
|
|
self.update()
|
|
|
|
def update(self):
|
|
logger.info("[Updates] Checking for updates...")
|
|
|
|
# Execute the update script asynchronously and update values
|
|
exec_shell_command_async(
|
|
f"{self.script_file} -arch",
|
|
lambda output: self.update_values(output),
|
|
)
|
|
|
|
return True
|