From 70d30ac72b93c19813f6574025fb4723845ed6a4 Mon Sep 17 00:00:00 2001
From: Victor Makarov <vitja.makarov@gmail.com>
Date: Thu, 28 Apr 2022 14:20:44 +0300
Subject: [PATCH] xkb_switch_layout: fix relative layout switches

Fixes #6011
---
 sway/commands/input/xkb_switch_layout.c | 36 ++++++++++++++++++++++---
 1 file changed, 32 insertions(+), 4 deletions(-)

diff --git a/sway/commands/input/xkb_switch_layout.c b/sway/commands/input/xkb_switch_layout.c
index 3be37daf..dabc6697 100644
--- a/sway/commands/input/xkb_switch_layout.c
+++ b/sway/commands/input/xkb_switch_layout.c
@@ -6,6 +6,11 @@
 #include "sway/input/input-manager.h"
 #include "log.h"
 
+struct xkb_switch_layout_action {
+	struct wlr_keyboard *keyboard;
+	xkb_layout_index_t layout;
+};
+
 static void switch_layout(struct wlr_keyboard *kbd, xkb_layout_index_t idx) {
 	xkb_layout_index_t num_layouts = xkb_keymap_num_layouts(kbd->keymap);
 	if (idx >= num_layouts) {
@@ -29,10 +34,10 @@ static xkb_layout_index_t get_current_layout_index(struct wlr_keyboard *kbd) {
 	return layout_idx;
 }
 
-static void switch_layout_relative(struct wlr_keyboard *kbd, int dir) {
+static xkb_layout_index_t get_layout_relative(struct wlr_keyboard *kbd, int dir) {
 	xkb_layout_index_t num_layouts = xkb_keymap_num_layouts(kbd->keymap);
 	xkb_layout_index_t idx = get_current_layout_index(kbd);
-	switch_layout(kbd, (idx + num_layouts + dir) % num_layouts);
+	return (idx + num_layouts + dir) % num_layouts;
 }
 
 struct cmd_results *input_cmd_xkb_switch_layout(int argc, char **argv) {
@@ -67,6 +72,18 @@ struct cmd_results *input_cmd_xkb_switch_layout(int argc, char **argv) {
 		relative = 0;
 	}
 
+	struct xkb_switch_layout_action *actions = calloc(
+		wl_list_length(&server.input->devices),
+		sizeof(struct xkb_switch_layout_action));
+	size_t actions_len = 0;
+
+	if (!actions) {
+		return cmd_results_new(CMD_FAILURE, "Unable to allocate actions");
+	}
+
+	/* Calculate new indexes first because switching a layout in one
+	   keyboard may result in a change on other keyboards as well because
+	   of keyboard groups. */
 	struct sway_input_device *dev;
 	wl_list_for_each(dev, &server.input->devices, link) {
 		if (strcmp(ic->identifier, "*") != 0 &&
@@ -77,12 +94,23 @@ struct cmd_results *input_cmd_xkb_switch_layout(int argc, char **argv) {
 		if (dev->wlr_device->type != WLR_INPUT_DEVICE_KEYBOARD) {
 			continue;
 		}
+
+		struct xkb_switch_layout_action *action =
+			&actions[actions_len++];
+
+		action->keyboard = dev->wlr_device->keyboard;
 		if (relative) {
-			switch_layout_relative(dev->wlr_device->keyboard, relative);
+			action->layout = get_layout_relative(
+				dev->wlr_device->keyboard, relative);
 		} else {
-			switch_layout(dev->wlr_device->keyboard, layout);
+			action->layout = layout;
 		}
 	}
 
+	for (size_t i = 0; i < actions_len; i++) {
+		switch_layout(actions[i].keyboard, actions[i].layout);
+	}
+	free(actions);
+
 	return cmd_results_new(CMD_SUCCESS, NULL);
 }