From af4a3a5f27857be350c1b45911c80c4aaf664358 Mon Sep 17 00:00:00 2001
From: k1ngsterr1 <ruslanmakhmatov@gmail.com>
Date: Thu, 19 Mar 2026 15:29:09 +0500
Subject: [PATCH] =?UTF-8?q?=D0=A1=D0=B8=D0=BD=D1=85=D1=80=D0=BE=D0=BD?=
 =?UTF-8?q?=D0=B8=D0=B7=D0=B0=D1=86=D0=B8=D1=8F=20=D1=81=D1=82=D0=B0=D1=82?=
 =?UTF-8?q?=D1=83=D1=81=D0=BE=D0=B2=20=D0=B8=20=D0=B2=D1=80=D0=B5=D0=BC?=
 =?UTF-8?q?=D0=B5=D0=BD=D0=B8=20=D1=81=D1=83=D1=89=D0=B5=D1=81=D1=82=D0=B2?=
 =?UTF-8?q?=D1=83=D1=8E=D1=89=D0=B8=D1=85=20=D1=81=D0=BE=D0=BE=D0=B1=D1=89?=
 =?UTF-8?q?=D0=B5=D0=BD=D0=B8=D0=B9,=20=D0=B4=D0=BE=D0=B1=D0=B0=D0=B2?=
 =?UTF-8?q?=D0=BB=D0=B5=D0=BD=D0=B8=D0=B5=20=D0=BD=D0=BE=D0=B2=D1=8B=D1=85?=
 =?UTF-8?q?=20=D1=81=D0=BE=D0=BE=D0=B1=D1=89=D0=B5=D0=BD=D0=B8=D0=B9=20?=
 =?UTF-8?q?=D0=B2=20=D0=BA=D1=8D=D1=88?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../messenger/ui/chats/ChatViewModel.kt       | 69 ++++++++++++-------
 1 file changed, 46 insertions(+), 23 deletions(-)

diff --git a/app/src/main/java/com/rosetta/messenger/ui/chats/ChatViewModel.kt b/app/src/main/java/com/rosetta/messenger/ui/chats/ChatViewModel.kt
index 1e42678..2549dd5 100644
--- a/app/src/main/java/com/rosetta/messenger/ui/chats/ChatViewModel.kt
+++ b/app/src/main/java/com/rosetta/messenger/ui/chats/ChatViewModel.kt
@@ -902,37 +902,60 @@ class ChatViewModel(application: Application) : AndroidViewModel(application) {
             // 🔥 Берём ТЕКУЩЕЕ состояние UI (может уже содержать новые сообщения)
             val currentMessages = _messages.value
             val existingIds = currentMessages.map { it.id }.toSet()
+            val entitiesById = entities.associateBy { it.messageId }
+
+            // 🔄 Синхронизируем статусы/время уже существующих сообщений из БД.
+            // Это критично для случая WAITING -> ERROR: без этого в открытом диалоге остаются "часики".
+            val reconciledMessages =
+                    currentMessages.map { message ->
+                        val entity = entitiesById[message.id] ?: return@map message
+
+                        val dbStatus =
+                                when (entity.delivered) {
+                                    0 -> MessageStatus.SENDING
+                                    1 -> if (entity.read == 1) MessageStatus.READ else MessageStatus.DELIVERED
+                                    2 -> MessageStatus.ERROR
+                                    3 -> MessageStatus.READ
+                                    else -> MessageStatus.SENT
+                                }
+
+                        var updatedMessage = message
+                        if (updatedMessage.status != dbStatus) {
+                            updatedMessage = updatedMessage.copy(status = dbStatus)
+                        }
+                        if (updatedMessage.timestamp.time != entity.timestamp) {
+                            updatedMessage = updatedMessage.copy(timestamp = Date(entity.timestamp))
+                        }
+                        updatedMessage
+                    }
+            val hasExistingUpdates = reconciledMessages != currentMessages
 
             // 🔥 Находим только НОВЫЕ сообщения (которых нет в текущем UI)
             val newEntities = entities.filter { it.messageId !in existingIds }
 
-            if (newEntities.isNotEmpty()) {
-                val semaphore = Semaphore(DECRYPT_PARALLELISM)
-                val newMessages = coroutineScope {
-                    newEntities
-                            .map { entity ->
-                                async { semaphore.withPermit { entityToChatMessage(entity) } }
-                            }
-                            .awaitAll()
-                }
+            val newMessages =
+                    if (newEntities.isNotEmpty()) {
+                        val semaphore = Semaphore(DECRYPT_PARALLELISM)
+                        coroutineScope {
+                            newEntities
+                                    .map { entity ->
+                                        async { semaphore.withPermit { entityToChatMessage(entity) } }
+                                    }
+                                    .awaitAll()
+                        }
+                    } else {
+                        emptyList()
+                    }
 
-                // 🔥 ДОБАВЛЯЕМ новые к текущим, а не заменяем!
-                // Сортируем по timestamp чтобы новые были в конце
+            if (hasExistingUpdates || newMessages.isNotEmpty()) {
+                // 🔥 ДОБАВЛЯЕМ новые + применяем статусные апдейты к существующим
                 val updatedMessages =
-                        sortMessagesAscending((currentMessages + newMessages).distinctBy { it.id })
+                        sortMessagesAscending((reconciledMessages + newMessages).distinctBy { it.id })
 
-                // 🔥 ИСПРАВЛЕНИЕ: Обновляем кэш сохраняя ВСЕ сообщения, не только отображаемые!
-                // Объединяем существующий кэш с новыми сообщениями
+                // 🔥 Обновляем кэш: сохраняем старые страницы + применяем новые статусы/время
                 val existingCache = dialogMessagesCache[cacheKey(account, dialogKey)] ?: emptyList()
-                val allCachedIds = existingCache.map { it.id }.toSet()
-                val trulyNewMessages = newMessages.filter { it.id !in allCachedIds }
-                if (trulyNewMessages.isNotEmpty()) {
-                    updateCacheWithLimit(
-                            account,
-                            dialogKey,
-                            sortMessagesAscending(existingCache + trulyNewMessages)
-                    )
-                }
+                val mergedCache = sortMessagesAscending((existingCache + updatedMessages).distinctBy { it.id })
+                updateCacheWithLimit(account, dialogKey, mergedCache)
 
                 withContext(Dispatchers.Main.immediate) { _messages.value = updatedMessages }
             }