File indexing completed on 2024-12-08 13:27:06

 /*
  * SPDX-FileCopyrightText: 2016 Red Hat Inc
  * SPDX-License-Identifier: LGPL-2.0-or-later
  *
  * SPDX-FileCopyrightText: 2016 Jan Grulich <jgrulich@redhat.com>
  */
 
 #include "notification.h"
 
 #include <QDBusConnection>
 #include <QDBusMessage>
 #include <QDBusMetaType>
 
 #include "fdo_application_interface.h"
 #include "notification_debug.h"
 #include "portalicon.h"
 
 using Action = QPair<QString, QVariant>;
 Q_DECLARE_METATYPE(QList<Action>);
 
 NotificationPortal::NotificationPortal(QObject *parent)
     : QDBusAbstractAdaptor(parent)
 {
     PortalIcon::registerDBusType();
 }
 
 NotificationPortal::~NotificationPortal()
 {
 }
 
 void NotificationPortal::AddNotification(const QString &app_id, const QString &id, const QVariantMap &notification)
 {
     qCDebug(XdgDesktopPortalKdeNotification) << "AddNotification called with parameters:";
     qCDebug(XdgDesktopPortalKdeNotification) << "    app_id: " << app_id;
     qCDebug(XdgDesktopPortalKdeNotification) << "    id: " << id;
     qCDebug(XdgDesktopPortalKdeNotification) << "    notification: " << notification;
 
     // We have to use "notification" as an ID because any other ID will not be configured
     KNotification *notify = new KNotification(QStringLiteral("notification"), KNotification::CloseOnTimeout | KNotification::DefaultEvent, this);
     if (notification.contains(QStringLiteral("title"))) {
         notify->setTitle(notification.value(QStringLiteral("title")).toString());
     }
     if (notification.contains(QStringLiteral("body"))) {
         notify->setText(notification.value(QStringLiteral("body")).toString());
     }
     if (notification.contains(QStringLiteral("icon"))) {
         QVariant iconVariant = notification.value(QStringLiteral("icon"));
         if (iconVariant.type() == QVariant::String) {
             notify->setIconName(iconVariant.toString());
         } else {
             QDBusArgument argument = iconVariant.value<QDBusArgument>();
             PortalIcon icon = qdbus_cast<PortalIcon>(argument);
             QVariant iconData = icon.data.variant();
             if (icon.str == QStringLiteral("themed") && iconData.type() == QVariant::StringList) {
                 notify->setIconName(iconData.toStringList().first());
             } else if (icon.str == QStringLiteral("bytes") && iconData.type() == QVariant::ByteArray) {
                 QPixmap pixmap;
                 if (pixmap.loadFromData(iconData.toByteArray(), "PNG")) {
                     notify->setPixmap(pixmap);
                 }
             }
         }
     }
 
     const QString priority = notification.value(QStringLiteral("priority")).toString();
     if (priority == QLatin1String("low")) {
         notify->setUrgency(KNotification::LowUrgency);
     } else if (priority == QLatin1String("normal")) {
         notify->setUrgency(KNotification::NormalUrgency);
     } else if (priority == QLatin1String("high")) {
         notify->setUrgency(KNotification::HighUrgency);
     } else if (priority == QLatin1String("urgent")) {
         notify->setUrgency(KNotification::CriticalUrgency);
     }
 
     if (notification.contains(QStringLiteral("default-action")) && notification.contains(QStringLiteral("default-action-target"))) {
         // default action is conveniently mapped to action number 0 so it uses the same action invocation method as the others
         notify->setDefaultAction(notification.value(QStringLiteral("default-action")).toString());
     }
 
     QList<Action> actionValues = {
         {notification.value(QStringLiteral("default-action")).toString(), notification.value(QStringLiteral("default-action-target"))}};
     if (notification.contains(QStringLiteral("buttons"))) {
         const QDBusArgument dbusArgument = notification.value(QStringLiteral("buttons")).value<QDBusArgument>();
         const auto buttons = qdbus_cast<QList<QVariantMap>>(dbusArgument);
 
         QStringList actions;
         actions.reserve(buttons.count());
         for (const QVariantMap &button : qAsConst(buttons)) {
             actions << button.value(QStringLiteral("label")).toString();
 
             actionValues.append({button.value(QStringLiteral("action")).toString(), button.value(QStringLiteral("target"))});
         }
 
         if (!actions.isEmpty()) {
             notify->setActions(actions);
         }
     }
     notify->setHint(QStringLiteral("desktop-entry"), app_id);
     notify->setHint(QStringLiteral("x-kde-xdgTokenAppId"), app_id);
 
     notify->setProperty("actionValues", QVariant::fromValue<QList<Action>>(actionValues));
     notify->setProperty("app_id", app_id);
     notify->setProperty("id", id);
     connect(notify, static_cast<void (KNotification::*)(uint)>(&KNotification::activated), this, &NotificationPortal::notificationActivated);
     connect(notify, &KNotification::closed, this, &NotificationPortal::notificationClosed);
 
     m_notifications.insert(QStringLiteral("%1:%2").arg(app_id, id), notify);
     notify->sendEvent();
 }
 
 static QString appPathFromId(const QString &app_id)
 {
     QString ret = QLatin1Char('/') + app_id;
     ret.replace('.', '/');
     ret.replace('-', '_');
     return ret;
 }
 
 void NotificationPortal::notificationActivated(uint action)
 {
     KNotification *notify = qobject_cast<KNotification *>(sender());
 
     if (!notify) {
         return;
     }
 
     const QString appId = notify->property("app_id").toString();
     const QString id = notify->property("id").toString();
     const auto actionValues = notify->property("actionValues").value<QList<Action>>();
     QVariantMap platformData;
     if (!notify->xdgActivationToken().isEmpty()) {
         platformData = {
             {QStringLiteral("activation-token"), notify->xdgActivationToken()},
 
             // apparently gtk uses "desktop-startup-id"
             {QStringLiteral("desktop-startup-id"), notify->xdgActivationToken()},
         };
     }
 
     qCDebug(XdgDesktopPortalKdeNotification) << "Notification activated:";
     qCDebug(XdgDesktopPortalKdeNotification) << "    app_id: " << appId;
     qCDebug(XdgDesktopPortalKdeNotification) << "    id: " << id;
     qCDebug(XdgDesktopPortalKdeNotification) << "    action: " << action << actionValues;
 
     const auto ourAction = actionValues.value(action);
 
     QVariantList params;
     if (ourAction.second.isValid()) {
         params += ourAction.second;
     }
 
     OrgFreedesktopApplicationInterface iface(appId, appPathFromId(appId), QDBusConnection::sessionBus());
     if (ourAction.first.startsWith("app.") && iface.isValid()) {
         iface.ActivateAction(ourAction.first.mid(4), params, platformData);
     } else {
         if (iface.isValid()) {
             iface.Activate(platformData);
         }
 
         QDBusMessage message = QDBusMessage::createSignal(QStringLiteral("/org/freedesktop/portal/desktop"),
                                                           QStringLiteral("org.freedesktop.impl.portal.Notification"),
                                                           QStringLiteral("ActionInvoked"));
         message << appId << id << ourAction.first << params;
         QDBusConnection::sessionBus().send(message);
     }
 }
 
 void NotificationPortal::RemoveNotification(const QString &app_id, const QString &id)
 {
     qCDebug(XdgDesktopPortalKdeNotification) << "RemoveNotification called with parameters:";
     qCDebug(XdgDesktopPortalKdeNotification) << "    app_id: " << app_id;
     qCDebug(XdgDesktopPortalKdeNotification) << "    id: " << id;
 
     KNotification *notify = m_notifications.take(QStringLiteral("%1:%2").arg(app_id, id));
     if (notify) {
         disconnect(notify, &KNotification::closed, this, &NotificationPortal::notificationClosed);
         notify->close();
     }
 }
 
 void NotificationPortal::notificationClosed()
 {
     KNotification *notify = qobject_cast<KNotification *>(sender());
     Q_ASSERT(notify);
     const QString appId = notify->property("app_id").toString();
     const QString id = notify->property("id").toString();
 
     auto n = m_notifications.take(QStringLiteral("%1:%2").arg(appId, id));
     if (n) {
         Q_ASSERT(n == notify);
         n->close();
     }
 }