Add screenshot-window action

1 files changed, 12 insertions, 0 deletions

diff --git a/src/input.rs b/src/input.rs
index c1e335dd..9f59c06e 100644
--- a/src/input.rs
+++ b/src/input.rs
@@ -167,6 +167,18 @@ impl State {
                                 }
                             }
                         }
+                        Action::ScreenshotWindow => {
+                            let active = self.niri.layout.active_window();
+                            if let Some((window, output)) = active {
+                                if let Some(renderer) = self.backend.renderer() {
+                                    if let Err(err) =
+                                        self.niri.screenshot_window(renderer, &output, &window)
+                                    {
+                                        warn!("error taking screenshot: {err:?}");
+                                    }
+                                }
+                            }
+                        }
                         Action::CloseWindow => {
                             if let Some(window) = self.niri.layout.focus() {
                                 window.toplevel().send_close();