如何hook delegate

有时，我们想要hook某个delegate的某个方法，做一些神奇的事情。

然而，这个delegate对象是哪个类的实例呢？编码阶段并不总是容易确定。

所以，该如何hook 它的方法呢？

答案就是 setDelegate

以UITableView为例，下面来hook一下其delegate的tableView:didSelectRowAtIndexPath:

@implementation UITableView (swizz)

- (void)swizz_setDelegate:(id)delegate;
{
    
    if ([delegate respondsToSelector:@selector(tableView:didSelectRowAtIndexPath:)])
    {
        [Swizzler swizzInstanceMethodOfClass:[delegate class] originMethod:@selector(tableView:didSelectRowAtIndexPath:) newMethod:@selector(swizz_tableView:didSelectRowAtIndexPath:)];
    }
    
    [self swizz_setDelegate:delegate];
}

@end

没错，拿到delegate的class，并施展swizzle大法。无需关心具体是哪个类。

剩下的唯一问题是：怎么给这个类添加上swizz_tableView:didSelectRowAtIndexPath:方法呢？

我想到的一个简单的解决办法是把这个方法直接添加到NSObject里。我们可以使用一个分类来做这件事情。

@implementation NSObject (swizz)

- (void)swizz_tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath
{
    // do something
    
    [self swizz_tableView:tableView didSelectRowAtIndexPath:indexPath];
}

@end

【完】