Implement removeHandler for WKWebViewJavascriptBridge.

marcuswestin · marcuswestin · commit f99c987292f4 · 2017-01-10T16:51:40.000-05:00
Fix marcuswestin#260
diff --git a/WebViewJavascriptBridge/WKWebViewJavascriptBridge.h b/WebViewJavascriptBridge/WKWebViewJavascriptBridge.h
@@ -21,6 +21,7 @@
 + (void)enableLogging;
 
 - (void)registerHandler:(NSString*)handlerName handler:(WVJBHandler)handler;
+- (void)removeHandler:(NSString*)handlerName;
 - (void)callHandler:(NSString*)handlerName;
 - (void)callHandler:(NSString*)handlerName data:(id)data;
 - (void)callHandler:(NSString*)handlerName data:(id)data responseCallback:(WVJBResponseCallback)responseCallback;
diff --git a/WebViewJavascriptBridge/WKWebViewJavascriptBridge.m b/WebViewJavascriptBridge/WKWebViewJavascriptBridge.m
@@ -53,6 +53,10 @@ - (void)registerHandler:(NSString *)handlerName handler:(WVJBHandler)handler {
     _base.messageHandlers[handlerName] = [handler copy];
 }
 
+- (void)removeHandler:(NSString *)handlerName {
+    [_base.messageHandlers removeObjectForKey:handlerName];
+}
+
 - (void)reset {
     [_base reset];
 }

Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,10 @@ - (void)registerHandler:(NSString *)handlerName handler:(WVJBHandler)handler {`
`53`	`53`	`_base.messageHandlers[handlerName] = [handler copy];`
`54`	`54`	`}`
`55`	`55`
	`56`	`+- (void)removeHandler:(NSString *)handlerName {`
	`57`	`+ [_base.messageHandlers removeObjectForKey:handlerName];`
	`58`	`+}`
	`59`	`+`
`56`	`60`	`- (void)reset {`
`57`	`61`	`[_base reset];`
`58`	`62`	`}`