scrolling between tags/workspaces

2 files changed, 83 insertions, 18 deletions

diff --git a/src/wm_info_provider/hyprland.rs b/src/wm_info_provider/hyprland.rs
index 29b86b3..a59a68f 100644
--- a/src/wm_info_provider/hyprland.rs
+++ b/src/wm_info_provider/hyprland.rs
@@ -1,3 +1,5 @@
+#![allow(clippy::collapsible_else_if)]
+
 use std::io::{self, Write};
 use std::os::fd::AsRawFd;
 use std::os::unix::net::UnixStream;
@@ -25,6 +27,10 @@ impl HyprlandInfoProvider {
             ipc,
         })
     }
+
+    fn set_workspace(&self, id: u32) {
+        let _ = self.ipc.exec(&format!("/dispatch workspace {id}"));
+    }
 }
 
 impl WmInfoProvider for HyprlandInfoProvider {
@@ -57,13 +63,42 @@ impl WmInfoProvider for HyprlandInfoProvider {
     fn click_on_tag(
         &mut self,
         _: &mut Connection<State>,
-        _: WlOutput,
+        output: &Output,
         _: WlSeat,
-        tag_id: u32,
+        tag_id: Option<u32>,
         btn: PointerBtn,
     ) {
-        if btn == PointerBtn::Left {
-            let _ = self.ipc.exec(&format!("/dispatch workspace {tag_id}"));
+        match btn {
+            PointerBtn::Left => {
+                if let Some(tag_id) = tag_id {
+                    self.set_workspace(tag_id);
+                }
+            }
+            PointerBtn::WheelUp | PointerBtn::WheelDown => {
+                if let Some(active_i) = self
+                    .workspaces
+                    .iter()
+                    .position(|ws| ws.monitor == output.name && self.active_id == ws.id)
+                {
+                    if btn == PointerBtn::WheelUp {
+                        if let Some(prev) = self.workspaces[..active_i]
+                            .iter()
+                            .rfind(|ws| ws.monitor == output.name)
+                        {
+                            self.set_workspace(prev.id);
+                        }
+                    } else {
+                        if let Some(next) = self.workspaces[active_i..]
+                            .iter()
+                            .skip(1)
+                            .find(|ws| ws.monitor == output.name)
+                        {
+                            self.set_workspace(next.id);
+                        }
+                    }
+                }
+            }
+            _ => (),
         }
     }
 
diff --git a/src/wm_info_provider/river.rs b/src/wm_info_provider/river.rs
index 4763f9f..f50eefe 100644
--- a/src/wm_info_provider/river.rs
+++ b/src/wm_info_provider/river.rs
@@ -52,6 +52,15 @@ impl RiverInfoProvider {
             },
         })
     }
+
+    fn set_focused_tags(&self, seat: WlSeat, conn: &mut Connection<State>, tags: u32) {
+        self.control
+            .add_argument(conn, c"set-focused-tags".to_owned());
+        self.control
+            .add_argument(conn, CString::new(tags.to_string()).unwrap());
+        self.control
+            .run_command_with_cb(conn, seat, river_command_cb);
+    }
 }
 
 impl WmInfoProvider for RiverInfoProvider {
@@ -109,23 +118,44 @@ impl WmInfoProvider for RiverInfoProvider {
     fn click_on_tag(
         &mut self,
         conn: &mut Connection<State>,
-        _output: WlOutput,
+        output: &Output,
         seat: WlSeat,
-        tag_id: u32,
+        tag_id: Option<u32>,
         btn: PointerBtn,
     ) {
-        let cmd = match btn {
-            PointerBtn::Left => c"set-focused-tags",
-            PointerBtn::Right => c"toggle-focused-tags",
-            _ => return,
-        };
-        self.control.add_argument(conn, cmd.to_owned());
-        self.control.add_argument(
-            conn,
-            CString::new((1u32 << (tag_id - 1)).to_string()).unwrap(),
-        );
-        self.control
-            .run_command_with_cb(conn, seat, river_command_cb);
+        match btn {
+            PointerBtn::Left => {
+                if let Some(tag_id) = tag_id {
+                    self.set_focused_tags(seat, conn, 1u32 << (tag_id - 1));
+                }
+            }
+            PointerBtn::Right => {
+                if let Some(tag_id) = tag_id {
+                    self.control
+                        .add_argument(conn, c"toggle-focused-tags".to_owned());
+                    self.control.add_argument(
+                        conn,
+                        CString::new((1u32 << (tag_id - 1)).to_string()).unwrap(),
+                    );
+                    self.control
+                        .run_command_with_cb(conn, seat, river_command_cb);
+                }
+            }
+            PointerBtn::WheelUp | PointerBtn::WheelDown => {
+                if let Some(status) = self.output_statuses.iter().find(|s| s.output == output.wl) {
+                    let mut new_tags = if btn == PointerBtn::WheelUp {
+                        status.focused_tags >> 1
+                    } else {
+                        status.focused_tags << 1
+                    };
+                    if new_tags == 0 {
+                        new_tags |= status.focused_tags & 0x8000_0001;
+                    }
+                    self.set_focused_tags(seat, conn, new_tags);
+                }
+            }
+            _ => (),
+        }
     }
 
     fn as_any(&mut self) -> &mut dyn Any {