feat(wayland): support image capture on wlroot wayland (#207)

* feat: support image capture on `wlroot` wayland

* fix: use correct method

* refactor: inline variables

* chore: descriptive error message

---------

Co-authored-by: Nikita Revenco <[email protected]>

Author: nik-rev

Cargo.toml

@@ -57,6 +57,7 @@ dbus = "0.9"
 lazy_static = "1.5"
 percent-encoding = "2.3"
 xcb = { version = "1.5", features = ["randr"] }
+libwayshot = "0.3"
 
 [dev-dependencies]
 fs_extra = "1.3"

src/error.rs

@@ -30,6 +30,9 @@ pub enum XCapError {
     #[cfg(target_os = "linux")]
     #[error(transparent)]
     StdTimeSystemTimeError(#[from] std::time::SystemTimeError),
+    #[cfg(target_os = "linux")]
+    #[error(transparent)]
+    LibwayshotError(#[from] libwayshot::Error),
 
     #[cfg(target_os = "macos")]
     #[error("Objc2CoreGraphicsCGError {:?}", 0)]

src/linux/wayland_capture.rs

@@ -172,12 +172,40 @@ fn org_freedesktop_portal_screenshot(
 
 static DBUS_LOCK: Mutex<()> = Mutex::new(());
 
+fn wlroots_screenshot(
+    x_coordinate: i32,
+    y_coordinate: i32,
+    width: i32,
+    height: i32,
+) -> XCapResult<RgbaImage> {
+    let wayshot_connection = libwayshot::WayshotConnection::new()?;
+    let capture_region = libwayshot::CaptureRegion {
+        x_coordinate,
+        y_coordinate,
+        width,
+        height,
+    };
+    let rgba_image = wayshot_connection.screenshot(capture_region, false)?;
+
+    // libwayshot returns image 0.24 RgbaImage
+    // we need image 0.25 RgbaImage
+    let image = image::RgbaImage::from_raw(
+        rgba_image.width(),
+        rgba_image.height(),
+        rgba_image.into_raw(),
+    )
+    .expect("Conversion of PNG -> Raw -> PNG does not fail");
+
+    Ok(image)
+}
+
 pub fn wayland_capture(x: i32, y: i32, width: i32, height: i32) -> XCapResult<RgbaImage> {
     let lock = DBUS_LOCK.lock();
 
     let conn = Connection::new_session()?;
     let res = org_gnome_shell_screenshot(&conn, x, y, width, height)
-        .or_else(|_| org_freedesktop_portal_screenshot(&conn, x, y, width, height));
+        .or_else(|_| org_freedesktop_portal_screenshot(&conn, x, y, width, height))
+        .or_else(|_| wlroots_screenshot(x, y, width, height));
 
     drop(lock);