Refactoring

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

@@ -0,0 +1,55 @@
+import Foundation
+import UIKit
+import AsyncDisplayKit
+import TelegramPresentationData
+import CheckNode
+import TelegramCore
+
+public final class ChatMessageSelectionNode: ASDisplayNode {
+    private let toggle: (Bool) -> Void
+    
+    public private(set) var selected = false
+    private let checkNode: CheckNode
+    
+    public init(wallpaper: TelegramWallpaper, theme: PresentationTheme, toggle: @escaping (Bool) -> Void) {
+        self.toggle = toggle
+        
+        let style: CheckNodeTheme.Style
+        if wallpaper == theme.chat.defaultWallpaper, case .color = wallpaper {
+            style = .plain
+        } else {
+            style = .overlay
+        }
+        
+        self.checkNode = CheckNode(theme: CheckNodeTheme(theme: theme, style: style, hasInset: true))
+        self.checkNode.isUserInteractionEnabled = false
+        
+        super.init()
+        
+        self.addSubnode(self.checkNode)
+    }
+    
+    override public func didLoad() {
+        super.didLoad()
+        
+        self.view.addGestureRecognizer(UITapGestureRecognizer(target: self, action: #selector(self.tapGesture(_:))))
+    }
+    
+    public func updateSelected(_ selected: Bool, animated: Bool) {
+        if self.selected != selected {
+            self.selected = selected
+            self.checkNode.setSelected(selected, animated: animated)
+        }
+    }
+    
+    @objc private func tapGesture(_ recognizer: UITapGestureRecognizer) {
+        if case .ended = recognizer.state {
+            self.toggle(!self.selected)
+        }
+    }
+    
+    public func updateLayout(size: CGSize, leftInset: CGFloat) {
+        let checkSize = CGSize(width: 28.0, height: 28.0)
+        self.checkNode.frame = CGRect(origin: CGPoint(x: 6.0 + leftInset, y: floor((size.height - checkSize.height) / 2.0)), size: checkSize)
+    }
+}