initial commit

modules/bar/components/volume.py

@@ -0,0 +1,81 @@
+from fabric import Fabricator
+from fabric.utils import exec_shell_command
+from fabric.widgets.box import Box
+from fabric.widgets.button import Button
+from snippets import MaterialIcon
+from services import audio
+
+VOLUME_ICONS = {
+    "muted": "volume_off",
+    "low": "volume_mute",
+    "medium-low": "volume_down",
+    "medium-high": "volume_up",
+    "high": "volume_up",
+}
+
+
+class VolumeIndicator(Box):
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        self.audio_service = audio
+        self.volume_icon_button = self.create_volume_icon()
+
+        self.volume_icon_button.connect("button-press-event", self.on_button_press)
+        self.audio_service.connect("changed", self.update_volume_status)
+
+        Fabricator(interval=1000, poll_from=self.update_volume_status)
+        self.children = (self.volume_icon_button,)
+
+    def create_volume_icon(self):
+        icon = MaterialIcon(VOLUME_ICONS["muted"], size=16)
+        button = Button(child=icon)
+        button.icon_widget = icon
+        return button
+
+    def on_button_press(self, widget, event):
+        if event.button == 1:
+            self.toggle_mute()
+        elif event.button == 3:
+            self.open_audio_settings()
+
+    def toggle_mute(self):
+        current_stream = self.audio_service.speaker
+        if current_stream:
+            current_stream.muted = not current_stream.muted
+            # self.update_volume_status()
+
+    def open_audio_settings(self):
+        exec_shell_command("pavucontrol")
+
+    def update_volume_status(self, *_):
+        current_stream = self.audio_service.speaker
+        if current_stream:
+            self.update_icon(current_stream)
+
+        return True
+
+    def update_icon(self, stream):
+        if stream.muted:
+            icon_name = VOLUME_ICONS["muted"]
+            volume_percentage = 0
+            tooltip_text = "Volume Muted"
+        else:
+            volume_level = stream.volume
+            icon_name = self.get_volume_icon(volume_level)
+            volume_percentage = int(volume_level)
+            tooltip_text = f"Volume: {volume_percentage}%"
+
+        self.volume_icon_button.icon_widget.set_label(icon_name)
+
+        self.volume_icon_button.set_tooltip_text(tooltip_text)
+
+    def get_volume_icon(self, volume_level):
+        if volume_level == 0:
+            return VOLUME_ICONS["low"]
+        elif volume_level <= 33:
+            return VOLUME_ICONS["medium-low"]
+        elif volume_level <= 66:
+            return VOLUME_ICONS["medium-high"]
+        else:
+            return VOLUME_ICONS["high"]