swaybar: Fix input device removal

Before swaybar would exit with a protocol error when a pointer or touch
device was removed.

1 files changed, 10 insertions, 8 deletions

diff --git a/swaybar/input.c b/swaybar/input.c
index 8b83eab4..7f68caec 100644
--- a/swaybar/input.c
+++ b/swaybar/input.c
@@ -405,15 +405,14 @@ static const struct wl_touch_listener touch_listener = {
 static void seat_handle_capabilities(void *data, struct wl_seat *wl_seat,
 		enum wl_seat_capability caps) {
 	struct swaybar_seat *seat = data;
-	if (seat->pointer.pointer != NULL) {
+
+	bool have_pointer = caps & WL_SEAT_CAPABILITY_POINTER;
+	bool have_touch = caps & WL_SEAT_CAPABILITY_TOUCH;
+
+	if (!have_pointer && seat->pointer.pointer != NULL) {
 		wl_pointer_release(seat->pointer.pointer);
 		seat->pointer.pointer = NULL;
-	}
-	if (seat->touch.touch != NULL) {
-		wl_touch_release(seat->touch.touch);
-		seat->touch.touch = NULL;
-	}
-	if ((caps & WL_SEAT_CAPABILITY_POINTER)) {
+	} else if (have_pointer && seat->pointer.pointer == NULL) {
 		seat->pointer.pointer = wl_seat_get_pointer(wl_seat);
 		if (seat->bar->running && !seat->pointer.cursor_surface) {
 			seat->pointer.cursor_surface =
@@ -422,7 +421,10 @@ static void seat_handle_capabilities(void *data, struct wl_seat *wl_seat,
 		}
 		wl_pointer_add_listener(seat->pointer.pointer, &pointer_listener, seat);
 	}
-	if ((caps & WL_SEAT_CAPABILITY_TOUCH)) {
+	if (!have_touch && seat->touch.touch != NULL) {
+		wl_touch_release(seat->touch.touch);
+		seat->touch.touch = NULL;
+	} else if (have_touch && seat->touch.touch == NULL) {
 		seat->touch.touch = wl_seat_get_touch(wl_seat);
 		wl_touch_add_listener(seat->touch.touch, &touch_listener, seat);
 	}