Add stackable on_key_down and on_key_up event handlers

examples/counter/src/main.rs

@@ -1,6 +1,5 @@
 use floem::{
-    event::{Event, EventListener},
-    keyboard::{Key, NamedKey},
+    keyboard::{Key, ModifiersState, NamedKey},
     peniko::Color,
     reactive::create_signal,
     unit::UnitExt,
@@ -74,13 +73,11 @@ fn app_view() -> impl View {
     });
 
     let id = view.id();
-    view.on_event_stop(EventListener::KeyUp, move |e| {
-        if let Event::KeyUp(e) = e {
-            if e.key.logical_key == Key::Named(NamedKey::F11) {
-                id.inspect();
-            }
-        }
-    })
+    view.on_key_up(
+        Key::Named(NamedKey::F11),
+        ModifiersState::empty(),
+        move |_| id.inspect(),
+    )
 }
 
 fn main() {

src/context.rs

@@ -901,6 +901,12 @@ impl<'a> EventCx<'a> {
             _ => (),
         }
 
+        for handler in &view.view_data().event_handlers {
+            if (handler)(&event).is_processed() {
+                return EventPropagation::Stop;
+            }
+        }
+
         if let Some(listener) = event.listener() {
             if let Some(action) = self.get_event_listener(id, &listener) {
                 let should_run = if let Some(pos) = event.point() {

src/view_data.rs

@@ -19,9 +19,16 @@ use std::{collections::HashMap, marker::PhantomData, time::Duration};
 use taffy::node::Node;
 
 /// A stack of view attributes. Each entry is associated with a view decorator call.
-#[derive(Default)]
 pub(crate) struct Stack<T> {
-    stack: SmallVec<[T; 1]>,
+    pub(crate) stack: SmallVec<[T; 1]>,
+}
+
+impl<T> Default for Stack<T> {
+    fn default() -> Self {
+        Stack {
+            stack: SmallVec::new(),
+        }
+    }
 }
 
 pub(crate) struct StackOffset<T> {
@@ -57,13 +64,15 @@ impl<T> Stack<T> {
 pub struct ViewData {
     pub(crate) id: Id,
     pub(crate) style: Stack<Style>,
+    pub(crate) event_handlers: Vec<Box<EventCallback>>,
 }
 
 impl ViewData {
     pub fn new(id: Id) -> Self {
         Self {
             id,
             style: Default::default(),
+            event_handlers: Default::default(),
         }
     }
     pub fn id(&self) -> Id {

src/views/decorator.rs

@@ -1,5 +1,6 @@
 use floem_reactive::{create_effect, create_updater};
 use kurbo::{Point, Rect};
+use winit::keyboard::{Key, ModifiersState};
 
 use crate::{
     action::{set_window_menu, set_window_title, update_window_scale},
@@ -91,6 +92,44 @@ pub trait Decorators: View + Sized {
         self
     }
 
+    /// Add an handler for pressing down a specific key.
+    fn on_key_down(
+        mut self,
+        key: Key,
+        modifiers: ModifiersState,
+        action: impl Fn(&Event) + 'static,
+    ) -> Self {
+        self.view_data_mut().event_handlers.push(Box::new(move |e| {
+            if let Event::KeyDown(ke) = e {
+                if ke.key.logical_key == key && ke.modifiers == modifiers {
+                    action(e);
+                    return EventPropagation::Stop;
+                }
+            }
+            EventPropagation::Continue
+        }));
+        self
+    }
+
+    /// Add an handler for a specific key being released.
+    fn on_key_up(
+        mut self,
+        key: Key,
+        modifiers: ModifiersState,
+        action: impl Fn(&Event) + 'static,
+    ) -> Self {
+        self.view_data_mut().event_handlers.push(Box::new(move |e| {
+            if let Event::KeyUp(ke) = e {
+                if ke.key.logical_key == key && ke.modifiers == modifiers {
+                    action(e);
+                    return EventPropagation::Stop;
+                }
+            }
+            EventPropagation::Continue
+        }));
+        self
+    }
+
     /// Add an event hanlder for the given EventListener
     ///
     /// This event will be handled with the given handler and the event will continue propagating

src/window_handle.rs

@@ -198,6 +198,13 @@ impl WindowHandle {
                 }
 
                 if !processed {
+                    for handler in &self.view.main.view_data().event_handlers {
+                        if (handler)(&event).is_processed() {
+                            processed = true;
+                            break;
+                        }
+                    }
+
                     if let Some(listener) = event.listener() {
                         if let Some(action) = cx.get_event_listener(self.view.main.id(), &listener)
                         {