eonist / callbackbutton Goto Github PK

class CallbackButton: UIButton {
    var onAction: ((CallbackButton) -> Void)?

    init(onAction: ((CallbackButton) -> Void)? = nil) {
        self.onAction = onAction
        super.init(frame: .zero)
        setup()
    }

    required init?(coder: NSCoder) {
        super.init(coder: coder)
        setup()
    }

    private func setup() {
        addTarget(self, action: #selector(onTap), for: .primaryActionTriggered)
    }

    @objc private func onTap() {
        onAction?(self)
    }
}

let button = CallbackButton { btn in
    print("Tapped!")
}