@@ -69,7 +69,7 @@ - (void)callHandler:(NSString *)handlerName data:(id)data responseCallback:(WVJB
6969}
7070
7171- (void )registerHandler : (NSString *)handlerName handler : (WVJBHandler)handler {
72- [ self .messageHandlers setObject: handler forKey: handlerName];
72+ self.messageHandlers [ handlerName] = handler ;
7373}
7474
7575- (void )reset {
@@ -83,12 +83,12 @@ - (void)_sendData:(NSDictionary *)data responseCallback:(WVJBResponseCallback)re
8383
8484 if (responseCallback) {
8585 NSString * callbackId = [NSString stringWithFormat: @" objc_cb_%d " , ++_uniqueId];
86- [ self .responseCallbacks setObject: responseCallback forKey: callbackId];
87- [ message setObject: callbackId forKey: @" callbackId" ];
86+ self.responseCallbacks [ callbackId] = responseCallback ;
87+ message[ @" callbackId" ] = callbackId ;
8888 }
8989
9090 if (handlerName) {
91- [ message setObject: handlerName forKey: @" handlerName" ];
91+ message[ @" handlerName" ] = handlerName ;
9292 }
9393 [self _queueMessage: message];
9494}
@@ -127,20 +127,17 @@ - (void)_flushMessageQueue {
127127
128128 NSDictionary * message = [self _deserializeMessageJSON: messageJSON];
129129
130- NSString * responseId = [ message objectForKey: @" responseId" ];
130+ NSString * responseId = message[ @" responseId" ];
131131 if (responseId) {
132- WVJBResponseCallback responseCallback = [ _responseCallbacks objectForKey: responseId];
133- responseCallback ([ message objectForKey: @" responseData" ]);
132+ WVJBResponseCallback responseCallback = _responseCallbacks[ responseId];
133+ responseCallback (message[ @" responseData" ]);
134134 [_responseCallbacks removeObjectForKey: responseId];
135135 } else {
136136 WVJBResponseCallback responseCallback = NULL ;
137- __block NSString * callbackId = [ message objectForKey: @" callbackId" ];
137+ __block NSString * callbackId = message[ @" callbackId" ];
138138 if (callbackId) {
139- __block bool wasCalled = false ;
140139 responseCallback = ^(id responseData) {
141- if (wasCalled) { return ; }
142- wasCalled = true ;
143- NSDictionary * message = [NSDictionary dictionaryWithObjectsAndKeys: callbackId, @" responseId" , responseData, @" responseData" , nil ];
140+ NSDictionary * message = @{ @" responseId" :callbackId, @" responseData" :responseData };
144141 [self _queueMessage: message];
145142 };
146143 } else {
@@ -149,20 +146,16 @@ - (void)_flushMessageQueue {
149146 };
150147 }
151148
152- WVJBHandler handler = self.messageHandler ;
153-
154- NSString * handlerName = [message objectForKey: @" handlerName" ];
155- if (handlerName) {
156- handler = [_messageHandlers objectForKey: handlerName];
157- }
158-
159- if (!handler) {
160- NSLog (@" WVJB Warning: No handler for %@ " , handlerName);
161- return ;
149+ WVJBHandler handler;
150+ if (message[@" handlerName" ]) {
151+ handler = _messageHandlers[message[@" handlerName" ]];
152+ if (!handler) { return NSLog (@" WVJB Warning: No handler for %@ " , message[@" handlerName" ]); }
153+ } else {
154+ handler = self.messageHandler ;
162155 }
163156
164157 @try {
165- NSDictionary * data = [ message objectForKey: @" data" ];
158+ NSDictionary * data = message[ @" data" ];
166159 if (!data || ((id )data) == [NSNull null ]) { data = [NSDictionary dictionary ]; }
167160 handler (data, responseCallback);
168161 }
0 commit comments