Fix notification not being cleared, when read event didn't cause a notification

This commit is contained in:
Nicolas Werner 2020-04-11 23:19:03 +02:00
parent 2b24a978e1
commit 82ec022f9c
2 changed files with 17 additions and 34 deletions

View File

@ -1,6 +1,5 @@
#pragma once #pragma once
#include <QHash>
#include <QImage> #include <QImage>
#include <QObject> #include <QObject>
#include <QString> #include <QString>
@ -16,27 +15,12 @@ struct roomEventId
QString eventId; QString eventId;
}; };
inline bool
operator<(const roomEventId &a, const roomEventId &b)
{
if (a.roomId == b.roomId)
return a.eventId < b.eventId;
else
return a.roomId < b.roomId;
}
inline bool inline bool
operator==(const roomEventId &a, const roomEventId &b) operator==(const roomEventId &a, const roomEventId &b)
{ {
return a.roomId == b.roomId && a.eventId == b.eventId; return a.roomId == b.roomId && a.eventId == b.eventId;
} }
inline uint
qHash(const roomEventId &v, uint seed)
{
return qHash(v.roomId, seed) ^ qHash(v.eventId, seed);
}
class NotificationsManager : public QObject class NotificationsManager : public QObject
{ {
Q_OBJECT Q_OBJECT
@ -67,7 +51,6 @@ private:
// notification ID to (room ID, event ID) // notification ID to (room ID, event ID)
QMap<uint, roomEventId> notificationIds; QMap<uint, roomEventId> notificationIds;
QHash<roomEventId, uint> eventToNotificationId;
#endif #endif
// these slots are platform specific (D-Bus only) // these slots are platform specific (D-Bus only)

View File

@ -40,7 +40,6 @@ NotificationsManager::postNotification(const QString &roomid,
{ {
uint id = showNotification(roomname, sender + ": " + text, icon); uint id = showNotification(roomname, sender + ": " + text, icon);
notificationIds[id] = roomEventId{roomid, eventid}; notificationIds[id] = roomEventId{roomid, eventid};
eventToNotificationId[roomEventId{roomid, eventid}] = id;
} }
/** /**
* This function is based on code from * This function is based on code from
@ -98,10 +97,9 @@ NotificationsManager::closeNotification(uint id)
void void
NotificationsManager::removeNotification(const QString &roomId, const QString &eventId) NotificationsManager::removeNotification(const QString &roomId, const QString &eventId)
{
roomEventId reId = {roomId, eventId}; roomEventId reId = {roomId, eventId};
if (eventToNotificationId.contains(reId)) { for (auto elem = notificationIds.begin(); elem != notificationIds.end(); ++elem) {
for (auto elem = notificationIds.begin(); elem != notificationIds.end(); ++elem) {
if (elem.value().roomId != roomId) if (elem.value().roomId != roomId)
continue; continue;
@ -112,10 +110,12 @@ NotificationsManager::removeNotification(const QString &roomId, const QString &e
// Otherwise we would need to store a 64bit counter instead. // Otherwise we would need to store a 64bit counter instead.
closeNotification(elem.key()); closeNotification(elem.key());
// FIXME: compare index of event id of the read receipt and the notification instead of just
// the id to prevent read receipts of events without notification clearing all notifications
// in that room!
if (elem.value() == reId) if (elem.value() == reId)
break; break;
} }
}
} }
void void
@ -131,7 +131,7 @@ void
NotificationsManager::notificationClosed(uint id, uint reason) NotificationsManager::notificationClosed(uint id, uint reason)
{ {
Q_UNUSED(reason); Q_UNUSED(reason);
eventToNotificationId.remove(notificationIds.take(id)); notificationIds.remove(id);
} }
/** /**