WKWebView 注入JS代碼

1. 初始化時進行注入

    WKWebViewConfiguration *config = [[WKWebViewConfiguration alloc] init];
	  //實際上注入js文件和注入js代碼字符串沒有區別,因爲也要把JS文件讀取爲字符串進行調用
    NSString *js = [[NSBundle mainBundle] pathForResource:@"demo/canvas.js" ofType:nil];
    NSData *jsData=[NSData dataWithContentsOfFile:js];
    NSString *jsString =  [[NSString alloc] initWithData:jsData encoding:NSUTF8StringEncoding];
    WKUserScript *jsScript = [[WKUserScript alloc]initWithSource:jsString injectionTime:WKUserScriptInjectionTimeAtDocumentStart forMainFrameOnly:YES];
 [config.userContentController addUserScript:jsScript];
 _webView = [[WKWebView alloc] initWithFrame:CGRectZero configuration:config];

2. 頁面加載完畢進行注入(當然也可以使用初始化時注入,injectionTime改爲WKUserScriptInjectionTimeAtDocumentEnd),一般是先在start時注入,end時進行調用

//便利宏: 避免書寫大量的反斜槓轉義或者換行
#define JS(...)  [[NSString alloc] initWithCString:#__VA_ARGS__ encoding:NSUTF8StringEncoding]

- (void)webView:(WKWebView *)webView didFinishNavigation:(null_unspecified WKNavigation *)navigation {
    self.title = webView.title;
    //注入js
    [webView evaluateJavaScript:JS(
       alert('你好,JS');
      console.log('hello JS');
    ) completionHandler:nil];
}