use crate::{ backend::render::{cursor::CursorState, SplitRenderElements}, shell::{ focus::target::PointerFocusTarget, grabs::{ReleaseMode, ResizeEdge}, }, state::State, utils::{ iced::{IcedElement, Program}, prelude::*, }, }; use calloop::LoopHandle; use cosmic::{ config::Density, iced::{Color, Command}, widget::mouse_area, Apply, }; use smithay::{ backend::{ input::KeyState, renderer::{ element::{ memory::MemoryRenderBufferRenderElement, surface::WaylandSurfaceRenderElement, AsRenderElements, }, ImportAll, ImportMem, Renderer, }, }, desktop::{space::SpaceElement, WindowSurfaceType}, input::{ keyboard::{KeyboardTarget, KeysymHandle, ModifiersState}, pointer::{ AxisFrame, ButtonEvent, CursorIcon, CursorImageStatus, GestureHoldBeginEvent, GestureHoldEndEvent, GesturePinchBeginEvent, GesturePinchEndEvent, GesturePinchUpdateEvent, GestureSwipeBeginEvent, GestureSwipeEndEvent, GestureSwipeUpdateEvent, MotionEvent, PointerTarget, RelativeMotionEvent, }, touch::{ DownEvent, MotionEvent as TouchMotionEvent, OrientationEvent, ShapeEvent, TouchTarget, UpEvent, }, Seat, }, output::Output, reexports::wayland_server::protocol::wl_surface::WlSurface, render_elements, utils::{IsAlive, Logical, Point, Rectangle, Serial, Size}, wayland::seat::WaylandFocus, }; use std::{ borrow::Cow, fmt, hash::Hash, sync::{ atomic::{AtomicBool, AtomicU8, Ordering}, Arc, Mutex, }, }; use wayland_backend::server::ObjectId; use super::{ surface::{RESIZE_BORDER, SSD_HEIGHT}, CosmicSurface, }; #[derive(Clone, PartialEq, Eq, Hash)] pub struct CosmicWindow(pub(super) IcedElement); impl fmt::Debug for CosmicWindow { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { f.debug_struct("CosmicWindow") .field("internal", &self.0) .finish_non_exhaustive() } } #[derive(Clone)] pub struct CosmicWindowInternal { pub(super) window: CosmicSurface, activated: Arc, /// TODO: This needs to be per seat pointer_entered: Arc, last_seat: Arc, Serial)>>>, last_title: Arc>, } impl fmt::Debug for CosmicWindowInternal { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { f.debug_struct("CosmicWindowInternal") .field("window", &self.window) .field("activated", &self.activated.load(Ordering::SeqCst)) .field("pointer_entered", &self.pointer_entered) // skip seat to avoid loop .field("last_seat", &"...") .finish() } } #[repr(u8)] #[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)] pub enum Focus { Header = 1, ResizeTop, ResizeLeft, ResizeRight, ResizeBottom, ResizeTopRight, ResizeTopLeft, ResizeBottomRight, ResizeBottomLeft, } impl Focus { pub fn under( surface: &CosmicSurface, header_height: i32, location: Point, ) -> Option { let geo = surface.geometry(); let loc = location.to_i32_round::() - geo.loc; if loc.y < 0 && loc.x < 0 { Some(Focus::ResizeTopLeft) } else if loc.y < 0 && loc.x >= geo.size.w { Some(Focus::ResizeTopRight) } else if loc.y < 0 { Some(Focus::ResizeTop) } else if loc.y >= header_height + geo.size.h && loc.x < 0 { Some(Focus::ResizeBottomLeft) } else if loc.y >= header_height + geo.size.h && loc.x >= geo.size.w { Some(Focus::ResizeBottomRight) } else if loc.y >= header_height + geo.size.h { Some(Focus::ResizeBottom) } else if loc.x < 0 { Some(Focus::ResizeLeft) } else if loc.x >= geo.size.w { Some(Focus::ResizeRight) } else if loc.y < header_height { Some(Focus::Header) } else { None } } pub fn cursor_shape(&self) -> CursorIcon { match self { Focus::ResizeTopLeft => CursorIcon::NwResize, Focus::ResizeTopRight => CursorIcon::NeResize, Focus::ResizeTop => CursorIcon::NResize, Focus::ResizeBottomLeft => CursorIcon::SwResize, Focus::ResizeBottomRight => CursorIcon::SeResize, Focus::ResizeBottom => CursorIcon::SResize, Focus::ResizeLeft => CursorIcon::WResize, Focus::ResizeRight => CursorIcon::EResize, Focus::Header => CursorIcon::Default, } } pub unsafe fn from_u8(value: u8) -> Option { match value { 0 => None, focus => unsafe { Some(std::mem::transmute::(focus)) }, } } } impl CosmicWindowInternal { pub fn swap_focus(&self, focus: Option) -> Option { let value = focus.map_or(0, |x| x as u8); unsafe { Focus::from_u8(self.pointer_entered.swap(value, Ordering::SeqCst)) } } pub fn current_focus(&self) -> Option { unsafe { Focus::from_u8(self.pointer_entered.load(Ordering::SeqCst)) } } /// returns if the window has any current or pending server-side decorations pub fn has_ssd(&self, pending: bool) -> bool { !self.window.is_decorated(pending) } } impl CosmicWindow { pub fn new( window: impl Into, handle: LoopHandle<'static, crate::state::State>, theme: cosmic::Theme, ) -> CosmicWindow { let window = window.into(); let width = window.geometry().size.w; let last_title = window.title(); CosmicWindow(IcedElement::new( CosmicWindowInternal { window, activated: Arc::new(AtomicBool::new(false)), pointer_entered: Arc::new(AtomicU8::new(0)), last_seat: Arc::new(Mutex::new(None)), last_title: Arc::new(Mutex::new(last_title)), }, (width, SSD_HEIGHT), handle, theme, )) } pub fn pending_size(&self) -> Option> { self.0.with_program(|p| { let mut size = p.window.pending_size()?; if p.has_ssd(true) { size.h += SSD_HEIGHT; } Some(size) }) } pub fn set_geometry(&self, geo: Rectangle) { self.0.with_program(|p| { let loc = ( geo.loc.x, geo.loc.y + if p.has_ssd(true) { SSD_HEIGHT } else { 0 }, ); let size = ( geo.size.w, std::cmp::max(geo.size.h - if p.has_ssd(true) { SSD_HEIGHT } else { 0 }, 0), ); p.window .set_geometry(Rectangle::from_loc_and_size(loc, size)); }); } pub fn on_commit(&self, surface: &WlSurface) { let mut geo = None; self.0.with_program(|p| { if &p.window == surface { p.window.0.on_commit(); geo = Some(p.window.geometry()); } }); if let Some(geo) = geo { self.0.resize(Size::from((geo.size.w, SSD_HEIGHT))); } } pub fn surface(&self) -> CosmicSurface { self.0.with_program(|p| p.window.clone()) } pub fn focus_under( &self, mut relative_pos: Point, ) -> Option<(PointerFocusTarget, Point)> { self.0.with_program(|p| { let mut offset = Point::from((0., 0.)); let mut window_ui = None; if p.has_ssd(false) { let geo = p.window.geometry(); let point_i32 = relative_pos.to_i32_round::(); if (point_i32.x - geo.loc.x >= -RESIZE_BORDER && point_i32.x - geo.loc.x < 0) || (point_i32.y - geo.loc.y >= -RESIZE_BORDER && point_i32.y - geo.loc.y < 0) || (point_i32.x - geo.loc.x >= geo.size.w && point_i32.x - geo.loc.x < geo.size.w + RESIZE_BORDER) || (point_i32.y - geo.loc.y >= geo.size.h && point_i32.y - geo.loc.y < geo.size.h + SSD_HEIGHT + RESIZE_BORDER) { window_ui = Some(( PointerFocusTarget::WindowUI(self.clone()), Point::from((0., 0.)), )); } if point_i32.y - geo.loc.y < SSD_HEIGHT { window_ui = Some(( PointerFocusTarget::WindowUI(self.clone()), Point::from((0., 0.)), )); } relative_pos.y -= SSD_HEIGHT as f64; offset.y += SSD_HEIGHT as f64; } p.window .0 .surface_under(relative_pos, WindowSurfaceType::ALL) .map(|(surface, surface_offset)| { ( PointerFocusTarget::WlSurface { surface, toplevel: Some(p.window.clone().into()), }, (offset + surface_offset.to_f64()), ) }) .or(window_ui) }) } pub fn contains_surface(&self, window: &CosmicSurface) -> bool { self.0.with_program(|p| &p.window == window) } pub fn offset(&self) -> Point { let has_ssd = self.0.with_program(|p| p.has_ssd(false)); if has_ssd { Point::from((0, SSD_HEIGHT)) } else { Point::from((0, 0)) } } pub(super) fn loop_handle(&self) -> LoopHandle<'static, crate::state::State> { self.0.loop_handle() } pub fn split_render_elements( &self, renderer: &mut R, location: smithay::utils::Point, scale: smithay::utils::Scale, alpha: f32, ) -> SplitRenderElements where R: Renderer + ImportAll + ImportMem, ::TextureId: Send + Clone + 'static, C: From>, { let has_ssd = self.0.with_program(|p| p.has_ssd(false)); let window_loc = if has_ssd { location + Point::from((0, (SSD_HEIGHT as f64 * scale.y) as i32)) } else { location }; let mut elements = SplitRenderElements::default(); elements.extend_map( self.0.with_program(|p| { p.window .split_render_elements::>( renderer, window_loc, scale, alpha, ) }), C::from, ); if has_ssd { let ssd_loc = location + self .0 .with_program(|p| p.window.geometry().loc) .to_physical_precise_round(scale); elements.w_elements.extend( AsRenderElements::::render_elements::>( &self.0, renderer, ssd_loc, scale, alpha, ) .into_iter() .map(C::from), ) } elements } pub(crate) fn set_theme(&self, theme: cosmic::Theme) { self.0.set_theme(theme); } pub(crate) fn force_redraw(&self) { self.0.force_redraw(); } } #[derive(Debug, Clone, Copy)] pub enum Message { DragStart, Minimize, Maximize, Close, Menu, } impl Program for CosmicWindowInternal { type Message = Message; fn update( &mut self, message: Self::Message, loop_handle: &LoopHandle<'static, crate::state::State>, ) -> Command { match message { Message::DragStart => { if let Some((seat, serial)) = self.last_seat.lock().unwrap().clone() { if let Some(surface) = self.window.wl_surface().map(Cow::into_owned) { loop_handle.insert_idle(move |state| { let res = state.common.shell.write().unwrap().move_request( &surface, &seat, serial, ReleaseMode::NoMouseButtons, false, &state.common.config, &state.common.event_loop_handle, &state.common.xdg_activation_state, false, ); if let Some((grab, focus)) = res { if grab.is_touch_grab() { seat.get_touch().unwrap().set_grab(state, grab, serial); } else { seat.get_pointer() .unwrap() .set_grab(state, grab, serial, focus); } } }); } } } Message::Minimize => { if let Some(surface) = self.window.wl_surface().map(Cow::into_owned) { loop_handle.insert_idle(move |state| { let mut shell = state.common.shell.write().unwrap(); if let Some(mapped) = shell.element_for_surface(&surface).cloned() { shell.minimize_request(&mapped) } }); } } Message::Maximize => { if let Some(surface) = self.window.wl_surface().map(Cow::into_owned) { loop_handle.insert_idle(move |state| { let mut shell = state.common.shell.write().unwrap(); if let Some(mapped) = shell.element_for_surface(&surface).cloned() { let seat = shell.seats.last_active().clone(); shell.maximize_toggle(&mapped, &seat) } }); } } Message::Close => self.window.close(), Message::Menu => { if let Some((seat, serial)) = self.last_seat.lock().unwrap().clone() { if let Some(surface) = self.window.wl_surface().map(Cow::into_owned) { loop_handle.insert_idle(move |state| { let shell = state.common.shell.read().unwrap(); if let Some(mapped) = shell.element_for_surface(&surface).cloned() { let position = if let Some((output, set)) = shell.workspaces.sets.iter().find(|(_, set)| { set.sticky_layer.mapped().any(|m| m == &mapped) }) { set.sticky_layer .element_geometry(&mapped) .unwrap() .loc .to_global(output) } else if let Some(workspace) = shell.space_for(&mapped) { let Some(elem_geo) = workspace.element_geometry(&mapped) else { return; }; elem_geo.loc.to_global(&workspace.output) } else { return; }; let pointer = seat.get_pointer().unwrap(); let mut cursor = pointer.current_location().to_i32_round(); cursor.y -= SSD_HEIGHT; let res = shell.menu_request( &surface, &seat, serial, cursor - position.as_logical(), false, &state.common.config, &state.common.event_loop_handle, ); std::mem::drop(shell); if let Some((grab, focus)) = res { pointer.set_grab(state, grab, serial, focus); } } }); } } } } Command::none() } fn background_color(&self, theme: &cosmic::Theme) -> Color { if self.window.is_maximized(false) { theme.cosmic().background.base.into() } else { Color::TRANSPARENT } } fn view(&self) -> cosmic::Element<'_, Self::Message> { let mut header = cosmic::widget::header_bar() .title(self.last_title.lock().unwrap().clone()) .on_drag(Message::DragStart) .on_close(Message::Close) .focused(self.window.is_activated(false)) .density(Density::Compact) .on_double_click(Message::Maximize); if cosmic::config::show_minimize() { header = header.on_minimize(Message::Minimize); } if cosmic::config::show_maximize() { header = header.on_maximize(Message::Maximize); } let header = header.apply(mouse_area).on_right_press(Message::Menu); header.into() } } impl IsAlive for CosmicWindow { fn alive(&self) -> bool { self.0.with_program(|p| p.window.alive()) } } impl SpaceElement for CosmicWindow { fn bbox(&self) -> Rectangle { self.0.with_program(|p| { let mut bbox = SpaceElement::bbox(&p.window); if p.has_ssd(false) { bbox.loc -= Point::from((RESIZE_BORDER, RESIZE_BORDER)); bbox.size += Size::from((RESIZE_BORDER * 2, RESIZE_BORDER * 2)); bbox.size.h += SSD_HEIGHT; } bbox }) } fn is_in_input_region(&self, point: &Point) -> bool { self.focus_under(*point).is_some() } fn set_activate(&self, activated: bool) { if self .0 .with_program(|p| p.activated.load(Ordering::SeqCst) != activated) { SpaceElement::set_activate(&self.0, activated); self.0.force_redraw(); self.0.with_program(|p| { p.activated.store(activated, Ordering::SeqCst); SpaceElement::set_activate(&p.window, activated); }); } } #[profiling::function] fn output_enter(&self, output: &Output, overlap: Rectangle) { SpaceElement::output_enter(&self.0, output, overlap); self.0 .with_program(|p| SpaceElement::output_enter(&p.window, output, overlap)); } #[profiling::function] fn output_leave(&self, output: &Output) { SpaceElement::output_leave(&self.0, output); self.0 .with_program(|p| SpaceElement::output_leave(&p.window, output)); } fn geometry(&self) -> Rectangle { self.0.with_program(|p| { let mut geo = SpaceElement::geometry(&p.window); if p.has_ssd(false) { geo.size.h += SSD_HEIGHT; } geo }) } fn z_index(&self) -> u8 { self.0.with_program(|p| SpaceElement::z_index(&p.window)) } #[profiling::function] fn refresh(&self) { SpaceElement::refresh(&self.0); if self.0.with_program(|p| { SpaceElement::refresh(&p.window); let title = p.window.title(); let mut last_title = p.last_title.lock().unwrap(); if *last_title != title { *last_title = title; true } else { false } }) { self.0.force_update(); } } } impl KeyboardTarget for CosmicWindow { fn enter( &self, seat: &Seat, data: &mut State, keys: Vec>, serial: Serial, ) { self.0 .with_program(|p| KeyboardTarget::enter(&p.window, seat, data, keys, serial)) } fn leave(&self, seat: &Seat, data: &mut State, serial: Serial) { self.0 .with_program(|p| KeyboardTarget::leave(&p.window, seat, data, serial)) } fn key( &self, seat: &Seat, data: &mut State, key: KeysymHandle<'_>, state: KeyState, serial: Serial, time: u32, ) { self.0 .with_program(|p| KeyboardTarget::key(&p.window, seat, data, key, state, serial, time)) } fn modifiers( &self, seat: &Seat, data: &mut State, modifiers: ModifiersState, serial: Serial, ) { self.0 .with_program(|p| KeyboardTarget::modifiers(&p.window, seat, data, modifiers, serial)) } } impl PointerTarget for CosmicWindow { fn enter(&self, seat: &Seat, data: &mut State, event: &MotionEvent) { let mut event = event.clone(); self.0.with_program(|p| { if p.has_ssd(false) { let Some(next) = Focus::under(&p.window, SSD_HEIGHT, event.location) else { return; }; let old_focus = p.swap_focus(Some(next)); assert_eq!(old_focus, None); let cursor_state = seat.user_data().get::().unwrap(); cursor_state.lock().unwrap().set_shape(next.cursor_shape()); let cursor_status = seat.user_data().get::>().unwrap(); *cursor_status.lock().unwrap() = CursorImageStatus::default_named(); } }); event.location -= self.0.with_program(|p| p.window.geometry().loc.to_f64()); PointerTarget::enter(&self.0, seat, data, &event) } fn motion(&self, seat: &Seat, data: &mut State, event: &MotionEvent) { let mut event = event.clone(); self.0.with_program(|p| { if p.has_ssd(false) { let Some(next) = Focus::under(&p.window, SSD_HEIGHT, event.location) else { return; }; let _previous = p.swap_focus(Some(next)); let cursor_state = seat.user_data().get::().unwrap(); cursor_state.lock().unwrap().set_shape(next.cursor_shape()); let cursor_status = seat.user_data().get::>().unwrap(); *cursor_status.lock().unwrap() = CursorImageStatus::default_named(); } }); event.location -= self.0.with_program(|p| p.window.geometry().loc.to_f64()); PointerTarget::motion(&self.0, seat, data, &event) } fn relative_motion( &self, _seat: &Seat, _data: &mut State, _event: &RelativeMotionEvent, ) { } fn button(&self, seat: &Seat, data: &mut State, event: &ButtonEvent) { match self.0.with_program(|p| p.current_focus()) { Some(Focus::Header) => { self.0.with_program(|p| { *p.last_seat.lock().unwrap() = Some((seat.clone(), event.serial)); }); PointerTarget::button(&self.0, seat, data, event) } Some(x) => { let serial = event.serial; let seat = seat.clone(); let Some(surface) = self.wl_surface().map(Cow::into_owned) else { return; }; self.0.loop_handle().insert_idle(move |state| { let res = state.common.shell.write().unwrap().resize_request( &surface, &seat, serial, match x { Focus::ResizeTop => ResizeEdge::TOP, Focus::ResizeTopLeft => ResizeEdge::TOP_LEFT, Focus::ResizeTopRight => ResizeEdge::TOP_RIGHT, Focus::ResizeBottom => ResizeEdge::BOTTOM, Focus::ResizeBottomLeft => ResizeEdge::BOTTOM_LEFT, Focus::ResizeBottomRight => ResizeEdge::BOTTOM_RIGHT, Focus::ResizeLeft => ResizeEdge::LEFT, Focus::ResizeRight => ResizeEdge::RIGHT, Focus::Header => unreachable!(), }, false, ); if let Some((grab, focus)) = res { if grab.is_touch_grab() { seat.get_touch().unwrap().set_grab(state, grab, serial); } else { seat.get_pointer() .unwrap() .set_grab(state, grab, serial, focus); } } }); } None => {} } } fn axis(&self, seat: &Seat, data: &mut State, frame: AxisFrame) { match self.0.with_program(|p| p.current_focus()) { Some(Focus::Header) => PointerTarget::axis(&self.0, seat, data, frame), _ => {} } } fn frame(&self, seat: &Seat, data: &mut State) { match self.0.with_program(|p| p.current_focus()) { Some(Focus::Header) => PointerTarget::frame(&self.0, seat, data), _ => {} } } fn leave(&self, seat: &Seat, data: &mut State, serial: Serial, time: u32) { self.0.with_program(|p| { let cursor_state = seat.user_data().get::().unwrap(); cursor_state.lock().unwrap().unset_shape(); let _previous = p.swap_focus(None); }); PointerTarget::leave(&self.0, seat, data, serial, time) } fn gesture_swipe_begin( &self, _seat: &Seat, _data: &mut State, _event: &GestureSwipeBeginEvent, ) { } fn gesture_swipe_update( &self, _seat: &Seat, _data: &mut State, _event: &GestureSwipeUpdateEvent, ) { } fn gesture_swipe_end( &self, _seat: &Seat, _data: &mut State, _event: &GestureSwipeEndEvent, ) { } fn gesture_pinch_begin( &self, _seat: &Seat, _data: &mut State, _event: &GesturePinchBeginEvent, ) { } fn gesture_pinch_update( &self, _seat: &Seat, _data: &mut State, _event: &GesturePinchUpdateEvent, ) { } fn gesture_pinch_end( &self, _seat: &Seat, _data: &mut State, _event: &GesturePinchEndEvent, ) { } fn gesture_hold_begin( &self, _seat: &Seat, _data: &mut State, _event: &GestureHoldBeginEvent, ) { } fn gesture_hold_end( &self, _seat: &Seat, _data: &mut State, _event: &GestureHoldEndEvent, ) { } } impl TouchTarget for CosmicWindow { fn down(&self, seat: &Seat, data: &mut State, event: &DownEvent, seq: Serial) { let mut event = event.clone(); self.0.with_program(|p| { event.location -= p.window.geometry().loc.to_f64(); *p.last_seat.lock().unwrap() = Some((seat.clone(), event.serial)); }); TouchTarget::down(&self.0, seat, data, &event, seq) } fn up(&self, seat: &Seat, data: &mut State, event: &UpEvent, seq: Serial) { TouchTarget::up(&self.0, seat, data, &event, seq) } fn motion(&self, seat: &Seat, data: &mut State, event: &TouchMotionEvent, seq: Serial) { let mut event = event.clone(); event.location -= self.0.with_program(|p| p.window.geometry().loc.to_f64()); TouchTarget::motion(&self.0, seat, data, &event, seq) } fn frame(&self, seat: &Seat, data: &mut State, seq: Serial) { TouchTarget::frame(&self.0, seat, data, seq) } fn cancel(&self, seat: &Seat, data: &mut State, seq: Serial) { TouchTarget::cancel(&self.0, seat, data, seq) } fn shape(&self, seat: &Seat, data: &mut State, event: &ShapeEvent, seq: Serial) { TouchTarget::shape(&self.0, seat, data, event, seq) } fn orientation( &self, _seat: &Seat, _data: &mut State, _event: &OrientationEvent, _seq: Serial, ) { } } impl WaylandFocus for CosmicWindow { fn wl_surface(&self) -> Option> { self.0.with_program(|p| { p.window .wl_surface() .map(|s| Cow::Owned(Cow::into_owned(s))) }) } fn same_client_as(&self, object_id: &ObjectId) -> bool { self.0.with_program(|p| p.window.same_client_as(object_id)) } } render_elements! { pub CosmicWindowRenderElement where R: ImportAll + ImportMem; Header = MemoryRenderBufferRenderElement, Window = WaylandSurfaceRenderElement, }