leyoda/cosmic-comp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(corner-radius): error handling and cleanup

Browse source
This commit is contained in:
Ashley Wulber 2025-10-03 09:44:32 -04:00 • committed by Victoria Brekenfeld
parent bef8c53069
commit 228af1037a
1 changed files with 14 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

43
src/wayland/protocols/corner_radius.rs
View file

@ -2,7 +2,6 @@ use cosmic_protocols::corner_radius::v1::server::cosmic_corner_radius_toplevel_v
use cosmic_protocols::corner_radius::v1::server::{
cosmic_corner_radius_manager_v1, cosmic_corner_radius_toplevel_v1,
};
use smithay::reexports::wayland_server::protocol::wl_surface::WlSurface;
use smithay::utils::HookId;
use smithay::wayland::compositor::add_pre_commit_hook;
use smithay::wayland::compositor::with_states;
@ -15,10 +14,11 @@ use smithay::{
},
wayland::shell::xdg::XdgShellHandler,
};
use std::collections::HashMap;
use std::sync::Mutex;
use wayland_backend::server::GlobalId;
type ToplevelHookId = Mutex<Option<(HookId, Weak<CosmicCornerRadiusToplevelV1>)>>;
#[derive(Debug)]
pub struct CornerRadiusState {
instances: Vec<cosmic_corner_radius_manager_v1::CosmicCornerRadiusManagerV1>,
@ -116,16 +116,11 @@ where
cosmic_corner_radius_manager_v1::Request::GetCornerRadius { id, toplevel } => {
if let Some(surface) = state.xdg_shell_state().get_toplevel(&toplevel) {
let radius_exists = with_states(surface.wl_surface(), |surface_data| {
let hook_ids = surface_data.data_map.get_or_insert_threadsafe(|| {
Mutex::new(HashMap::<
WlSurface,
(HookId, Weak<CosmicCornerRadiusToplevelV1>),
>::new())
});
let guard = hook_ids.lock().unwrap();
guard
.get(surface.wl_surface())
.map(|(_, t)| t.upgrade().is_ok())
let hook_id = surface_data
.data_map
.get_or_insert_threadsafe(|| ToplevelHookId::new(None));
let guard = hook_id.lock().unwrap();
guard.as_ref().map(|(_, t)| t.upgrade().is_ok())
});
if radius_exists.unwrap_or_default() {
resource.post_error(
@ -179,14 +174,11 @@ where
);
with_states(surface.wl_surface(), |surface_data| {
let hook_ids = surface_data.data_map.get_or_insert_threadsafe(|| {
Mutex::new(HashMap::<
WlSurface,
(HookId, Weak<CosmicCornerRadiusToplevelV1>),
>::new())
});
let hook_ids = surface_data
.data_map
.get_or_insert_threadsafe(|| ToplevelHookId::new(None));
let mut guard = hook_ids.lock().unwrap();
guard.insert(surface.wl_surface().clone(), (hook_id, obj_downgrade));
*guard = Some((hook_id, obj_downgrade));
});
}
}
@ -231,22 +223,15 @@ where
guard.corners = None;
let Ok(toplevel) = guard.toplevel.upgrade() else {
resource.post_error(
cosmic_corner_radius_toplevel_v1::Error::ToplevelDestroyed as u32,
format!("{:?} No toplevel found", resource),
);
return;
};
if let Some(surface) = state.xdg_shell_state().get_toplevel(&toplevel) {
with_states(surface.wl_surface(), |surface_data| {
if let Some(hook_ids_mutex) =
surface_data.data_map.get::<Mutex<
HashMap<WlSurface, (HookId, Weak<CosmicCornerRadiusToplevelV1>)>,
>>()
if let Some(hook_ids_mutex) = surface_data.data_map.get::<ToplevelHookId>()
{
let mut hook_ids = hook_ids_mutex.lock().unwrap();
hook_ids.remove(surface.wl_surface());
let mut hook_id = hook_ids_mutex.lock().unwrap();
*hook_id = None;
}
});
}