Working D-Bus desktop notifications (#361)

* Working D-Bus desktop notifications * Remove return type on constructor * Fix the Windows placeholder class * Fix wrong variable name * Fix windows and macOS versions of notificationsmanager
2018-07-11 17:33:02 +03:00 · 2018-07-11 17:33:02 +03:00 · 80ebe3f29d
commit 80ebe3f29d
parent e7f30b57e8
7 changed files with 286 additions and 18 deletions
--- a/src/notifications/ManagerLinux.cpp
+++ b/src/notifications/ManagerLinux.cpp
@ -1,7 +1,158 @@
 #include "notifications/Manager.h"

-void
-NotificationsManager::postNotification(const QString &, const QString &, const QString &)
+#include <QImage>
+#include <QDebug>
+#include <QtDBus/QDBusMessage>
+#include <QtDBus/QDBusMetaType>
+#include <QtDBus/QDBusConnection>
+
+NotificationsManager::NotificationsManager(QObject *parent) :
+    QObject(parent),
+    dbus(
+        "org.freedesktop.Notifications",
+        "/org/freedesktop/Notifications",
+        "org.freedesktop.Notifications",
+        QDBusConnection::sessionBus(),
+        this)
 {
-        // TODO: To be implemented
+        qDBusRegisterMetaType<QImage>();
+
+        //connectSlot("ActionInvoked", SLOT(actionInvoked(uint, QString)));
+        //connectSlot("NotificationClosed", SLOT(notificationClosed(uint, uint)));
+        QDBusConnection::sessionBus().connect(
+                "org.freedesktop.Notifications",
+                "/org/freedesktop/Notifications",
+                "org.freedesktop.Notifications",
+                "ActionInvoked",
+                this,
+                SLOT(actionInvoked(uint, QString)));
+        QDBusConnection::sessionBus().connect(
+                "org.freedesktop.Notifications",
+                "/org/freedesktop/Notifications",
+                "org.freedesktop.Notifications",
+                "NotificationClosed",
+                this,
+                SLOT(notificationClosed(uint, uint)));
+}
+
+void
+NotificationsManager::postNotification(const QString &roomid,
+                                const QString &eventid,
+                                const QString &roomname,
+                                const QString &sender,
+                                const QString &text,
+                                const QImage &icon)
+{
+    uint id = showNotification(roomname, sender+": "+text, icon);
+    notificationIds[id] = roomEventId{roomid,eventid};
+}
+/**
+ * This function is based on code from
+ * https://github.com/rohieb/StratumsphereTrayIcon
+ * Copyright (C) 2012 Roland Hieber <rohieb@rohieb.name>
+ * Licensed under the GNU General Public License, version 3
+ */
+uint
+NotificationsManager::showNotification(const QString summary, const QString text, const QImage image)
+{
+        QVariantMap hints;
+        hints["image_data"] = image;
+        QList<QVariant> argumentList;
+        argumentList << "nheko"; //app_name
+        argumentList << (uint)0;  // replace_id
+        argumentList << "";  // app_icon
+        argumentList << summary; // summary
+        argumentList << text; // body
+        argumentList << (QStringList("default")<<"reply");  // actions
+        argumentList << hints;  // hints
+        argumentList << (int)0; // timeout in ms
+
+        static QDBusInterface notifyApp(
+                    "org.freedesktop.Notifications",
+                    "/org/freedesktop/Notifications",
+                    "org.freedesktop.Notifications");
+        QDBusMessage reply = notifyApp.callWithArgumentList(
+                    QDBus::AutoDetect,
+                    "Notify",
+                    argumentList);
+        if(reply.type() == QDBusMessage::ErrorMessage) {
+            qDebug() << "D-Bus Error:" << reply.errorMessage();
+            return 0;
+        } else {
+            return reply.arguments().first().toUInt();
+        }
+        return true;
+}
+
+void
+NotificationsManager::actionInvoked(uint id, QString action)
+{
+        if (action == "default" && notificationIds.contains(id)) {
+                roomEventId idEntry = notificationIds[id];
+                emit notificationClicked(idEntry.roomId, idEntry.eventId);
+        }
+}
+
+void
+NotificationsManager::notificationClosed(uint id, uint reason)
+{
+        Q_UNUSED(reason);
+        notificationIds.remove(id);
+}
+
+/**
+ * Automatic marshaling of a QImage for org.freedesktop.Notifications.Notify
+ *
+ * This function is from the Clementine project (see
+ * http://www.clementine-player.org) and licensed under the GNU General Public
+ * License, version 3 or later.
+ *
+ * Copyright 2010, David Sansome <me@davidsansome.com>
+ */
+QDBusArgument& operator<<(QDBusArgument& arg, const QImage& image) {
+  if(image.isNull()) {
+    arg.beginStructure();
+    arg << 0 << 0 << 0 << false << 0 << 0 << QByteArray();
+    arg.endStructure();
+    return arg;
+  }
+
+  QImage scaled = image.scaledToHeight(100, Qt::SmoothTransformation);
+  scaled = scaled.convertToFormat(QImage::Format_ARGB32);
+
+#if Q_BYTE_ORDER == Q_LITTLE_ENDIAN
+  // ABGR -> ARGB
+  QImage i = scaled.rgbSwapped();
+#else
+  // ABGR -> GBAR
+  QImage i(scaled.size(), scaled.format());
+  for (int y = 0; y < i.height(); ++y) {
+    QRgb* p = (QRgb*) scaled.scanLine(y);
+    QRgb* q = (QRgb*) i.scanLine(y);
+    QRgb* end = p + scaled.width();
+    while (p < end) {
+      *q = qRgba(qGreen(*p), qBlue(*p), qAlpha(*p), qRed(*p));
+      p++;
+      q++;
+    }
+  }
+#endif
+
+  arg.beginStructure();
+  arg << i.width();
+  arg << i.height();
+  arg << i.bytesPerLine();
+  arg << i.hasAlphaChannel();
+  int channels = i.isGrayscale() ? 1 : (i.hasAlphaChannel() ? 4 : 3);
+  arg << i.depth() / channels;
+  arg << channels;
+  arg << QByteArray(reinterpret_cast<const char*>(i.bits()), i.byteCount());
+  arg.endStructure();
+  return arg;
+}
+
+const QDBusArgument& operator>>(const QDBusArgument& arg, QImage&) {
+  // This is needed to link but shouldn't be called.
+  Q_ASSERT(0);
+  return arg;
 }