Made the shuffle and repeat buttons actual toggle buttons

sublime/ui/common/icon_button.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Any, Optional
 
 import gi
 gi.require_version('Gtk', '3.0')
@@ -15,6 +15,7 @@ class IconButton(Gtk.Button):
         **kwargs,
     ):
         Gtk.Button.__init__(self, **kwargs)
+
         self.icon_size = icon_size
         box = Gtk.Box(
             orientation=Gtk.Orientation.HORIZONTAL, name='icon-button-box')
@@ -33,3 +34,39 @@ class IconButton(Gtk.Button):
 
     def set_icon(self, icon_name: Optional[str]):
         self.image.set_from_icon_name(icon_name, self.icon_size)
+
+
+class IconToggleButton(Gtk.ToggleButton):
+    def __init__(
+        self,
+        icon_name: Optional[str],
+        relief: bool = False,
+        icon_size: Gtk.IconSize = Gtk.IconSize.BUTTON,
+        label: str = None,
+        **kwargs,
+    ):
+        Gtk.ToggleButton.__init__(self, **kwargs)
+        self.icon_size = icon_size
+        box = Gtk.Box(
+            orientation=Gtk.Orientation.HORIZONTAL, name='icon-button-box')
+
+        self.image = Gtk.Image()
+        self.image.set_from_icon_name(icon_name, self.icon_size)
+        box.add(self.image)
+
+        if label is not None:
+            box.add(Gtk.Label(label=label))
+
+        if not relief:
+            self.props.relief = Gtk.ReliefStyle.NONE
+
+        self.add(box)
+
+    def set_icon(self, icon_name: Optional[str]):
+        self.image.set_from_icon_name(icon_name, self.icon_size)
+
+    def get_active(self):
+        return super().get_active()
+
+    def set_active(self, active):
+        super().set_active(active)