iOS Method Swizzling 的一个实际应用

本文 Method Swizzling 部分参考了 《iOS黑魔法－Method Swizzling》

一、问题

最近在维护公司一个久远的项目时，发现当时使用了 UIWebView 展示 HTML 页面，为了解决 JavaScript 中 alert 和 confirm 样式不能自定义的问题，所以通过实现以下方法，

@interface UIWebView (JSConfirmAlert)

- (void)webView:(UIWebView *)sender runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(id)frame;
- (BOOL)webView:(UIWebView *)sender runJavaScriptConfirmPanelWithMessage:(NSString *)message initiatedByFrame:(id)frame;

@end

来拦截 alert 和 confirm ，并通过 UIAlertView 重新实现。

但是原来的代码在实现上有些问题，这部分功能又是做成了 framework 集成到工程里面的，而且由于项目几经易手，framework 部分的源码已经没有了。由于原来是通过给 UIWebView 写 Category 来实现的功能，所以即便不引用 .h 文件，只要集成了 framework，这部分代码就会生效。（这一点是刚发现的，原来一直以为要引用了 .h 才会生效。）

二、解决方案

无奈之下，就考虑用 Method Swizzling 来改写之前的方法了。

// UIWebView+SwizzlingAlertAndConfirm.h

#import 

@interface UIWebView (SwizzlingAlertAndConfirm)

@end

// UIWebView+SwizzlingAlertAndConfirm.m

#import 
#import 
#import "UIWebView+SwizzlingAlertAndConfirm.h"

@implementation UIWebView (SwizzlingAlertAndConfirm)

+ (void)exchangeMethod:(SEL)fromSelector toMethod:(SEL)toSelector {

    // 通过class_getInstanceMethod()函数从当前对象中的method list获取method结构体，如果是类方法就使用class_getClassMethod()函数获取。
    Method fromMethod = class_getInstanceMethod([self class], fromSelector);
    Method toMethod = class_getInstanceMethod([self class], toSelector);

    /**
     *  我们在这里使用class_addMethod()函数对Method Swizzling做了一层验证，如果self没有实现被交换的方法，会导致失败。
     *  而且self没有交换的方法实现，但是父类有这个方法，这样就会调用父类的方法，结果就不是我们想要的结果了。
     *  所以我们在这里通过class_addMethod()的验证，如果self实现了这个方法，class_addMethod()函数将会返回NO，我们就可以对其进行交换了。
     */
    if (!class_addMethod([self class], fromSelector, method_getImplementation(toMethod), method_getTypeEncoding(toMethod))) {
        method_exchangeImplementations(fromMethod, toMethod);
    }

}


+ (void)load {
    [super load];

    [self exchangeMethod:@selector(webView:runJavaScriptAlertPanelWithMessage:initiatedByFrame:)
                toMethod:@selector(swizzlingWebView:runJavaScriptAlertPanelWithMessage:initiatedByFrame:)];

    [self exchangeMethod:@selector(webView:runJavaScriptConfirmPanelWithMessage:initiatedByFrame:)
                toMethod:@selector(swizzlingWebView:runJavaScriptConfirmPanelWithMessage:initiatedByFrame:)];

    [self exchangeMethod:@selector(alertView:clickedButtonAtIndex:)
                toMethod:@selector(swizzlingAlertView:clickedButtonAtIndex:)];

}



- (void)swizzlingWebView:(UIWebView *)sender runJavaScriptAlertPanelWithMessage:(NSString *)message initiatedByFrame:(id)frame {
    UIAlertView* customAlert = [[UIAlertView alloc] initWithTitle:@"助手提示" message:message delegate:nil cancelButtonTitle:@"确定bbb" otherButtonTitles:nil];
    [customAlert show];
}

static BOOL diagStat = NO;
static NSInteger bIdx = -1;

- (BOOL)swizzlingWebView:(UIWebView *)sender runJavaScriptConfirmPanelWithMessage:(NSString *)message initiatedByFrame:(id)frame {
    UIAlertView *confirmDiag = [[UIAlertView alloc] initWithTitle:@"助手提示"
                                                          message:message
                                                         delegate:self
                                                cancelButtonTitle:@"取消13"
                                                otherButtonTitles:@"确定13", nil];

    [confirmDiag show];
    bIdx = -1;

    while (bIdx==-1) {
        //[NSThread sleepForTimeInterval:0.2];
        [[NSRunLoop mainRunLoop] runUntilDate:[NSDate dateWithTimeIntervalSinceNow:0.1f]];
    }
    if (bIdx == 0){//取消;
        diagStat = NO;
    }
    else if (bIdx == 1) {//确定;
        diagStat = YES;
    }
    return diagStat;
}

- (void)swizzlingAlertView:(UIAlertView *)alertView clickedButtonAtIndex:(NSInteger)buttonIndex{
    bIdx = buttonIndex;
}


@end

工程引用这两个文件后，问题解决。

三、讨论

1、Method Swizzling 的封装

这里对 Method Swizzling 做了个简单的封装，不过只是为了写着方便随便整了下。真正在项目中我们肯定会在很多地方用到 Method Swizzling，而且在使用这个特性时有很多需要注意的地方。我们可以将 Method Swizzling 封装起来，也可以使用一些比较成熟的第三方。在这里我推荐Github上星最多的一个第三方——JRSwizzle

2、改进

都折腾完之后才发现，就这个项目本身的问题而言，其实用不到 Method Swizzling，只要写个 UIWebView 的子类，在子类中重新实现这几个方法，然后直接使用子类就好了。当时没想起来。不过使用子类化的方法，还是需要修改原来代码中对 UIWebView 引用的那部分代码。而使用 Method Swizzling 还是更酷一些，原来的代码不用做任何修改，只要在工程中引入这两个文件就可以了。