//

// WKJavaScriptBridge.m

// WKJavaScriptBridge-demo

//

// Created by Kevin on 2019/10/22.

// Copyright © 2019 王凯. All rights reserved.

//

#import "WKJavaScriptBridge.h"

#import "WKCommandImpl.h"

#import "WKWeakScriptMessageHandlerDelegate.h"

#import "WKJavaScriptMessageDispatcher.h"

#import "WKJavaScriptBridgePluginAnnotation.h"

#import "WKBasePlugin.h"

#import "WKCommandProtocol.h"

#import "WKPluginSecurityProtocol.h"

static NSString * const WKJSBridge = @"WKJSBridgeMessageName";

@interface WKJavaScriptBridge ()<WKScriptMessageHandler>

@property (nonatomic, strong) WKJavaScriptBridgePluginAnnotation *annotation;//模块注册白名单

@property (nonatomic, strong) WKJavaScriptMessageDispatcher *dispatcher;//插件模块派发

@property (nonatomic, strong) WKCommandImpl *commandDelegate;//插件模块可执行函数实现

@property (nonatomic, strong) NSMutableDictionary *pluginsByName;//插件模块实例对象

@property (nonatomic, weak, readwrite) WKWebView *webView;

//安全相关

@property (nonatomic, strong) id <WKPluginSecurityProtocol>securityPlugin;//白名单插件

@property (nonatomic, strong) NSArray *secInfo;//业务注册的安全信息

@end

@implementation WKJavaScriptBridge

- (void)dealloc {

#ifdef DEBUG

NSLog(@"WKJavaScriptBridge dealloc");

#endif

WKWebView *webView = (WKWebView *)self.webView;

[webView.configuration.userContentController removeScriptMessageHandlerForName:WKJSBridge];

[[self.pluginsByName allValues] makeObjectsPerformSelector:@selector(dispose)];

[self.pluginsByName removeAllObjects];

}

#pragma mark - Public

+ (instancetype)bindBridgeWithWebView:(WKWebView *)webView {

if ([webView isKindOfClass:[WKWebView class]]) {

WKJavaScriptBridge *bridge = [[self alloc] initBridgeWithWebView:webView];

return bridge;

}

[NSException raise:@"BadWebView" format:@"Unknown web view."];

return nil;

}

#pragma mark - Init

- (instancetype)initBridgeWithWebView:(WKWebView *)webView {

if (self = [super init]) {

self.webView = webView;

[self commonInit];

[self setup];

}

return self;

}

- (void)commonInit {

self.secInfo = [NSArray array];

self.dispatcher = [[WKJavaScriptMessageDispatcher alloc] initWithBridge:self];

self.annotation = [[WKJavaScriptBridgePluginAnnotation alloc] initWithBridge:self];

self.commandDelegate = [[WKCommandImpl alloc] initWithBridge:self];

}

#pragma mark - Privite

- (void)setup {

[self setupJSBridge];

[self setupPlugin];

}

- (void)setupJSBridge {

WKWebViewConfiguration *webViewConfiguration = self.webView.configuration;

if (webViewConfiguration && !webViewConfiguration.userContentController) {

self.webView.configuration.userContentController = [WKUserContentController new];

}

// 防止内存泄露

[self.webView.configuration.userContentController addScriptMessageHandler:[[WKWeakScriptMessageHandlerDelegate alloc] initWithDelegate:self] name:WKJSBridge];

}

- (void)setupPlugin {

[self.annotation getAllRegisterPluginName];

}

- (void)registerPlugin:(WKBasePlugin *)plugin {

if ([plugin respondsToSelector:@selector(setCommandDelegate:)]) {

[plugin setCommandDelegate:_commandDelegate];

}

}

- (Class)pluginImplClass:(NSString *)pluginName {

if (pluginName.length > 0) {

return NSClassFromString(pluginName);

}

return nil;

}

- (BOOL)pluginShouldCache:(Class)implClass {

BOOL shouldCache = NO;

if ([[implClass class] respondsToSelector:@selector(shouldCache)]) {

if ([[implClass class] shouldCache]) {

shouldCache = YES;

}

}

return shouldCache;

}

- (void)pluginInstall:(WKBasePlugin *)plugin {

[plugin pluginInitialize];

[self registerPlugin:plugin];

}

#pragma mark - WKJavaScriptBridgeProtocol

- (void)registerSecurityPlugin:(NSString *)name {

id<WKPluginSecurityProtocol>securityPlugin = [self getCommandInstance:name];

if ([securityPlugin isKindOfClass:[WKBasePlugin class]] &&

[securityPlugin conformsToProtocol:@protocol(WKPluginSecurityProtocol)]) {

self.securityPlugin = securityPlugin;

}

}

- (void)registerPluginSecurityInfomation:(NSArray<NSString *>*)secInfo {

self.secInfo = secInfo;

}

- (void)pluginSecurityVerityWithCommand:(WKMsgCommand *)command completionHandler:(void (^)(BOOL passed))completionHandler {

if (self.securityPlugin && [self.securityPlugin respondsToSelector:@selector(applyAuthorityWithService:action:securityInfomation:complete:)]) {

[self.securityPlugin applyAuthorityWithService:command.className action:command.methodName securityInfomation:self.secInfo complete:^(BOOL allowed) {

completionHandler(allowed);

}];

}else {

completionHandler(YES);

}

}

- (id)getCommandInstance:(NSString*)pluginName {

WKBasePlugin <WKPluginProtocol> *plugin = nil;

Class implClass = [self pluginImplClass:pluginName];

BOOL shouldCache = [self pluginShouldCache:implClass];

if (shouldCache) {

plugin = [self.pluginsByName objectForKey:pluginName];

if (plugin) {

return plugin;

}

}

if ([[implClass class] respondsToSelector:@selector(singleton)]) {

if ([[implClass class] singleton]) {

if ([[implClass class] respondsToSelector:@selector(shareInstance)])

plugin = [[implClass class] shareInstance];

else

plugin = [[implClass alloc] init];

if (plugin) {

[self pluginInstall:plugin];

if (shouldCache) {

[self.pluginsByName setObject:plugin forKey:pluginName];

}

}else {

#ifdef DEBUG

NSLog(@"PHJavaScriptBridge Error : 模块: (%@) 不存在。", pluginName);

#endif

}

return plugin;

}

}

plugin = [[implClass alloc] init];

if (plugin) {

[self pluginInstall:plugin];

if (shouldCache) {

[self.pluginsByName setObject:plugin forKey:pluginName];

}

}else {

#ifdef DEBUG

NSLog(@"PHJavaScriptBridge Error : 模块: (%@) 不存在。", pluginName);

#endif

}

return plugin;

}

- (void)evaluateJavaScript:(NSString *)javaScriptString completionHandler:(void (^)(id , NSError *))completionHandler {

WKWebView *webView = (WKWebView *)self.webView;

[webView evaluateJavaScript:javaScriptString completionHandler:^(id obj, NSError * _Nullable error) {

if (completionHandler) {

completionHandler(obj, error);

}

}];

}

#pragma mark - WKScriptMessageHandler

- (void)userContentController:(WKUserContentController *)userContentController didReceiveScriptMessage:(WKScriptMessage *)message {

if ([message.name isEqualToString:WKJSBridge]) {

if ([message.body isKindOfClass:[NSString class]]) {

[self.dispatcher handleMsgCommand:message.body];

}else {

#ifdef DEBUG

NSLog(@"WKJavaScriptBridge Error : 通信参数不合法，请将JS侧参数转为字符串通信。");

#endif

}

}else {

#ifdef DEBUG

NSLog(@"WKJavaScriptBridge Error : 本次调用所使用的JS函数非（WKJavaScriptBridge）注入。请使用（%@）",WKJSBridge);

#endif

}

}

#pragma mark - Set or Get

- (NSMutableDictionary *)pluginsByName {

if (!_pluginsByName) {

_pluginsByName = [[NSMutableDictionary alloc] initWithCapacity:1];

}

return _pluginsByName;

}

@end