use crate::ecs::{Entity, World};
use crate::surface::InputEvent;
use crate::widget::Parent;
use super::gesture::GestureEvent;
pub struct Focusable;
#[derive(Default)]
pub struct FocusState {
pub focused: Option<Entity>,
}
pub struct KeyHandler {
pub on_key: fn(&mut World, Entity, &InputEvent) -> bool,
}
pub fn focus_on_tap(world: &mut World, event: &GestureEvent) {
let target = match event {
GestureEvent::Tap { target, .. } => *target,
_ => return,
};
let focused_entity = find_focusable(world, target);
if let Some(fs) = world.resource_mut::<FocusState>() {
fs.focused = focused_entity;
}
}
fn find_focusable(world: &World, entity: Entity) -> Option<Entity> {
if super::entity_or_ancestor_disabled(world, entity) {
return None;
}
let mut cur = entity;
loop {
if world.get::<Focusable>(cur).is_some() {
return Some(cur);
}
match world.get::<Parent>(cur) {
Some(p) => cur = p.0,
None => return None,
}
}
}
pub fn key_dispatch(world: &mut World, event: &InputEvent) {
let target = match world.resource::<FocusState>() {
Some(fs) => match fs.focused {
Some(e) => e,
None => return,
},
None => return,
};
match event {
InputEvent::Key { .. } | InputEvent::CharInput { .. } => {}
_ => return,
}
if super::entity_or_ancestor_disabled(world, target) {
return;
}
let handler_fn = world.get::<KeyHandler>(target).map(|h| h.on_key);
if let Some(f) = handler_fn {
f(world, target, event);
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::widget::UserState;
#[test]
fn focus_skips_disabled_subtree() {
let mut world = World::new();
let parent = world.spawn();
let child = world.spawn();
world.insert(child, Parent(parent));
world.insert(child, Focusable);
world.insert(parent, UserState::Disabled);
assert_eq!(find_focusable(&world, child), None);
}
#[test]
fn focus_ignores_disabled_self() {
let mut world = World::new();
let e = world.spawn();
world.insert(e, Focusable);
world.insert(e, UserState::Disabled);
assert_eq!(find_focusable(&world, e), None);
}
#[test]
fn focus_walks_through_non_disabled() {
let mut world = World::new();
let parent = world.spawn();
let child = world.spawn();
world.insert(child, Parent(parent));
world.insert(parent, Focusable);
assert_eq!(find_focusable(&world, child), Some(parent));
}
}