Add support for WKWebView.

carlosaiello · Sep 7, 2016 · 7c71ef9 · 7c71ef9
1 parent 3717b95
commit 7c71ef9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/ios/screenshot.m b/src/ios/screenshot.m
@@ -13,7 +13,11 @@ - (void)pluginInitialize
                            queue:mainQueue
                       usingBlock:^(NSNotification *note) {
                         if ([self.webView respondsToSelector:@selector(stringByEvaluatingJavaScriptFromString:)]) {
+                            // UIWebView
                             [self.webView performSelectorOnMainThread:@selector(stringByEvaluatingJavaScriptFromString:) withObject:@"cordova.fireDocumentEvent('screenshot');" waitUntilDone:NO];
+                        } else if ([self.webView respondsToSelector:@selector(evaluateJavaScript:completionHandler:)]) {
+                            // WKWebView
+                            [self.webView performSelector:@selector(evaluateJavaScript:completionHandler:) withObject:@"cordova.fireDocumentEvent('screenshot');" withObject:nil];
                         } else {
                             // cordova lib version is > 4
                             [self.webView performSelectorOnMainThread:@selector(evaluateJavaScript:completionHandler:) withObject:@"cordova.fireDocumentEvent('screenshot');"  waitUntilDone:NO];
@@ -22,4 +26,4 @@ - (void)pluginInitialize
         }
     }
 
-    @end
+    @end