fix workspace splits

1 files changed, 21 insertions, 17 deletions

diff --git a/sway/tree/layout.c b/sway/tree/layout.c
index e13c2ac4..bea15f8a 100644
--- a/sway/tree/layout.c
+++ b/sway/tree/layout.c
@@ -405,7 +405,12 @@ void apply_vert_layout(struct sway_container *container,
                         wlr_log(L_DEBUG,
                                 "Calculating arrangement for %p:%d (will scale %f by %f)",
                                 child, child->type, height, scale);
-                        view_set_position(child->sway_view, x, child_y);
+                        if (child->type == C_VIEW) {
+                                view_set_position(child->sway_view, x, child_y);
+                        } else {
+                                child->x = x;
+                                child->y = child_y;
+                        }
 
                         if (i == end - 1) {
                                 double remaining_height = y + height - child_y;
@@ -691,34 +696,33 @@ struct sway_container *container_split(struct sway_container *child,
         wlr_log(L_DEBUG, "creating container %p around %p", cont, child);
 
         cont->prev_layout = L_NONE;
-        cont->layout = layout;
         cont->width = child->width;
         cont->height = child->height;
         cont->x = child->x;
         cont->y = child->y;
 
-        /* Container inherits all of workspaces children, layout and whatnot */
         if (child->type == C_WORKSPACE) {
+                struct sway_seat *seat = sway_input_manager_get_default_seat(input_manager);
                 struct sway_container *workspace = child;
-                // reorder focus
-                int i;
-                for (i = 0; i < workspace->children->length; ++i) {
-                        ((struct sway_container *)workspace->children->items[i])->parent =
-                                cont;
+                bool set_focus = (sway_seat_get_focus(seat) == workspace);
+
+                while (workspace->children->length) {
+                        struct sway_container *ws_child = workspace->children->items[0];
+                        container_remove_child(ws_child);
+                        container_add_child(cont, ws_child);
                 }
 
-                // Swap children
-                list_t  *tmp_list  = workspace->children;
-                workspace->children = cont->children;
-                cont->children = tmp_list;
-                // add container to workspace chidren
                 container_add_child(workspace, cont);
-                // give them proper layouts
-                cont->layout = workspace->workspace_layout;
-                cont->prev_layout = workspace->prev_layout;
-        } else { // Or is built around container
+                container_set_layout(workspace, layout);
+
+                if (set_focus) {
+                        sway_seat_set_focus(seat, cont);
+                }
+        } else {
+                cont->layout = layout;
                 container_replace_child(child, cont);
                 container_add_child(cont, child);
         }
+
         return cont;
 }