//
//  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