From f7669a55ad5e268243e9a1b057ae9aa3ac73af67 Mon Sep 17 00:00:00 2001
From: Ivan Molodetskikh <yalterz@gmail.com>
Date: Sat, 14 Oct 2023 20:42:10 +0400
Subject: Add binds to move workspaces up and down

---
 src/config.rs |  2 ++
 src/input.rs  | 10 +++++++
 src/layout.rs | 85 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 97 insertions(+)

(limited to 'src')

diff --git a/src/config.rs b/src/config.rs
index 654b0557..1570b737 100644
--- a/src/config.rs
+++ b/src/config.rs
@@ -258,6 +258,8 @@ pub enum Action {
     MoveWindowToWorkspaceDown,
     MoveWindowToWorkspaceUp,
     MoveWindowToWorkspace(#[knuffel(argument)] u8),
+    MoveWorkspaceDown,
+    MoveWorkspaceUp,
     FocusMonitorLeft,
     FocusMonitorRight,
     FocusMonitorDown,
diff --git a/src/input.rs b/src/input.rs
index 6b6baa34..cdb9ce1b 100644
--- a/src/input.rs
+++ b/src/input.rs
@@ -271,6 +271,16 @@ impl State {
                             // FIXME: granular
                             self.niri.queue_redraw_all();
                         }
+                        Action::MoveWorkspaceDown => {
+                            self.niri.layout.move_workspace_down();
+                            // FIXME: granular
+                            self.niri.queue_redraw_all();
+                        }
+                        Action::MoveWorkspaceUp => {
+                            self.niri.layout.move_workspace_up();
+                            // FIXME: granular
+                            self.niri.queue_redraw_all();
+                        }
                         Action::ConsumeWindowIntoColumn => {
                             self.niri.layout.consume_into_column();
                             // FIXME: granular
diff --git a/src/layout.rs b/src/layout.rs
index 99a6cd93..365e9f54 100644
--- a/src/layout.rs
+++ b/src/layout.rs
@@ -1363,6 +1363,20 @@ impl<W: LayoutElement> Layout<W> {
 
         None
     }
+
+    pub fn move_workspace_down(&mut self) {
+        let Some(monitor) = self.active_monitor() else {
+            return;
+        };
+        monitor.move_workspace_down();
+    }
+
+    pub fn move_workspace_up(&mut self) {
+        let Some(monitor) = self.active_monitor() else {
+            return;
+        };
+        monitor.move_workspace_up();
+    }
 }
 
 impl Layout<Window> {
@@ -1640,6 +1654,46 @@ impl<W: LayoutElement> Monitor<W> {
     fn set_column_width(&mut self, change: SizeChange) {
         self.active_workspace().set_column_width(change);
     }
+
+    fn move_workspace_down(&mut self) {
+        let new_idx = min(self.active_workspace_idx + 1, self.workspaces.len() - 1);
+        if new_idx == self.active_workspace_idx {
+            return;
+        }
+
+        self.workspaces.swap(self.active_workspace_idx, new_idx);
+
+        if new_idx == self.workspaces.len() - 1 {
+            // Insert a new empty workspace.
+            let ws = Workspace::new(self.output.clone(), self.options.clone());
+            self.workspaces.push(ws);
+        }
+
+        self.activate_workspace(new_idx);
+        self.workspace_switch = None;
+
+        self.clean_up_workspaces();
+    }
+
+    fn move_workspace_up(&mut self) {
+        let new_idx = self.active_workspace_idx.saturating_sub(1);
+        if new_idx == self.active_workspace_idx {
+            return;
+        }
+
+        self.workspaces.swap(self.active_workspace_idx, new_idx);
+
+        if self.active_workspace_idx == self.workspaces.len() - 1 {
+            // Insert a new empty workspace.
+            let ws = Workspace::new(self.output.clone(), self.options.clone());
+            self.workspaces.push(ws);
+        }
+
+        self.activate_workspace(new_idx);
+        self.workspace_switch = None;
+
+        self.clean_up_workspaces();
+    }
 }
 
 impl Monitor<Window> {
@@ -2864,6 +2918,8 @@ mod tests {
         MoveWindowToWorkspaceDown,
         MoveWindowToWorkspaceUp,
         MoveWindowToWorkspace(#[proptest(strategy = "1..=5u8")] u8),
+        MoveWorkspaceDown,
+        MoveWorkspaceUp,
         MoveWindowToOutput(#[proptest(strategy = "1..=5u8")] u8),
         SwitchPresetColumnWidth,
         MaximizeColumn,
@@ -2917,6 +2973,29 @@ mod tests {
                     layout.focus_output(&output);
                 }
                 Op::AddWindow { id, bbox, activate } => {
+                    match &mut layout.monitor_set {
+                        MonitorSet::Normal { monitors, .. } => {
+                            for mon in monitors {
+                                for ws in &mut mon.workspaces {
+                                    for win in ws.windows() {
+                                        if win.0.id == id {
+                                            return;
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                        MonitorSet::NoOutputs { workspaces, .. } => {
+                            for ws in workspaces {
+                                for win in ws.windows() {
+                                    if win.0.id == id {
+                                        return;
+                                    }
+                                }
+                            }
+                        }
+                    }
+
                     let win = TestWindow::new(id, bbox);
                     layout.add_window(win, activate);
                 }
@@ -2952,6 +3031,8 @@ mod tests {
 
                     layout.move_to_output(&output);
                 }
+                Op::MoveWorkspaceDown => layout.move_workspace_down(),
+                Op::MoveWorkspaceUp => layout.move_workspace_up(),
                 Op::SwitchPresetColumnWidth => layout.toggle_width(),
                 Op::MaximizeColumn => layout.toggle_full_width(),
                 Op::SetColumnWidth(change) => layout.set_column_width(change),
@@ -3049,6 +3130,8 @@ mod tests {
             Op::MoveWindowToWorkspace(1),
             Op::MoveWindowToWorkspace(2),
             Op::MoveWindowToWorkspace(3),
+            Op::MoveWindowDown,
+            Op::MoveWindowUp,
         ];
 
         for third in every_op {
@@ -3162,6 +3245,8 @@ mod tests {
             Op::MoveWindowToWorkspace(1),
             Op::MoveWindowToWorkspace(2),
             Op::MoveWindowToWorkspace(3),
+            Op::MoveWindowDown,
+            Op::MoveWindowUp,
         ];
 
         for third in every_op {
-- 
cgit