Reactions update [skip ci]

2025-12-22 14:20:20 +00:00 · 2022-01-25 14:08:52 +04:00
parent e694aa319d
commit f47fa686e9
18 changed files with 497 additions and 152 deletions
--- a/submodules/TelegramCore/Sources/State/ManagedConsumePersonalMessagesActions.swift
+++ b/submodules/TelegramCore/Sources/State/ManagedConsumePersonalMessagesActions.swift
@@ -283,40 +283,71 @@ private func synchronizeConsumeMessageContents(transaction: Transaction, postbox
 }

 private func synchronizeReadMessageReactions(transaction: Transaction, postbox: Postbox, network: Network, stateManager: AccountStateManager, id: MessageId) -> Signal<Void, NoError> {
-    guard let inputPeer = transaction.getPeer(id.peerId).flatMap(apiInputPeer) else {
-        return .complete()
-    }
-    return network.request(Api.functions.messages.readReactions(peer: inputPeer))
-    |> map(Optional.init)
-    |> `catch` { _ -> Signal<Api.messages.AffectedHistory?, NoError> in
-        return .single(nil)
-    }
-    |> mapToSignal { result -> Signal<Void, NoError> in
-        if let result = result {
-            switch result {
-            case let .affectedHistory(pts, ptsCount, _):
-                stateManager.addUpdateGroups([.updatePts(pts: pts, ptsCount: ptsCount)])
+    if id.peerId.namespace == Namespaces.Peer.CloudUser || id.peerId.namespace == Namespaces.Peer.CloudGroup {
+        return network.request(Api.functions.messages.readMessageContents(id: [id.id]))
+        |> map(Optional.init)
+        |> `catch` { _ -> Signal<Api.messages.AffectedMessages?, NoError> in
+            return .single(nil)
+        }
+        |> mapToSignal { result -> Signal<Void, NoError> in
+            if let result = result {
+                switch result {
+                    case let .affectedMessages(pts, ptsCount):
+                        stateManager.addUpdateGroups([.updatePts(pts: pts, ptsCount: ptsCount)])
+                }
+            }
+            return postbox.transaction { transaction -> Void in
+                transaction.setPendingMessageAction(type: .readReaction, id: id, action: nil)
+                transaction.updateMessage(id, update: { currentMessage in
+                    var storeForwardInfo: StoreMessageForwardInfo?
+                    if let forwardInfo = currentMessage.forwardInfo {
+                        storeForwardInfo = StoreMessageForwardInfo(authorId: forwardInfo.author?.id, sourceId: forwardInfo.source?.id, sourceMessageId: forwardInfo.sourceMessageId, date: forwardInfo.date, authorSignature: forwardInfo.authorSignature, psaType: forwardInfo.psaType, flags: forwardInfo.flags)
+                    }
+                    var attributes = currentMessage.attributes
+                    loop: for j in 0 ..< attributes.count {
+                        if let attribute = attributes[j] as? ReactionsMessageAttribute, attribute.hasUnseen {
+                            attributes[j] = attribute.withAllSeen()
+                            break loop
+                        }
+                    }
+                    var updatedTags = currentMessage.tags
+                    updatedTags.remove(.unseenReaction)
+                    return .update(StoreMessage(id: currentMessage.id, globallyUniqueId: currentMessage.globallyUniqueId, groupingKey: currentMessage.groupingKey, threadId: currentMessage.threadId, timestamp: currentMessage.timestamp, flags: StoreMessageFlags(currentMessage.flags), tags: updatedTags, globalTags: currentMessage.globalTags, localTags: currentMessage.localTags, forwardInfo: storeForwardInfo, authorId: currentMessage.author?.id, text: currentMessage.text, attributes: attributes, media: currentMessage.media))
+                })
            }
        }
-        return postbox.transaction { transaction -> Void in
-            transaction.setPendingMessageAction(type: .readReaction, id: id, action: nil)
-            transaction.updateMessage(id, update: { currentMessage in
-                var storeForwardInfo: StoreMessageForwardInfo?
-                if let forwardInfo = currentMessage.forwardInfo {
-                    storeForwardInfo = StoreMessageForwardInfo(authorId: forwardInfo.author?.id, sourceId: forwardInfo.source?.id, sourceMessageId: forwardInfo.sourceMessageId, date: forwardInfo.date, authorSignature: forwardInfo.authorSignature, psaType: forwardInfo.psaType, flags: forwardInfo.flags)
+    } else if id.peerId.namespace == Namespaces.Peer.CloudChannel {
+        if let peer = transaction.getPeer(id.peerId), let inputChannel = apiInputChannel(peer) {
+            return network.request(Api.functions.channels.readMessageContents(channel: inputChannel, id: [id.id]))
+            |> `catch` { _ -> Signal<Api.Bool, NoError> in
+                return .single(.boolFalse)
+            }
+            |> mapToSignal { result -> Signal<Void, NoError> in
+                return postbox.transaction { transaction -> Void in
+                    transaction.setPendingMessageAction(type: .readReaction, id: id, action: nil)
+                    transaction.updateMessage(id, update: { currentMessage in
+                        var storeForwardInfo: StoreMessageForwardInfo?
+                        if let forwardInfo = currentMessage.forwardInfo {
+                            storeForwardInfo = StoreMessageForwardInfo(authorId: forwardInfo.author?.id, sourceId: forwardInfo.source?.id, sourceMessageId: forwardInfo.sourceMessageId, date: forwardInfo.date, authorSignature: forwardInfo.authorSignature, psaType: forwardInfo.psaType, flags: forwardInfo.flags)
+                        }
+                        var attributes = currentMessage.attributes
+                        loop: for j in 0 ..< attributes.count {
+                            if let attribute = attributes[j] as? ReactionsMessageAttribute, attribute.hasUnseen {
+                                attributes[j] = attribute.withAllSeen()
+                                break loop
+                            }
+                        }
+                        var updatedTags = currentMessage.tags
+                        updatedTags.remove(.unseenReaction)
+                        return .update(StoreMessage(id: currentMessage.id, globallyUniqueId: currentMessage.globallyUniqueId, groupingKey: currentMessage.groupingKey, threadId: currentMessage.threadId, timestamp: currentMessage.timestamp, flags: StoreMessageFlags(currentMessage.flags), tags: updatedTags, globalTags: currentMessage.globalTags, localTags: currentMessage.localTags, forwardInfo: storeForwardInfo, authorId: currentMessage.author?.id, text: currentMessage.text, attributes: attributes, media: currentMessage.media))
+                    })
                }
-                var attributes = currentMessage.attributes
-                loop: for j in 0 ..< attributes.count {
-                    if let attribute = attributes[j] as? ReactionsMessageAttribute, attribute.hasUnseen {
-                        attributes[j] = attribute.withAllSeen()
-                        break loop
-                    }
-                }
-                var updatedTags = currentMessage.tags
-                updatedTags.remove(.unseenReaction)
-                return .update(StoreMessage(id: currentMessage.id, globallyUniqueId: currentMessage.globallyUniqueId, groupingKey: currentMessage.groupingKey, threadId: currentMessage.threadId, timestamp: currentMessage.timestamp, flags: StoreMessageFlags(currentMessage.flags), tags: updatedTags, globalTags: currentMessage.globalTags, localTags: currentMessage.localTags, forwardInfo: storeForwardInfo, authorId: currentMessage.author?.id, text: currentMessage.text, attributes: attributes, media: currentMessage.media))
-            })
+            }
+        } else {
+            return .complete()
        }
+    } else {
+        return .complete()
    }
 }