Switch restore workspaces to a nested for-loop

1 files changed, 21 insertions, 18 deletions

diff --git a/sway/tree/output.c b/sway/tree/output.c
index 1ab8bed2..ed7e941e 100644
--- a/sway/tree/output.c
+++ b/sway/tree/output.c
@@ -8,26 +8,30 @@
 #include "sway/tree/workspace.h"
 #include "log.h"
 
-static void restore_workspace(struct sway_container *ws, void *output) {
-        if (ws->parent == output) {
-                return;
-        }
-
-        struct sway_container *highest = workspace_output_get_highest_available(
-                        ws, NULL);
-        if (!highest) {
-                return;
-        }
+static void restore_workspaces(struct sway_container *output) {
+        for (int i = 0; i < root_container.children->length; i++) {
+                struct sway_container *other = root_container.children->items[i];
+                if (other == output) {
+                        continue;
+                }
 
-        if (highest == output) {
-                struct sway_container *other = container_remove_child(ws);
-                container_add_child(output, ws);
-                ipc_event_workspace(ws, NULL, "move");
+                for (int j = 0; j < other->children->length; j++) {
+                        struct sway_container *ws = other->children->items[j];
+                        struct sway_container *highest =
+                                workspace_output_get_highest_available(ws, NULL);
+                        if (highest == output) {
+                                container_remove_child(ws);
+                                container_add_child(output, ws);
+                                ipc_event_workspace(ws, NULL, "move");
+                                j--;
+                        }
+                }
 
-                container_sort_workspaces(output);
-                arrange_output(output);
                 arrange_output(other);
         }
+
+        container_sort_workspaces(output);
+        arrange_output(output);
 }
 
 struct sway_container *output_create(
@@ -80,8 +84,7 @@ struct sway_container *output_create(
         output->width = size.width;
         output->height = size.height;
 
-        container_descendants(&root_container, C_WORKSPACE, restore_workspace,
-                        output);
+        restore_workspaces(output);
 
         if (!output->children->length) {
                 // Create workspace