When a message is hard deleted, make sure to reload the previous cell

GetStream · Aug 17, 2022 · 6492e7d · 6492e7d
1 parent 12e1d5a
commit 6492e7d
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/Sources/StreamChatUI/ChatMessageList/ChatMessageListView.swift b/Sources/StreamChatUI/ChatMessageList/ChatMessageListView.swift
@@ -222,6 +222,17 @@ open class ChatMessageListView: UITableView, Customizable, ComponentsProvider {
                             self?.reloadRows(at: [movedIndexPath], with: .none)
                         }
                     }
+
+                    // When there are deletions, we should update the previous message, so that we add the
+                    // avatar image back and the timestamp too. Since we have an inverted list, the previous
+                    // message has the same index of the deleted message after the deletion has been executed.
+                    let visibleRemoves = changes.filter {
+                        $0.isRemove && self?.indexPathsForVisibleRows?.contains($0.indexPath) == true
+                    }
+                    visibleRemoves.forEach {
+                        self?.reloadRows(at: [$0.indexPath], with: .none)
+                    }
+
                     completion?()
                 }
             )