wrap container in direction (if possible)

Introduces container wrapping - if there is no other viable move, the
selection wraps on the first container where such action is possible.

1 files changed, 18 insertions, 4 deletions

diff --git a/sway/layout.c b/sway/layout.c
index 6012af66..f10e21bd 100644
--- a/sway/layout.c
+++ b/sway/layout.c
@@ -1084,13 +1084,17 @@ swayc_t *get_swayc_in_direction_under(swayc_t *container, enum movement_directio
 		return container->fullscreen;
 	}
 
+	swayc_t *wrap_candidate = NULL;
 	while (true) {
 		// Test if we can even make a difference here
 		bool can_move = false;
 		int diff = 0;
 		if (parent->type == C_ROOT) {
-			sway_log(L_DEBUG, "Moving between outputs");
 			swayc_t *output = swayc_adjacent_output(container, dir, &abs_pos, true);
+			if (!output || output == container) {
+				return wrap_candidate;
+			}
+			sway_log(L_DEBUG, "Moving between outputs");
 			return get_swayc_in_output_direction(output, dir);
 		} else {
 			if (dir == MOVE_LEFT || dir == MOVE_RIGHT) {
@@ -1108,8 +1112,18 @@ swayc_t *get_swayc_in_direction_under(swayc_t *container, enum movement_directio
 
 		if (can_move) {
 			int desired = index_child(container) + diff;
-			if (container->is_floating || desired < 0 || desired >= parent->children->length) {
+			if (container->is_floating) {
 				can_move = false;
+			} else if (desired < 0 || desired >= parent->children->length) {
+				can_move = false;
+				int len = parent->children->length;
+				if (!wrap_candidate && len > 1) {
+					if (desired < 0) {
+						wrap_candidate = parent->children->items[len-1];
+					} else {
+						wrap_candidate = parent->children->items[0];
+					}
+				}
 			} else {
 				return parent->children->items[desired];
 			}
@@ -1118,8 +1132,8 @@ swayc_t *get_swayc_in_direction_under(swayc_t *container, enum movement_directio
 			container = parent;
 			parent = parent->parent;
 			if (!parent || container == limit) {
-				// Nothing we can do
-				return NULL;
+				// wrapping is the last chance
+				return wrap_candidate;
 			}
 		}
 	}