Telegram-iOS/submodules/ListSectionHeaderNode/Sources/ListSectionHeaderNode.swift at master · sclcoder/Telegram-iOS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
import Foundation
import UIKit
import AsyncDisplayKit
import Display
import TelegramPresentationData

private let titleFont = Font.bold(13.0)
private let actionFont = Font.medium(13.0)

public final class ListSectionHeaderNode: ASDisplayNode {
    private let label: ImmediateTextNode
    private var actionButtonLabel: ImmediateTextNode?
    private var actionButton: HighlightableButtonNode?
    private var theme: PresentationTheme

    private var validLayout: (size: CGSize, leftInset: CGFloat, rightInset: CGFloat)?

    public var title: String? {
        didSet {
            self.label.attributedText = NSAttributedString(string: self.title ?? "", font: titleFont, textColor: self.theme.chatList.sectionHeaderTextColor)

            if let (size, leftInset, rightInset) = self.validLayout {
                self.updateLayout(size: size, leftInset: leftInset, rightInset: rightInset)
            }
        }
    }

    public var action: String? {
        didSet {
            if (self.action != nil) != (self.actionButton != nil) {
                if let _ = self.action {
                    let actionButtonLabel = ImmediateTextNode()
                    self.addSubnode(actionButtonLabel)
                    self.actionButtonLabel = actionButtonLabel
                    let actionButton = HighlightableButtonNode()
                    self.addSubnode(actionButton)
                    self.actionButton = actionButton
                    actionButton.addTarget(self, action: #selector(self.actionButtonPressed), forControlEvents: .touchUpInside)
                } else {
                    if let actionButtonLabel = self.actionButtonLabel {
                        self.actionButtonLabel = nil
                        actionButtonLabel.removeFromSupernode()
                    }
                    if let actionButton = self.actionButton {
                        self.actionButton = nil
                        actionButton.removeFromSupernode()
                    }
                }
            }
            if let action = self.action {
                self.actionButtonLabel?.attributedText = NSAttributedString(string: action, font: actionFont, textColor: self.theme.chatList.sectionHeaderTextColor)
            }

            if let (size, leftInset, rightInset) = self.validLayout {
                self.updateLayout(size: size, leftInset: leftInset, rightInset: rightInset)
            }
        }
    }

    public var activateAction: (() -> Void)?

    public init(theme: PresentationTheme) {
        self.theme = theme

        self.label = ImmediateTextNode()
        self.label.isUserInteractionEnabled = false

        super.init()

        self.addSubnode(self.label)

        self.backgroundColor = theme.chatList.sectionHeaderFillColor
    }

    public func updateTheme(theme: PresentationTheme) {
        if self.theme !== theme {
            self.theme = theme

            self.label.attributedText = NSAttributedString(string: self.title ?? "", font: titleFont, textColor: self.theme.chatList.sectionHeaderTextColor)

            self.backgroundColor = theme.chatList.sectionHeaderFillColor
            if let action = self.action {
                self.actionButtonLabel?.attributedText = NSAttributedString(string: action, font: actionFont, textColor: self.theme.chatList.sectionHeaderTextColor)
            }

            if let (size, leftInset, rightInset) = self.validLayout {
                self.updateLayout(size: size, leftInset: leftInset, rightInset: rightInset)
            }
        }
    }

    public func updateLayout(size: CGSize, leftInset: CGFloat, rightInset: CGFloat) {
        self.validLayout = (size, leftInset, rightInset)
        let labelSize = self.label.updateLayout(CGSize(width: max(0.0, size.width - leftInset - rightInset - 18.0), height: size.height))
        self.label.frame = CGRect(origin: CGPoint(x: leftInset + 16.0, y: 6.0 + UIScreenPixel), size: labelSize)

        if let actionButton = self.actionButton, let actionButtonLabel = self.actionButtonLabel {
            let buttonSize = actionButtonLabel.updateLayout(CGSize(width: size.width, height: size.height))
            actionButtonLabel.frame = CGRect(origin: CGPoint(x: size.width - rightInset - 16.0 - buttonSize.width, y: 6.0 + UIScreenPixel), size: buttonSize)
            actionButton.frame = CGRect(origin: CGPoint(x: size.width - rightInset - 16.0 - buttonSize.width, y: 6.0 + UIScreenPixel), size: buttonSize)
        }
    }

    @objc private func actionButtonPressed() {
        self.activateAction?()
    }
}