Fix focusing after splitting tabs/stacks

sway/tree/layout.c

@@ -874,11 +874,10 @@ struct sway_container *container_split(struct sway_container *child,
 	cont->x = child->x;
 	cont->y = child->y;
 
+	struct sway_seat *seat = input_manager_get_default_seat(input_manager);
+	bool set_focus = (seat_get_focus(seat) == child);
 	if (child->type == C_WORKSPACE) {
-		struct sway_seat *seat = input_manager_get_default_seat(input_manager);
 		struct sway_container *workspace = child;
-		bool set_focus = (seat_get_focus(seat) == workspace);
-
 		while (workspace->children->length) {
 			struct sway_container *ws_child = workspace->children->items[0];
 			container_remove_child(ws_child);
@@ -890,10 +889,6 @@ struct sway_container *container_split(struct sway_container *child,
 		enum sway_container_layout old_layout = workspace->layout;
 		workspace->layout = layout;
 		cont->layout = old_layout;
-
-		if (set_focus) {
-			seat_set_focus(seat, cont);
-		}
 	} else {
 		struct sway_container *old_parent = child->parent;
 		cont->layout = layout;
@@ -902,6 +897,14 @@ struct sway_container *container_split(struct sway_container *child,
 		wl_signal_emit(&child->events.reparent, old_parent);
 	}
 
+	if (set_focus) {
+		seat_set_focus(seat, cont);
+		if (cont->parent->layout == L_TABBED
+				|| cont->parent->layout == L_STACKED) {
+			seat_set_focus(seat, child);
+		}
+	}
+
 	container_notify_subtree_changed(cont);
 
 	return cont;