screenshot_ui: add ctrl+a command to select entire output

Author: lineargraph

niri-config/src/binds.rs

@@ -118,6 +118,8 @@ pub enum Action {
     #[knuffel(skip)]
     CancelScreenshot,
     #[knuffel(skip)]
+    ScreenshotSelectAll,
+    #[knuffel(skip)]
     ScreenshotTogglePointer,
     Screenshot(
         #[knuffel(property(name = "show-pointer"), default = true)] bool,

src/input/mod.rs

@@ -733,6 +733,10 @@ impl State {
                     .set_cursor_image(CursorImageStatus::default_named());
                 self.niri.queue_redraw_all();
             }
+            Action::ScreenshotSelectAll => {
+                self.niri.screenshot_ui.select_entire_output();
+                self.niri.queue_redraw_all();
+            }
             Action::ScreenshotTogglePointer => {
                 self.niri.screenshot_ui.toggle_pointer();
                 self.niri.queue_redraw_all();

src/ui/screenshot_ui.rs

@@ -38,9 +38,11 @@ const FONT: &str = "sans 14px";
 const BORDER: i32 = 4;
 const TEXT_HIDE_P: &str =
     "Press <span face='mono' bgcolor='#2C2C2C'> Space </span> to save the screenshot.\n\
+     Press <span face='mono' bgcolor='#2C2C2C'> Ctrl+A </span> to select the entire output.\n\
      Press <span face='mono' bgcolor='#2C2C2C'> P </span> to hide the pointer.";
 const TEXT_SHOW_P: &str =
     "Press <span face='mono' bgcolor='#2C2C2C'> Space </span> to save the screenshot.\n\
+     Press <span face='mono' bgcolor='#2C2C2C'> Ctrl+A </span> to select the entire output.\n\
      Press <span face='mono' bgcolor='#2C2C2C'> P </span> to show the pointer.";
 
 // Ideally the screenshot UI should support cross-output selections. However, that poses some
@@ -460,6 +462,23 @@ impl ScreenshotUi {
         self.update_buffers();
     }
 
+
+    pub fn select_entire_output(&mut self) {
+        let Self::Open {
+            selection,
+            output_data,
+            ..
+        } = self else { return };
+
+        let current_data = &output_data[&selection.0];
+        let size = current_data.size;
+
+        selection.1 = Point::new(0, 0);
+        selection.2 = Point::new(size.w - 1, size.h - 1);
+
+        self.update_buffers();
+    }
+
     pub fn set_width(&mut self, change: SizeChange) {
         let Self::Open {
             selection: (output, a, b),
@@ -1067,6 +1086,10 @@ fn action(raw: Keysym, mods: ModifiersState) -> Option<Action> {
         });
     }
 
+    if mods.ctrl && raw == Keysym::a {
+        return Some(Action::ScreenshotSelectAll);
+    }
+
     if !mods.ctrl && raw == Keysym::p {
         return Some(Action::ScreenshotTogglePointer);
     }