Swiftgram/submodules/TelegramUI/Components/ContextMenuScreen/Sources/ContextMenuActionNode.swift

import Foundation
import UIKit
import AsyncDisplayKit
import Display

final private class ContextMenuActionButton: HighlightTrackingButton {
    override func convert(_ point: CGPoint, from view: UIView?) -> CGPoint {
        if view is UIWindow {
            return super.convert(point, from: nil)
        } else {
            return super.convert(point, from: view)
        }
    }
}

final class ContextMenuActionNode: ASDisplayNode {
    private let textNode: ImmediateTextNode?
    private let subtitleNode: ImmediateTextNode?
    private var textSize: CGSize?
    private var subtitleSize: CGSize?
    private let iconView: UIImageView?
    private let action: () -> Void
    private let button: ContextMenuActionButton
    private let actionArea: AccessibilityAreaNode
    
    var dismiss: (() -> Void)?
    
    init(action: ContextMenuAction, blurred: Bool, isDark: Bool) {
        self.actionArea = AccessibilityAreaNode()
        self.actionArea.accessibilityTraits = .button
        
        switch action.content {
        case let .text(title, accessibilityLabel):
            self.actionArea.accessibilityLabel = accessibilityLabel
            
            let textNode = ImmediateTextNode()
            textNode.isUserInteractionEnabled = false
            textNode.displaysAsynchronously = false
            textNode.attributedText = NSAttributedString(string: title, font: Font.regular(14.0), textColor: isDark ? .white : .black)
            textNode.isAccessibilityElement = false
            
            self.textNode = textNode
            self.subtitleNode = nil
            self.iconView = nil
        case let .textWithIcon(title, icon):
            let textNode = ImmediateTextNode()
            textNode.isUserInteractionEnabled = false
            textNode.displaysAsynchronously = false
            textNode.attributedText = NSAttributedString(string: title, font: Font.regular(17.0), textColor: isDark ? .white : .black)
            textNode.isAccessibilityElement = false
            
            let iconView = UIImageView()
            iconView.tintColor = isDark ? .white : .black
            iconView.image = icon
            
            self.textNode = textNode
            self.subtitleNode = nil
            self.iconView = iconView
        case let .textWithSubtitleAndIcon(title, subtitle, icon):
            let textNode = ImmediateTextNode()
            textNode.isUserInteractionEnabled = false
            textNode.displaysAsynchronously = false
            textNode.attributedText = NSAttributedString(string: title, font: Font.regular(17.0), textColor: isDark ? .white : .black)
            textNode.isAccessibilityElement = false
            
            let subtitleNode = ImmediateTextNode()
            subtitleNode.isUserInteractionEnabled = false
            subtitleNode.displaysAsynchronously = false
            subtitleNode.attributedText = NSAttributedString(string: subtitle, font: Font.regular(12.0), textColor: (isDark ? UIColor.white : UIColor.black).withAlphaComponent(0.5))
            subtitleNode.isAccessibilityElement = false
            
            let iconView = UIImageView()
            iconView.tintColor = isDark ? .white : .black
            iconView.image = icon
            
            self.textNode = textNode
            self.subtitleNode = subtitleNode
            self.iconView = iconView
        case let .icon(image):
            let iconView = UIImageView()
            iconView.tintColor = isDark ? .white : .black
            iconView.image = image
            
            self.iconView = iconView
            self.textNode = nil
            self.subtitleNode = nil
        }
        self.action = action.action
        
        self.button = ContextMenuActionButton()
        self.button.isAccessibilityElement = false
        
        super.init()
        
        if !blurred {
            self.backgroundColor = isDark ? UIColor(rgb: 0x2f2f2f) : nil
        }
        
        if let textNode = self.textNode {
            self.addSubnode(textNode)
        }
        if let subtitleNode = self.subtitleNode {
            self.addSubnode(subtitleNode)
        }
        if let iconView = self.iconView {
            self.view.addSubview(iconView)
        }
        
        self.button.highligthedChanged = { [weak self] highlighted in
            if isDark {
                if blurred {
                    self?.backgroundColor = highlighted ? UIColor(rgb: 0xffffff, alpha: 0.5) : .clear
                } else {
                    self?.backgroundColor = highlighted ? UIColor(rgb: 0x8c8e8e) : UIColor(rgb: 0x2f2f2f)
                }
            } else {
                self?.backgroundColor = highlighted ? UIColor(rgb: 0xDCE3DC) : .clear
            }
        }
        self.view.addSubview(self.button)
        self.addSubnode(self.actionArea)
        
        self.actionArea.activate = { [weak self] in
            self?.buttonPressed()
            return true
        }
    }
    
    override func didLoad() {
        super.didLoad()
        
        self.button.addTarget(self, action: #selector(self.buttonPressed), for: [.touchUpInside])
    }
    
    @objc private func buttonPressed() {
        self.backgroundColor = UIColor(white: 0.0, alpha: 0.4)
        
        self.action()
        if let dismiss = self.dismiss {
            dismiss()
        }
    }
    
    override func calculateSizeThatFits(_ constrainedSize: CGSize) -> CGSize {
        if let textNode = self.textNode {
            let constrainedSize = CGSize(width: constrainedSize.width - 36.0 - 24.0, height: constrainedSize.height)
            let textSize = textNode.updateLayout(constrainedSize)
            self.textSize = textSize
            
            var totalWidth = 0.0
            var totalHeight: CGFloat = 54.0
            totalWidth += textSize.width
            
            if let subtitleNode = self.subtitleNode {
                let subtitleSize = subtitleNode.updateLayout(CGSize(width: constrainedSize.width * 0.75, height: constrainedSize.height))
                self.subtitleSize = subtitleSize
                totalWidth = max(totalWidth, subtitleSize.width)
                totalHeight += 14.0
            }
            
            if let image = self.iconView?.image {
                if totalWidth > 0.0 {
                    totalWidth += 11.0
                }
                totalWidth += image.size.width
                totalWidth += 24.0
            } else {
                totalWidth += 36.0
            }
            
            return CGSize(width: totalWidth, height: totalHeight)
        } else if let iconView = self.iconView, let image = iconView.image {
            return CGSize(width: image.size.width + 36.0, height: 54.0)
        } else {
            return CGSize(width: 36.0, height: 54.0)
        }
    }
    
    override func layout() {
        super.layout()
        
        self.button.frame = self.bounds
        self.actionArea.frame = self.bounds
        
        var totalWidth = 0.0
        if let textSize = self.textSize {
            totalWidth += textSize.width
        }
        if let subtitleSize = self.subtitleSize {
            totalWidth = max(totalWidth, subtitleSize.width)
        }
        if let image = self.iconView?.image {
            if totalWidth > 0.0 {
                totalWidth += 11.0
            }
            totalWidth += image.size.width
        }
        
        var totalTextHeight: CGFloat = 0.0
        if let textSize = self.textSize {
            totalTextHeight += textSize.height
        }
        if let subtitleSize = self.subtitleSize {
            totalTextHeight += subtitleSize.height
        }
        if let textNode = self.textNode, let textSize = self.textSize {
            textNode.frame = CGRect(origin: CGPoint(x: floor((self.bounds.size.width - totalWidth) / 2.0), y: floor((self.bounds.size.height - totalTextHeight) / 2.0)), size: textSize)
        }
        if let subtitleNode = self.subtitleNode, let subtitleSize = self.subtitleSize {
            subtitleNode.frame = CGRect(origin: CGPoint(x: floor((self.bounds.size.width - totalWidth) / 2.0), y: floor((self.bounds.size.height - totalTextHeight) / 2.0) + totalTextHeight - subtitleSize.height), size: subtitleSize)
        }
        if let iconView = self.iconView, let image = iconView.image {
            let iconSize = image.size
            iconView.frame = CGRect(origin: CGPoint(x: floor((self.bounds.size.width - totalWidth) / 2.0) + totalWidth - iconSize.width, y: floorToScreenPixels((self.bounds.size.height - iconSize.height) / 2.0)), size: iconSize)
        }
    }
}