From a27af527066fb5c05a8064a8342864f33def0d99 Mon Sep 17 00:00:00 2001
From: Michael Stapelberg <michael@stapelberg.de>
Date: Sun, 14 Nov 2010 22:35:44 +0100
Subject: [PATCH] fix floating focus behaviour, extend testcase

---
 src/con.c                       | 19 ++++++++++--
 src/floating.c                  |  4 +++
 src/x.c                         |  2 +-
 testcases/t/35-floating-focus.t | 51 ++++++++++++++++++++++++++++++++-
 4 files changed, 72 insertions(+), 4 deletions(-)

diff --git a/src/con.c b/src/con.c
index e9232273..2487dd83 100644
--- a/src/con.c
+++ b/src/con.c
@@ -422,9 +422,24 @@ Con *con_next_focused(Con *con) {
     /* floating containers are attached to a workspace, so we focus either the
      * next floating container (if any) or the workspace itself. */
     if (con->type == CT_FLOATING_CON) {
+        DLOG("selecting next for CT_FLOATING_CON\n");
         next = TAILQ_NEXT(con, floating_windows);
-        if (next == TAILQ_END(&(parent->floating_head)))
-            next = con_get_workspace(con);
+        if (next == TAILQ_END(&(parent->floating_head))) {
+            Con *ws = con_get_workspace(con);
+            next = ws;
+            DLOG("no more floating containers for next = %p, restoring workspace focus\n", next);
+            while (next != TAILQ_END(&(ws->focus_head)) && !TAILQ_EMPTY(&(next->focus_head))) {
+                next = TAILQ_FIRST(&(next->focus_head));
+                if (next == con) {
+                    DLOG("skipping container itself, we want the next client\n");
+                    next = TAILQ_NEXT(next, focused);
+                }
+            }
+            if (next == TAILQ_END(&(ws->focus_head))) {
+                DLOG("Focus list empty, returning NULL\n");
+                next = NULL;
+            }
+        }
         return next;
     }
 
diff --git a/src/floating.c b/src/floating.c
index 005f2092..2292d89b 100644
--- a/src/floating.c
+++ b/src/floating.c
@@ -78,6 +78,8 @@ void floating_enable(Con *con, bool automatic) {
 
     TAILQ_INSERT_TAIL(&(nc->nodes_head), con, nodes);
     TAILQ_INSERT_TAIL(&(nc->focus_head), con, focused);
+    // TODO: don’t influence focus handling when Con was not focused before.
+    con_focus(con);
 }
 
 void floating_disable(Con *con, bool automatic) {
@@ -105,6 +107,8 @@ void floating_disable(Con *con, bool automatic) {
     con->floating = FLOATING_USER_OFF;
 
     con_fix_percent(con->parent, WINDOW_ADD);
+    // TODO: don’t influence focus handling when Con was not focused before.
+    con_focus(con);
 }
 
 
diff --git a/src/x.c b/src/x.c
index fe83a848..5c854d01 100644
--- a/src/x.c
+++ b/src/x.c
@@ -372,7 +372,7 @@ static void x_push_node(Con *con) {
     state = state_for_frame(con->frame);
 
     if (state->name != NULL) {
-        DLOG("pushing name %s\n", state->name);
+        DLOG("pushing name %s for con %p\n", state->name, con);
 
         xcb_change_property(conn, XCB_PROP_MODE_REPLACE, con->frame,
                             WM_NAME, STRING, 8, strlen(state->name), state->name);
diff --git a/testcases/t/35-floating-focus.t b/testcases/t/35-floating-focus.t
index 7efe2472..3b6cfdae 100644
--- a/testcases/t/35-floating-focus.t
+++ b/testcases/t/35-floating-focus.t
@@ -1,7 +1,7 @@
 #!perl
 # vim:ts=4:sw=4:expandtab
 
-use i3test tests => 3;
+use i3test tests => 9;
 use Time::HiRes qw(sleep);
 
 my $i3 = i3("/tmp/nestedcons");
@@ -9,6 +9,10 @@ my $i3 = i3("/tmp/nestedcons");
 my $tmp = get_unused_workspace();
 $i3->command("workspace $tmp")->recv;
 
+#############################################################################
+# 1: see if focus stays the same when toggling tiling/floating mode
+#############################################################################
+
 $i3->command("open")->recv;
 $i3->command("open")->recv;
 
@@ -21,3 +25,48 @@ $i3->command("mode tiling")->recv;
 
 @content = @{get_ws_content($tmp)};
 cmp_ok($content[1]->{focused}, '==', 1, 'Second container still focused after mode toggle');
+
+#############################################################################
+# 2: see if the focus gets reverted correctly when closing floating clients
+# (first to the next floating client, then to the last focused tiling client)
+#############################################################################
+
+$tmp = get_unused_workspace();
+$i3->command("workspace $tmp")->recv;
+
+$i3->command("open")->recv;
+$i3->command("open")->recv;
+$i3->command("open")->recv;
+
+@content = @{get_ws_content($tmp)};
+cmp_ok(@content, '==', 3, 'two containers opened');
+cmp_ok($content[2]->{focused}, '==', 1, 'Last container focused');
+
+my $last_id = $content[2]->{id};
+my $second_id = $content[1]->{id};
+my $first_id = $content[0]->{id};
+diag("last_id = $last_id, second_id = $second_id, first_id = $first_id");
+
+$i3->command(qq|[con_id="$second_id"] focus|)->recv;
+@content = @{get_ws_content($tmp)};
+cmp_ok($content[1]->{focused}, '==', 1, 'Second container focused');
+
+$i3->command("mode floating")->recv;
+
+$i3->command(qq|[con_id="$last_id"] focus|)->recv;
+@content = @{get_ws_content($tmp)};
+cmp_ok($content[1]->{focused}, '==', 1, 'Last container focused');
+
+$i3->command("mode floating")->recv;
+
+diag("focused = " . get_focused($tmp));
+
+$i3->command("kill")->recv;
+
+diag("focused = " . get_focused($tmp));
+# TODO: this test result is actually not right. the focus reverts to where the mouse pointer is.
+cmp_ok(get_focused($tmp), '==', $second_id, 'Focus reverted to second floating container');
+
+$i3->command("kill")->recv;
+@content = @{get_ws_content($tmp)};
+cmp_ok($content[0]->{focused}, '==', 1, 'Focus reverted to tiling container');