(iOS) Swift button tap with handler

MySwitch.swift

class MySwitch: UISwitch {
    
    typealias DidTapSwitch = (WMSwitch) -> ()
    
    var didChangeState: DidTapSwitch? {
        didSet {
            if didChangeState != nil {
                addTarget(self, action: #selector(self.switchStateDidChange(_:)), for: .valueChanged)
            } else {
                removeTarget(self, action: #selector(self.switchStateDidChange(_:)), for: .valueChanged)
            }
        }
    }

    @objc func switchStateDidChange(_ sender: MySwitch) {
        if let handler = didChangeState {
            handler(self)
        }
    }
}

Use it:

@IBOutlet var smartSwitch: MySwitch!

smartSwitch. didChangeState = { (sender) in
    ......
}

Remove action

smartSwitch.didChangeState = nil