mod event_handling; mod kb_owner; mod pointer_owner; pub mod wl_keyboard; pub mod wl_pointer; pub mod wl_touch; pub mod zwp_relative_pointer_manager_v1; pub mod zwp_relative_pointer_v1; pub use event_handling::NodeSeatState; use { crate::{ async_engine::SpawnedFuture, client::{Client, ClientError, ClientId}, cursor::{Cursor, KnownCursor}, fixed::Fixed, globals::{Global, GlobalName}, ifs::{ ipc, ipc::{ wl_data_device::{ClipboardIpc, WlDataDevice}, wl_data_source::WlDataSource, zwp_primary_selection_device_v1::{ PrimarySelectionIpc, ZwpPrimarySelectionDeviceV1, }, zwp_primary_selection_source_v1::ZwpPrimarySelectionSourceV1, IpcError, }, wl_seat::{ kb_owner::KbOwnerHolder, pointer_owner::PointerOwnerHolder, wl_keyboard::{WlKeyboard, WlKeyboardError, REPEAT_INFO_SINCE}, wl_pointer::WlPointer, wl_touch::WlTouch, zwp_relative_pointer_v1::ZwpRelativePointerV1, }, wl_surface::WlSurface, }, leaks::Tracker, object::Object, state::State, tree::{ generic_node_visitor, ContainerNode, ContainerSplit, Direction, FloatNode, FoundNode, Node, OutputNode, WorkspaceNode, }, utils::{ asyncevent::AsyncEvent, buffd::{MsgParser, MsgParserError}, clonecell::CloneCell, copyhashmap::CopyHashMap, errorfmt::ErrorFmt, linkedlist::LinkedNode, numcell::NumCell, rc_eq::rc_eq, }, wire::{ wl_seat::*, WlDataDeviceId, WlKeyboardId, WlPointerId, WlSeatId, ZwpPrimarySelectionDeviceV1Id, ZwpRelativePointerV1Id, }, xkbcommon::{XkbKeymap, XkbState}, }, ahash::{AHashMap, AHashSet}, jay_config::keyboard::mods::Modifiers, smallvec::SmallVec, std::{ cell::{Cell, RefCell}, collections::hash_map::Entry, mem, ops::{Deref, DerefMut}, rc::Rc, }, thiserror::Error, uapi::{c, Errno, OwnedFd}, }; const POINTER: u32 = 1; const KEYBOARD: u32 = 2; #[allow(dead_code)] const TOUCH: u32 = 4; #[allow(dead_code)] const MISSING_CAPABILITY: u32 = 0; pub const BTN_LEFT: u32 = 0x110; pub const SEAT_NAME_SINCE: u32 = 2; pub const PX_PER_SCROLL: f64 = 15.0; #[derive(Clone)] pub struct Dnd { pub seat: Rc, client: Rc, src: Option>, } pub struct DroppedDnd { dnd: Dnd, } impl Drop for DroppedDnd { fn drop(&mut self) { if let Some(src) = self.dnd.src.take() { ipc::detach_seat::(&src); } } } linear_ids!(SeatIds, SeatId); pub struct WlSeatGlobal { id: SeatId, name: GlobalName, state: Rc, seat_name: String, move_: Cell, move_start_pos: Cell<(Fixed, Fixed)>, extents_start_pos: Cell<(i32, i32)>, pos_time_usec: Cell, pos: Cell<(Fixed, Fixed)>, pointer_stack: RefCell>>, pointer_stack_modified: Cell, found_tree: RefCell>, keyboard_node: CloneCell>, pressed_keys: RefCell>, bindings: RefCell>>>, data_devices: RefCell>>>, primary_selection_devices: RefCell< AHashMap< ClientId, AHashMap>, >, >, repeat_rate: Cell<(i32, i32)>, kb_map: CloneCell>, kb_state: RefCell, cursor: CloneCell>>, tree_changed: Rc, selection: CloneCell>>, selection_serial: Cell, primary_selection: CloneCell>>, primary_selection_serial: Cell, pointer_owner: PointerOwnerHolder, kb_owner: KbOwnerHolder, dropped_dnd: RefCell>, shortcuts: CopyHashMap<(u32, u32), Modifiers>, queue_link: Cell>>>, tree_changed_handler: Cell>>, output: CloneCell>, desired_known_cursor: Cell>, changes: NumCell, } const CHANGE_CURSOR_MOVED: u32 = 1 << 0; const CHANGE_TREE: u32 = 1 << 1; impl WlSeatGlobal { pub fn new(name: GlobalName, seat_name: &str, state: &Rc) -> Rc { let slf = Rc::new(Self { id: state.seat_ids.next(), name, state: state.clone(), seat_name: seat_name.to_string(), move_: Cell::new(false), move_start_pos: Cell::new((Fixed(0), Fixed(0))), extents_start_pos: Cell::new((0, 0)), pos_time_usec: Cell::new(0), pos: Cell::new((Fixed(0), Fixed(0))), pointer_stack: RefCell::new(vec![]), pointer_stack_modified: Cell::new(false), found_tree: RefCell::new(vec![]), keyboard_node: CloneCell::new(state.root.clone()), pressed_keys: RefCell::new(Default::default()), bindings: Default::default(), data_devices: RefCell::new(Default::default()), primary_selection_devices: RefCell::new(Default::default()), repeat_rate: Cell::new((25, 250)), kb_map: CloneCell::new(state.default_keymap.clone()), kb_state: RefCell::new(state.default_keymap.state().unwrap()), cursor: Default::default(), tree_changed: Default::default(), selection: Default::default(), selection_serial: Cell::new(0), primary_selection: Default::default(), primary_selection_serial: Cell::new(0), pointer_owner: Default::default(), kb_owner: Default::default(), dropped_dnd: RefCell::new(None), shortcuts: Default::default(), queue_link: Cell::new(None), tree_changed_handler: Cell::new(None), output: CloneCell::new(state.dummy_output.get().unwrap()), desired_known_cursor: Cell::new(None), changes: NumCell::new(CHANGE_CURSOR_MOVED | CHANGE_TREE), }); let seat = slf.clone(); let future = state.eng.spawn(async move { loop { seat.tree_changed.triggered().await; seat.state.tree_changed_sent.set(false); seat.changes.or_assign(CHANGE_TREE); // log::info!("tree_changed"); seat.apply_changes(); } }); slf.tree_changed_handler.set(Some(future)); slf } pub fn add_data_device(&self, device: &Rc) { let mut dd = self.data_devices.borrow_mut(); dd.entry(device.client.id) .or_default() .insert(device.id, device.clone()); } pub fn remove_data_device(&self, device: &WlDataDevice) { let mut dd = self.data_devices.borrow_mut(); if let Entry::Occupied(mut e) = dd.entry(device.client.id) { e.get_mut().remove(&device.id); if e.get().is_empty() { e.remove(); } } } pub fn add_primary_selection_device(&self, device: &Rc) { let mut dd = self.primary_selection_devices.borrow_mut(); dd.entry(device.client.id) .or_default() .insert(device.id, device.clone()); } pub fn remove_primary_selection_device(&self, device: &ZwpPrimarySelectionDeviceV1) { let mut dd = self.primary_selection_devices.borrow_mut(); if let Entry::Occupied(mut e) = dd.entry(device.client.id) { e.get_mut().remove(&device.id); if e.get().is_empty() { e.remove(); } } } pub fn get_output(&self) -> Rc { self.output.get() } pub fn set_workspace(&self, ws: &Rc) { let tl = match self.keyboard_node.get().node_toplevel() { Some(tl) => tl, _ => return, }; if tl.tl_data().is_fullscreen.get() { return; } let old_ws = match tl.tl_data().workspace.get() { Some(ws) => ws, _ => return, }; if old_ws.id == ws.id { return; } let cn = match tl .tl_data() .parent .get() .and_then(|p| p.node_into_containing_node()) { Some(cn) => cn, _ => return, }; let kb_foci = collect_kb_foci(tl.clone().tl_into_node()); cn.cnode_remove_child2(tl.tl_as_node(), true); if !ws.visible.get() { for focus in kb_foci { old_ws.clone().node_do_focus(&focus, Direction::Unspecified); } } if tl.tl_data().is_floating.get() { self.state.map_floating( tl.clone(), tl.tl_data().float_width.get(), tl.tl_data().float_height.get(), ws, ); } else { self.state.map_tiled_on(tl, ws); } } pub fn mark_last_active(self: &Rc) { self.queue_link .set(Some(self.state.seat_queue.add_last(self.clone()))); } pub fn set_fullscreen(&self, fullscreen: bool) { if let Some(tl) = self.keyboard_node.get().node_toplevel() { tl.tl_set_fullscreen(fullscreen); } } pub fn get_fullscreen(&self) -> bool { if let Some(tl) = self.keyboard_node.get().node_toplevel() { return tl.tl_data().is_fullscreen.get(); } false } pub fn set_keymap(&self, keymap: &Rc) { let state = match keymap.state() { Ok(s) => s, Err(e) => { log::error!("Could not create keymap state: {}", ErrorFmt(e)); return; } }; self.kb_map.set(keymap.clone()); *self.kb_state.borrow_mut() = state; let bindings = self.bindings.borrow_mut(); for (id, client) in bindings.iter() { for seat in client.values() { let kbs = seat.keyboards.lock(); for kb in kbs.values() { let fd = match seat.keymap_fd(keymap) { Ok(fd) => fd, Err(e) => { log::error!("Could not creat a file descriptor to transfer the keymap to client {}: {}", id, ErrorFmt(e)); continue; } }; kb.send_keymap(wl_keyboard::XKB_V1, fd, keymap.map_len as _); } } } } pub fn prepare_for_lock(self: &Rc) { self.pointer_owner.revert_to_default(self); self.kb_owner.ungrab(self); } pub fn set_position(&self, x: i32, y: i32) { self.pos.set((Fixed::from_int(x), Fixed::from_int(y))); self.trigger_tree_changed(); 'set_output: { let outputs = self.state.outputs.lock(); for output in outputs.values() { if output.node.global.pos.get().contains(x, y) { self.output.set(output.node.clone()); break 'set_output; } } self.output.set(self.state.dummy_output.get().unwrap()); } } pub fn position(&self) -> (Fixed, Fixed) { self.pos.get() } pub fn render_ctx_changed(&self) { if let Some(cursor) = self.desired_known_cursor.get() { self.set_known_cursor(cursor); } } pub fn kb_parent_container(&self) -> Option> { if let Some(tl) = self.keyboard_node.get().node_toplevel() { if let Some(parent) = tl.tl_data().parent.get() { if let Some(container) = parent.node_into_container() { return Some(container); } } } None } pub fn get_mono(&self) -> Option { self.kb_parent_container() .map(|c| c.mono_child.get().is_some()) } pub fn get_split(&self) -> Option { self.kb_parent_container().map(|c| c.split.get()) } pub fn set_mono(&self, mono: bool) { if let Some(tl) = self.keyboard_node.get().node_toplevel() { if let Some(parent) = tl.tl_data().parent.get() { if let Some(container) = parent.node_into_container() { let node = if mono { Some(tl.deref()) } else { None }; container.set_mono(node); } } } } pub fn set_split(&self, axis: ContainerSplit) { if let Some(c) = self.kb_parent_container() { c.set_split(axis); } } pub fn create_split(&self, axis: ContainerSplit) { let tl = match self.keyboard_node.get().node_toplevel() { Some(tl) => tl, _ => return, }; if tl.tl_data().is_fullscreen.get() { return; } let ws = match tl.tl_data().workspace.get() { Some(ws) => ws, _ => return, }; let pn = match tl.tl_data().parent.get() { Some(pn) => pn, _ => return, }; if let Some(pn) = pn.node_into_containing_node() { let cn = ContainerNode::new(&self.state, &ws, pn.clone(), tl.clone(), axis); pn.cnode_replace_child(tl.tl_as_node(), cn); } } pub fn focus_parent(self: &Rc) { if let Some(tl) = self.keyboard_node.get().node_toplevel() { if let Some(parent) = tl.tl_data().parent.get() { self.focus_node(parent.cnode_into_node()); } } } pub fn get_floating(self: &Rc) -> Option { match self.keyboard_node.get().node_toplevel() { Some(tl) => Some(tl.tl_data().is_floating.get()), _ => None, } } pub fn set_floating(self: &Rc, floating: bool) { let tl = match self.keyboard_node.get().node_toplevel() { Some(tl) => tl, _ => return, }; let data = tl.tl_data(); if data.is_fullscreen.get() { return; } if data.is_floating.get() == floating { return; } let parent = match data.parent.get() { Some(p) => p, _ => return, }; if let Some(cn) = parent.node_into_containing_node() { if !floating { cn.cnode_remove_child2(tl.tl_as_node(), true); self.state.map_tiled(tl); } else if let Some(ws) = data.workspace.get() { cn.cnode_remove_child2(tl.tl_as_node(), true); let (width, height) = data.float_size(&ws); self.state.map_floating(tl, width, height, &ws); } } } pub fn get_rate(&self) -> (i32, i32) { self.repeat_rate.get() } pub fn set_rate(&self, rate: i32, delay: i32) { self.repeat_rate.set((rate, delay)); let bindings = self.bindings.borrow_mut(); for client in bindings.values() { for seat in client.values() { if seat.version >= REPEAT_INFO_SINCE { let kbs = seat.keyboards.lock(); for kb in kbs.values() { kb.send_repeat_info(rate, delay); } } } } } pub fn close(self: &Rc) { let kb_node = self.keyboard_node.get(); if let Some(tl) = kb_node.node_toplevel() { tl.tl_close(); } } pub fn move_focus(self: &Rc, direction: Direction) { let tl = match self.keyboard_node.get().node_toplevel() { Some(tl) => tl, _ => return, }; if direction == Direction::Down && tl.node_is_container() { tl.node_do_focus(self, direction); } else if let Some(p) = tl.tl_data().parent.get() { if let Some(c) = p.node_into_container() { c.move_focus_from_child(self, tl.deref(), direction); } } } pub fn move_focused(self: &Rc, direction: Direction) { let kb_node = self.keyboard_node.get(); if let Some(tl) = kb_node.node_toplevel() { if let Some(parent) = tl.tl_data().parent.get() { if let Some(c) = parent.node_into_container() { c.move_child(tl, direction); } } } } fn set_selection_( self: &Rc, field: &CloneCell>>, src: Option>, ) -> Result<(), WlSeatError> { if let Some(new) = &src { ipc::attach_seat::(new, self, ipc::Role::Selection)?; } if let Some(old) = field.set(src.clone()) { ipc::detach_seat::(&old); } if let Some(client) = self.keyboard_node.get().node_client() { match src { Some(src) => ipc::offer_source_to::(&src, &client), _ => T::for_each_device(self, client.id, |device| { T::send_selection(device, None); }), } // client.flush(); } Ok(()) } pub fn start_drag( self: &Rc, origin: &Rc, source: Option>, icon: Option>, serial: u32, ) -> Result<(), WlSeatError> { self.pointer_owner .start_drag(self, origin, source, icon, serial) } pub fn cancel_dnd(self: &Rc) { self.pointer_owner.cancel_dnd(self); } pub fn unset_selection(self: &Rc) { let _ = self.set_selection(None, None); } pub fn set_selection( self: &Rc, selection: Option>, serial: Option, ) -> Result<(), WlSeatError> { if let Some(serial) = serial { self.selection_serial.set(serial); } self.set_selection_::(&self.selection, selection) } pub fn may_modify_selection(&self, client: &Rc, serial: u32) -> bool { let dist = serial.wrapping_sub(self.selection_serial.get()) as i32; if dist < 0 { return false; } self.keyboard_node.get().node_client_id() == Some(client.id) } pub fn may_modify_primary_selection(&self, client: &Rc, serial: Option) -> bool { if let Some(serial) = serial { let dist = serial.wrapping_sub(self.primary_selection_serial.get()) as i32; if dist < 0 { return false; } } self.keyboard_node.get().node_client_id() == Some(client.id) || self.pointer_node().and_then(|n| n.node_client_id()) == Some(client.id) } pub fn unset_primary_selection(self: &Rc) { let _ = self.set_primary_selection(None, None); } pub fn set_primary_selection( self: &Rc, selection: Option>, serial: Option, ) -> Result<(), WlSeatError> { if let Some(serial) = serial { self.primary_selection_serial.set(serial); } self.set_selection_::(&self.primary_selection, selection) } pub fn set_known_cursor(&self, cursor: KnownCursor) { self.desired_known_cursor.set(Some(cursor)); let cursors = match self.state.cursors.get() { Some(c) => c, None => { self.set_cursor2(None); return; } }; let tpl = match cursor { KnownCursor::Default => &cursors.default, KnownCursor::ResizeLeftRight => &cursors.resize_left_right, KnownCursor::ResizeTopBottom => &cursors.resize_top_bottom, KnownCursor::ResizeTopLeft => &cursors.resize_top_left, KnownCursor::ResizeTopRight => &cursors.resize_top_right, KnownCursor::ResizeBottomLeft => &cursors.resize_bottom_left, KnownCursor::ResizeBottomRight => &cursors.resize_bottom_right, }; self.set_cursor2(Some(tpl.instantiate())); } pub fn set_app_cursor(&self, cursor: Option>) { self.set_cursor2(cursor); self.desired_known_cursor.set(None); } fn set_cursor2(&self, cursor: Option>) { if let Some(old) = self.cursor.get() { if let Some(new) = cursor.as_ref() { if rc_eq(&old, new) { return; } } old.handle_unset(); } if let Some(cursor) = cursor.as_ref() { let (x, y) = self.pos.get(); cursor.set_position(x.round_down(), y.round_down()); } self.cursor.set(cursor); } pub fn dnd_icon(&self) -> Option> { self.pointer_owner.dnd_icon() } pub fn remove_dnd_icon(&self) { self.pointer_owner.remove_dnd_icon(); } pub fn get_cursor(&self) -> Option> { self.cursor.get() } pub fn clear(self: &Rc) { mem::take(self.pointer_stack.borrow_mut().deref_mut()); mem::take(self.found_tree.borrow_mut().deref_mut()); self.keyboard_node.set(self.state.root.clone()); self.state .root .clone() .node_visit(&mut generic_node_visitor(|node| { node.node_seat_state().on_seat_remove(self); })); self.bindings.borrow_mut().clear(); self.data_devices.borrow_mut().clear(); self.primary_selection_devices.borrow_mut().clear(); self.cursor.set(None); self.selection.set(None); self.primary_selection.set(None); self.pointer_owner.clear(); self.kb_owner.clear(); *self.dropped_dnd.borrow_mut() = None; self.queue_link.set(None); self.tree_changed_handler.set(None); self.output.set(self.state.dummy_output.get().unwrap()); } pub fn id(&self) -> SeatId { self.id } pub fn seat_name(&self) -> &str { &self.seat_name } fn bind_( self: Rc, id: WlSeatId, client: &Rc, version: u32, ) -> Result<(), WlSeatError> { let obj = Rc::new(WlSeat { global: self.clone(), id, client: client.clone(), pointers: Default::default(), relative_pointers: Default::default(), keyboards: Default::default(), version, tracker: Default::default(), }); track!(client, obj); client.add_client_obj(&obj)?; obj.send_capabilities(); if version >= SEAT_NAME_SINCE { obj.send_name(&self.seat_name); } { let mut bindings = self.bindings.borrow_mut(); let bindings = bindings.entry(client.id).or_insert_with(Default::default); bindings.insert(id, obj.clone()); } Ok(()) } } global_base!(WlSeatGlobal, WlSeat, WlSeatError); impl Global for WlSeatGlobal { fn singleton(&self) -> bool { false } fn version(&self) -> u32 { 8 } fn break_loops(&self) { self.bindings.borrow_mut().clear(); self.queue_link.take(); self.tree_changed_handler.take(); } } dedicated_add_global!(WlSeatGlobal, seats); pub struct WlSeat { pub global: Rc, pub id: WlSeatId, pub client: Rc, pointers: CopyHashMap>, relative_pointers: CopyHashMap>, keyboards: CopyHashMap>, version: u32, tracker: Tracker, } const READ_ONLY_KEYMAP_SINCE: u32 = 7; impl WlSeat { fn send_capabilities(self: &Rc) { self.client.event(Capabilities { self_id: self.id, capabilities: POINTER | KEYBOARD, }) } fn send_name(self: &Rc, name: &str) { self.client.event(Name { self_id: self.id, name, }) } pub fn move_(&self, node: &Rc) { self.global.move_(node); } fn get_pointer(self: &Rc, parser: MsgParser<'_, '_>) -> Result<(), WlSeatError> { let req: GetPointer = self.client.parse(&**self, parser)?; let p = Rc::new(WlPointer::new(req.id, self)); track!(self.client, p); self.client.add_client_obj(&p)?; self.pointers.set(req.id, p); Ok(()) } fn get_keyboard(self: &Rc, parser: MsgParser<'_, '_>) -> Result<(), WlSeatError> { let req: GetKeyboard = self.client.parse(&**self, parser)?; let p = Rc::new(WlKeyboard::new(req.id, self)); track!(self.client, p); self.client.add_client_obj(&p)?; self.keyboards.set(req.id, p.clone()); let keymap = self.global.kb_map.get(); p.send_keymap( wl_keyboard::XKB_V1, self.keymap_fd(&keymap)?, keymap.map_len as _, ); if self.version >= REPEAT_INFO_SINCE { let (rate, delay) = self.global.repeat_rate.get(); p.send_repeat_info(rate, delay); } Ok(()) } pub fn keymap_fd(&self, keymap: &XkbKeymap) -> Result, WlKeyboardError> { if self.version >= READ_ONLY_KEYMAP_SINCE { return Ok(keymap.map.clone()); } let fd = match uapi::memfd_create("shared-keymap", c::MFD_CLOEXEC) { Ok(fd) => fd, Err(e) => return Err(WlKeyboardError::KeymapMemfd(e.into())), }; let target = keymap.map_len as c::off_t; let mut pos = 0; while pos < target { let rem = target - pos; let res = uapi::sendfile(fd.raw(), keymap.map.raw(), Some(&mut pos), rem as usize); match res { Ok(_) | Err(Errno(c::EINTR)) => {} Err(e) => return Err(WlKeyboardError::KeymapCopy(e.into())), } } Ok(Rc::new(fd)) } fn get_touch(self: &Rc, parser: MsgParser<'_, '_>) -> Result<(), WlSeatError> { let req: GetTouch = self.client.parse(&**self, parser)?; let p = Rc::new(WlTouch::new(req.id, self)); track!(self.client, p); self.client.add_client_obj(&p)?; Ok(()) } fn release(&self, parser: MsgParser<'_, '_>) -> Result<(), WlSeatError> { let _req: Release = self.client.parse(self, parser)?; { let mut bindings = self.global.bindings.borrow_mut(); if let Entry::Occupied(mut hm) = bindings.entry(self.client.id) { hm.get_mut().remove(&self.id); if hm.get().is_empty() { hm.remove(); } } } self.client.remove_obj(self)?; Ok(()) } } object_base! { WlSeat; GET_POINTER => get_pointer, GET_KEYBOARD => get_keyboard, GET_TOUCH => get_touch, RELEASE => release, } impl Object for WlSeat { fn num_requests(&self) -> u32 { if self.version < 5 { GET_TOUCH + 1 } else { RELEASE + 1 } } fn break_loops(&self) { { let mut bindings = self.global.bindings.borrow_mut(); if let Entry::Occupied(mut hm) = bindings.entry(self.client.id) { hm.get_mut().remove(&self.id); if hm.get().is_empty() { hm.remove(); } } } self.pointers.clear(); self.relative_pointers.clear(); self.keyboards.clear(); } } dedicated_add_obj!(WlSeat, WlSeatId, seats); #[derive(Debug, Error)] pub enum WlSeatError { #[error(transparent)] ClientError(Box), #[error(transparent)] IpcError(#[from] IpcError), #[error("Parsing failed")] MsgParserError(#[source] Box), #[error(transparent)] WlKeyboardError(Box), } efrom!(WlSeatError, ClientError); efrom!(WlSeatError, MsgParserError); efrom!(WlSeatError, WlKeyboardError); pub fn collect_kb_foci2(node: Rc, seats: &mut SmallVec<[Rc; 3]>) { node.node_visit(&mut generic_node_visitor(|node| { node.node_seat_state().for_each_kb_focus(|s| seats.push(s)); })); } pub fn collect_kb_foci(node: Rc) -> SmallVec<[Rc; 3]> { let mut res = SmallVec::new(); collect_kb_foci2(node, &mut res); res }