File indexing completed on 2024-04-28 05:36:53

 /*
  * SPDX-FileCopyrightText: 2018 Red Hat Inc
  *
  * SPDX-License-Identifier: LGPL-2.0-or-later
  *
  * SPDX-FileCopyrightText: 2018 Jan Grulich <jgrulich@redhat.com>
  * SPDX-FileCopyrightText: 2022 Harald Sitter <sitter@kde.org>
  */
 
 #include "screenshot.h"
 #include "request.h"
 #include "screenshot_debug.h"
 #include "screenshotdialog.h"
 #include "utils.h"
 
 #include <QDBusArgument>
 #include <QDBusReply>
 #include <QDateTime>
 #include <QPointer>
 #include <QStandardPaths>
 #include <QWindow>
 #include <QtDBus>
 
 // Keep in sync with qflatpakcolordialog from Qt flatpak platform theme
 Q_DECLARE_METATYPE(ScreenshotPortal::ColorRGB)
 
 QDBusArgument &operator<<(QDBusArgument &arg, const ScreenshotPortal::ColorRGB &color)
 {
     arg.beginStructure();
     arg << color.red << color.green << color.blue;
     arg.endStructure();
     return arg;
 }
 
 const QDBusArgument &operator>>(const QDBusArgument &arg, ScreenshotPortal::ColorRGB &color)
 {
     double red, green, blue;
     arg.beginStructure();
     arg >> red >> green >> blue;
     color.red = red;
     color.green = green;
     color.blue = blue;
     arg.endStructure();
 
     return arg;
 }
 
 QDBusArgument &operator<<(QDBusArgument &argument, const QColor &color)
 {
     argument.beginStructure();
     argument << color.rgba();
     argument.endStructure();
     return argument;
 }
 
 const QDBusArgument &operator>>(const QDBusArgument &argument, QColor &color)
 {
     argument.beginStructure();
     QRgb rgba;
     argument >> rgba;
     argument.endStructure();
     color = QColor::fromRgba(rgba);
     return argument;
 }
 
 ScreenshotPortal::ScreenshotPortal(QObject *parent)
     : QDBusAbstractAdaptor(parent)
 {
     qDBusRegisterMetaType<QColor>();
     qDBusRegisterMetaType<ColorRGB>();
 }
 
 ScreenshotPortal::~ScreenshotPortal()
 {
 }
 
 uint ScreenshotPortal::Screenshot(const QDBusObjectPath &handle,
                                   const QString &app_id,
                                   const QString &parent_window,
                                   const QVariantMap &options,
                                   QVariantMap &results)
 {
     qCDebug(XdgDesktopPortalKdeScreenshot) << "Screenshot called with parameters:";
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    handle: " << handle.path();
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    app_id: " << app_id;
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    parent_window: " << parent_window;
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    options: " << options;
 
     QPointer<ScreenshotDialog> screenshotDialog = new ScreenshotDialog;
     Utils::setParentWindow(screenshotDialog->windowHandle(), parent_window);
     Request::makeClosableDialogRequest(handle, screenshotDialog.data());
 
     const bool modal = options.value(QStringLiteral("modal"), false).toBool();
     screenshotDialog->windowHandle()->setModality(modal ? Qt::WindowModal : Qt::NonModal);
 
     const bool interactive = options.value(QStringLiteral("interactive"), false).toBool();
     if (!interactive) {
         screenshotDialog->takeScreenshotNonInteractive();
     } else {
         screenshotDialog->exec();
     }
 
     const QImage screenshot = screenshotDialog->image();
 
     if (screenshotDialog) {
         screenshotDialog->deleteLater();
     }
 
     if (screenshot.isNull()) {
         return 1;
     }
 
     const QString filename =
         QStringLiteral("%1/Screenshot_%2.png") //
             .arg(QStandardPaths::writableLocation(QStandardPaths::PicturesLocation), QDateTime::currentDateTime().toString(QStringLiteral("yyyyMMdd_hhmmss")));
 
     if (!screenshot.save(filename, "PNG")) {
         return 1;
     }
 
     results.insert(QStringLiteral("uri"), QUrl::fromLocalFile(filename).toString(QUrl::FullyEncoded));
 
     return 0;
 }
 
 uint ScreenshotPortal::PickColor(const QDBusObjectPath &handle,
                                  const QString &app_id,
                                  const QString &parent_window,
                                  const QVariantMap &options,
                                  QVariantMap &results)
 {
     qCDebug(XdgDesktopPortalKdeScreenshot) << "PickColor called with parameters:";
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    handle: " << handle.path();
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    app_id: " << app_id;
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    parent_window: " << parent_window;
     qCDebug(XdgDesktopPortalKdeScreenshot) << "    options: " << options;
 
     QDBusMessage msg = QDBusMessage::createMethodCall(QStringLiteral("org.kde.KWin"),
                                                       QStringLiteral("/ColorPicker"),
                                                       QStringLiteral("org.kde.kwin.ColorPicker"),
                                                       QStringLiteral("pick"));
     QDBusReply<QColor> reply = QDBusConnection::sessionBus().call(msg);
     if (reply.isValid() && !reply.error().isValid()) {
         QColor selectedColor = reply.value();
         ColorRGB color;
         color.red = selectedColor.redF();
         color.green = selectedColor.greenF();
         color.blue = selectedColor.blueF();
 
         results.insert(QStringLiteral("color"), QVariant::fromValue<ScreenshotPortal::ColorRGB>(color));
         return 0;
     }
 
     return 1;
 }