Add shouldSafeMethodCall option

DavinAhn · DavinAhn · commit c19ca4bdfba5 · 2017-01-26T16:49:43.000+09:00
diff --git a/WKJavaScriptController/WKJavaScriptController.swift b/WKJavaScriptController/WKJavaScriptController.swift
@@ -59,6 +59,10 @@ open class JSFloat: JSValueType {
 }
 
 open class WKJavaScriptController: NSObject {
+    // If true, do not allow NSNull(If pass undefined in JavaScript) for method arguments.
+    // That is, if get NSNull as arguments, do not call method.
+    open var shouldSafeMethodCall = true
+    
     fileprivate let name: String
     fileprivate weak var target: AnyObject?
     fileprivate let bridgeProtocol: Protocol
@@ -256,6 +260,14 @@ extension WKJavaScriptController: WKScriptMessageHandler {
             return
         }
         
+        if shouldSafeMethodCall {
+            for arg in args {
+                if arg is NSNull {
+                    return
+                }
+            }
+        }
+        
         let method = class_getInstanceMethod(target.classForCoder, bridge.nativeSelector)
         if method == nil {
             log("An unimplemented method has been called. (selector: \(bridge.nativeSelector))")
