Fix message editing

app/src/main/java/chat/rocket/android/chatroom/adapter/ChatRoomAdapter.kt

@@ -87,12 +87,14 @@ class ChatRoomAdapter(
     }
 
     fun prependData(dataSet: List<BaseViewModel<*>>) {
-        val lastMessage = this.dataSet.first()
-        val lastId = lastMessage.messageId
-        val lastType = lastMessage.viewType
-        this.dataSet.addAll(0, dataSet.filterNot { lastType == it.viewType && lastId == it.messageId })
+        val item = dataSet.firstOrNull { newItem ->
+            this.dataSet.indexOfFirst { it.messageId == newItem.messageId && it.viewType == newItem.viewType } > -1
+        }
+        if (item == null) {
+            this.dataSet.addAll(0, dataSet)
             notifyItemRangeInserted(0, dataSet.size)
         }
+    }
 
     fun updateItem(message: BaseViewModel<*>) {
         val index = dataSet.indexOfLast { it.messageId == message.messageId }

app/src/main/java/chat/rocket/android/chatroom/ui/ChatRoomFragment.kt

@@ -213,9 +213,8 @@ class ChatRoomFragment : Fragment(), ChatRoomView, EmojiKeyboardPopup.Listener {
     }
 
     override fun dispatchUpdateMessage(index: Int, message: List<BaseViewModel<*>>) {
-        if (message.size == 1) {
         adapter.updateItem(message.last())
-        } else {
+        if (message.size > 1) {
             adapter.updateItem(message.last())
             adapter.prependData(listOf(message.first()))
         }