Huge Updates

modules/bar/components/updates.py

@@ -0,0 +1,103 @@
+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