From 1963aaa7755759442a2d25fffdc5af0938bf7c78 Mon Sep 17 00:00:00 2001
From: Ivan Molodetskikh <yalterz@gmail.com>
Date: Wed, 20 Sep 2023 09:28:23 +0400
Subject: Add initial impl of org.gnome.Shell.Screenshot

Enough to make the portal all-outputs screenshot work. With this,
Flameshot kinda-works.
---
 src/dbus/gnome_shell_screenshot.rs | 57 ++++++++++++++++++++++++++++++++++++++
 src/dbus/mod.rs                    |  1 +
 2 files changed, 58 insertions(+)
 create mode 100644 src/dbus/gnome_shell_screenshot.rs

(limited to 'src/dbus')

diff --git a/src/dbus/gnome_shell_screenshot.rs b/src/dbus/gnome_shell_screenshot.rs
new file mode 100644
index 00000000..fad3649b
--- /dev/null
+++ b/src/dbus/gnome_shell_screenshot.rs
@@ -0,0 +1,57 @@
+use std::path::PathBuf;
+
+use smithay::reexports::calloop;
+use zbus::{dbus_interface, fdo};
+
+pub struct Screenshot {
+    to_niri: calloop::channel::Sender<ScreenshotToNiri>,
+    from_niri: async_channel::Receiver<NiriToScreenshot>,
+}
+
+pub enum ScreenshotToNiri {
+    TakeScreenshot { include_cursor: bool },
+}
+
+pub enum NiriToScreenshot {
+    ScreenshotResult(Option<PathBuf>),
+}
+
+#[dbus_interface(name = "org.gnome.Shell.Screenshot")]
+impl Screenshot {
+    async fn screenshot(
+        &self,
+        include_cursor: bool,
+        _flash: bool,
+        _filename: PathBuf,
+    ) -> fdo::Result<(bool, PathBuf)> {
+        if let Err(err) = self
+            .to_niri
+            .send(ScreenshotToNiri::TakeScreenshot { include_cursor })
+        {
+            warn!("error sending message to niri: {err:?}");
+            return Err(fdo::Error::Failed("internal error".to_owned()));
+        }
+
+        let filename = match self.from_niri.recv().await {
+            Ok(NiriToScreenshot::ScreenshotResult(Some(filename))) => filename,
+            Ok(NiriToScreenshot::ScreenshotResult(None)) => {
+                return Err(fdo::Error::Failed("internal error".to_owned()));
+            }
+            Err(err) => {
+                warn!("error receiving message from niri: {err:?}");
+                return Err(fdo::Error::Failed("internal error".to_owned()));
+            }
+        };
+
+        Ok((true, filename))
+    }
+}
+
+impl Screenshot {
+    pub fn new(
+        to_niri: calloop::channel::Sender<ScreenshotToNiri>,
+        from_niri: async_channel::Receiver<NiriToScreenshot>,
+    ) -> Self {
+        Self { to_niri, from_niri }
+    }
+}
diff --git a/src/dbus/mod.rs b/src/dbus/mod.rs
index 84a8bc30..459af258 100644
--- a/src/dbus/mod.rs
+++ b/src/dbus/mod.rs
@@ -1,3 +1,4 @@
+pub mod gnome_shell_screenshot;
 pub mod mutter_display_config;
 pub mod mutter_screen_cast;
 pub mod mutter_service_channel;
-- 
cgit