xkb_switch_layout: fix relative layout switches

Fixes #6011

1 files 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);
 }