Y3WebView.m

Go to the documentation of this file.

//
//  Y3WebView.m
//  TodaysMenu
//
//  Created by Don Zeek on 5/17/18.
//  Copyright © 2018 net.dzeek.y2015.ios.portfolio. All rights reserved.
//
#import "Y3WebView.h"
#import "U4WebControlsView.h"
#import "W7WebSnippetViewController.h"
#import <JavaScriptCore/JavaScriptCore.h>
#import "WKWebViewConfiguration+Console.h"


@interface WKWebView () <WKScriptMessageHandler>
@end

@implementation Y3WebViewResponder {NSString *selectedText;
    id<Y3WebViewDelegate> y3rDelegate;
}

- (instancetype)initWithFrame:(CGRect)frame
                configuration:(WKWebViewConfiguration *)configuration
        andControllerDelegate:(id<Y3WebViewDelegate>)delegate
{
    self = [super initWithFrame:frame configuration:configuration];
    if (self) {
        y3rDelegate = delegate;
        
        if (configuration.showConsole) {
            WKUserContentController *userCC = configuration.userContentController;
            [userCC addScriptMessageHandler:self name:@"log"];
            
            [userCC addScriptMessageHandler:self name:@"logging"];
            [userCC addScriptMessageHandler:self name:@"newSelectionDetected"] ;

            [self showConsole];
        }
    }
    return self;
}

- (BOOL) becomeFirstResponder
{
    NSLog(@"Y3WebView.becomeFirstResponder");
    [y3rDelegate becomeResponder:YES];
    BOOL ret = [super becomeFirstResponder];
    return ret;
}
- (BOOL) resignFirstResponder
{
    NSLog(@"Y3WebView.resignFirstResponder");
     [y3rDelegate becomeResponder:NO];
    BOOL ret = [super resignFirstResponder];
    return ret;
}
- (void)copy:(id)sender {
    NSLog(@"Y3WebView.copyClicked:  start:sender class: %@", NSStringFromClass([sender class]));
     
    [self exerciseJavascriptSendSelection];
}

- (void (^)(id, NSError *error)) provideQueryCopyBlock:(MenuItemSegment) itemSegment
{
    void (^queryCopyBlock)(id, NSError *error) = ^void(id jsReturn, NSError *error)
    {
        if (nil == error) {

             // [self->y3rDelegate jsReturn:jsReturn];
            switch (itemSegment) {
                case ReturnBasic:
                    [self->y3rDelegate jsReturnIndication:jsReturn];
                    NSLog(@"Y3WebView.provideQueryCopyBlock: ReturnBasic: %@", jsReturn);
                    break;
                case ReturnTitle:
                    [self->y3rDelegate jsTitleReturn:jsReturn];
                    NSLog(@"Y3WebView.provideQueryCopyBlock: ReturnTitle: %@", jsReturn);
                    break;
                case ReturnIngredients:
                    [self->y3rDelegate jsIngredientsReturn:jsReturn];
                    NSLog(@"Y3WebView.provideQueryCopyBlock: ReturnIngredients: %@", jsReturn);
                   break;
                case ReturnInstructions:
                    [self->y3rDelegate jsInstructionReturn:jsReturn];
                    NSLog(@"Y3WebView.provideQueryCopyBlock: ReturnInstructions: %@", jsReturn);
                   break;
                default:
                    NSLog(@"Y3WebView.provideQueryCopyBlock: default?: ret: %@", jsReturn);
                    break;
            }
            [self->y3rDelegate setControlbarColor:[UIColor greenColor]];
        } else {
            NSLog(@"Y3WebView.provideQueryCopyBlock: completion: ERROR: %@", error);
            [self->y3rDelegate setControlbarColor:[UIColor redColor]];
        }
    };
    return queryCopyBlock;
}
       
- (void) exerciseJavascriptBasic: (NSString *)js
{
    void (^queryCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnBasic];
    [self evaluateJavaScript:js completionHandler:queryCopyBlock];
}
- (void) exerciseJavascriptSendSelection
{
    NSBundle *mb = [NSBundle mainBundle];
    NSString *jsPathStr = [mb pathForResource:@"detectSelection"
                                       ofType:@"js"];
    NSString* content = [NSString stringWithContentsOfFile:jsPathStr
                                                  encoding:NSUTF8StringEncoding
                                                     error:NULL];

    [self exerciseJavascriptBasic:content];
}
- (void) exerciseJavascriptScanForRecipe
{
    NSBundle *mb = [NSBundle mainBundle];
    NSString *jsPathStr = [mb pathForResource:@"scanForRecipe"
                                       ofType:@"js"
                                  inDirectory:@"webviewJS"];
    NSString* content = [NSString stringWithContentsOfFile:jsPathStr
                                                  encoding:NSUTF8StringEncoding
                                                     error:NULL];
    NSLog(@"Y3WebView.exerciseJavascriptScanForRecipe: content: %@", content);
    void (^queryTitleCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnTitle];
    [self evaluateJavaScript:content completionHandler:queryTitleCopyBlock];
    
/*   
    void (^queryIngredientsCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnIngredients];
    [self evaluateJavaScript:content completionHandler:queryIngredientsCopyBlock];
 
    void (^queryInstructionCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnInstructions];
    [self evaluateJavaScript:content completionHandler:queryInstructionCopyBlock];
 */
}
- (void) exerciseJavascriptScanForTitle
{
    NSBundle *mb = [NSBundle mainBundle];
    NSString *jsPathStr = [mb pathForResource:@"scanForTitle"
                                       ofType:@"js"
                                  inDirectory:@"webviewJS"];
    NSString* content = [NSString stringWithContentsOfFile:jsPathStr
                                                  encoding:NSUTF8StringEncoding
                                                     error:NULL];
    NSLog(@"Y3WebView.exerciseJavascriptScanForTitle: content: %@", content);
    void (^queryTitleCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnTitle];
    [self evaluateJavaScript:content completionHandler:queryTitleCopyBlock];
}
    
- (void) exerciseJavascriptScanForIngredients
{
    NSBundle *mb = [NSBundle mainBundle];
    NSError *error;
    NSString *jsPathStr = [mb pathForResource:@"scanForIngredients"
                                       ofType:@"js"
                                  inDirectory:@"webviewJS"];
    NSString* content = [NSString stringWithContentsOfFile:jsPathStr
                                                  encoding:NSUTF8StringEncoding
                                                     error:&error];
    if (nil == error) {
        NSLog(@"Y3WebView.exerciseJavascriptScanForIngredients: content: %@", content);
        void (^queryIngredientsCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnIngredients];
        [self evaluateJavaScript:content completionHandler:queryIngredientsCopyBlock];
    } else {
        NSLog(@"Y3WebView.exerciseJavascriptScanForIngredients: error: %@", error);
    }
}
- (void) exerciseJavascriptScanForInstruction
{
    NSBundle *mb = [NSBundle mainBundle];
    NSString *jsPathStr = [mb pathForResource:@"scanForInstruction"
                                       ofType:@"js"
                                  inDirectory:@"webviewJS"];
    NSString* content = [NSString stringWithContentsOfFile:jsPathStr
                                                  encoding:NSUTF8StringEncoding
                                                     error:NULL];
    void (^queryInstructionCopyBlock)(id, NSError *error) = [self provideQueryCopyBlock:ReturnInstructions];
    [self evaluateJavaScript:content completionHandler:queryInstructionCopyBlock];
}

- (void)showConsole {
    //rewrite the method of console.log
    NSString *jsCode = @"console.log = (function(oriLogFunc){\
    return function(str)\
    {\
    window.webkit.messageHandlers.log.postMessage(str);\
    oriLogFunc.call(console,str);\
    }\
    })(console.log);";
    //injected the method when H5 starts to create the DOM tree
    [self.configuration.userContentController addUserScript:[[WKUserScript alloc] initWithSource:jsCode injectionTime:WKUserScriptInjectionTimeAtDocumentStart forMainFrameOnly:YES]];
}
- (void)userContentController:(WKUserContentController *)userContentController didReceiveScriptMessage:(WKScriptMessage *)message {
    NSLog(@"OK> %@",message.body);
}


@end
@implementation Y3WebView {
    id<Y3WebViewDelegate> y3Delegate;
    NSURL   *fileURL;
}
- (instancetype) initWithFrame:(CGRect)frame andDelegate:(id<Y3WebViewDelegate>)delegate
{
    self = [super initWithFrame:frame];
    if (self) {
        CGRect totalFrame = self.frame;
        
        CGRect controlsViewFrame = CGRectMake(10, 0, totalFrame.size.width - 20, 68.0);
        _webControls = [[U4WebControlsView alloc] initWithFrame:controlsViewFrame andDelegate:self];
        [_webControls appear:controlsViewFrame];
        [_webControls setBackgroundColor:[UIColor lightGrayColor]];
        
        float controlsHeight = 70.0;
        CGRect webViewFrame = CGRectMake(totalFrame.size.width * 0.01,
                                         controlsHeight,
                                         totalFrame.size.width * 0.98,
                                         totalFrame.size.height - controlsHeight);
        
        WKPreferences *prefers = [[WKPreferences alloc] init];
        [prefers setJavaScriptEnabled:YES];
        [prefers setJavaScriptCanOpenWindowsAutomatically:YES];


//let contentController = WKUserContentController()
//let scriptSource = "document.body.style.backgroundColor = `red`;"
//let script = WKUserScript(source: scriptSource, injectionTime: .atDocumentEnd, forMainFrameOnly: true)
//contentController.addUserScript(script)
//
//let config = WKWebViewConfiguration()
//config.userContentController = contentController
//
//let webView = WKWebView(frame: .zero, configuration: config)

        // create message handlers named "logging" and "newSelectionDetected"
        WKUserContentController *ucc = [[WKUserContentController alloc] init];
//        [ucc addScriptMessageHandler:delegate name:@"logging"];
//        [ucc addScriptMessageHandler:delegate name:@"newSelectionDetected"] ;

        // assign usercontentcontroller to configuration
        WKWebViewConfiguration *configur = [[WKWebViewConfiguration alloc] init];
        [configur setPreferences:prefers];
        [configur setUserContentController:ucc];
        configur.showConsole = YES;

        _webView = [[Y3WebViewResponder alloc] initWithFrame:webViewFrame configuration:configur andControllerDelegate:delegate];
        
        y3Delegate = delegate;
        _webView.navigationDelegate = delegate;
        _webView.UIDelegate = delegate;
        
        
//  Terminating app due to uncaught exception 'NSUnknownKeyException', reason: '[<Y3WebViewResponder 0x7f98bd002e00> valueForUndefinedKey:]: this class is not key value coding-compliant for the key documentView.'
//        JSContext *ctx = [_webView valueForKeyPath:@"documentView.webView.mainFrame.javaScriptContext"];
//        ctx[@"console"][@"log"] = ^(JSValue * msg) {
//            NSLog(@"JavaScript %@ log message: %@", [JSContext currentContext], msg);
//        };
        
        
        
        // UITapGestureRecognizer *tapRecognizer = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(handleTapGesture:)];
        // [webView addGestureRecognizer:tapRecognizer];

        [self addSubview:_webControls];
        [self addSubview:_webView];
    }
    return self;
}
- (void) updateWebControls
{
    [_webControls update:_webView];
}

- (void) captureClicked:(id)sender
{
    NSLog(@"captureClicked");
}

- (void) handleTapGesture:(UIGestureRecognizer *)gesture
{
    NSLog(@"Y3WebView.handleTapGesture: tap recognized");
    
    CGPoint tapLocation = [gesture locationInView:_webView];
    
    CGRect targetRect = CGRectMake(tapLocation.x, tapLocation.y, 0.0f, 0.0f);
    [[UIMenuController sharedMenuController] setTargetRect:targetRect inView:_webView];
    
    UIMenuItem *menuItem = [[UIMenuItem alloc] initWithTitle:@"Capture" action:@selector(captureClicked:)];
    [[UIMenuController sharedMenuController] setMenuItems:@[menuItem]];
    
    [[UIMenuController sharedMenuController] setMenuVisible:YES animated:YES];
}

- (void) becomeResponder: (BOOL)flag {
    
    [_webControls focusAvailable:flag];
}
- (void) reformForKeyboard
{
    NSLog(@"Y3WebView.reformForKeyboard: NO-OP");
//
//    CGRect totalFrame = self.frame;
//    CGFloat height;
//    if (flag) {
//        height = totalFrame.size.height * 0.38;
//    } else {
//        height = totalFrame.size.height * 0.88;
//    }
//
//    CGRect webViewFrame = CGRectMake(totalFrame.size.width * 0.01,
//                                     totalFrame.size.height * 0.10,
//                                     totalFrame.size.width * 0.98,
//                                     height);
//
//    [_webView setFrame:webViewFrame];
//    [self setNeedsDisplay];
}
- (void) goBack
{
    NSLog(@"Y3WebView.goBack");
    [_webView goBack];
}
- (void) goForward
{
    NSLog(@"Y3WebView.goForward");
    [_webView goForward];
}
- (void) stirJavascript
{
    NSLog(@"Y3WebView.stirJavascript");
    [_webView exerciseJavascriptSendSelection];
 }
- (void) loadRequest: (NSURLRequest *)requestURL
{
    NSLog(@"Y3WebView.loadRequest");
    [_webView loadRequest:requestURL];
}
- (void) goEditCopy
{
    NSLog(@"Y3WebView.goEditCopy");
    [y3Delegate fireSnippetScreen];
}
#pragma - mark
#pragma called from didFinishNavigation
- (void) scanPageForTitle
{
    NSLog(@"Y3WebView.scanPageForTitle");
    [_webView exerciseJavascriptScanForTitle];
}
- (void) scanPageForIngredients
{
    NSLog(@"Y3WebView.scanPageForIngredients");
    [_webView exerciseJavascriptScanForIngredients];
}
- (void) scanPageForInstruction
{
    NSLog(@"Y3WebView.scanPageForInstruction");
    [_webView exerciseJavascriptScanForInstruction];
}
#pragma mark - WKUIDelegate

- (void)webView:(WKWebView *)webViewHere didFinishNavigation:(WKNavigation *)navigation
{
    NSLog(@"Y3WebView.didFinishNavigation");
    assert(NO);
    [_webControls update: webViewHere];
}

- (void)webView:(WKWebView *)webView
didCommitNavigation:(WKNavigation *)navigation
{
    NSLog(@"Y3WebView.didCommitNavigation: active URL: %@", webView.URL.description);
}

- (void)webView:(WKWebView *)webView didStartProvisionalNavigation:(null_unspecified WKNavigation *)navigation
{
    NSLog(@"Y3WebView.didStartProvisionalNavigation: active URL: %@", webView.URL.description);
}

- (void)webView:(WKWebView *)webView
        decidePolicyForNavigationResponse:(WKNavigationResponse *)navigationResponse
        decisionHandler:(void (^)(WKNavigationResponsePolicy))decisionHandler{

           NSHTTPURLResponse *response = (NSHTTPURLResponse *)navigationResponse.response;
           NSArray *cookies =[NSHTTPCookie cookiesWithResponseHeaderFields:[response allHeaderFields] forURL:response.URL];
           for (NSHTTPCookie *cookie in cookies) {
            [[NSHTTPCookieStorage sharedHTTPCookieStorage] setCookie:cookie];
           }
    
    decisionHandler(WKNavigationResponsePolicyAllow);
    NSLog(@"Y3WebView.decidePolicyForNavigationResponse: ALLOW");
}


//Download manager
- (void)webView:(WKWebView *)webView
        decidePolicyForNavigationAction:(WKNavigationAction *)navigationAction
        decisionHandler:(void (^)(WKNavigationActionPolicy))decisionHandler {
    
    NSURLRequest *request = navigationAction.request;
    fileURL = request.URL;
    
    //    HCDownloadViewController *dlvc = [[HCDownloadViewController alloc] init];
    //    UINavigationController *vc = [[UINavigationController alloc] initWithRootViewController:dlvc];
    //    vc.transitioningDelegate  = self;
    //    dlvc.delegate = self;
    
    NSLog(@"Y3WebView.decidePolicyForNavigationAction: ~~~~~~~~~~~~~~~~~~~~~~");
    [self logNavigationAction:navigationAction];
    if (navigationAction.navigationType == WKNavigationTypeLinkActivated) {
        
    }
    
    if (decisionHandler) {
        decisionHandler(WKNavigationActionPolicyAllow);
    }
}

- (void) logNavigationAction: (WKNavigationAction *)navigationAction
{
    NSLog(@"     request: %@", navigationAction.request);
    NSLog(@"    source frame: %@", navigationAction.sourceFrame);
    NSLog(@"    target frame: %@", navigationAction.targetFrame);
    if (navigationAction.navigationType == WKNavigationTypeLinkActivated) {
        NSLog(@"     WKNavigationTypeLinkActivated");
    }
    if (navigationAction.navigationType == WKNavigationTypeFormSubmitted) {
        NSLog(@"     WKNavigationTypeFormSubmitted");
    }
    if (navigationAction.navigationType == WKNavigationTypeBackForward) {
        NSLog(@"     WKNavigationTypeBackForward");
    }
    if (navigationAction.navigationType == WKNavigationTypeReload) {
        NSLog(@"     WKNavigationTypeReload");
    }
    if (navigationAction.navigationType == WKNavigationTypeFormResubmitted) {
        NSLog(@"     WKNavigationTypeFormResubmitted");
    }
    if (navigationAction.navigationType == WKNavigationTypeOther) {
        NSLog(@"     WKNavigationTypeOther");
    }
}

/*
// Only override drawRect: if you perform custom drawing.
// An empty implementation adversely affects performance during animation.
- (void)drawRect:(CGRect)rect {
    // Drawing code
}
*/

@end