Merge pull request #1404 from nextcloud/reload-referencing-messages

Update referencing messages when a message was updated
nextcloud · Nov 3, 2023 · ba2409c · ba2409c
2 parents 0ec421f + b3b6350
commit ba2409c
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/NextcloudTalk/BaseChatViewController.swift b/NextcloudTalk/BaseChatViewController.swift
@@ -595,14 +595,28 @@ import QuickLook
     internal func updateMessage(withMessageId messageId: Int, updatedMessage: NCChatMessage) {
         DispatchQueue.main.async {
             guard let (indexPath, message) = self.indexPathAndMessage(forMessageId: messageId) else { return }
+            var reloadIndexPaths = [indexPath]
 
             let isAtBottom = self.shouldScrollOnNewMessages()
             let keyDate = self.dateSections[indexPath.section]
             updatedMessage.isGroupMessage = message.isGroupMessage && message.actorType != "bots"
             self.messages[keyDate]?[indexPath.row] = updatedMessage
 
+            // Check if there are any messages that reference our message as a parent -> these need to be reloaded as well
+            if let visibleIndexPaths = self.tableView?.indexPathsForVisibleRows {
+                let referencingIndexPaths = visibleIndexPaths.filter({
+                    guard let message = self.message(for: $0),
+                          let parentMessage = message.parent()
+                    else { return false }
+
+                    return parentMessage.messageId == messageId
+                })
+
+                reloadIndexPaths.append(contentsOf: referencingIndexPaths)
+            }
+
             self.tableView?.beginUpdates()
-            self.tableView?.reloadRows(at: [indexPath], with: .none)
+            self.tableView?.reloadRows(at: reloadIndexPaths, with: .none)
             self.tableView?.endUpdates()
 
             if isAtBottom {