Changing Touch Bar icon immediately after toggling mode

insidegui · Nov 14, 2019 · 3412e92 · 3412e92
1 parent cad872a
commit 3412e92
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/NoiseBuddy/Controllers/TouchBarUIController.swift b/NoiseBuddy/Controllers/TouchBarUIController.swift
@@ -39,6 +39,14 @@ final class TouchBarUIController: NoiseControlUIController {
         DFRElementSetControlStripPresenceForIdentifier(.noiseControl, shouldShow(for: currentDevice))
     }
 
+    override func toggleNoiseControlMode(_ sender: Any) {
+        let nextMode = preferences.nextListeningMode(from: listeningModeController.listeningMode)
+
+        listeningModeController.listeningMode = nextMode
+
+        button.image = nextMode.touchBarImage
+    }
+
     override func handleListeningModeDidChange(_ device: NCDevice) {
         button.image = device.listeningMode.touchBarImage
     }