PKHUD custom loader with cancel button and action on cancel button click

LoaderWithCancelView+PKHUBCustomView.swift

class LoaderWithCancelView: UIView,PKHUDAnimating{

    let progressView: PKHUDProgressView = {
        let pkHUDPV = PKHUDProgressView()
        pkHUDPV.translatesAutoresizingMaskIntoConstraints = false
        return pkHUDPV
    }()
    let buttonCancel:UIButton = {
        let button = UIButton()
        button.setTitle("Cancel", for: .normal)
        button.setTitleColor(.white, for: .normal)
        button.backgroundColor = UIColor.black.withAlphaComponent(0.3)
        button.translatesAutoresizingMaskIntoConstraints = false
//        button.heightAnchor.constraint(equalToConstant: 30).isActive = true
        return button
    }()


    let pKHUDFrame = CGRect(origin: CGPoint.zero, size: CGSize(width: 156.0, height: 156.0))

    var onCancelButtonTapp : (() -> Swift.Void)?


    override init(frame: CGRect) {
        super.init(frame: pKHUDFrame)
        setupSubviews()
    }

    public required init?(coder aDecoder: NSCoder) {
        super.init(coder: aDecoder)
        setupSubviews()
    }

    func setupSubviews(){

        buttonCancel.addTarget(self, action: #selector(buttonCancelTapped), for: .touchUpInside)

        addSubview(progressView)
        addSubview(buttonCancel)

        let contentView = self
        progressView.leftAnchor.constraint(equalTo: contentView.leftAnchor, constant: 0).isActive = true
        progressView.rightAnchor.constraint(equalTo: contentView.rightAnchor, constant: 0).isActive = true
        progressView.topAnchor.constraint(equalTo: contentView.topAnchor, constant: 0).isActive = true

        progressView.bottomAnchor.constraint(equalTo: buttonCancel.topAnchor, constant: 0).isActive = true

        buttonCancel.leftAnchor.constraint(equalTo: contentView.leftAnchor, constant: 0).isActive = true
        buttonCancel.rightAnchor.constraint(equalTo: contentView.rightAnchor, constant: 0).isActive = true
        buttonCancel.bottomAnchor.constraint(equalTo: contentView.bottomAnchor, constant: 0).isActive = true


    }


    public func startAnimation() {
        progressView.startAnimation()
    }

    public func stopAnimation() {
        progressView.stopAnimation()
    }


    //action
    @objc func buttonCancelTapped() {
        onCancelButtonTapp?()
    }

}

LoaderWithCancelViewIntegration.swift

        let customHudWithButton = LoaderWithCancelView()
        customHudWithButton.onCancelButtonTapp = {
            // on click action
            print("Cancel tapped")
        }
        PKHUD.sharedHUD.contentView = customHudWithButton
        PKHUD.sharedHUD.show()
        PKHUD.sharedHUD.hide(afterDelay: 2.0) { success in
            // Completion Handler
        }