//
//  YHExplorerViewController.m
//  YohoExplorerDemo
//
//  Created by gaoqiang xu on 2/27/15.
//  Copyright (c) 2015 gaoqiang xu. All rights reserved.
//

static float kprogressViewHeight = 2.0f;
static float kprogressViewRadius = 2.0f;

#import "YHExplorerViewController.h"
#import "YHWebViewProgress.h"
#import "YHWebViewProgressView.h"

@interface YHExplorerViewController ()
@property (strong, nonatomic) YHWebViewProgress *progressBar;
@property (strong, nonatomic) YHWebViewProgressView *progressView;

@property (nonatomic) BOOL isAppeared;
@property (strong, nonatomic) NSMutableArray *linkParsers;

@property (strong, nonatomic) NSValue *pointerValue;

@end

@implementation YHExplorerViewController


- (void)dealloc
{
    [self.webView stopLoading];
    self.webView.delegate = nil;
    [self.webView removeFromSuperview];
    
    self.progressBar = nil;
    [self removeProgress];
    self.progressView = nil;

    
    self.webView = nil;
    
    if ([[NSURLCache sharedURLCache] isKindOfClass:[YHURLCache class]]) {
        YHURLCache *cache = (YHURLCache *)[YHURLCache sharedURLCache];
        [cache removeUrlsForKey:self.pointerValue];
    }
}

- (void)initializeExplorer
{
    _commandDelegate = [[YHExplorerCommandDelegate alloc] initWithViewController:self];
    _commandQueue = [[YHExplorerCommandQueue alloc] initWithViewController:self];
    _progressBarEnabled = NO;
    _automaticallySchemeHandle = YES;
}

- (instancetype)initWithNibName:(NSString*)nibNameOrNil bundle:(NSBundle*)nibBundleOrNil
{
    self = [super initWithNibName:nibNameOrNil bundle:nibBundleOrNil];
    if (self) {
        [self initializeExplorer];
    }
    return self;
}

- (instancetype)initWithCurrentNaviBar:(UINavigationBar *)currentNaviBar
{
    self = [self init];
    if (self) {
        self.currentNaviBar = currentNaviBar;
    }
    return self;
}


- (void)viewDidLoad
{
    [super viewDidLoad];
    self.isAppeared = NO;
    
    YHLinkDefaultParser *parser = [[YHLinkDefaultParser alloc] init];
    [self registerLinkParser:parser];
    
    if (self.progressBar && !self.progressBar.progressView.superview) {
        [self.currentNaviBar addSubview:self.progressBar.progressView];
    }
}

- (BOOL)shouldAutorotateToInterfaceOrientation:(UIInterfaceOrientation)interfaceOrientation
{
    return [super shouldAutorotateToInterfaceOrientation:interfaceOrientation];
}

- (void)viewDidAppear:(BOOL)animated
{
    [super viewDidAppear:animated];
    self.isAppeared = YES;
    if (self.progressBar) {
        if (!self.progressBar.progressView.superview) {
            [self.currentNaviBar addSubview:self.progressBar.progressView];
        } else {
            [self.currentNaviBar bringSubviewToFront:self.progressBar.progressView];
        }
    }
}


/**
 *  移除滚动条
 */
- (void)removeProgress
{
    if (self.progressView) {
        [self.progressView removeFromSuperview];
    }
}

#pragma mark - Override Setter & Getter
- (NSMutableArray *)linkParsers
{
    if (!_linkParsers) {
        _linkParsers = [[NSMutableArray alloc] initWithCapacity:1];
    }
    return _linkParsers;
}

- (void)setProgressBarEnabled:(BOOL)progressBarEnabled
{
    if (_progressBarEnabled == progressBarEnabled) {
        return;
    }
    
    _progressBarEnabled = progressBarEnabled;
    
    if (progressBarEnabled) {
        self.progressView = [[YHWebViewProgressView alloc]init];
        self.progressView.layer.cornerRadius = kprogressViewRadius;
        
        [self.progressView setFrame:CGRectMake(0, 41 , CGRectGetWidth(self.view.frame), kprogressViewHeight)];
        self.progressView.autoresizingMask = UIViewAutoresizingFlexibleWidth|UIViewAutoresizingFlexibleBottomMargin;
        if (self.isAppeared) {
            [_currentNaviBar addSubview:self.progressView];
        }
        _progressBar.progressView = self.progressView;
        
        _progressBar = [[YHWebViewProgress alloc] init];
        _progressBar.progressView = self.progressView;
        
    } else {
        if (self.progressBar) {
            [self.progressBar.progressView removeFromSuperview];
            self.progressBar = nil;
        }
    }
}

-(YHWebViewProgressView *)progressView
{
    static YHWebViewProgressView *sharedProgressViewInstance = nil;
    static dispatch_once_t predicate;
    dispatch_once(&predicate, ^{
        sharedProgressViewInstance = [[YHWebViewProgressView alloc] init];
    });
    return sharedProgressViewInstance;
}

- (NSMutableArray *)autoSchemes
{
    if (!_autoSchemes) {
        _autoSchemes = [[NSMutableArray alloc] initWithObjects:@"itms-apps", @"itms-appss", nil];
    }
    return _autoSchemes;
}

- (NSValue *)pointerValue
{
    if (!_pointerValue) {
        _pointerValue = [NSValue valueWithPointer:(__bridge const void *)(self)];
    }
    return _pointerValue;
}

- (NSString*)userAgent
{
    _userAgent = [super userAgent];
    if ([self.delegate respondsToSelector:@selector(extraUserAgent)]) {
        NSString *ua = [self.delegate extraUserAgent];
        if (ua.length) {
            return [NSString stringWithFormat:@"%@ %@", _userAgent, ua];
        }
    }
    return _userAgent;
}

#pragma mark - Public
- (instancetype)initWithUrl:(NSString *)url
{
    self = [self init];
    if (self) {
        self.startPage = url;
    }
    
    return self;
}

- (void)loadWebUrl:(NSString *)url
{
    self.startPage = url;
    NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:[NSURL URLWithString:url]];
    if ([self.delegate respondsToSelector:@selector(explorerWillLoadRequest:)]) {
        [self.delegate explorerWillLoadRequest:request];
    }
    [self.webView loadRequest:request];
}

- (void)loadHTMLString:(NSString *)html
{
    [self.webView loadHTMLString:html baseURL:[[NSBundle mainBundle] bundleURL]];
}

- (void)loadHTMLString:(NSString *)html baseURL:(nullable NSURL *)baseURL
{
    [self.webView loadHTMLString:html baseURL:baseURL];
}

- (void)setProgressBarColor:(UIColor *)color
{
    if (self.progressBar) {
        YHWebViewProgressView *view = (YHWebViewProgressView *)self.progressBar.progressView;
        view.progressBarColor = color;
    }
}

- (void)registerLinkParser:(id <YHLinkParserDelegate>)parser
{
    if ([self.linkParsers containsObject:parser]) {
        return;
    }
    
    __weak typeof(self)weakSelf = self;
    parser.completion = ^(NSDictionary *result) {
        [weakSelf parseFinishedWithResult:result];
    };
    
    [self.linkParsers addObject:parser];
}

- (void)unRegisterLinkParser:(id <YHLinkParserDelegate>)parser
{
    [self.linkParsers removeObject:parser];
}

#pragma mark - Private
/**
 *  动态注入Js
 *
 *  @param resource js的名称
 */
- (BOOL)injectJavascript:(NSString *)resource {
    NSString *jsPath = [[NSBundle mainBundle] pathForResource:resource ofType:@"js"];
    NSString *js = [NSString stringWithContentsOfFile:jsPath encoding:NSUTF8StringEncoding error:NULL];
    
    return ([self.webView stringByEvaluatingJavaScriptFromString:js] != nil);
}

/**
 @brief 循环注入直到成功
 
 @since 1.0.2
 */
- (void)cycleInject {
    
    BOOL bSuc = NO;
    
    // 向网页中注入cordova
    bSuc = [self injectJavascript:@"cordova"];
    
    if (!bSuc) {
        __weak typeof(self) weakSelf = self;
        
        dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.2 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
            [weakSelf cycleInject];
        });
        
    }
}

- (void)parseFinishedWithResult:(NSDictionary *)result
{
    if ([self.delegate respondsToSelector:@selector(explorerDidDetectParameters:)])
    {
        [self.delegate explorerDidDetectParameters:result];
    }
    
    if ([self.delegate respondsToSelector:@selector(explorer:didDetectParameters:)]) {
        [self.delegate explorer:self didDetectParameters:result];
    }
}

#pragma mark - UIWebDelegate

- (void)webViewDidFinishLoad:(UIWebView*)theWebView
{
    NSString *title = [theWebView stringByEvaluatingJavaScriptFromString: @"document.title"];
    _pageTitle = title;
    
    if (self.delegate && [self.delegate respondsToSelector:@selector(pageTitleUpdated:)]) {
        [self.delegate pageTitleUpdated:title];
    }
    
    [super webViewDidFinishLoad:theWebView];
    
    [self cycleInject];
    
    if (self.delegate && [self.delegate respondsToSelector:@selector(webViewDidFinishLoad:)]) {
        [self.delegate webViewDidFinishLoad:theWebView];
    }
    
    if (self.progressBar) {
        [self.progressBar webViewDidFinishLoad:theWebView];
    }
    
    NSString *key = [[NSString alloc] initWithFormat:@"WebKit%@%@", @"CacheModel", @"PreferenceKey"];//WebKitCacheModelPreferenceKey
    [[NSUserDefaults standardUserDefaults] setInteger:0 forKey:key];
    [[NSUserDefaults standardUserDefaults] synchronize];
}

- (void) webViewDidStartLoad:(UIWebView*)theWebView
{
    [super webViewDidStartLoad:theWebView];
    
    if (self.delegate && [self.delegate respondsToSelector:@selector(webViewDidStartLoad:)]) {
        [self.delegate webViewDidStartLoad:theWebView];
    }
    
    if (self.progressBar) {
        [self.progressBar webViewDidStartLoad:theWebView];
    }
    
    [self cycleInject];
}

- (void) webView:(UIWebView*)theWebView didFailLoadWithError:(NSError*)error
{
    [super webView:theWebView didFailLoadWithError:error];
    
    if (self.delegate && [self.delegate respondsToSelector:@selector(webView:didFailLoadWithError:)]) {
        [self.delegate webView:theWebView didFailLoadWithError:error];
    }
    
    if (self.progressBar) {
        [self.progressBar webView:theWebView didFailLoadWithError:error];
    }
}

- (BOOL) webView:(UIWebView*)theWebView shouldStartLoadWithRequest:(NSURLRequest*)request navigationType:(UIWebViewNavigationType)navigationType
{
    if (self.progressBar && [self.progressBar checkIfRPCURL:request]) {
        return NO;
    }
    
    if ([self.delegate respondsToSelector:@selector(webView:didCatchRequest:navigationType:)]) {
        [self.delegate webView:theWebView didCatchRequest:request navigationType:navigationType];
    }
    
    BOOL b = [super webView:theWebView shouldStartLoadWithRequest:request navigationType:navigationType];
    
    if (self.delegate && [self.delegate respondsToSelector:@selector(webView:shouldStartLoadWithRequest:navigationType:)]) {
        b = [self.delegate webView:theWebView shouldStartLoadWithRequest:request navigationType:navigationType];
    }
    
    if (self.progressBar && b) {
        b = [self.progressBar webView:theWebView shouldStartLoadWithRequest:request navigationType:navigationType];
    }
    
    if (self.automaticallySchemeHandle && b) {
        if ([[UIApplication sharedApplication] canOpenURL:request.URL]) {
            NSArray *array = [self.autoSchemes filteredArrayUsingPredicate:[NSPredicate predicateWithFormat:@"SELF = %@", request.URL.scheme]];
            if (array.count > 0) {
                [[UIApplication sharedApplication] openURL:request.URL];
            }
        }
    }
    
    // 尝试把当前url添加至缓存
    if (b && [[request.URL scheme] hasPrefix:@"http"] && [[NSURLCache sharedURLCache] isKindOfClass:[YHURLCache class]]) {
        YHURLCache *cache = (YHURLCache *)[YHURLCache sharedURLCache];
        [cache addCacheUrl:request.URL.absoluteString forKey:self.pointerValue];
    }
    
    return b;
}

#pragma mark - CDVYHInterfaceDelegate
- (BOOL)parseLink:(NSString *)linkUrl
{
    for (id <YHLinkParserDelegate> parser in self.linkParsers) {
        if ([parser parseString:linkUrl]) {
            return YES;
        }
    }
    
    return NO;
}

- (void)nativeActionDidTriggerEventWithUserInfo:(NSDictionary *)userInfo completion:(void (^)(id, BOOL))completion {
    
    dispatch_async(dispatch_get_main_queue(), ^{
        if ([self.delegate respondsToSelector:@selector(didGetMessageFromWeb:completion:)]) {
            [self.delegate didGetMessageFromWeb:userInfo completion:completion];
        }
    });
}

#pragma mark - Method Forwarding
- (BOOL)respondsToSelector:(SEL)aSelector
{
    if ([super respondsToSelector:aSelector]) {
        return YES;
    }
    
    if ([self.webView respondsToSelector:aSelector]) {
        return YES;
    }
    
    return NO;
}

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector
{
    NSMethodSignature *signature = [super methodSignatureForSelector:aSelector];
    if (!signature) {
        if (self.webView && [self.webView respondsToSelector:aSelector]) {
            return [(NSObject *)self.webView methodSignatureForSelector:aSelector];
        }
    }
    
    return signature;
}

- (void)forwardInvocation:(NSInvocation *)anInvocation
{
    if (self.webView && [self.webView respondsToSelector:[anInvocation selector]]) {
        [anInvocation invokeWithTarget:self.webView];
    }
}

@end


@implementation YHExplorerCommandDelegate

#pragma mark CDVCommandDelegate implementation

- (id)getCommandInstance:(NSString*)className
{
    return [super getCommandInstance:className];
}

- (NSString*)pathForResource:(NSString*)resourcepath
{
    return [super pathForResource:resourcepath];
}

@end

@implementation YHExplorerCommandQueue

- (BOOL)execute:(CDVInvokedUrlCommand*)command
{
    return [super execute:command];
}

@end