在 Swift 中使用 Bool 扩展视图控制器

Question

我正在尝试为视图控制器进行扩展。这是一个带有点击识别器的简单视图。如果按下一次，背景颜色应该从洋红色变为橙色。到下一次推送时，它应该又是洋红色了。

如果我为具体的 VC 做这件事，比如说 StartVC，它就可以工作。代码如下：

import UIKit

class StartVC: UIViewController {
    // bool
    var colorChanged = false

    override func viewDidLoad() {
        super.viewDidLoad()

        view.backgroundColor = UIColor.magenta
        setMyTouchableView2()
    }
}

extension StartVC {
    // make touchable view
    func setMyTouchableView2() {
        let myView = UIImageView(frame: CGRect(x: 100, y: 100, width: 50, height: 50))
        myView.backgroundColor = UIColor.yellow

        // add gesture recognizer
        let tapGestureRecognizer = UITapGestureRecognizer(target: self, action: #selector(changeColor2(_:)))
        myView.addGestureRecognizer(tapGestureRecognizer)
        myView.isUserInteractionEnabled = true

        view.addSubview(myView)
    }

    func changeColor2(_ recognizer:UITapGestureRecognizer){
        if(colorChanged) {
            view.backgroundColor = UIColor.magenta
        } else {
            view.backgroundColor = UIColor.orange
        }

        colorChanged = !colorChanged
    }
}

但是，如果我尝试将这个扩展推广到任何视图控制器，我会遇到以下问题：我不能将我的布尔值 (colorChanged) 放在视图控制器中，因为扩展不会看到它。我该如何解决？

代码如下：

import UIKit

class ViewController: UIViewController {
    // bool
    var colorChanged = false

    override func viewDidLoad() {
        super.viewDidLoad()

        view.backgroundColor = UIColor.magenta
        setMyTouchableView()
    }
}

extension UIViewController {
    // make touchable view
    func setMyTouchableView() {
        let myView = UIImageView(frame: CGRect(x: 100, y: 100, width: 50, height: 50))
        myView.backgroundColor = UIColor.yellow

        // add gesture recognizer
        let tapGestureRecognizer = UITapGestureRecognizer(target: self, action: #selector(changeColor(_:)))
        myView.addGestureRecognizer(tapGestureRecognizer)
        myView.isUserInteractionEnabled = true

        view.addSubview(myView)
    }

    func changeColor(_ recognizer:UITapGestureRecognizer){
        if(colorChanged) {
            view.backgroundColor = UIColor.magenta
        } else {
            view.backgroundColor = UIColor.orange
        }
    }

Answer 1

默认情况下，扩展不能具有存储属性。但是，如果你真的需要这个，你可以通过 objc 运行时创建它

private var associationKey: UInt8 = 111 // Some constant

extension ViewController {
    var colorChanged: Bool {
        get {
            return objc_getAssociatedObject(self, &associationKey) as! Bool
        }
        set {
            objc_setAssociatedObject(self, &associationKey, newValue, .OBJC_ASSOCIATION_ASSIGN)
        }
    }
}

但是您可能会遇到一个问题，即多个控制器会更改此标志。所以你也可以尝试使用协议

protocol ColorChangable {
    var colorChange: Bool { get set }
}

extension ColorChangable where Self: UIViewController {
    func setMyTouchableView {
        // Your logic
    }
}

class ViewController: UIViewController {
    var colorChange: Bool

    func viewDidLoad() {
        self.viewDidLoad()

        setMyTouchableView()
    }
}

extension ViewController: ColorChangable {}

Answer 2

你可以使用struct来做一个全局变量-

   struct GlobalStruct {
    static var colorChanged = false
}

class ViewController: UIViewController {

    override func viewDidLoad() {
        super.viewDidLoad()

        view.backgroundColor = UIColor.magenta
        setMyTouchableView()
    }
}

extension UIViewController {
    // make touchable view
    func setMyTouchableView() {
        let myView = UIImageView(frame: CGRect(x: 100, y: 100, width: 50, height: 50))
        myView.backgroundColor = UIColor.yellow

        // add gesture recognizer
        let tapGestureRecognizer = UITapGestureRecognizer(target: self, action: #selector(changeColor(_:)))
        myView.addGestureRecognizer(tapGestureRecognizer)
        myView.isUserInteractionEnabled = true

        view.addSubview(myView)
    }

    func changeColor(_ recognizer:UITapGestureRecognizer){
        if(GlobalStruct.colorChanged) {
            view.backgroundColor = UIColor.magenta
        } else {
            view.backgroundColor = UIColor.orange
        }
    }
}