diff --git a/src/lock.rs b/src/lock.rs index 087228b..d087220 100644 --- a/src/lock.rs +++ b/src/lock.rs @@ -262,6 +262,17 @@ impl LockManager { action } + pub fn remove_surface_by_output(&mut self, output: &wl_output::WlOutput) -> Option { + use wayland_client::Proxy; + let output_id = Proxy::id(output); + let idx = self + .surfaces + .iter() + .position(|s| Proxy::id(s.output()) == output_id)?; + self.surfaces.remove(idx); + Some(idx) + } + pub fn set_system_status(&mut self, status: SystemStatus) { for surface in &mut self.surfaces { surface.set_system_status(status.clone()); diff --git a/src/main.rs b/src/main.rs index ebfbdbd..0bab7a3 100644 --- a/src/main.rs +++ b/src/main.rs @@ -296,9 +296,61 @@ impl OutputHandler for WaylandLock { fn output_state(&mut self) -> &mut OutputState { &mut self.output_state } - fn new_output(&mut self, _conn: &Connection, _qh: &QueueHandle, _output: WlOutput) {} - fn update_output(&mut self, _conn: &Connection, _qh: &QueueHandle, _output: WlOutput) {} - fn output_destroyed(&mut self, _conn: &Connection, _qh: &QueueHandle, _output: WlOutput) { + fn new_output(&mut self, _conn: &Connection, qh: &QueueHandle, output: WlOutput) { + // If we are already locked, we must create a lock surface for this newly + // available output. This happens e.g. when a monitor powers back on after + // "niri msg action power-off-monitors" — niri re-advertises the output and + // the compositor requires a lock surface on every output or it shows a + // compositor-defined fallback (typically a solid red/black screen). + if let Some(session_lock) = &self.session_lock { + let surface = self.compositor_state.create_surface(qh); + let (width, height) = self.get_output_dimensions(&output); + let lock_surface = session_lock.create_lock_surface(surface.clone(), &output, qh); + self.lock_surfaces.push(lock_surface); + if !self.outputs.contains(&output) { + self.outputs.push(output.clone()); + } + if let Ok(mut lm) = self.lock_manager.lock() { + lm.add_surface(width, height, output); + let count = lm.surface_count(); + if let Some(ls) = lm.get_surface_mut(count - 1) { + ls.set_wayland_surface(surface); + } + } + log::info!("Created lock surface for newly available output"); + } + } + fn update_output(&mut self, _conn: &Connection, _qh: &QueueHandle, _output: WlOutput) { + // Dimension changes while locked are handled by the compositor sending a configure + // event on the lock surface, which the SessionLockHandler::configure callback + // already processes via locked_surface.resize(). + } + fn output_destroyed(&mut self, _conn: &Connection, _qh: &QueueHandle, output: WlOutput) { + // Clean up the lock surface and associated state for this output. + // This happens e.g. when a monitor is powered off with its physical power switch. + // When the monitor comes back on, new_output() will fire and recreate everything. + // + // outputs and captured_backgrounds are kept at the same indices, so we remove + // from both using the same position. + let output_id = Proxy::id(&output); + if let Some(idx) = self.outputs.iter().position(|o| Proxy::id(o) == output_id) { + self.outputs.remove(idx); + if idx < self.captured_backgrounds.len() { + self.captured_backgrounds.remove(idx); + } + } + + // lock_manager.surfaces and lock_surfaces are built in tandem and share indices, + // so the index returned from the lock_manager removal applies to lock_surfaces too. + if let Ok(mut lm) = self.lock_manager.lock() { + if let Some(idx) = lm.remove_surface_by_output(&output) { + if idx < self.lock_surfaces.len() { + drop(self.lock_surfaces.remove(idx)); + } + } + } + + log::info!("Removed lock surface for destroyed output"); } }