From 06da5cc9b61d2265e66f5ab40a49ba92d0a772ec Mon Sep 17 00:00:00 2001
From: Victoria Brekenfeld <victoria@system76.com>
Date: Thu, 7 Jul 2022 22:41:17 +0200
Subject: [PATCH] workspace: Allow tiling to be toggled

---
 src/shell/workspace.rs | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/src/shell/workspace.rs b/src/shell/workspace.rs
index 56e7406b..5a7617ea 100644
--- a/src/shell/workspace.rs
+++ b/src/shell/workspace.rs
@@ -24,6 +24,7 @@ pub struct Workspace {
     pub space: Space,
     pub tiling_layer: TilingLayout,
     pub floating_layer: FloatingLayout,
+    tiling_enabled: bool,
     pub fullscreen: HashMap<String, Window>,
     pub handle: WorkspaceHandle,
 }
@@ -35,6 +36,7 @@ impl Workspace {
             space: Space::new(None),
             tiling_layer: TilingLayout::new(),
             floating_layer: FloatingLayout::new(),
+            tiling_enabled: true,
             fullscreen: HashMap::new(),
             handle,
         }
@@ -172,4 +174,36 @@ impl Workspace {
         }
         self.fullscreen.get(&output.name()).filter(|w| w.alive())
     }
+
+    pub fn toggle_tiling(&mut self, seat: &Seat<State>) {
+        if self.tiling_enabled {
+            for window in self.tiling_layer.windows.clone().into_iter() {
+                self.tiling_layer.unmap_window(&mut self.space, &window);
+                self.floating_layer.map_window(&mut self.space, window, seat);
+            }
+            self.tiling_enabled = false;
+        } else {
+            let focus_stack = self.focus_stack(seat);
+            for window in self.floating_layer.windows.clone().into_iter() {
+                self.floating_layer.unmap_window(&mut self.space, &window);
+                self.tiling_layer.map_window(&mut self.space, window, seat, focus_stack.iter())
+            }
+            self.tiling_enabled = true;
+        }
+    }
+
+    pub fn toggle_floating_window(&mut self, seat: &Seat<State>) {
+        if self.tiling_enabled {
+            if let Some(window) = self.focus_stack(seat).iter().next().cloned() {
+                if self.tiling_layer.windows.contains(&window) {
+                    self.tiling_layer.unmap_window(&mut self.space, &window);
+                    self.floating_layer.map_window(&mut self.space, window, seat);
+                } else if self.floating_layer.windows.contains(&window) {
+                    let focus_stack = self.focus_stack(seat);
+                    self.floating_layer.unmap_window(&mut self.space, &window);
+                    self.tiling_layer.map_window(&mut self.space, window, seat, focus_stack.iter())
+                }
+            }
+        }
+    }
 }