1 files changed, 6 insertions, 9 deletions

diff --git a/sway/input/switch.c b/sway/input/switch.c
index ac4baece..831f4dbf 100644
--- a/sway/input/switch.c
+++ b/sway/input/switch.c
@@ -1,6 +1,5 @@
 #include "sway/config.h"
 #include "sway/input/switch.h"
-#include <wlr/types/wlr_idle.h>
 #include "log.h"
 
 struct sway_switch *sway_switch_create(struct sway_seat *seat,
@@ -11,6 +10,7 @@ struct sway_switch *sway_switch_create(struct sway_seat *seat,
                 return NULL;
         }
         device->switch_device = switch_device;
+        switch_device->wlr = wlr_switch_from_input_device(device->input_device->wlr_device);
         switch_device->seat_device = device;
         switch_device->state = WLR_SWITCH_STATE_OFF;
         wl_list_init(&switch_device->switch_toggle.link);
@@ -33,9 +33,8 @@ static bool sway_switch_trigger_test(enum sway_switch_trigger trigger,
 }
 
 static void execute_binding(struct sway_switch *sway_switch) {
-        struct sway_seat* seat = sway_switch->seat_device->sway_seat;
-        bool input_inhibited = seat->exclusive_client != NULL ||
-                server.session_lock.locked;
+        struct sway_seat *seat = sway_switch->seat_device->sway_seat;
+        bool locked = server.session_lock.lock;
 
         list_t *bindings = config->current_mode->switch_bindings;
         struct sway_switch_binding *matched_binding = NULL;
@@ -52,13 +51,13 @@ static void execute_binding(struct sway_switch *sway_switch) {
                         continue;
                 }
                 bool binding_locked = binding->flags & BINDING_LOCKED;
-                if (!binding_locked && input_inhibited) {
+                if (!binding_locked && locked) {
                         continue;
                 }
 
                 matched_binding = binding;
 
-                if (binding_locked == input_inhibited) {
+                if (binding_locked == locked) {
                         break;
                 }
         }
@@ -95,10 +94,8 @@ static void handle_switch_toggle(struct wl_listener *listener, void *data) {
 }
 
 void sway_switch_configure(struct sway_switch *sway_switch) {
-        struct wlr_input_device *wlr_device =
-                sway_switch->seat_device->input_device->wlr_device;
         wl_list_remove(&sway_switch->switch_toggle.link);
-        wl_signal_add(&wlr_device->switch_device->events.toggle,
+        wl_signal_add(&sway_switch->wlr->events.toggle,
                         &sway_switch->switch_toggle);
         sway_switch->switch_toggle.notify = handle_switch_toggle;
         sway_log(SWAY_DEBUG, "Configured switch for device");