fix framework add injectJavascript

Author: Lision

WKWebViewJavascriptBridge/WKWebViewJavascriptBridge.swift

@@ -11,12 +11,17 @@ import WebKit
 
 @available(iOS 9.0, *)
 public class WKWebViewJavascriptBridge: NSObject {
+    private let injectJavascript = "iOS_Native_InjectJavascript"
+    
     private weak var webView: WKWebView!
     private var base: WKWebViewJavascriptBridgeBase!
 
     public init(webView: WKWebView) {
-        self.webView = webView
         super.init()
+        self.webView = webView
+        self.webView.configuration.userContentController.add(self, name: injectJavascript)
+        base = WKWebViewJavascriptBridgeBase()
+        base.delegate = self
     }
 
     public func reset() {
@@ -47,8 +52,18 @@ public class WKWebViewJavascriptBridge: NSObject {
     }
 }
 
+extension WKWebViewJavascriptBridge: WKWebViewJavascriptBridgeBaseDelegate {
+    func evaluateJavascript(javascript: String) {
+        self.webView.evaluateJavaScript(javascript, completionHandler: nil)
+    }
+}
+
 extension WKWebViewJavascriptBridge: WKScriptMessageHandler {
     public func userContentController(_ userContentController: WKUserContentController, didReceive message: WKScriptMessage) {
-        self.flushMessageQueue()
+        if message.name == injectJavascript {
+            self.base.injectJavascriptFile()
+        } else {
+            self.flushMessageQueue()
+        }
     }
 }

WKWebViewJavascriptBridge/WKWebViewJavascriptBridgeBase.swift

@@ -66,9 +66,7 @@ public class WKWebViewJavascriptBridgeBase: NSObject {
                 var callback: Callback?
                 if let callbackID = message["callbackID"] {
                     callback = { (_ responseData: Any?) -> Void in
-                        guard responseData != nil else {
-                            return
-                        }
+                        guard responseData != nil else { return }
 
                         let msg = ["responseID": callbackID, "responseData": responseData!] as Message
                         self.queue(message: msg)
@@ -79,9 +77,7 @@ public class WKWebViewJavascriptBridgeBase: NSObject {
                     }
                 }
 
-                guard let handlerName = message["handlerName"] as? String else {
-                    return
-                }
+                guard let handlerName = message["handlerName"] as? String else { return }
                 guard let handler = self.messageHandlers[handlerName] else {
                     log("NoHandlerException, No handler for message from JS: \(message)")
                     return
@@ -91,6 +87,11 @@ public class WKWebViewJavascriptBridgeBase: NSObject {
         }
     }
 
+    func injectJavascriptFile() {
+        let js = WKWebViewJavascriptBridgeJS
+        self.delegate?.evaluateJavascript(javascript: js)
+    }
+    
     // MARK: - Private
     private func queue(message: Message) {
         if self.startupMessageQueue.isEmpty {
@@ -101,9 +102,7 @@ public class WKWebViewJavascriptBridgeBase: NSObject {
     }
 
     private func dispatch(message: Message) {
-        guard var messageJSON = self.serialize(message: message, pretty: false) else {
-            return
-        }
+        guard var messageJSON = self.serialize(message: message, pretty: false) else { return }
 
         messageJSON = messageJSON.replacingOccurrences(of: "\\", with: "\\\\")
         messageJSON = messageJSON.replacingOccurrences(of: "\"", with: "\\\"")
@@ -138,9 +137,7 @@ public class WKWebViewJavascriptBridgeBase: NSObject {
 
     private func deserialize(messageJSON: String) -> [Message]? {
         var result = [Message]()
-        guard let data = messageJSON.data(using: .utf8) else {
-            return nil
-        }
+        guard let data = messageJSON.data(using: .utf8) else { return nil }
         do {
             result = try JSONSerialization.jsonObject(with: data, options: .allowFragments) as! [WKWebViewJavascriptBridgeBase.Message]
         } catch let error {