From e43c0f648df04395fb804cd98aeccb4ef656e431 Mon Sep 17 00:00:00 2001
From: Victoria Brekenfeld <victoria@system76.com>
Date: Wed, 7 Feb 2024 21:32:37 +0100
Subject: [PATCH] shell: Fix edge-cases on added output

---
 src/shell/mod.rs | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/shell/mod.rs b/src/shell/mod.rs
index 03b4c329..712e25bc 100644
--- a/src/shell/mod.rs
+++ b/src/shell/mod.rs
@@ -530,6 +530,10 @@ impl Workspaces {
                 .partition(|w| w.preferrs_output(output));
             moved_workspaces.extend(preferrs);
             set.workspaces = doesnt;
+            if set.workspaces.is_empty() {
+                set.add_empty_workspace(workspace_state);
+            }
+            set.active = set.active.min(set.workspaces.len() - 1);
         }
         {
             let set = self.sets.get_mut(output).unwrap();
@@ -2821,9 +2825,7 @@ impl Shell {
                         .map(mapped.clone(), Some(focus_stack.iter()), None);
                 }
                 ManagedLayer::Sticky => unreachable!(),
-                _ => {
-                    workspace.floating_layer.map(mapped.clone(), geometry.loc)
-                }
+                _ => workspace.floating_layer.map(mapped.clone(), geometry.loc),
             }
         }
     }