JSBridge 完善: OC 调用 JS, JS 并且回调给 OC

NJHu · NJHu · commit 5aff05fbb18e · 2018-03-20T11:10:20.000+08:00
OC 调用 JS, 并给 JS传递数据, JS并且回调给 OC
==================================
JS动态创建OC对象, 并调用对象方法, 并回调 给JS
JS调用OC类方法, 并回调 给JS
JS开启OC页面, 并设置数据
JS关闭OC页面
JS设置OC数据偏好设置存储
JS读取OC数据, 并回调 给JS
diff --git a/iOSProject/iOSProject/Classes/Preview2/LMJH5JSBridgeViewController.m b/iOSProject/iOSProject/Classes/Preview2/LMJH5JSBridgeViewController.m
@@ -35,12 +35,29 @@ - (void)viewDidLoad {
     
     self.webView.lmj_height = self.view.lmj_height * 0.7;
     self.view.backgroundColor = [UIColor greenColor];
+    
+    UIButton *btn = [UIButton initWithFrame:CGRectMake(0, self.view.lmj_height * 0.8, kScreenWidth, 44) buttonTitle:@"oc 调用 js 注册的方法" normalBGColor:[UIColor whiteColor] selectBGColor:[UIColor redColor] normalColor:[UIColor blackColor] selectColor:[UIColor greenColor] buttonFont:[UIFont boldSystemFontOfSize:17] cornerRadius:5 doneBlock:^(UIButton *btn) {
+        [weakself callJS];
+    }];
+    
+    [self.view addSubview:btn];
 }
 
 - (void)addData
 {
     [[NSUserDefaults standardUserDefaults] setObject:@"我是存在偏好设置里边value" forKey:@"sesstionId"];
-    
+}
+
+- (void)callJS
+{
+    [self.webView.scrollView scrollToTop];
+    static int i = 0;
+    i+=2;
+    LMJWeakSelf(self);
+    [self.jsBridge callHandler:@"insertContent" data:[NSString stringWithFormat:@"我是oc调用js传给js的内容, 内容: %zd", i] responseCallback:^(id responseData) {
+        NSLog(@"%@", responseData);
+        [weakself.view makeToast:responseData duration:3 position:CSToastPositionCenter];
+    }];
 }
 
 #pragma mark - LMJNavUIBaseViewControllerDataSource
@@ -62,5 +79,8 @@ -(void)rightButtonEvent:(UIButton *)sender navigationBar:(LMJNavigationBar *)nav
     
 }
 
+- (void)dealloc {
+    _jsBridge = nil;
+}
 
 @end
diff --git a/iOSProject/iOSProject/Classes/Preview2/Tool/H5Bridge.js b/iOSProject/iOSProject/Classes/Preview2/Tool/H5Bridge.js
@@ -21,6 +21,7 @@ function _createMessageIframe (src) {
 // JS 注册原生的事件, handlerName: 原生的方法名字标识   handler: 回调
 function registerHandler (handlerName, handler) {
     messageHandlers[handlerName] = handler;
+    console.log(messageHandlers);
 }
 
 
@@ -41,31 +42,36 @@ function _doSend (message, responseCallback) {
 }
 
 function _dispatchMessageFromApp (messageJSON) {
-    
+    console.log(messageJSON);
     messageJSON = messageJSON.replace(/(\t)/g, '    ');
     messageJSON = messageJSON.replace(/(\r\n)|(\n)/g, '<br>');
     const message = JSON.parse(messageJSON);
-    var responseCallback;
+    
     if (message.responseId) {
+        var responseCallback;
         responseCallback = responseCallbacks[message.responseId];
         if (!responseCallback) {
             return;
         }
         responseCallback(message.responseData);
         delete responseCallbacks[message.responseId];
     } else {
+        var messageCallback;
         if (message.callbackId) {
             const callbackResponseId = message.callbackId;
-            responseCallback = function (responseData) {
+            messageCallback = function (responseData) {
                 _doSend({responseId: callbackResponseId, responseData: responseData});
             };
         }
         var handler;
+        console.log(message.handlerName);
+        console.log(messageHandlers);
         if (message.handlerName) {
             handler = messageHandlers[message.handlerName];
         }
+        console.log(handler);
         if (handler) {
-            handler(message.data, responseCallback);
+            handler(message.data, messageCallback);
         }
     }
     
@@ -74,6 +80,7 @@ function _dispatchMessageFromApp (messageJSON) {
 const app = {
     callHandler: callHandler,
     _dispatchMessageFromApp: _dispatchMessageFromApp,
+    registerHandler: registerHandler
 };
 
 
diff --git a/iOSProject/iOSProject/Classes/Preview2/Tool/LMJH5JSBridgeViewController.html b/iOSProject/iOSProject/Classes/Preview2/Tool/LMJH5JSBridgeViewController.html
@@ -12,6 +12,8 @@
     <body>
         <ul id="tableUlId">
             
+            <div id="insertContentId" style="color: red"></div>
+            
             <li onclick="alertToast()">toast</li><br>
             
             <li onclick="closepage()">close</li><br>
@@ -129,6 +131,23 @@
         app.callHandler('alertToast', '我是 H5 给的内容', null);
     }
     
-    // =======     数据示范    =======
+                            // =======     数据示范    =======
+                            </script>
+    
+    <script>
+        
+        function registJS() {
+            console.log('setTimeout');
+            app.registerHandler('insertContent', function (data, responseCallback) {
+                                console.log(data);
+                                document.getElementById('insertContentId').innerHTML = data;
+                                    if (responseCallback) {
+                                        responseCallback('我是 js 返回给 OC 的内容 回调');
+                                    }
+                                });
+                        }
+                                
+            registJS();
+        
         </script>
 </html>
diff --git a/iOSProject/iOSProject/Classes/Preview2/Tool/WKWebViewJsBridge.m b/iOSProject/iOSProject/Classes/Preview2/Tool/WKWebViewJsBridge.m
@@ -8,9 +8,11 @@ @interface WKWebViewJsBridge ()
    __weak WKWebView *_webView;
     __weak id<WKNavigationDelegate> _delegate;
 }
-/** js 调用 oc */
+/** js 调用 oc , oc 需要注册*/
 @property (nonatomic, strong) NSMutableDictionary *messageHandles;
 
+/** js 回调 OC  */
+@property (nonatomic, strong) NSMutableDictionary *responseCallbacks;
 @end
 
 @implementation WKWebViewJsBridge
@@ -52,12 +54,11 @@ - (void)continueAction:(NSURL *)url
         return;
     }
     
+    
+    
     NSString *handleName = messageJsonDict[khandlerName];
     // 转换成小写啦啦啦啦啦
     void(^handlerBlock)(id data, void(^)(id responseData)) = self.messageHandles[handleName.lowercaseString];
-    if (!handlerBlock) {
-        return;
-    }
     
     // 回调给 H5!!!!关键步骤!!!!!================================
     void(^responseCallback)(id responseData) = nil;
@@ -68,33 +69,59 @@ - (void)continueAction:(NSURL *)url
             if (LMJIsEmpty(responseData)) {
                 return ;
             }
-            
             NSDictionary *responseMsg = @{kResponseId: callbackId, kResponseData: responseData};
-            
             // 回调 H5
             [WKBridgeTool dispatchMsgToh5:responseMsg webView:_webView];
         };
     }else {
-        
         responseCallback = ^(id responseData) {
-            
         };
     }
     // =============================================================
     // 传递的数据
     id data = messageJsonDict[kData];
+    !handlerBlock ?: handlerBlock(data, responseCallback);
     
     
-    handlerBlock(data, responseCallback);
+#define KResponseData @"responseData"
+    // response==========================================================================
+    NSString *responseId = messageJsonDict[kResponseId];
+    if (!LMJIsEmpty(responseId)) {
+        void (^responseCallBack)(id responseData) = self.responseCallbacks[responseId];
+        
+        id data = messageJsonDict[KResponseData];
+        
+        !responseCallBack ?: responseCallBack(data);
+    }
 }
 
-#pragma mark - handel
-- (void)registerHandler:(NSString *)handleName handle:(void(^)(id data, void(^)(id responseData)))handle {
-    self.messageHandles[handleName.lowercaseString] = [handle copy];
-}
 
 - (void)callHandler:(NSString*)handlerName data:(id)data responseCallback:(void(^)(id responseData))responseCallback {
+
+#define kHandlerName @"handlerName"
+#define kCallbackId @"callbackId"
+    
+    NSDictionary *message01 = @{kHandlerName: handlerName, kData: data};
+    
+    NSMutableDictionary *message = [NSMutableDictionary dictionaryWithDictionary:message01];
+    
+    if (responseCallback) {
+        
+        NSString* callbackSpecialId = [NSString stringWithFormat:@"objc_cb_%lf", CFAbsoluteTimeGetCurrent()];
+        // 传递需要回调的 id
+        message[kCallbackId] = callbackSpecialId;
+        
+        // 保存 block
+        self.responseCallbacks[callbackSpecialId] = [responseCallback copy];
+    }
     
+    [WKBridgeTool dispatchMsgToh5:message webView:_webView];
+    
+}
+
+#pragma mark - handel
+- (void)registerHandler:(NSString *)handleName handle:(void(^)(id data, void(^)(id responseData)))handle {
+    self.messageHandles[handleName.lowercaseString] = [handle copy];
 }
 
 #pragma mark - init
@@ -119,12 +146,14 @@ - (instancetype)init
     self = [super init];
     if (self) {
         _messageHandles = [NSMutableDictionary dictionary];
+        _responseCallbacks = [NSMutableDictionary dictionary];
     }
     return self;
 }
 
 - (void)dealloc {
     _messageHandles = nil;
+    _responseCallbacks = nil;
     _delegate = nil;
     _webView = nil;
 }
