利用isa-swizzling hook UITableViewCell的点击事件

最近在作无痕埋点相关的事情，须要对用户的操做进行插桩进行上报，其余事件都还好说，cell点击事件遇到了点问题，最初的想法是hook UITableViewCell的setSelected(_ selected: Bool, animated: Bool)方法。

可是此方法有2个问题：

1. 不太好获取cell所在的位置
2. 即便UITableView的代理方法没实现didSelectRowAtIndexPath方法，也会上报埋点

后来再与同事的讨论中迸发出来一个想法，可否利用KVO中用到的isa-swizzling进行hookUITableViewCell的点击，这个场景和KVO的场景其实差很少，KVO是对某个值观察，当值改变的时候，调用某个固定的方法，而我如今的需求是对UITableViewCell的点击进行观察，当点击的时候，调用咱们上报埋点的方法

简单介绍下KVO的原理：

1. 当某个类的属性被观察时，系统会在运行时动态的建立一个该类的子类。而且把改对象的isa指向这个子类

2. 假设被观察的属性名是name，若父类里有setName:或这_setName:,那么在子类里重写这2个方法，若2个方法同时存在，则只会重写setName:一个（这里和KVCset时的搜索顺序是同样的）

3. 若被观察的类型是NSString,那么重写的方法的实现会指向_NSSetObjectValueAndNotify这个函数，如果Bool类型，那么重写的方法的实现会指向_NSSetBoolValueAndNotify这个函数，这个函数里会调用willChangeValueForKey:和didChangevlueForKey:,而且会在这2个方法调用之间，调用父类set方法的实现

4. 系统会在willChangeValueForKey:对observe里的change[old]赋值，取值是用valueForKey:取值的,didChangevlueForKey:对observe里的change[new]赋值，而后调用observe的这个方法- (void)observeValueForKeyPath:(nullable NSString *)keyPath ofObject:(nullable id)object change:(nullable NSDictionary<NSKeyValueChangeKey, id> *)change context:(nullable void *)context;

5. 当使用KVC赋值的时候,在NSObject里的setValue:forKey:方法里,若父类不存在setName:或这_setName:这些方法,会调用_NSSetValueAndNotifyForKeyInIvar这个函数，这个函数里一样也会调用willChangeValueForKey:和didChangevlueForKey:,若存在则调用

hook Cell的点击事件步骤以下：

注：生成子类类型的名字的规则为当前的类名+"_sub_czb_tableview_delegate_analysis"

1. hook UITableView的setDelegate方法
2. 在setDelegate方法中判断要设置delegate是否为nil或者delegate是否没实现了tableView:didSelectRowAtIndexPath:方法
3. 如果则设置UITableView的delegate并结束，不然进行下一步
4. 判断当前类的类名是否知足生成子类类型的规则，如果则设置UITableView的delegate并结束，不然进行下一步
5. 判断须要生成的子类类型是否已经注册过，若没注册过跳到第7不，不然进行下一步
6. 若注册过，将delegate的isa指向已经注册过的子类类型，而后设置UITableView的delegate，结束
7. 建立一个delegate类型的子类，并注册
8. 为此子类添加一个与tableView点击事件代理同名的方法，并在此方法中调用父类此方法的实现
9. 将delegate的isa指向刚刚建立的子类类型

代码以下：

typealias TableviewDidSelectRow = @convention(c) (NSObject, Selector, UITableView, IndexPath) -> Void

let czb_didSelectRow:@convention(block) (NSObject, UITableView, IndexPath) -> Void = {
    (this, tableView, indexPath) in
    let superClass: AnyClass? = this.superclass
    let sel = NSSelectorFromString("tableView:didSelectRowAtIndexPath:")
    let method = class_getInstanceMethod(superClass, sel)
    if let impl = class_getMethodImplementation(superClass, sel) {
        let fn = unsafeBitCast(impl, to: TableviewDidSelectRow.self)
        fn(this, sel, tableView, indexPath)
    }
}

extension UITableView {
    static func enableAutoAnalysis () {
       let originalSelector = NSSelectorFromString("setDelegate:")
        let swizzledSelector = #selector(czb_setDelegate(_:))
        /// 此方法是对对应的方法进行hook
        swizzlingForClass(UITableView.classForCoder(),originalSelector: originalSelector,swizzledSelector: swizzledSelector)  
    }
    
    @objc func czb_setDelegate(_ delegate: NSObject?) {
        let sel = NSSelectorFromString("tableView:didSelectRowAtIndexPath:")
        guard let delegate = delegate,delegate.responds(to: sel) else {
            czb_setDelegate(nil)
            return
        }
        var className = NSStringFromClass(delegate.classForCoder)
        if className.hasSuffix("_sub_czb_tableview_delegate_analysis") {
            czb_setDelegate(delegate)
            return
        }
        className += "_sub_czb_tableview_delegate_analysis"
        if let analysisClass = NSClassFromString(className) {
            object_setClass(delegate, analysisClass)
            czb_setDelegate(delegate)
            return
        }
        
        if let customClass = objc_allocateClassPair(delegate.classForCoder, className, 0),
            let method = class_getInstanceMethod(delegate.classForCoder, sel) {
            objc_registerClassPair(customClass)
            let type = method_getTypeEncoding(method)
            let imp = imp_implementationWithBlock(unsafeBitCast(czb_didSelectRow, to: AnyObject.self))
            class_addMethod(customClass, sel, imp, type)
            object_setClass(delegate, customClass)
            czb_setDelegate(delegate)
        }else {
            czb_setDelegate(delegate)
        }
    }
}

复制代码

其余收获：

1. @convention的使用

   • @convention(swift) : 代表这个是一个swift的闭包
   • @convention(block) ：代表这个是一个兼容oc的block的闭包
   • @convention(c) : 代表这个是兼容c的函数指针的闭包

2. 在Swift中如何把IMP转成func以及如何经过一个block建立一个IMP

   • 如何把IMP转成func

     经过typealias和@convention(c)声明一个和IMP相同参数的闭包，例：

     typealias TableviewDidSelectRow = @convention(c) (NSObject, Selector, UITableView, IndexPath) -> Void

     利用unsafeBitCast函数转换，例：let fn = unsafeBitCast(impl, to: TableviewDidSelectRow.self)

   • 如何经过一个block建立一个IMP

     创一个用建@convention(block)修饰的闭包，例：

     let czb_didSelectRow:@convention(block) (NSObject, UITableView, IndexPath) -> Void = {
         (this, tableView, indexPath) in
         ///实现代码
     }
     复制代码

     利用imp_implementationWithBlock和unsafeBitCast，例：

     let block = unsafeBitCast(czb_didSelectRow, to: AnyObject.self)
     let imp = imp_implementationWithBlock(block)
     复制代码