Swiftgram/submodules/ReactionSelectionNode/Sources/ReactionSelectionParentNode.swift

import Foundation
import AsyncDisplayKit
import Display
import Postbox
import TelegramCore

public final class ReactionSelectionParentNode: ASDisplayNode {
    private let account: Account
    
    private var currentNode: ReactionSelectionNode?
    private var currentLocation: (CGPoint, CGFloat)?
    
    private var validLayout: (size: CGSize, insets: UIEdgeInsets)?
    
    public init(account: Account) {
        self.account = account
        
        super.init()
    }
    
    func displayReactions(_ reactions: [ReactionGestureItem], at point: CGPoint) {
        if let currentNode = self.currentNode {
            currentNode.removeFromSupernode()
            self.currentNode = nil
        }
        
        let reactionNode = ReactionSelectionNode(account: self.account, reactions: reactions)
        self.addSubnode(reactionNode)
        self.currentNode = reactionNode
        self.currentLocation = (point, point.x)
        
        if let (size, insets) = self.validLayout {
            self.update(size: size, insets: insets, isInitial: true)
            
            reactionNode.animateIn()
        }
    }
    
    func dismissReactions() {
        if let currentNode = self.currentNode {
            currentNode.animateOut(completion: { [weak currentNode] in
                currentNode?.removeFromSupernode()
            })
            self.currentNode = nil
        }
    }
    
    func updateReactionsAnchor(point: CGPoint) {
        if let (currentPoint, _) = self.currentLocation {
            self.currentLocation = (currentPoint, point.x)
            
            if let (size, insets) = self.validLayout {
                self.update(size: size, insets: insets, isInitial: false)
            }
        }
    }
    
    public func updateLayout(size: CGSize, insets: UIEdgeInsets, transition: ContainedViewLayoutTransition) {
        self.validLayout = (size, insets)
        
        self.update(size: size, insets: insets, isInitial: false)
    }
    
    private func update(size: CGSize, insets: UIEdgeInsets, isInitial: Bool) {
        if let currentNode = self.currentNode, let (point, offset) = currentLocation {
            currentNode.updateLayout(constrainedSize: size, startingPoint: point, offsetFromStart: offset, isInitial: isInitial)
            currentNode.frame = CGRect(origin: CGPoint(), size: size)
        }
    }
    
    override public func hitTest(_ point: CGPoint, with event: UIEvent?) -> UIView? {
        return nil
    }
}