ios - Swift:UIButton 自定义

Question

作为最佳实践，我想将代码从我的 viewcontroller 类中移出以自定义 UIButtons。我下面的代码是向 UIButtons 添加一个白色边框，我想在整个项目的按钮上轻松调用它。

    //White Border
    let passwordBorder = CALayer()

    let width = CGFloat(5.0)

    passwordBorder.borderColor = UIColor.whiteColor().CGColor

    passwordBorder.frame = CGRect(x: 0, y: 0, width:  passwordField.frame.size.width, height: passwordField.frame.size.height)

    passwordBorder.borderWidth = width

    passwordField.layer.addSublayer(passwordBorder)

    passwordField.layer.masksToBounds = true

如何将此代码放入辅助函数中以便我可以轻松地调用它？

我是编码新手，在任何UI 上使用辅助函数时遇到问题。谢谢!

Answer 1

看看 Swift 的扩展。你可以很容易地做类似的事情

extension UIButton {
      func setPasswordBorderColor(borderColor: UIColor) {
        //White Border
        let passwordBorder = CALayer()

        let width = CGFloat(5.0)

        passwordBorder.borderColor = borderColor.CGColor

        passwordBorder.frame = CGRect(x: 0, y: 0, width:  self.frame.size.width, height: self.frame.size.height)

        passwordBorder.borderWidth = width

        self.layer.addSublayer(passwordBorder)

        self.layer.masksToBounds = true
  }
}