Swiftgram/submodules/TelegramUI/TelegramUI/GalleryNavigationCheckNode.swift

import Foundation
import UIKit
import AsyncDisplayKit
import Display
import TelegramPresentationData

final class GalleryNavigationCheckNode: ASDisplayNode, NavigationButtonCustomDisplayNode {
    private var checkNode: CheckNode
    
    init(theme: PresentationTheme) {
        self.checkNode = CheckNode(strokeColor: theme.list.itemCheckColors.strokeColor, fillColor: theme.list.itemCheckColors.fillColor, foregroundColor: theme.list.itemCheckColors.foregroundColor, style: .navigation)
    
        super.init()
        
        self.addSubnode(self.checkNode)
    }
    
    var isHighlightable: Bool {
        return false
    }
    
    var isChecked: Bool {
        return self.checkNode.isChecked
    }
    
    func setIsChecked(_ isChecked: Bool, animated: Bool) {
        self.checkNode.setIsChecked(isChecked, animated: animated)
    }
    
    func addTarget(target: AnyObject?, action: Selector) {
        self.checkNode.addTarget(target: target, action: action)
    }
    
    override func calculateSizeThatFits(_ constrainedSize: CGSize) -> CGSize {
        return CGSize(width: 39.0, height: 39.0)
    }
    
    override func layout() {
        super.layout()
        
        let size = self.bounds.size
        let checkSize = CGSize(width: 39.0, height: 39.0)
        self.checkNode.frame = CGRect(origin: CGPoint(x: floor((size.width - checkSize.width) / 2.0) + 11.0, y: floor((size.height - checkSize.height) / 2.0) + 3.0), size: checkSize)
    }
}