Implement input-method keyboard grab

Author: xdavidwu

sway/input/keyboard.c

@@ -466,6 +466,39 @@ static void handle_key_event(struct sway_keyboard *keyboard,
 	}
 
 	if (!handled || event->state == WLR_KEY_RELEASED) {
+		if (seat->im_relay.input_method) {
+			struct wlr_input_method_keyboard_grab_v2 *kb_grab =
+				seat->im_relay.input_method->im_keyboard_grab;
+			struct wlr_virtual_keyboard_v1 *virtual_keyboard =
+				wlr_device->keyboard->virtual_keyboard;
+
+			// If event is from virtual keyboard of the same client as grab,
+			// do not send it back
+			if (kb_grab && !(virtual_keyboard &&
+					wl_resource_get_client(virtual_keyboard->resource) ==
+					wl_resource_get_client(kb_grab->resource))) {
+				// Do not grab the release event if the press event was not
+				// sent to grab.
+				if (event->state == WLR_KEY_RELEASED) {
+					bool pressed_sent = update_shortcut_state(
+						&keyboard->state_pressed_sent, event, (uint32_t)keycode, 0);
+					if (pressed_sent) {
+						wlr_seat_set_keyboard(wlr_seat, wlr_device);
+						wlr_seat_keyboard_notify_key(wlr_seat, event->time_msec,
+							event->keycode, event->state);
+						goto end;
+					}
+
+				}
+
+				wlr_input_method_keyboard_grab_v2_set_keyboard(kb_grab,
+					wlr_device->keyboard);
+				wlr_input_method_keyboard_grab_v2_send_key(kb_grab,
+					event->time_msec, event->keycode, event->state);
+				goto end;
+			}
+		}
+
 		bool pressed_sent = update_shortcut_state(
 				&keyboard->state_pressed_sent, event, (uint32_t)keycode, 0);
 		if (pressed_sent || event->state == WLR_KEY_PRESSED) {
@@ -475,6 +508,7 @@ static void handle_key_event(struct sway_keyboard *keyboard,
 		}
 	}
 
+end:
 	transaction_commit_dirty();
 
 	free(device_identifier);
@@ -538,10 +572,32 @@ static void handle_modifier_event(struct sway_keyboard *keyboard) {
 	struct wlr_input_device *wlr_device =
 		keyboard->seat_device->input_device->wlr_device;
 	if (!wlr_device->keyboard->group) {
-		struct wlr_seat *wlr_seat = keyboard->seat_device->sway_seat->wlr_seat;
-		wlr_seat_set_keyboard(wlr_seat, wlr_device);
-		wlr_seat_keyboard_notify_modifiers(wlr_seat,
-				&wlr_device->keyboard->modifiers);
+		struct sway_seat *seat = keyboard->seat_device->sway_seat;
+		bool kb_grab_sent = false;
+		if (seat->im_relay.input_method) {
+			struct wlr_input_method_keyboard_grab_v2 *kb_grab =
+				seat->im_relay.input_method->im_keyboard_grab;
+			struct wlr_virtual_keyboard_v1 *virtual_keyboard =
+				wlr_device->keyboard->virtual_keyboard;
+
+			// If event is from virtual keyboard of the same client as grab,
+			// do not send it back
+			if (kb_grab && !(virtual_keyboard &&
+					wl_resource_get_client(virtual_keyboard->resource) ==
+					wl_resource_get_client(kb_grab->resource))) {
+				wlr_input_method_keyboard_grab_v2_set_keyboard(kb_grab,
+						wlr_device->keyboard);
+				wlr_input_method_keyboard_grab_v2_send_modifiers(kb_grab,
+						&wlr_device->keyboard->modifiers);
+				kb_grab_sent = true;
+			}
+		}
+		if (!kb_grab_sent) {
+			struct wlr_seat *wlr_seat = keyboard->seat_device->sway_seat->wlr_seat;
+			wlr_seat_set_keyboard(wlr_seat, wlr_device);
+			wlr_seat_keyboard_notify_modifiers(wlr_seat,
+					&wlr_device->keyboard->modifiers);
+		}
 
 		uint32_t modifiers = wlr_keyboard_get_modifiers(wlr_device->keyboard);
 		determine_bar_visibility(modifiers);