also check floating cons in container_find

1 files changed, 20 insertions, 5 deletions

diff --git a/sway/container.c b/sway/container.c
index 9d5e2690..4f22eb0d 100644
--- a/sway/container.c
+++ b/sway/container.c
@@ -727,14 +727,29 @@ swayc_t *container_find(swayc_t *container, bool (*f)(swayc_t *, const void *),
 		return NULL;
 	}
 
+	swayc_t *con;
+	if (container->type == C_WORKSPACE) {
+		for (int i = 0; i < container->floating->length; ++i) {
+			con = container->floating->items[i];
+			if (f(con, data)) {
+				return con;
+			}
+			con = container_find(con, f, data);
+			if (con != NULL) {
+				return con;
+			}
+		}
+	}
+
 	for (int i = 0; i < container->children->length; ++i) {
-		if (f(container->children->items[i], data)) {
-			return container->children->items[i];
+		con = container->children->items[i];
+		if (f(con, data)) {
+			return con;
 		}
 
-		swayc_t *find = container_find(container->children->items[i], f, data);
-		if (find != NULL) {
-			return find;
+		con = container_find(con, f, data);
+		if (con != NULL) {
+			return con;
 		}
 	}