Reapply prev layout when exiting tabbed/stacked

include/container.h

@@ -56,6 +56,7 @@ struct sway_container {
 
 	enum swayc_types type;
 	enum swayc_layouts layout;
+	enum swayc_layouts prev_layout;
 
 	/**
 	 * Width and height of this container, without borders or gaps.

sway/commands.c

@@ -1760,9 +1760,17 @@ static struct cmd_results *cmd_layout(int argc, char **argv) {
 	}
 
 	if (strcasecmp(argv[0], "default") == 0) {
+		parent->layout = parent->prev_layout;
+		if (parent->layout == L_NONE) {
 			swayc_t *output = swayc_parent_by_type(parent, C_OUTPUT);
 			parent->layout = default_layout(output);
-	} else if (strcasecmp(argv[0], "tabbed") == 0) {
+		}
+	} else {
+		if (parent->layout != L_TABBED && parent->layout != L_STACKED) {
+			parent->prev_layout = parent->layout;
+		}
+
+		if (strcasecmp(argv[0], "tabbed") == 0) {
 			if (parent->type != C_CONTAINER) {
 				parent = new_container(parent, L_TABBED);
 			}
@@ -1785,6 +1793,7 @@ static struct cmd_results *cmd_layout(int argc, char **argv) {
 				parent->layout = L_VERT;
 			}
 		}
+	}
 
 	arrange_windows(parent, parent->width, parent->height);
 

sway/container.c

@@ -163,6 +163,7 @@ swayc_t *new_workspace(swayc_t *output, const char *name) {
 	sway_log(L_DEBUG, "Added workspace %s for output %u", name, (unsigned int)output->handle);
 	swayc_t *workspace = new_swayc(C_WORKSPACE);
 
+	workspace->prev_layout = L_NONE;
 	workspace->layout = default_layout(output);
 
 	workspace->x = output->x;
@@ -203,6 +204,7 @@ swayc_t *new_container(swayc_t *child, enum swayc_layouts layout) {
 
 	sway_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;
@@ -229,6 +231,7 @@ swayc_t *new_container(swayc_t *child, enum swayc_layouts layout) {
 		add_child(workspace, cont);
 		// give them proper layouts
 		cont->layout = workspace->layout;
+		cont->prev_layout = workspace->prev_layout;
 		/* TODO: might break shit in move_container!!! workspace->layout = layout; */
 		set_focused_container_for(workspace, get_focused_view(workspace));
 	} else { // Or is built around container