wip: snowcap: make WlrTaskList output aware

Author: Ph4ntomas

snowcap/src/widget.rs

@@ -1,4 +1,5 @@
 pub mod input_region;
+pub mod output;
 pub mod wlr_tasklist;
 
 use iced::{Color, Theme, event::Status};

snowcap/src/widget/output.rs

@@ -0,0 +1,77 @@
+use smithay_client_toolkit::reexports::client::protocol::wl_output::WlOutput;
+
+#[derive(Default, Debug, Clone)]
+pub struct OutputState {
+    pub output: Option<WlOutput>,
+}
+
+impl OutputState {
+    pub fn enter(&mut self, output: WlOutput) {
+        self.output = Some(output);
+    }
+
+    pub fn leave(&mut self, output: WlOutput) {
+        if self.output == Some(output) {
+            self.output = None;
+        }
+    }
+}
+
+pub mod operation {
+    use iced_wgpu::core::widget::Operation;
+    use smithay_client_toolkit::reexports::client::protocol::wl_output::WlOutput;
+
+    pub fn enter_output(handle: WlOutput) -> impl Operation {
+        struct EnterOutput {
+            handle: WlOutput,
+        }
+
+        impl Operation for EnterOutput {
+            fn traverse(&mut self, operate: &mut dyn FnMut(&mut dyn Operation)) {
+                operate(self);
+            }
+
+            fn custom(
+                &mut self,
+                _id: Option<&iced::widget::Id>,
+                _bounds: iced::Rectangle,
+                state: &mut dyn std::any::Any,
+            ) {
+                let Some(state) = state.downcast_mut::<super::OutputState>() else {
+                    return;
+                };
+
+                state.enter(self.handle.clone());
+            }
+        }
+
+        EnterOutput { handle }
+    }
+
+    pub fn leave_output(handle: WlOutput) -> impl Operation {
+        struct LeaveOutput {
+            handle: WlOutput,
+        }
+
+        impl Operation for LeaveOutput {
+            fn traverse(&mut self, operate: &mut dyn FnMut(&mut dyn Operation)) {
+                operate(self);
+            }
+
+            fn custom(
+                &mut self,
+                _id: Option<&iced::widget::Id>,
+                _bounds: iced::Rectangle,
+                state: &mut dyn std::any::Any,
+            ) {
+                let Some(state) = state.downcast_mut::<super::OutputState>() else {
+                    return;
+                };
+
+                state.leave(self.handle.clone());
+            }
+        }
+
+        LeaveOutput { handle }
+    }
+}

snowcap/src/widget/wlr_tasklist.rs

@@ -16,9 +16,12 @@ use smithay_client_toolkit::{
     },
 };
 
-use crate::handlers::foreign_toplevel_management::{
-    ForeignToplevelData, ForeignToplevelInfo, ToplevelState,
-    WeakZwlrForeignToplevelManagementState, ZwlrForeignToplevelManagementState,
+use crate::{
+    handlers::foreign_toplevel_management::{
+        ForeignToplevelData, ForeignToplevelInfo, ToplevelState,
+        WeakZwlrForeignToplevelManagementState, ZwlrForeignToplevelManagementState,
+    },
+    widget::output::OutputState,
 };
 
 pub mod operation {
@@ -136,6 +139,7 @@ pub struct WlrTaskList<'a, Message, Theme = iced::Theme, Renderer = iced::Render
 /// Local state of the [`WlrTaskList`].
 #[derive(Default)]
 pub struct State {
+    output_state: OutputState,
     toplevel_list: Vec<Weak<ZwlrForeignToplevelHandleV1>>,
 
     pending_enter: Vec<(WlrTaskState, Weak<ZwlrForeignToplevelHandleV1>)>,
@@ -272,6 +276,7 @@ where
     ) {
         let state = tree.state.downcast_mut::<State>();
 
+        operation.custom(None, layout.bounds(), &mut state.output_state);
         operation.custom(None, layout.bounds(), state);
 
         operation.traverse(&mut |operation| {