Message preview improvements

submodules/TelegramUI/Components/Chat/ChatSendAudioMessageContextPreview/Sources/ChatSendAudioMessageContextPreview.swift

@@ -14,6 +14,180 @@ import WallpaperBackgroundNode
 import AudioWaveform
 import ChatMessageItemView
 
+public final class ChatSendContactMessageContextPreview: UIView, ChatSendMessageContextScreenMediaPreview {
+    private let context: AccountContext
+    private let presentationData: PresentationData
+    private let wallpaperBackgroundNode: WallpaperBackgroundNode?
+    private let contactPeers: [ContactListPeer]
+    
+    private var messageNodes: [ListViewItemNode]?
+    private let messagesContainer: UIView
+    
+    public var isReady: Signal<Bool, NoError> {
+        return .single(true)
+    }
+
+    public var view: UIView {
+        return self
+    }
+    
+    public var globalClippingRect: CGRect? {
+        return nil
+    }
+
+    public var layoutType: ChatSendMessageContextScreenMediaPreviewLayoutType {
+        return .message
+    }
+    
+    public init(context: AccountContext, presentationData: PresentationData, wallpaperBackgroundNode: WallpaperBackgroundNode?, contactPeers: [ContactListPeer]) {
+        self.context = context
+        self.presentationData = presentationData
+        self.wallpaperBackgroundNode = wallpaperBackgroundNode
+        self.contactPeers = contactPeers
+        
+        self.messagesContainer = UIView()
+        self.messagesContainer.layer.sublayerTransform = CATransform3DMakeScale(-1.0, -1.0, 1.0)
+        
+        super.init(frame: CGRect())
+        
+        self.addSubview(self.messagesContainer)
+    }
+    
+    required public init?(coder: NSCoder) {
+        fatalError("init(coder:) has not been implemented")
+    }
+    
+    deinit {
+    }
+    
+    public func animateIn(transition: Transition) {
+        transition.animateAlpha(view: self.messagesContainer, from: 0.0, to: 1.0)
+        transition.animateScale(view: self.messagesContainer, from: 0.001, to: 1.0)
+    }
+
+    public func animateOut(transition: Transition) {
+        transition.setAlpha(view: self.messagesContainer, alpha: 0.0)
+        transition.setScale(view: self.messagesContainer, scale: 0.001)
+    }
+
+    public func animateOutOnSend(transition: Transition) {
+        transition.setAlpha(view: self.messagesContainer, alpha: 0.0)
+    }
+
+    public func update(containerSize: CGSize, transition: Transition) -> CGSize {
+        var contactsMedia: [TelegramMediaContact] = []
+        for peer in self.contactPeers {
+            switch peer {
+            case let .peer(contact, _, _):
+                guard let contact = contact as? TelegramUser, let phoneNumber = contact.phone else {
+                    continue
+                }
+                let contactData = DeviceContactExtendedData(basicData: DeviceContactBasicData(firstName: contact.firstName ?? "", lastName: contact.lastName ?? "", phoneNumbers: [DeviceContactPhoneNumberData(label: "_$!<Mobile>!$_", value: phoneNumber)]), middleName: "", prefix: "", suffix: "", organization: "", jobTitle: "", department: "", emailAddresses: [], urls: [], addresses: [], birthdayDate: nil, socialProfiles: [], instantMessagingProfiles: [], note: "")
+                
+                let phone = contactData.basicData.phoneNumbers[0].value
+                contactsMedia.append(TelegramMediaContact(firstName: contactData.basicData.firstName, lastName: contactData.basicData.lastName, phoneNumber: phone, peerId: contact.id, vCardData: nil))
+            case let .deviceContact(_, basicData):
+                guard !basicData.phoneNumbers.isEmpty else {
+                    continue
+                }
+                let contactData = DeviceContactExtendedData(basicData: basicData, middleName: "", prefix: "", suffix: "", organization: "", jobTitle: "", department: "", emailAddresses: [], urls: [], addresses: [], birthdayDate: nil, socialProfiles: [], instantMessagingProfiles: [], note: "")
+                
+                let phone = contactData.basicData.phoneNumbers[0].value
+                contactsMedia.append(TelegramMediaContact(firstName: contactData.basicData.firstName, lastName: contactData.basicData.lastName, phoneNumber: phone, peerId: nil, vCardData: nil))
+            }
+        }
+        
+        var items: [ListViewItem] = []
+        for contactMedia in contactsMedia {
+            let message = Message(stableId: 1, stableVersion: 0, id: MessageId(peerId: self.context.account.peerId, namespace: 0, id: 1), globallyUniqueId: nil, groupingKey: nil, groupInfo: nil, threadId: nil, timestamp: 0, flags: [.Incoming], tags: [], globalTags: [], localTags: [], customTags: [], forwardInfo: nil, author: nil, text: "", attributes: [], media: [contactMedia], peers: SimpleDictionary(), associatedMessages: SimpleDictionary(), associatedMessageIds: [], associatedMedia: [:], associatedThreadInfo: nil, associatedStories: [:])
+            
+            let item = self.context.sharedContext.makeChatMessagePreviewItem(
+                context: self.context,
+                messages: [message],
+                theme: presentationData.theme,
+                strings: presentationData.strings,
+                wallpaper: presentationData.chatWallpaper,
+                fontSize: presentationData.chatFontSize,
+                chatBubbleCorners: presentationData.chatBubbleCorners,
+                dateTimeFormat: presentationData.dateTimeFormat,
+                nameOrder: presentationData.nameDisplayOrder,
+                forcedResourceStatus: FileMediaResourceStatus(mediaStatus: .fetchStatus(.Local), fetchStatus: .Local),
+                tapMessage: nil,
+                clickThroughMessage: nil,
+                backgroundNode: self.wallpaperBackgroundNode,
+                availableReactions: nil,
+                accountPeer: nil,
+                isCentered: false,
+                isPreview: true,
+                isStandalone: true
+            )
+            items.append(item)
+        }
+        
+        let params = ListViewItemLayoutParams(width: containerSize.width, leftInset: 0.0, rightInset: 0.0, availableHeight: containerSize.height)
+        if let messageNodes = self.messageNodes {
+            for i in 0 ..< items.count {
+                let itemNode = messageNodes[i]
+                items[i].updateNode(async: { $0() }, node: {
+                    return itemNode
+                }, params: params, previousItem: i == 0 ? nil : items[i - 1], nextItem: i == (items.count - 1) ? nil : items[i + 1], animation: .None, completion: { (layout, apply) in
+                    let nodeFrame = CGRect(origin: CGPoint(x: itemNode.frame.minX, y: itemNode.frame.minY), size: CGSize(width: containerSize.width, height: layout.size.height))
+                    
+                    itemNode.contentSize = layout.contentSize
+                    itemNode.insets = layout.insets
+                    itemNode.frame = nodeFrame
+                    itemNode.isUserInteractionEnabled = false
+                    
+                    apply(ListViewItemApply(isOnScreen: true))
+                })
+            }
+        } else {
+            var messageNodes: [ListViewItemNode] = []
+            for i in 0 ..< items.count {
+                var itemNode: ListViewItemNode?
+                items[i].nodeConfiguredForParams(async: { $0() }, params: params, synchronousLoads: false, previousItem: i == 0 ? nil : items[i - 1], nextItem: i == (items.count - 1) ? nil : items[i + 1], completion: { node, apply in
+                    itemNode = node
+                    apply().1(ListViewItemApply(isOnScreen: true))
+                })
+                itemNode!.isUserInteractionEnabled = false
+                messageNodes.append(itemNode!)
+                self.messagesContainer.addSubview(itemNode!.view)
+            }
+            self.messageNodes = messageNodes
+        }
+        
+        var contentSize = CGSize()
+        for messageNode in self.messageNodes ?? [] {
+            guard let messageNode = messageNode as? ChatMessageItemView else {
+                continue
+            }
+            if !contentSize.height.isZero {
+                contentSize.height += 2.0
+            }
+            let contentFrame = messageNode.contentFrame()
+            contentSize.height += contentFrame.height
+            contentSize.width = max(contentSize.width, contentFrame.width)
+        }
+        
+        var contentOffsetY: CGFloat = 0.0
+        for messageNode in self.messageNodes ?? [] {
+            guard let messageNode = messageNode as? ChatMessageItemView else {
+                continue
+            }
+            if !contentOffsetY.isZero {
+                contentOffsetY += 2.0
+            }
+            let contentFrame = messageNode.contentFrame()
+            messageNode.frame = CGRect(origin: CGPoint(x: contentFrame.minX + contentSize.width - contentFrame.width + 6.0, y: 3.0 + contentOffsetY), size: CGSize(width: contentFrame.width, height: contentFrame.height))
+            contentOffsetY += contentFrame.height
+        }
+        
+        self.messagesContainer.frame = CGRect(origin: CGPoint(x: 6.0, y: 3.0), size: CGSize(width: contentSize.width, height: contentSize.height))
+        
+        return CGSize(width: contentSize.width - 4.0, height: contentSize.height + 2.0)
+    }
+}
+
 public final class ChatSendAudioMessageContextPreview: UIView, ChatSendMessageContextScreenMediaPreview {
     private let context: AccountContext
     private let presentationData: PresentationData