Refactoring

2025-12-22 22:25:57 +00:00 · 2021-07-08 02:35:51 +04:00
parent 8f656dbce1
commit b4f35689f0
38 changed files with 171 additions and 85 deletions
--- a/submodules/TelegramCore/Sources/TelegramEngine/Messages/InstallInteractiveReadMessagesAction.swift
+++ b/submodules/TelegramCore/Sources/TelegramEngine/Messages/InstallInteractiveReadMessagesAction.swift
@@ -0,0 +1,62 @@
+import Foundation
+import Postbox
+import TelegramApi
+import SwiftSignalKit
+
+import SyncCore
+
+func _internal_installInteractiveReadMessagesAction(postbox: Postbox, stateManager: AccountStateManager, peerId: PeerId) -> Disposable {
+    return postbox.installStoreMessageAction(peerId: peerId, { messages, transaction in
+        var consumeMessageIds: [MessageId] = []
+        
+        var readMessageIndexByNamespace: [MessageId.Namespace: MessageIndex] = [:]
+        
+        for message in messages {
+            if case let .Id(id) = message.id {
+                var hasUnconsumedMention = false
+                var hasUnconsumedContent = false
+                
+                if message.tags.contains(.unseenPersonalMessage) {
+                    inner: for attribute in message.attributes {
+                        if let attribute = attribute as? ConsumablePersonalMentionMessageAttribute, !attribute.consumed, !attribute.pending {
+                            hasUnconsumedMention = true
+                        } else if let attribute = attribute as? ConsumableContentMessageAttribute, !attribute.consumed {
+                            hasUnconsumedContent = true
+                        }
+                    }
+                }
+                
+                if hasUnconsumedMention && !hasUnconsumedContent {
+                    consumeMessageIds.append(id)
+                }
+                
+                if !message.flags.intersection(.IsIncomingMask).isEmpty {
+                    let index = MessageIndex(id: id, timestamp: message.timestamp)
+                    let current = readMessageIndexByNamespace[id.namespace]
+                    if current == nil || current! < index {
+                        readMessageIndexByNamespace[id.namespace] = index
+                    }
+                }
+            }
+        }
+        
+        for id in consumeMessageIds {
+            transaction.updateMessage(id, update: { currentMessage in
+                var attributes = currentMessage.attributes
+                loop: for j in 0 ..< attributes.count {
+                    if let attribute = attributes[j] as? ConsumablePersonalMentionMessageAttribute {
+                        attributes[j] = ConsumablePersonalMentionMessageAttribute(consumed: attribute.consumed, pending: true)
+                        break loop
+                    }
+                }
+                return .update(StoreMessage(id: currentMessage.id, globallyUniqueId: currentMessage.globallyUniqueId, groupingKey: currentMessage.groupingKey, threadId: currentMessage.threadId, timestamp: currentMessage.timestamp, flags: StoreMessageFlags(currentMessage.flags), tags: currentMessage.tags, globalTags: currentMessage.globalTags, localTags: currentMessage.localTags, forwardInfo: currentMessage.forwardInfo.flatMap(StoreMessageForwardInfo.init), authorId: currentMessage.author?.id, text: currentMessage.text, attributes: attributes, media: currentMessage.media))
+            })
+            
+            transaction.setPendingMessageAction(type: .consumeUnseenPersonalMessage, id: id, action: ConsumePersonalMessageAction())
+        }
+        
+        for (_, index) in readMessageIndexByNamespace {
+            _internal_applyMaxReadIndexInteractively(transaction: transaction, stateManager: stateManager, index: index)
+        }
+    })
+}