TakeV
/
dino
mirror of https://github.com/TakeV-Lambda/dino.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix reactions from MAM getting attributed to the wrong message

This commit is contained in:
fiaxh 2023-01-25 19:42:19 +01:00
parent e35df88d4a
commit 26be9d4bb4
2 changed files with 26 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libdino/src/service/content_item_store.vala
View File

@ -139,7 +139,7 @@ public class ContentItemStore : StreamInteractionModule, Object {
return message.from; return message.from;
} }
private Message? get_message_for_content_item(Conversation conversation, ContentItem content_item) { public Message? get_message_for_content_item(Conversation conversation, ContentItem content_item) {
FileItem? file_item = content_item as FileItem; FileItem? file_item = content_item as FileItem;
if (file_item != null) { if (file_item != null) {
if (file_item.file_transfer.provider != 0 || file_item.file_transfer.info == null) return null; if (file_item.file_transfer.provider != 0 || file_item.file_transfer.info == null) return null;

62
libdino/src/service/reactions.vala
View File

@ -26,7 +26,7 @@ public class Dino.Reactions : StreamInteractionModule, Object {
this.db = database; this.db = database;
stream_interactor.account_added.connect(on_account_added); stream_interactor.account_added.connect(on_account_added);
stream_interactor.get_module(MessageProcessor.IDENTITY).message_sent_or_received.connect(on_new_message); stream_interactor.get_module(ContentItemStore.IDENTITY).new_item.connect(on_new_item);
} }
public void add_reaction(Conversation conversation, ContentItem content_item, string reaction) { public void add_reaction(Conversation conversation, ContentItem content_item, string reaction) {
@ -248,7 +248,7 @@ public class Dino.Reactions : StreamInteractionModule, Object {
Conversation conversation = stream_interactor.get_module(ConversationManager.IDENTITY).get_conversation_for_message(reaction_message); Conversation conversation = stream_interactor.get_module(ConversationManager.IDENTITY).get_conversation_for_message(reaction_message);
int content_item_id = stream_interactor.get_module(ContentItemStore.IDENTITY).get_content_item_id_for_message_id(conversation, message_id); int content_item_id = stream_interactor.get_module(ContentItemStore.IDENTITY).get_content_item_id_for_message_id(conversation, message_id);
var reaction_info = new ReactionInfo() { account=account, from_jid=from_jid, reactions=reactions, stanza=stanza, received_time=new DateTime.now() }; var reaction_info = new ReactionInfo() { conversation=conversation, from_jid=from_jid, reactions=reactions, stanza=stanza, received_time=new DateTime.now() };
if (content_item_id != -1) { if (content_item_id != -1) {
process_reaction_for_message(content_item_id, reaction_info); process_reaction_for_message(content_item_id, reaction_info);
@ -263,45 +263,33 @@ public class Dino.Reactions : StreamInteractionModule, Object {
reaction_infos[message_id].add(reaction_info); reaction_infos[message_id].add(reaction_info);
} }
private void on_new_message(Message message, Conversation conversation) { /*
Gee.List<ReactionInfo>? reaction_info_list = null; * When we get a new ContentItem, check if we have any reactions cached that apply to it.
if (conversation.type_ == Conversation.Type.CHAT) { * If so, process the reactions, map and store them.
reaction_info_list = reaction_infos[message.stanza_id]; */
} else { private void on_new_item(ContentItem item, Conversation conversation) {
reaction_info_list = reaction_infos[message.server_id]; string? stanza_id = stream_interactor.get_module(ContentItemStore.IDENTITY).get_message_id_for_content_item(conversation, item);
} if (stanza_id == null) return;
Gee.List<ReactionInfo>? reaction_info_list = reaction_infos[stanza_id];
if (reaction_info_list == null) return; if (reaction_info_list == null) return;
Message? message = stream_interactor.get_module(ContentItemStore.IDENTITY).get_message_for_content_item(conversation, item);
if (message == null) return;
// Check if the (or potentially which) reaction fits the message // Check if the (or potentially which) reaction fits the message
ReactionInfo? reaction_info = null; var applicable_reactions = new ArrayList<ReactionInfo>();
foreach (ReactionInfo info in reaction_info_list) { applicable_reactions.add_all_iterator(reaction_info_list.filter(info => info.conversation.equals(conversation)));
if (!info.account.equals(conversation.account)) return;
switch (info.stanza.type_) { foreach (ReactionInfo applicable_reaction in applicable_reactions) {
case MessageStanza.TYPE_CHAT: reaction_info_list.remove(applicable_reaction);
Jid counterpart = message.from.equals_bare(conversation.account.bare_jid) ? info.stanza.from: info.stanza.to; if (reaction_info_list.is_empty) {
if (message.type_ != Message.Type.CHAT || !counterpart.equals_bare(conversation.counterpart)) continue; reaction_infos.unset(stanza_id);
break;
case MessageStanza.TYPE_GROUPCHAT:
if (message.type_ != Message.Type.GROUPCHAT || !message.from.equals_bare(conversation.counterpart)) continue;
break;
default:
break;
} }
reaction_info = info; debug("Got ContentItem for reaction %s", stanza_id);
process_reaction_for_message(item.id, applicable_reaction);
} }
if (reaction_info == null) return;
reaction_info_list.remove(reaction_info);
if (reaction_info_list.is_empty) {
if (conversation.type_ == Conversation.Type.GROUPCHAT) {
reaction_infos.unset(message.server_id);
} else {
reaction_infos.unset(message.stanza_id);
}
}
debug("Got message for reaction %s", message.stanza_id);
process_reaction_for_message(message.id, reaction_info);
} }
private Message? get_message_for_reaction(Conversation conversation, string message_id) { private Message? get_message_for_reaction(Conversation conversation, string message_id) {
@ -314,7 +302,7 @@ public class Dino.Reactions : StreamInteractionModule, Object {
} }
private void process_reaction_for_message(int content_item_id, ReactionInfo reaction_info) { private void process_reaction_for_message(int content_item_id, ReactionInfo reaction_info) {
Account account = reaction_info.account; Account account = reaction_info.conversation.account;
MessageStanza stanza = reaction_info.stanza; MessageStanza stanza = reaction_info.stanza;
Jid from_jid = reaction_info.from_jid; Jid from_jid = reaction_info.from_jid;
Gee.List<string> reactions = reaction_info.reactions; Gee.List<string> reactions = reaction_info.reactions;
@ -463,7 +451,7 @@ public class Dino.ReactionUsers {
} }
public class Dino.ReactionInfo { public class Dino.ReactionInfo {
public Account account { get; set; } public Conversation conversation { get; set; }
public Jid from_jid { get; set; } public Jid from_jid { get; set; }
public Gee.List<string> reactions { get; set; } public Gee.List<string> reactions { get; set; }
public MessageStanza stanza { get; set; } public MessageStanza stanza { get; set; }