BSButton's style updates on light/dark appearance

beeminder · Oct 6, 2024 · ef2356c · ef2356c
1 parent fc3423f
commit ef2356c
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/BeeKit/UI/BSButton.swift b/BeeKit/UI/BSButton.swift
@@ -13,11 +13,25 @@ public class BSButton : UIButton {
 
     required init?(coder aDecoder: NSCoder) {
         super.init(coder: aDecoder)
+
+        registerForTraitChanges(
+            [UITraitUserInterfaceLevel.self]) {
+                (self: Self, previousTraitCollection: UITraitCollection) in
+                self.resetStyle()
+            }
+
         self.setUp()
     }
 
     override init(frame: CGRect) {
         super.init(frame: frame)
+
+        registerForTraitChanges(
+            [UITraitUserInterfaceLevel.self]) {
+                (self: Self, previousTraitCollection: UITraitCollection) in
+                self.resetStyle()
+            }
+
         self.setUp()
     }
 
@@ -31,6 +45,10 @@ public class BSButton : UIButton {
         self.layer.borderWidth = 1
     }
 
+    private func resetStyle() {
+        self.tintColor = dynamicTintFillColor
+    }
+
     private let dynamicTintFillColor = UIColor { traitCollection in
         switch traitCollection.userInterfaceStyle {
         case .dark: