Notification improvements.

2018-10-19 22:02:12 +08:00
parent a0fcd00a6f
commit bb069197d6
16 changed files with 1559 additions and 48 deletions
--- a/src/notifications/managerlinux.cpp
+++ b/src/notifications/managerlinux.cpp
@@ -0,0 +1,133 @@
+#include "manager.h"
+
+#include <QDebug>
+#include <QImage>
+#include <QtDBus/QDBusConnection>
+#include <QtDBus/QDBusMessage>
+#include <QtDBus/QDBusMetaType>
+
+NotificationsManager::NotificationsManager(QObject *parent)
+    : QObject(parent),
+      dbus("org.freedesktop.Notifications", "/org/freedesktop/Notifications",
+           "org.freedesktop.Notifications", QDBusConnection::sessionBus(),
+           this) {
+  qDBusRegisterMetaType<QImage>();
+
+  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 << "Spectral";                              // 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(-1);                              // 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();
+  }
+}
+
+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.sizeInBytes());
+  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;
+}