Temp

submodules/TelegramUI/Sources/ChatMessageContextControllerContentSource.swift

@@ -0,0 +1,55 @@
+import Foundation
+import UIKit
+import Display
+import ContextUI
+import Postbox
+
+final class ChatMessageContextExtractedContentSource: ContextExtractedContentSource {
+    private weak var chatNode: ChatControllerNode?
+    private let message: Message
+    
+    init(chatNode: ChatControllerNode, message: Message) {
+        self.chatNode = chatNode
+        self.message = message
+    }
+    
+    func takeView() -> ContextControllerTakeViewInfo? {
+        guard let chatNode = self.chatNode else {
+            return nil
+        }
+        
+        var result: ContextControllerTakeViewInfo?
+        chatNode.historyNode.forEachItemNode { itemNode in
+            guard let itemNode = itemNode as? ChatMessageItemView else {
+                return
+            }
+            guard let item = itemNode.item else {
+                return
+            }
+            if item.content.contains(where: { $0.0.stableId == self.message.stableId }), let contentNode = itemNode.getMessageContextSourceNode() {
+                result = ContextControllerTakeViewInfo(contentContainingNode: contentNode, contentAreaInScreenSpace: chatNode.convert(chatNode.frameForVisibleArea(), to: nil))
+            }
+        }
+        return result
+    }
+    
+    func putBack() -> ContextControllerPutBackViewInfo? {
+        guard let chatNode = self.chatNode else {
+            return nil
+        }
+        
+        var result: ContextControllerPutBackViewInfo?
+        chatNode.historyNode.forEachItemNode { itemNode in
+            guard let itemNode = itemNode as? ChatMessageItemView else {
+                return
+            }
+            guard let item = itemNode.item else {
+                return
+            }
+            if item.content.contains(where: { $0.0.stableId == self.message.stableId }) {
+                result = ContextControllerPutBackViewInfo(contentAreaInScreenSpace: chatNode.convert(chatNode.frameForVisibleArea(), to: nil))
+            }
+        }
+        return result
+    }
+}