From b135a1c7b09182dd48b42cc81db6e088bb561901 Mon Sep 17 00:00:00 2001
From: Vlad Zahorodnii <vlad.zahorodnii@kde.org>
Date: Fri, 29 Oct 2021 11:55:10 +0300
Subject: [PATCH] effects/screenshot: Add
 org.kde.KWin.ScreenShot2.CaptureActiveWindow

BUG: 386271
---
 .../screenshot/org.kde.KWin.ScreenShot2.xml   | 42 +++++++++++++++++++
 .../screenshot/screenshotdbusinterface2.cpp   | 26 ++++++++++++
 .../screenshot/screenshotdbusinterface2.h     |  2 +
 3 files changed, 70 insertions(+)

diff --git a/src/effects/screenshot/org.kde.KWin.ScreenShot2.xml b/src/effects/screenshot/org.kde.KWin.ScreenShot2.xml
index ba12d5765f..a84cdc8a0c 100644
--- a/src/effects/screenshot/org.kde.KWin.ScreenShot2.xml
+++ b/src/effects/screenshot/org.kde.KWin.ScreenShot2.xml
@@ -63,6 +63,48 @@
             <arg name="results" type="a{sv}" direction="out" />
         </method>
 
+        <!--
+            CaptureActiveWindow:
+            @options: Optional vardict with screenshot options
+            @pipe: The pipe file descriptor where the screenshot will be written
+
+            Take a screenshot of the active window. The application that
+            requests the screenshot must have the org.kde.KWin.ScreenShot2
+            interface listed in the X-KDE-DBUS-Restricted-Interfaces desktop
+            file entry.
+
+            Supported since version 2.
+
+            Available @options include:
+
+            * "include-cursor" (b): Whether the cursor should be included.
+                                    Defaults to false
+            * "include-decoration" (b): Whether the decoration should be included.
+                                        Defaults to false
+            * "native-resolution" (b): Whether the screenshot should be in
+                                       native size. Defaults to false
+
+            The following results get returned via the @results vardict:
+
+            * "type" (s): The type of the image written to the pipe. Currently,
+                          the only supported type is "raw"
+            * "width" (u): The width of the image. Available only if the image
+                           type is "raw"
+            * "height" (u): The height of the image. Available only if the image
+                            type is "raw"
+            * "stride" (u): The number of bytes per row. Available only if the
+                            image type is "raw"
+            * "format" (u): The image format, as defined in QImage::Format.
+                            Available only if the image type is "raw"
+        -->
+        <method name="CaptureActiveWindow">
+            <annotation name="org.qtproject.QtDBus.QtTypeName.In0" value="QVariantMap" />
+            <arg name="options" type="a{sv}" direction="in" />
+            <arg name="pipe" type="h" direction="in" />
+            <annotation name="org.qtproject.QtDBus.QtTypeName.Out0" value="QVariantMap" />
+            <arg name="results" type="a{sv}" direction="out" />
+        </method>
+
         <!--
             CaptureArea:
             @x: The x coordinate of the upper left corner of the area
diff --git a/src/effects/screenshot/screenshotdbusinterface2.cpp b/src/effects/screenshot/screenshotdbusinterface2.cpp
index 23e22011bd..d6bb7cf935 100644
--- a/src/effects/screenshot/screenshotdbusinterface2.cpp
+++ b/src/effects/screenshot/screenshotdbusinterface2.cpp
@@ -299,6 +299,32 @@ bool ScreenShotDBusInterface2::checkPermissions() const
     return true;
 }
 
+QVariantMap ScreenShotDBusInterface2::CaptureActiveWindow(const QVariantMap &options,
+                                                          QDBusUnixFileDescriptor pipe)
+{
+    if (!checkPermissions()) {
+        return QVariantMap();
+    }
+
+    EffectWindow *window = effects->activeWindow();
+    if (!window) {
+        sendErrorReply(s_errorInvalidWindow, s_errorInvalidWindowMessage);
+        return QVariantMap();
+    }
+
+    const int fileDescriptor = dup(pipe.fileDescriptor());
+    if (fileDescriptor == -1) {
+        sendErrorReply(s_errorFileDescriptor, s_errorFileDescriptorMessage);
+        return QVariantMap();
+    }
+
+    takeScreenShot(window, screenShotFlagsFromOptions(options),
+                   new ScreenShotSinkPipe2(fileDescriptor, message()));
+
+    setDelayedReply(true);
+    return QVariantMap();
+}
+
 QVariantMap ScreenShotDBusInterface2::CaptureWindow(const QString &handle,
                                                     const QVariantMap &options,
                                                     QDBusUnixFileDescriptor pipe)
diff --git a/src/effects/screenshot/screenshotdbusinterface2.h b/src/effects/screenshot/screenshotdbusinterface2.h
index 9142e3393e..ac768a16ea 100644
--- a/src/effects/screenshot/screenshotdbusinterface2.h
+++ b/src/effects/screenshot/screenshotdbusinterface2.h
@@ -42,6 +42,8 @@ public:
 public Q_SLOTS:
     QVariantMap CaptureWindow(const QString &handle, const QVariantMap &options,
                               QDBusUnixFileDescriptor pipe);
+    QVariantMap CaptureActiveWindow(const QVariantMap &options,
+                                    QDBusUnixFileDescriptor pipe);
     QVariantMap CaptureArea(int x, int y, int width, int height,
                             const QVariantMap &options,
                             QDBusUnixFileDescriptor pipe);