Use ContentItems for NotificationEvents

1 files changed, 29 insertions, 17 deletions

diff --git a/libdino/src/service/notification_events.vala b/libdino/src/service/notification_events.vala
index c0efd7de..010341e3 100644
--- a/libdino/src/service/notification_events.vala
+++ b/libdino/src/service/notification_events.vala
@@ -9,13 +9,13 @@ public class NotificationEvents : StreamInteractionModule, Object {
     public static ModuleIdentity<NotificationEvents> IDENTITY = new ModuleIdentity<NotificationEvents>("notification_events");
     public string id { get { return IDENTITY.id; } }
 
-    public signal void notify_message(Message message, Conversation conversation);
+    public signal void notify_content_item(ContentItem content_item, Conversation conversation);
     public signal void notify_subscription_request(Conversation conversation);
     public signal void notify_connection_error(Account account, ConnectionManager.ConnectionError error);
 
     private StreamInteractor stream_interactor;
 
-    private HashMap<Account, HashMap<Conversation, Entities.Message>> mam_potential_new = new HashMap<Account, HashMap<Conversation, Entities.Message>>(Account.hash_func, Account.equals_func);
+    private HashMap<Account, HashMap<Conversation, ContentItem>> mam_potential_new = new HashMap<Account, HashMap<Conversation, ContentItem>>(Account.hash_func, Account.equals_func);
     private Gee.List<Account> synced_accounts = new ArrayList<Account>(Account.equals_func);
 
     public static void start(StreamInteractor stream_interactor) {
@@ -26,16 +26,16 @@ public class NotificationEvents : StreamInteractionModule, Object {
     public NotificationEvents(StreamInteractor stream_interactor) {
         this.stream_interactor = stream_interactor;
 
-        stream_interactor.get_module(MessageProcessor.IDENTITY).message_received.connect(on_message_received);
+        stream_interactor.get_module(ContentItemStore.IDENTITY).new_item.connect(on_content_item_received);
         stream_interactor.get_module(PresenceManager.IDENTITY).received_subscription_request.connect(on_received_subscription_request);
         stream_interactor.get_module(MessageProcessor.IDENTITY).history_synced.connect((account) => {
             synced_accounts.add(account);
             if (!mam_potential_new.has_key(account)) return;
             foreach (Conversation c in mam_potential_new[account].keys) {
-                Entities.Message m = mam_potential_new[account][c];
-                Entities.Message last_message = stream_interactor.get_module(MessageStorage.IDENTITY).get_last_message(c);
-                if (m.equals(last_message) && !c.read_up_to.equals(m)) {
-                    on_message_received(m, c);
+                ContentItem last_mam_item = mam_potential_new[account][c];
+                ContentItem last_item = stream_interactor.get_module(ContentItemStore.IDENTITY).get_latest(c);
+                if (last_mam_item == last_item /* && !c.read_up_to.equals(m) */) {
+                    on_content_item_received(last_mam_item, c);
                 }
             }
             mam_potential_new[account].clear();
@@ -43,26 +43,38 @@ public class NotificationEvents : StreamInteractionModule, Object {
         stream_interactor.connection_manager.connection_error.connect((account, error) => notify_connection_error(account, error));
     }
 
-    private void on_message_received(Entities.Message message, Conversation conversation) {
-        bool is_mam_message = Xep.MessageArchiveManagement.MessageFlag.get_flag(message.stanza) != null;
-        if (!synced_accounts.contains(conversation.account) && is_mam_message) {
+    private void on_content_item_received(ContentItem item, Conversation conversation) {
+        if (!synced_accounts.contains(conversation.account)) {
             if (!mam_potential_new.has_key(conversation.account)) {
-                mam_potential_new[conversation.account] = new HashMap<Conversation, Entities.Message>(Conversation.hash_func, Conversation.equals_func);
+                mam_potential_new[conversation.account] = new HashMap<Conversation, ContentItem>(Conversation.hash_func, Conversation.equals_func);
             }
-            mam_potential_new[conversation.account][conversation] = message;
+            mam_potential_new[conversation.account][conversation] = item;
             return;
         }
-        if (!should_notify_message(message, conversation)) return;
+        if (!should_notify(item, conversation)) return;
         if (stream_interactor.get_module(ChatInteraction.IDENTITY).is_active_focus()) return;
-        notify_message(message, conversation);
+        notify_content_item(item, conversation);
     }
 
-    private bool should_notify_message(Entities.Message message, Conversation conversation) {
+    private bool should_notify(ContentItem content_item, Conversation conversation) {
         Conversation.NotifySetting notify = conversation.get_notification_setting(stream_interactor);
+        switch (content_item.type_) {
+            case MessageItem.TYPE:
+                Message message = (content_item as MessageItem).message;
+                if (message.direction == Message.DIRECTION_SENT) return false;
+                break;
+            case FileItem.TYPE:
+                FileTransfer file_transfer = (content_item as FileItem).file_transfer;
+                if (file_transfer.direction == FileTransfer.DIRECTION_SENT) return false;
+                break;
+        }
         if (notify == Conversation.NotifySetting.OFF) return false;
         Jid? nick = stream_interactor.get_module(MucManager.IDENTITY).get_own_jid(conversation.counterpart, conversation.account);
-        if (notify == Conversation.NotifySetting.HIGHLIGHT && nick != null) {
-            return Regex.match_simple("\\b" + Regex.escape_string(nick.resourcepart) + "\\b", message.body, RegexCompileFlags.CASELESS);
+        if (content_item.type_ == MessageItem.TYPE) {
+            Entities.Message message = (content_item as MessageItem).message;
+            if (notify == Conversation.NotifySetting.HIGHLIGHT && nick != null) {
+                return Regex.match_simple("\\b" + Regex.escape_string(nick.resourcepart) + "\\b", message.body, RegexCompileFlags.CASELESS);
+            }
         }
         return true;
     }